diff --git a/aibridge/internal/integrationtest/circuit_breaker_internal_test.go b/aibridge/internal/integrationtest/circuit_breaker_internal_test.go index 1f5ddd71f3..990382c73e 100644 --- a/aibridge/internal/integrationtest/circuit_breaker_internal_test.go +++ b/aibridge/internal/integrationtest/circuit_breaker_internal_test.go @@ -20,6 +20,7 @@ import ( "github.com/coder/coder/v2/aibridge/internal/testutil" "github.com/coder/coder/v2/aibridge/metrics" "github.com/coder/coder/v2/aibridge/provider" + "github.com/coder/coder/v2/testutil" ) // Common response bodies for circuit breaker tests. @@ -126,7 +127,7 @@ func TestCircuitBreaker_FullRecoveryCycle(t *testing.T) { cbConfig := &config.CircuitBreaker{ FailureThreshold: 2, Interval: time.Minute, - Timeout: 50 * time.Millisecond, + Timeout: testutil.IntervalMedium, MaxRequests: 1, } @@ -283,7 +284,7 @@ func TestCircuitBreaker_HalfOpenFailure(t *testing.T) { cbConfig := &config.CircuitBreaker{ FailureThreshold: 2, Interval: time.Minute, - Timeout: 50 * time.Millisecond, + Timeout: testutil.IntervalMedium, MaxRequests: 1, } @@ -431,7 +432,7 @@ func TestCircuitBreaker_HalfOpenMaxRequests(t *testing.T) { cbConfig := &config.CircuitBreaker{ FailureThreshold: 2, Interval: time.Minute, - Timeout: 50 * time.Millisecond, + Timeout: testutil.IntervalMedium, MaxRequests: maxRequests, // Allow only 2 concurrent requests in half-open }