@@ -302,7 +302,7 @@ func TestProxyHandle_RequestDeadlineExceeded_GRPC(t *testing.T) {
302302 select {
303303 case <- tunnel .Done ():
304304 t .Log ("Tunnel closed successfully" )
305- case <- time .After (wait .ForeverTestTimeout ):
305+ case <- time .After (framework .ForeverTestTimeout ):
306306 t .Errorf ("Timed out waiting for tunnel to close" )
307307 }
308308 }()
@@ -352,7 +352,7 @@ func TestProxyDial_RequestCancelled_GRPC(t *testing.T) {
352352
353353 select {
354354 case <- tunnel .Done ():
355- case <- time .After (wait .ForeverTestTimeout ):
355+ case <- time .After (framework .ForeverTestTimeout ):
356356 t .Errorf ("Timed out waiting for tunnel to close" )
357357 }
358358 }()
@@ -415,7 +415,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
415415
416416 select {
417417 case <- tunnel .Done ():
418- case <- time .After (wait .ForeverTestTimeout ):
418+ case <- time .After (framework .ForeverTestTimeout ):
419419 t .Errorf ("Timed out waiting for tunnel to close" )
420420 }
421421 }
@@ -441,7 +441,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
441441
442442 // Wait for the closed connections to propogate
443443 var endpointConnsErr , goLeaksErr error
444- wait .PollImmediate (time .Second , wait .ForeverTestTimeout , func () (done bool , err error ) {
444+ wait .PollImmediate (time .Second , framework .ForeverTestTimeout , func () (done bool , err error ) {
445445 endpointConnsErr = a .Metrics ().ExpectAgentEndpointConnections (0 )
446446 goLeaksErr = goleak .Find (ignoredGoRoutines ... )
447447 return endpointConnsErr == nil && goLeaksErr == nil , nil
@@ -503,7 +503,7 @@ func TestProxyDial_AgentTimeout_GRPC(t *testing.T) {
503503
504504 select {
505505 case <- tunnel .Done ():
506- case <- time .After (wait .ForeverTestTimeout ):
506+ case <- time .After (framework .ForeverTestTimeout ):
507507 t .Errorf ("Timed out waiting for tunnel to close" )
508508 }
509509 }()
@@ -979,7 +979,7 @@ func waitForConnectedServerCount(t testing.TB, expectedServerCount int, a framew
979979// agents (backends). This assumes the ProxyServer is using a single ProxyStrategy.
980980func waitForConnectedAgentCount (t testing.TB , expectedAgentCount int , ps framework.ProxyServer ) {
981981 t .Helper ()
982- err := wait .PollImmediate (100 * time .Millisecond , wait .ForeverTestTimeout , func () (bool , error ) {
982+ err := wait .PollImmediate (100 * time .Millisecond , framework .ForeverTestTimeout , func () (bool , error ) {
983983 count , err := ps .ConnectedBackends ()
984984 if err != nil {
985985 return false , err
0 commit comments