@@ -251,7 +251,12 @@ func TestProxyHandle_RequestDeadlineExceeded_GRPC(t *testing.T) {
251
251
252
252
func () {
253
253
// Ensure that tunnels aren't leaked with long-running servers.
254
- defer goleak .VerifyNone (t , goleak .IgnoreCurrent (), goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ))
254
+ ignoredGoRoutines := []goleak.Option {
255
+ goleak .IgnoreCurrent (),
256
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ),
257
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransportAndUnlock" ),
258
+ }
259
+ defer goleak .VerifyNone (t , ignoredGoRoutines ... )
255
260
256
261
// run test client
257
262
tunnel , err := createSingleUseGrpcTunnel (context .Background (), ps .FrontAddr ())
@@ -309,7 +314,12 @@ func TestProxyDial_RequestCancelled_GRPC(t *testing.T) {
309
314
310
315
func () {
311
316
// Ensure that tunnels aren't leaked with long-running servers.
312
- defer goleak .VerifyNone (t , goleak .IgnoreCurrent (), goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ))
317
+ ignoredGoRoutines := []goleak.Option {
318
+ goleak .IgnoreCurrent (),
319
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ),
320
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransportAndUnlock" ),
321
+ }
322
+ defer goleak .VerifyNone (t , ignoredGoRoutines ... )
313
323
314
324
// run test client
315
325
tunnel , err := createSingleUseGrpcTunnel (context .Background (), ps .FrontAddr ())
@@ -404,6 +414,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
404
414
ignoredGoRoutines := []goleak.Option {
405
415
goleak .IgnoreCurrent (),
406
416
goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ),
417
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransportAndUnlock" ),
407
418
}
408
419
409
420
const concurrentConns = 50
@@ -444,7 +455,12 @@ func TestProxyDial_AgentTimeout_GRPC(t *testing.T) {
444
455
445
456
func () {
446
457
// Ensure that tunnels aren't leaked with long-running servers.
447
- defer goleak .VerifyNone (t , goleak .IgnoreCurrent (), goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ))
458
+ ignoredGoRoutines := []goleak.Option {
459
+ goleak .IgnoreCurrent (),
460
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ),
461
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransportAndUnlock" ),
462
+ }
463
+ defer goleak .VerifyNone (t , ignoredGoRoutines ... )
448
464
449
465
// run test client
450
466
tunnel , err := createSingleUseGrpcTunnel (context .Background (), ps .FrontAddr ())
@@ -975,9 +991,13 @@ func resetAllMetrics() {
975
991
976
992
func expectCleanShutdown (t testing.TB ) {
977
993
resetAllMetrics ()
978
- currentGoRoutines := goleak .IgnoreCurrent ()
994
+ ignoredGoRoutines := []goleak.Option {
995
+ goleak .IgnoreCurrent (),
996
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransport" ),
997
+ goleak .IgnoreTopFunction ("google.golang.org/grpc.(*addrConn).resetTransportAndUnlock" ),
998
+ }
979
999
t .Cleanup (func () {
980
- goleak .VerifyNone (t , currentGoRoutines , goleak . IgnoreTopFunction ( "google.golang.org/grpc.(*addrConn).resetTransport" ) )
1000
+ goleak .VerifyNone (t , ignoredGoRoutines ... )
981
1001
if err := clientmetricstest .ExpectClientDialFailures (nil ); err != nil {
982
1002
t .Errorf ("Unexpected %s metric: %v" , "dial_failure_total" , err )
983
1003
}
0 commit comments