@@ -138,7 +138,7 @@ func makeMockStream(t *testing.T) *mockStream {
138138 t .Helper ()
139139 return & mockStream {
140140 t : t ,
141- ctx : context . Background (),
141+ ctx : t . Context (),
142142 sent : make (chan * discovery.DiscoveryResponse , 10 ),
143143 recv : make (chan * discovery.DiscoveryRequest , 10 ),
144144 }
@@ -255,7 +255,7 @@ func TestServerShutdown(t *testing.T) {
255255 config := makeMockConfigWatcher ()
256256 config .responses = makeResponses ()
257257 shutdown := make (chan bool )
258- ctx , cancel := context .WithCancel (context . Background ())
258+ ctx , cancel := context .WithCancel (t . Context ()) //nolint:gosec // G118
259259 s := server .NewServer (ctx , config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
260260
261261 // make a request
@@ -304,7 +304,7 @@ func TestResponseHandlers(t *testing.T) {
304304 for _ , typ := range testTypes {
305305 t .Run (typ , func (t * testing.T ) {
306306 done := make (chan struct {})
307- ctx , cancel := context .WithCancel (context . Background ())
307+ ctx , cancel := context .WithCancel (t . Context ())
308308
309309 config := makeMockConfigWatcher ()
310310 config .responses = makeResponses ()
@@ -383,88 +383,88 @@ func TestFetch(t *testing.T) {
383383 },
384384 }
385385
386- s := server .NewServer (context . Background (), config , cb , sotw .WithLogger (log .NewTestLogger (t )))
387- out , err := s .FetchEndpoints (context . Background (), & discovery.DiscoveryRequest {Node : node })
386+ s := server .NewServer (t . Context (), config , cb , sotw .WithLogger (log .NewTestLogger (t )))
387+ out , err := s .FetchEndpoints (t . Context (), & discovery.DiscoveryRequest {Node : node })
388388 assert .NotNil (t , out )
389389 require .NoError (t , err )
390390
391- out , err = s .FetchClusters (context . Background (), & discovery.DiscoveryRequest {Node : node })
391+ out , err = s .FetchClusters (t . Context (), & discovery.DiscoveryRequest {Node : node })
392392 assert .NotNil (t , out )
393393 require .NoError (t , err )
394394
395- out , err = s .FetchRoutes (context . Background (), & discovery.DiscoveryRequest {Node : node })
395+ out , err = s .FetchRoutes (t . Context (), & discovery.DiscoveryRequest {Node : node })
396396 assert .NotNil (t , out )
397397 require .NoError (t , err )
398398
399- out , err = s .FetchListeners (context . Background (), & discovery.DiscoveryRequest {Node : node })
399+ out , err = s .FetchListeners (t . Context (), & discovery.DiscoveryRequest {Node : node })
400400 assert .NotNil (t , out )
401401 require .NoError (t , err )
402402
403- out , err = s .FetchSecrets (context . Background (), & discovery.DiscoveryRequest {Node : node })
403+ out , err = s .FetchSecrets (t . Context (), & discovery.DiscoveryRequest {Node : node })
404404 assert .NotNil (t , out )
405405 require .NoError (t , err )
406406
407- out , err = s .FetchRuntime (context . Background (), & discovery.DiscoveryRequest {Node : node })
407+ out , err = s .FetchRuntime (t . Context (), & discovery.DiscoveryRequest {Node : node })
408408 assert .NotNil (t , out )
409409 require .NoError (t , err )
410410
411411 // try again and expect empty results
412- out , err = s .FetchEndpoints (context . Background (), & discovery.DiscoveryRequest {Node : node })
412+ out , err = s .FetchEndpoints (t . Context (), & discovery.DiscoveryRequest {Node : node })
413413 assert .Nil (t , out )
414414 require .Error (t , err )
415415
416- out , err = s .FetchClusters (context . Background (), & discovery.DiscoveryRequest {Node : node })
416+ out , err = s .FetchClusters (t . Context (), & discovery.DiscoveryRequest {Node : node })
417417 assert .Nil (t , out )
418418 require .Error (t , err )
419419
420- out , err = s .FetchRoutes (context . Background (), & discovery.DiscoveryRequest {Node : node })
420+ out , err = s .FetchRoutes (t . Context (), & discovery.DiscoveryRequest {Node : node })
421421 assert .Nil (t , out )
422422 require .Error (t , err )
423423
424- out , err = s .FetchListeners (context . Background (), & discovery.DiscoveryRequest {Node : node })
424+ out , err = s .FetchListeners (t . Context (), & discovery.DiscoveryRequest {Node : node })
425425 assert .Nil (t , out )
426426 require .Error (t , err )
427427
428428 // try empty requests: not valid in a real gRPC server
429- out , err = s .FetchEndpoints (context . Background (), nil )
429+ out , err = s .FetchEndpoints (t . Context (), nil )
430430 assert .Nil (t , out )
431431 require .Error (t , err )
432432
433- out , err = s .FetchClusters (context . Background (), nil )
433+ out , err = s .FetchClusters (t . Context (), nil )
434434 assert .Nil (t , out )
435435 require .Error (t , err )
436436
437- out , err = s .FetchRoutes (context . Background (), nil )
437+ out , err = s .FetchRoutes (t . Context (), nil )
438438 assert .Nil (t , out )
439439 require .Error (t , err )
440440
441- out , err = s .FetchListeners (context . Background (), nil )
441+ out , err = s .FetchListeners (t . Context (), nil )
442442 assert .Nil (t , out )
443443 require .Error (t , err )
444444
445- out , err = s .FetchSecrets (context . Background (), nil )
445+ out , err = s .FetchSecrets (t . Context (), nil )
446446 assert .Nil (t , out )
447447 require .Error (t , err )
448448
449- out , err = s .FetchRuntime (context . Background (), nil )
449+ out , err = s .FetchRuntime (t . Context (), nil )
450450 assert .Nil (t , out )
451451 require .Error (t , err )
452452
453453 // send error from callback
454454 callbackError = true
455- out , err = s .FetchEndpoints (context . Background (), nil )
455+ out , err = s .FetchEndpoints (t . Context (), nil )
456456 assert .Nil (t , out )
457457 require .Error (t , err )
458458
459- out , err = s .FetchClusters (context . Background (), nil )
459+ out , err = s .FetchClusters (t . Context (), nil )
460460 assert .Nil (t , out )
461461 require .Error (t , err )
462462
463- out , err = s .FetchRoutes (context . Background (), nil )
463+ out , err = s .FetchRoutes (t . Context (), nil )
464464 assert .Nil (t , out )
465465 require .Error (t , err )
466466
467- out , err = s .FetchListeners (context . Background (), nil )
467+ out , err = s .FetchListeners (t . Context (), nil )
468468 assert .Nil (t , out )
469469 require .Error (t , err )
470470
@@ -478,7 +478,7 @@ func TestSendError(t *testing.T) {
478478 t .Run (typ , func (t * testing.T ) {
479479 config := makeMockConfigWatcher ()
480480 config .responses = makeResponses ()
481- s := server .NewServer (context . Background (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
481+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
482482
483483 // make a request
484484 resp := makeMockStream (t )
@@ -502,7 +502,7 @@ func TestStaleNonce(t *testing.T) {
502502 t .Run (typ , func (t * testing.T ) {
503503 config := makeMockConfigWatcher ()
504504 config .responses = makeResponses ()
505- s := server .NewServer (context . Background (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
505+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
506506
507507 resp := makeMockStream (t )
508508 resp .recv <- & discovery.DiscoveryRequest {
@@ -577,7 +577,7 @@ func TestAggregatedHandlers(t *testing.T) {
577577
578578 // We create the server with the optional ordered ADS flag so we guarantee resource
579579 // ordering over the stream.
580- s := server .NewServer (context . Background (), config , server.CallbackFuncs {}, sotw .WithOrderedADS (), sotw .WithLogger (log .NewTestLogger (t )))
580+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {}, sotw .WithOrderedADS (), sotw .WithLogger (log .NewTestLogger (t )))
581581 go func () {
582582 err := s .StreamAggregatedResources (resp )
583583 require .NoError (t , err )
@@ -612,7 +612,7 @@ func TestAggregatedHandlers(t *testing.T) {
612612
613613func TestAggregateRequestType (t * testing.T ) {
614614 config := makeMockConfigWatcher ()
615- s := server .NewServer (context . Background (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
615+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
616616 resp := makeMockStream (t )
617617 resp .recv <- & discovery.DiscoveryRequest {Node : node }
618618 err := s .StreamAggregatedResources (resp )
@@ -629,7 +629,7 @@ func TestCancellations(t *testing.T) {
629629 }
630630 }
631631 close (resp .recv )
632- s := server .NewServer (context . Background (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
632+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
633633 err := s .StreamAggregatedResources (resp )
634634 require .NoError (t , err )
635635 assert .Equal (t , 0 , config .watches )
@@ -647,7 +647,7 @@ func TestOpaqueRequestsChannelMuxing(t *testing.T) {
647647 }
648648 }
649649 close (resp .recv )
650- s := server .NewServer (context . Background (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
650+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {}, sotw .WithLogger (log .NewTestLogger (t )))
651651 err := s .StreamAggregatedResources (resp )
652652 require .NoError (t , err )
653653 assert .Equal (t , 0 , config .watches )
@@ -665,7 +665,7 @@ func TestNilPropagationOverResponseChannelShouldCloseTheStream(t *testing.T) {
665665 }
666666 }
667667 close (resp .recv )
668- s := server .NewServer (context . Background (), config , server.CallbackFuncs {})
668+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {})
669669 err := s .StreamAggregatedResources (resp )
670670 require .Error (t , err )
671671 assert .Equal (t , 0 , config .watches )
@@ -677,7 +677,7 @@ func TestCallbackError(t *testing.T) {
677677 config := makeMockConfigWatcher ()
678678 config .responses = makeResponses ()
679679
680- s := server .NewServer (context . Background (), config , server.CallbackFuncs {
680+ s := server .NewServer (t . Context (), config , server.CallbackFuncs {
681681 StreamOpenFunc : func (context.Context , int64 , string ) error {
682682 return errors .New ("stream open error" )
683683 },
0 commit comments