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
48
49
50
51

//! Tests for .check() vs .check_and_record() methods

use tiny_counter::{EventStore, TimeUnit};

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

    let result = store
        .limit()
        .at_most("action", 10, TimeUnit::Hours)
        .check("action");
    assert!(result.is_ok());

    // Event should not be recorded
    assert_eq!(store.query("action").last_hours(1).sum(), None);
}

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

    let result = store
        .limit()
        .at_most("action", 10, TimeUnit::Hours)
        .check_and_record("action");
    assert!(result.is_ok());

    // Event should be recorded
    assert_eq!(store.query("action").last_hours(1).sum(), Some(1));
}

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

    // Fill to limit
    for _ in 0..10 {
        store.record("action");
    }

    // This should fail and not record
    let result = store
        .limit()
        .at_most("action", 10, TimeUnit::Hours)
        .check_and_record("action");
    assert!(result.is_err());

    // Count should remain at limit
    assert_eq!(store.query("action").last_hours(1).sum(), Some(10));
}