@@ -34,6 +34,7 @@ import (
3434 "github.com/elastic/beats/v7/filebeat/input/filestream/internal/task"
3535 input "github.com/elastic/beats/v7/filebeat/input/v2"
3636 "github.com/elastic/beats/v7/libbeat/beat"
37+ "github.com/elastic/beats/v7/libbeat/management/status"
3738 "github.com/elastic/beats/v7/libbeat/tests/resources"
3839 "github.com/elastic/elastic-agent-libs/logp"
3940 "github.com/elastic/elastic-agent-libs/logp/logptest"
@@ -115,7 +116,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
115116 defer goroutinesChecker .WaitUntilOriginalCount ()
116117
117118 wg .Add (1 )
118- hg .Start (input.Context {Logger : logp .L (), Cancelation : context . Background () }, source )
119+ hg .Start (input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }, source )
119120
120121 // wait until harvester.Run is done
121122 wg .Wait ()
@@ -167,10 +168,10 @@ func TestDefaultHarvesterGroup(t *testing.T) {
167168 source2 := & testSource {name : "/path/to/test/2" }
168169 wg .Add (2 )
169170 hg .Start (
170- input.Context {Logger : logp .L (), Cancelation : context . Background () },
171+ input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} },
171172 source1 )
172173 hg .Start (
173- input.Context {Logger : logp .L (), Cancelation : context . Background () },
174+ input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} },
174175 source2 )
175176
176177 assert .Eventually (t ,
@@ -221,7 +222,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
221222
222223 goroutinesChecker := resources .NewGoroutinesChecker ()
223224
224- hg .Start (input.Context {Logger : logp .L (), Cancelation : context . Background () }, source )
225+ hg .Start (input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }, source )
225226
226227 goroutinesChecker .WaitUntilIncreased (1 )
227228 // wait until harvester is started
@@ -241,7 +242,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
241242 t .Run ("assert a harvester for same source cannot be started" , func (t * testing.T ) {
242243 mockHarvester := & mockHarvester {onRun : blockUntilCancelOnRun }
243244 hg := testDefaultHarvesterGroup (t , mockHarvester )
244- inputCtx := input.Context {Logger : logp .L (), Cancelation : context . Background () }
245+ inputCtx := input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }
245246
246247 goroutinesChecker := resources .NewGoroutinesChecker ()
247248 defer goroutinesChecker .WaitUntilOriginalCount ()
@@ -274,7 +275,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
274275
275276 goroutinesChecker := resources .NewGoroutinesChecker ()
276277
277- hg .Start (input.Context {Logger : logp .L (), Cancelation : context . Background () }, source )
278+ hg .Start (input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }, source )
278279
279280 // wait until harvester is stopped
280281 goroutinesChecker .WaitUntilOriginalCount ()
@@ -295,7 +296,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
295296 goroutinesChecker := resources .NewGoroutinesChecker ()
296297 defer goroutinesChecker .WaitUntilOriginalCount ()
297298
298- hg .Start (input.Context {Logger : logp .L (), Cancelation : context . Background () }, source )
299+ hg .Start (input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }, source )
299300
300301 goroutinesChecker .WaitUntilOriginalCount ()
301302
@@ -311,7 +312,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
311312 var wg sync.WaitGroup
312313 mockHarvester := & mockHarvester {onRun : correctOnRun , wg : & wg }
313314 hg := testDefaultHarvesterGroup (t , mockHarvester )
314- inputCtx := input.Context {Logger : logp .L (), Cancelation : context . Background () }
315+ inputCtx := input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }
315316
316317 r , err := lock (inputCtx , hg .store , hg .identifier .ID (source ))
317318 if err != nil {
@@ -346,7 +347,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
346347 mockHarvester := & mockHarvester {onRun : correctOnRun }
347348 hg := testDefaultHarvesterGroup (t , mockHarvester )
348349 hg .tg = task .NewGroup (0 , 50 * time .Millisecond , testLog , "" )
349- inputCtx := input.Context {Logger : logp .L (), Cancelation : context . Background () }
350+ inputCtx := input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }
350351
351352 goroutinesChecker := resources .NewGoroutinesChecker ()
352353 defer goroutinesChecker .WaitUntilOriginalCount ()
@@ -369,7 +370,7 @@ func TestDefaultHarvesterGroup(t *testing.T) {
369370 var wg sync.WaitGroup
370371 mockHarvester := & mockHarvester {onRun : blockUntilCancelOnRun , wg : & wg }
371372 hg := testDefaultHarvesterGroup (t , mockHarvester )
372- inputCtx := input.Context {Logger : logp .L (), Cancelation : context . Background () }
373+ inputCtx := input.Context {Logger : logp .L (), Cancelation : t . Context (), StatusReporter : mockStatusReporter {} }
373374
374375 goroutinesChecker := resources .NewGoroutinesChecker ()
375376 defer goroutinesChecker .WaitUntilOriginalCount ()
@@ -588,10 +589,10 @@ func panicOnRun(_ input.Context, _ Source, _ Cursor, _ Publisher) error {
588589
589590type testLogger strings.Builder
590591
591- func (tl * testLogger ) Errorf (format string , args ... interface {} ) {
592+ func (tl * testLogger ) Errorf (format string , args ... any ) {
592593 sb := (* strings .Builder )(tl )
593- sb . WriteString ( fmt .Sprintf ( format , args ... ) )
594- sb . WriteString ( "\n " )
594+ fmt .Fprintf ( sb , format , args ... )
595+ fmt . Fprint ( sb , "\n " )
595596}
596597
597598func (tl * testLogger ) String () string {
@@ -681,3 +682,9 @@ func (mp *MockPipeline) ConnectWith(config beat.ClientConfig) (beat.Client, erro
681682func (mp * MockPipeline ) Connect () (beat.Client , error ) {
682683 return mp .ConnectWith (beat.ClientConfig {})
683684}
685+
686+ type mockStatusReporter struct {}
687+
688+ // UpdateStatus is a no-op
689+ func (m mockStatusReporter ) UpdateStatus (status status.Status , msg string ) {
690+ }
0 commit comments