use SendContext to ensure no goroutine leaks

grpc · Oct 2, 2024 · 2cf64a5 · 2cf64a5
1 parent 7c9f578
commit 2cf64a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xds/internal/xdsclient/tests/ads_stream_ack_nack_test.go b/xds/internal/xdsclient/tests/ads_stream_ack_nack_test.go
@@ -246,7 +246,7 @@ func (s) TestADS_ACK_NACK_InvalidFirstResponse(t *testing.T) {
 			return nil
 		},
 		OnStreamResponse: func(_ context.Context, _ int64, _ *v3discoverypb.DiscoveryRequest, resp *v3discoverypb.DiscoveryResponse) {
-			streamResponseCh.Send(resp)
+			streamResponseCh.SendContext(ctx, resp)
 		},
 	})
 
@@ -351,7 +351,7 @@ func (s) TestADS_ACK_NACK_ResourceIsNotRequestedAnymore(t *testing.T) {
 			return nil
 		},
 		OnStreamResponse: func(_ context.Context, _ int64, _ *v3discoverypb.DiscoveryRequest, resp *v3discoverypb.DiscoveryResponse) {
-			streamResponseCh.Send(resp)
+			streamResponseCh.SendContext(ctx, resp)
 		},
 	})