trueno 0.17.3

use super::super::super::super::*;

/// F234: InferencePhase tests
#[test]
fn test_f234_inference_phase() {
    // All variants
    assert!(!format!("{:?}", InferencePhase::Prefill).is_empty());
    assert!(!format!("{:?}", InferencePhase::Decode).is_empty());

    // PartialEq
    assert_eq!(InferencePhase::Prefill, InferencePhase::Prefill);
    assert_ne!(InferencePhase::Prefill, InferencePhase::Decode);

    // Clone
    let phase = InferencePhase::Prefill;
    let cloned = phase;
    assert_eq!(phase, cloned);

    // Default
    let phase = InferencePhase::default();
    assert_eq!(phase, InferencePhase::Prefill);
}

/// F235: CircuitBreaker comprehensive tests
#[test]
fn test_f235_circuit_breaker_comprehensive() {
    use std::time::Duration;

    let mut breaker = CircuitBreaker::new(2, Duration::from_millis(50));

    // Initial state - closed
    assert_eq!(breaker.state(), CircuitState::Closed);
    assert!(breaker.allow_request());

    // Record failures to open
    breaker.record_failure();
    assert_eq!(breaker.state(), CircuitState::Closed);
    breaker.record_failure();
    assert_eq!(breaker.state(), CircuitState::Open);
    assert!(!breaker.allow_request());

    // Wait for half-open transition
    std::thread::sleep(Duration::from_millis(60));
    // allow_request triggers the state transition
    assert!(breaker.allow_request()); // This transitions to HalfOpen
    assert_eq!(breaker.state(), CircuitState::HalfOpen);

    // Success closes it
    breaker.record_success();
    assert_eq!(breaker.state(), CircuitState::Closed);

    // Reset
    breaker.record_failure();
    breaker.record_failure();
    breaker.reset();
    assert_eq!(breaker.state(), CircuitState::Closed);

    // Default trait
    let breaker = CircuitBreaker::default();
    assert_eq!(breaker.state(), CircuitState::Closed);
}

/// F236: ManagedConnection tests
#[test]
fn test_f236_managed_connection() {
    use std::time::Duration;

    let mut conn = ManagedConnection::new("test", Duration::from_secs(60), Duration::from_secs(30));

    // Initial state
    assert!(conn.is_valid());
    assert!(!conn.is_expired());
    assert!(!conn.is_idle());

    // Access inner
    assert_eq!(*conn.inner(), "test");
    *conn.inner_mut() = "modified";
    assert_eq!(*conn.inner(), "modified");

    // Touch updates idle time
    conn.touch();

    // Health tracking
    conn.record_health_failure();
    conn.reset_health();

    // Age and idle time
    let _age = conn.age();
    let _idle = conn.idle_time();

    // Into inner
    let inner = conn.into_inner();
    assert_eq!(inner, "modified");
}

/// F237: BoundedQueue comprehensive tests
#[test]
fn test_f237_bounded_queue_comprehensive() {
    let mut queue: BoundedQueue<i32> = BoundedQueue::new(3);

    // Initial state
    assert!(queue.is_empty());
    assert!(!queue.is_full());
    assert_eq!(queue.capacity(), 3);
    assert_eq!(queue.remaining(), 3);

    // Push items
    assert!(queue.try_push(1).is_ok());
    assert!(queue.try_push(2).is_ok());
    assert_eq!(queue.len(), 2);
    assert_eq!(queue.remaining(), 1);

    // Peek
    assert_eq!(queue.peek(), Some(&1));

    // Fill queue
    assert!(queue.try_push(3).is_ok());
    assert!(queue.is_full());

    // Push to full queue fails
    assert_eq!(queue.try_push(4), Err(4));

    // Pop
    assert_eq!(queue.pop(), Some(1));
    assert!(!queue.is_full());

    // Clear
    queue.clear();
    assert!(queue.is_empty());

    // Default trait
    let queue: BoundedQueue<i32> = BoundedQueue::default();
    assert!(queue.is_empty());
}

/// F238: StrategicBuffer tests
#[test]
fn test_f238_strategic_buffer() {
    // With strategy
    let mut buffer = StrategicBuffer::new(ReserveStrategy::Double);
    buffer.write(&[1, 2, 3]);
    assert_eq!(buffer.len(), 3);
    assert!(!buffer.is_empty());
    assert_eq!(buffer.as_slice(), &[1, 2, 3]);
    assert!(buffer.capacity() >= 3);

    // Reserve
    buffer.reserve(100);
    assert!(buffer.capacity() >= 103);

    // Clear
    buffer.clear();
    assert!(buffer.is_empty());

    // With capacity
    let buffer = StrategicBuffer::with_capacity(100, ReserveStrategy::Grow50);
    assert!(buffer.capacity() >= 100);

    // Default trait
    let buffer = StrategicBuffer::default();
    assert!(buffer.is_empty());

    // Different strategies
    let _buffer = StrategicBuffer::new(ReserveStrategy::Exact);
    let _buffer = StrategicBuffer::new(ReserveStrategy::PowerOfTwo);
}

/// F239: GraphReuseCounter tests
#[test]
fn test_f239_graph_reuse_counter() {
    let mut counter = GraphReuseCounter::new(5);

    // Initial state
    assert!(!counter.is_hot());
    assert!(!counter.should_cache());
    assert_eq!(counter.count(), 0);

    // Record uses
    counter.record_use();
    counter.record_use();
    counter.record_use();
    assert!(!counter.is_hot());
    assert_eq!(counter.count(), 3);

    // Reach hot threshold
    counter.record_use();
    counter.record_use();
    assert!(counter.is_hot());
    assert!(counter.should_cache());

    // Reset
    counter.reset();
    assert!(!counter.is_hot());
    assert_eq!(counter.count(), 0);
}

/// F240: KvCacheSlot and KvCacheManager
#[test]
fn test_f240_kv_cache() {
    // Create cache manager
    let mut mgr = KvCacheManager::new(3);
    assert_eq!(mgr.capacity(), 3);
    assert_eq!(mgr.valid_count(), 0);

    // Allocate slots
    let idx0 = mgr.allocate(0, 100, 0, 0).unwrap();
    let idx1 = mgr.allocate(1, 101, 0, 0).unwrap();
    assert_eq!(mgr.valid_count(), 2);

    // Access
    let slot = mgr.access(idx0).unwrap();
    assert_eq!(slot.token_id, 100);

    // Step advances global step
    mgr.step();

    // Evict LRU
    let _evicted = mgr.evict_lru();

    // Access
    assert!(mgr.access(idx1).is_some());
}

/// F241: SequentialBatchOrderer tests
#[test]
fn test_f241_sequential_batch_orderer() {
    // Forward order
    let mut orderer = SequentialBatchOrderer::new(3);
    assert!(!orderer.is_done());
    assert_eq!(orderer.remaining(), 3);

    assert_eq!(orderer.next_batch(), Some(0));
    assert_eq!(orderer.next_batch(), Some(1));
    assert_eq!(orderer.next_batch(), Some(2));
    assert_eq!(orderer.next_batch(), None);
    assert!(orderer.is_done());

    // Reset
    orderer.reset();
    assert!(!orderer.is_done());
    assert_eq!(orderer.remaining(), 3);

    // Reversed order
    let mut orderer = SequentialBatchOrderer::reversed(3);
    assert_eq!(orderer.next_batch(), Some(2));
    assert_eq!(orderer.next_batch(), Some(1));
    assert_eq!(orderer.next_batch(), Some(0));

    // Interleaved order
    let mut orderer = SequentialBatchOrderer::interleaved(4);
    let batches: Vec<_> = orderer.by_ref().collect();
    assert_eq!(batches.len(), 4);

    // Iterator trait
    let orderer = SequentialBatchOrderer::new(3);
    let batches: Vec<_> = orderer.collect();
    assert_eq!(batches, vec![0, 1, 2]);
}

/// F242: reserve_capacity and ReserveStrategy
#[test]
fn test_f242_reserve_capacity() {
    // Exact strategy
    assert_eq!(reserve_capacity(10, ReserveStrategy::Exact), 10);

    // Grow50 strategy - adds 50% headroom
    let cap = reserve_capacity(10, ReserveStrategy::Grow50);
    assert!(cap >= 15); // 10 + 50%

    // Double strategy
    let cap = reserve_capacity(10, ReserveStrategy::Double);
    assert_eq!(cap, 20);

    // PowerOfTwo strategy
    let cap = reserve_capacity(10, ReserveStrategy::PowerOfTwo);
    assert_eq!(cap, 16); // next power of two
}

/// F243: ServeLimits tests
#[test]
fn test_f243_serve_limits() {
    // Default limits
    let limits = ServeLimits::default();
    assert!(limits.max_request_size > 0);
    assert!(limits.max_headers > 0);
    assert!(limits.max_header_size > 0);
    assert!(limits.max_pipelined > 0);
    assert!(limits.max_connections > 0);

    // Custom limits
    let limits = ServeLimits {
        max_request_size: 1024,
        max_headers: 10,
        max_header_size: 4096,
        keep_alive_timeout: std::time::Duration::from_secs(30),
        client_timeout: std::time::Duration::from_secs(60),
        max_pipelined: 5,
        max_connections: 100,
    };
    assert_eq!(limits.max_request_size, 1024);
}

/// F244: LimitError Display
#[test]
fn test_f244_limit_error_display() {
    let err = LimitError::BodyTooLarge { size: 2000, max: 1000 };
    let msg = format!("{}", err);
    assert!(msg.contains("2000"));

    let err = LimitError::TooManyHeaders { count: 50, max: 10 };
    let msg = format!("{}", err);
    assert!(msg.contains("50"));

    let err = LimitError::ConnectionLimitReached { current: 200, max: 100 };
    let msg = format!("{}", err);
    assert!(msg.contains("200"));

    let err = LimitError::HeaderTooLarge { size: 5000, max: 1000 };
    let msg = format!("{}", err);
    assert!(msg.contains("5000"));

    let err = LimitError::TooManyPipelined { count: 20, max: 10 };
    let msg = format!("{}", err);
    assert!(msg.contains("20"));
}

/// F245: GracefulShutdown tests
#[test]
fn test_f245_graceful_shutdown() {
    use std::time::Duration;

    let shutdown = GracefulShutdown::new(Duration::from_millis(100));

    // Initial state
    assert!(!shutdown.is_shutdown_requested());
    assert_eq!(shutdown.active_count(), 0);

    // Register guard
    let guard = shutdown.register();
    assert!(guard.is_some());
    assert_eq!(shutdown.active_count(), 1);
    drop(guard);
    assert_eq!(shutdown.active_count(), 0);

    // Shutdown
    let result = shutdown.shutdown();
    assert!(matches!(result, ShutdownResult::Clean));
    assert!(shutdown.is_shutdown_requested());

    // Can't register after shutdown
    let guard = shutdown.register();
    assert!(guard.is_none());

    // Reset
    shutdown.reset();
    assert!(!shutdown.is_shutdown_requested());

    // Default trait
    let shutdown = GracefulShutdown::default();
    assert!(!shutdown.is_shutdown_requested());
}

/// F246: ResourcePool tests
#[test]
fn test_f246_resource_pool() {
    let pool: ResourcePool<i32> = ResourcePool::new(3, || 42);

    // Initial state
    assert_eq!(pool.max_resources(), 3);
    assert_eq!(pool.available(), 3);

    // Acquire resource
    let resource = pool.try_acquire();
    assert!(resource.is_some());
    assert_eq!(pool.available(), 2);

    // Use resource via Deref
    let mut resource = resource.unwrap();
    assert_eq!(*resource, 42);
    *resource = 100;
    assert_eq!(*resource, 100);

    // Drop returns to pool
    drop(resource);
    assert_eq!(pool.available(), 3);

    // Debug trait
    let pool: ResourcePool<i32> = ResourcePool::new(2, || 0);
    let debug = format!("{:?}", pool);
    assert!(debug.contains("ResourcePool"));
}