//! Incoming rend accepts: 30 / 60s, burst 10.

use std::time::{Duration, Instant};

use onionwire::ratelimit::TokenBucket;

#[test]
fn burst_allows_then_denies() {
    let t0 = Instant::now();
    let mut b = TokenBucket::new(30, Duration::from_secs(60), 10);
    for _ in 0..10 {
        assert!(b.try_acquire_at(t0), "burst of 10 must pass");
    }
    assert!(!b.try_acquire_at(t0), "11th in the burst must drop");
}

#[test]
fn refill_one_token_after_two_seconds() {
    let t0 = Instant::now();
    let mut b = TokenBucket::new(30, Duration::from_secs(60), 10);
    for _ in 0..10 {
        assert!(b.try_acquire_at(t0));
    }
    // 30 tokens / 60s = 0.5/s → 2s yields one token.
    let t1 = t0 + Duration::from_secs(2);
    assert!(b.try_acquire_at(t1));
    assert!(!b.try_acquire_at(t1));
}