@@ -293,7 +293,7 @@ func TestProxyHandle_RequestDeadlineExceeded_GRPC(t *testing.T) {
293
293
select {
294
294
case <- tunnel .Done ():
295
295
t .Log ("Tunnel closed successfully" )
296
- case <- time .After (wait .ForeverTestTimeout ):
296
+ case <- time .After (framework .ForeverTestTimeout ):
297
297
t .Errorf ("Timed out waiting for tunnel to close" )
298
298
}
299
299
}()
@@ -343,7 +343,7 @@ func TestProxyDial_RequestCancelled_GRPC(t *testing.T) {
343
343
344
344
select {
345
345
case <- tunnel .Done ():
346
- case <- time .After (wait .ForeverTestTimeout ):
346
+ case <- time .After (framework .ForeverTestTimeout ):
347
347
t .Errorf ("Timed out waiting for tunnel to close" )
348
348
}
349
349
}()
@@ -405,7 +405,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
405
405
406
406
select {
407
407
case <- tunnel .Done ():
408
- case <- time .After (wait .ForeverTestTimeout ):
408
+ case <- time .After (framework .ForeverTestTimeout ):
409
409
t .Errorf ("Timed out waiting for tunnel to close" )
410
410
}
411
411
}
@@ -427,7 +427,7 @@ func TestProxyDial_RequestCancelled_Concurrent_GRPC(t *testing.T) {
427
427
428
428
// Wait for the closed connections to propogate
429
429
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 ) {
431
431
endpointConnsErr = a .Metrics ().ExpectAgentEndpointConnections (0 )
432
432
goLeaksErr = goleak .Find (ignoredGoRoutines ... )
433
433
return endpointConnsErr == nil && goLeaksErr == nil , nil
@@ -489,7 +489,7 @@ func TestProxyDial_AgentTimeout_GRPC(t *testing.T) {
489
489
490
490
select {
491
491
case <- tunnel .Done ():
492
- case <- time .After (wait .ForeverTestTimeout ):
492
+ case <- time .After (framework .ForeverTestTimeout ):
493
493
t .Errorf ("Timed out waiting for tunnel to close" )
494
494
}
495
495
}()
@@ -965,7 +965,7 @@ func waitForConnectedServerCount(t testing.TB, expectedServerCount int, a framew
965
965
// agents (backends). This assumes the ProxyServer is using a single ProxyStrategy.
966
966
func waitForConnectedAgentCount (t testing.TB , expectedAgentCount int , ps framework.ProxyServer ) {
967
967
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 ) {
969
969
count , err := ps .ConnectedBackends ()
970
970
if err != nil {
971
971
return false , err
0 commit comments