tiny-counter 0.1.0

Track event counts across time windows with fixed memory and fast queries
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

//! Tests for .cooldown() rate limiting constraint

use chrono::Duration;
use std::sync::Arc;
use tiny_counter::{EventStore, TestClock};

#[test]
fn cooldown_enforces_wait_period() {
    let store = EventStore::new();

    // First call should succeed
    let result = store
        .limit()
        .cooldown("action", Duration::hours(1))
        .check_and_record("action");
    assert!(result.is_ok());

    // Immediate second call should fail (still in cooldown)
    let result = store
        .limit()
        .cooldown("action", Duration::hours(1))
        .check_and_record("action");
    assert!(result.is_err());
}

#[test]
fn cooldown_allows_after_duration() {
    let fixed_time = chrono::Utc::now();
    let clock = TestClock::build_for_testing_at(fixed_time);

    let store = EventStore::builder()
        .with_clock(Arc::new(clock.clone()))
        .build()
        .unwrap();

    store.record("action");

    // Advance clock past cooldown
    clock.advance(Duration::hours(2));

    // Should now be allowed
    let result = store
        .limit()
        .cooldown("action", Duration::hours(1))
        .check_and_record("action");
    assert!(result.is_ok());
}