Function test_circuit_breaker_functionality

tests/rate_limiter_test.rs:53–95

()

#[test]
fn test_circuit_breaker_functionality() {
    let config = RateLimitConfig {
        max_requests: 1000, // High limit to focus on circuit breaker
        window_duration: Duration::from_secs(60),
        per_ip_limiting: false,
        max_tracked_ips: 100,
        cleanup_interval: Duration::from_secs(60),
    };

    let circuit_config = CircuitBreakerConfig {
        failure_threshold: 3,
        timeout_duration: Duration::from_millis(100),
        success_threshold: 2,
        enabled: true,
    };

    let limiter = RateLimiter::with_config(config, circuit_config);

    // Initially circuit should be closed
    assert_eq!(limiter.get_circuit_state(), CircuitState::Closed);

    // Record failures to open the circuit
    for _ in 0..3 {
        limiter.record_failure();
    }

    // Circuit should now be open
    assert_eq!(limiter.get_circuit_state(), CircuitState::Open);

    // Requests should be rejected
    assert!(limiter.check_request(None).is_err());

    // Wait for timeout
    std::thread::sleep(Duration::from_millis(150));

    // Should transition to half-open on next request
    assert!(limiter.check_request(None).is_ok());
    assert_eq!(limiter.get_circuit_state(), CircuitState::HalfOpen);

    // Record success to close circuit
    assert!(limiter.check_request(None).is_ok());
    assert_eq!(limiter.get_circuit_state(), CircuitState::Closed);
}

#[test]
fn test_per_ip_isolation() {