@@ -293,7 +293,7 @@ func TestProxyHandle_RequestDeadlineExceeded_GRPC(t *testing.T) {
293293 select {
294294 case <- tunnel .Done ():
295295 t .Log ("Tunnel closed successfully" )
296- case <- time .After (wait .ForeverTestTimeout ):
296+ case <- time .After (framework .ForeverTestTimeout ):
297297 t .Errorf ("Timed out waiting for tunnel to close" )
298298 }
299299 }()
@@ -343,7 +343,7 @@ func TestProxyDial_RequestCancelled_GRPC(t *testing.T) {
343343
344344 select {
345345 case <- tunnel .Done ():
346- case <- time .After (wait .ForeverTestTimeout ):
346+ case <- time .After (framework .ForeverTestTimeout ):
347347 t .Errorf ("Timed out waiting for tunnel to close" )
348348 }
349349 }()
@@ -405,7 +405,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
405405
406406 select {
407407 case <- tunnel .Done ():
408- case <- time .After (wait .ForeverTestTimeout ):
408+ case <- time .After (framework .ForeverTestTimeout ):
409409 t .Errorf ("Timed out waiting for tunnel to close" )
410410 }
411411 }
@@ -427,7 +427,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
427427
428428 // Wait for the closed connections to propogate
429429 var endpointConnsErr , goLeaksErr error
430- wait .PollImmediate (time .Second , wait .ForeverTestTimeout , func () (done bool , err error ) {
430+ wait .PollImmediate (time .Second , framework .ForeverTestTimeout , func () (done bool , err error ) {
431431 endpointConnsErr = a .Metrics ().ExpectAgentEndpointConnections (0 )
432432 goLeaksErr = goleak .Find (ignoredGoRoutines ... )
433433 return endpointConnsErr == nil && goLeaksErr == nil , nil
@@ -489,7 +489,7 @@ func TestProxyDial_AgentTimeout_GRPC(t *testing.T) {
489489
490490 select {
491491 case <- tunnel .Done ():
492- case <- time .After (wait .ForeverTestTimeout ):
492+ case <- time .After (framework .ForeverTestTimeout ):
493493 t .Errorf ("Timed out waiting for tunnel to close" )
494494 }
495495 }()
@@ -965,7 +965,7 @@ func waitForConnectedServerCount(t testing.TB, expectedServerCount int, a framew
965965// agents (backends). This assumes the ProxyServer is using a single ProxyStrategy.
966966func waitForConnectedAgentCount (t testing.TB , expectedAgentCount int , ps framework.ProxyServer ) {
967967 t .Helper ()
968- err := wait .PollImmediate (100 * time .Millisecond , wait .ForeverTestTimeout , func () (bool , error ) {
968+ err := wait .PollImmediate (100 * time .Millisecond , framework .ForeverTestTimeout , func () (bool , error ) {
969969 count , err := ps .ConnectedBackends ()
970970 if err != nil {
971971 return false , err
0 commit comments