throttle-ro 0.1.2

A configurable throttling service for rate limiting IP-based requests
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use cache_ro::{Cache, CacheConfig};
use std::thread::sleep;
use std::time::Duration;
use throttle_ro::ThrottlesService;

#[test]
fn test_all() {
    test_initial_can_go_is_true();
    test_hit_increments_value();
    test_can_go_blocks_after_max_attempts();
    test_remove_clears_cache();
    test_expire_returns_default_when_none_set();
    test_expire_returns_custom_when_set()
}
fn test_initial_can_go_is_true() {
    let ip = "127.0.0.1".to_string();
    let cache = Cache::new(CacheConfig {
        persistent: false,
        ..Default::default()
    })
    .unwrap();
    let mut service = ThrottlesService::new(ip, 3, Duration::from_secs(60), "test_");

    assert!(service.can_go(&cache));
    Cache::drop()
}


fn test_hit_increments_value() {
    let ip = "127.0.0.2".to_string();
    let cache = Cache::new(CacheConfig {
        persistent: false,
        ..Default::default()
    })
    .unwrap();
    let mut service = ThrottlesService::new(ip.clone(), 3, Duration::from_secs(60), "test_");

    service.hit(&cache);
    assert_eq!(cache.get::<u32>(&service.key()), Some(1));
    let mut service = ThrottlesService::new(ip, 3, Duration::from_secs(60), "test_");

    service.hit(&cache);
    assert_eq!(cache.get::<u32>(&service.key()), Some(2));
    Cache::drop()
}

fn test_can_go_blocks_after_max_attempts() {
    let ip = "127.0.0.3".to_string();
    let cache = Cache::new(CacheConfig {
        persistent: false,
        ..Default::default()
    })
    .unwrap();
    let mut service = ThrottlesService::new(ip, 3, Duration::from_secs(60), "test_");

    service.hit(&cache);
    service.hit(&cache);
    service.hit(&cache);
    assert!(!service.can_go(&cache));
    Cache::drop()
}


fn test_remove_clears_cache() {
    let ip = "127.0.0.4".to_string();
    let cache = Cache::new(CacheConfig {
        persistent: false,
        ..Default::default()
    })
    .unwrap();
    let mut service = ThrottlesService::new(ip, 3, Duration::from_secs(60), "test_");

    service.hit(&cache);
    assert_eq!(cache.get::<u32>(&service.key()), Some(1));

    service.remove(&cache);
    assert_eq!(cache.get::<i32>(&service.key()), None);
    Cache::drop()
}

fn test_expire_returns_default_when_none_set() {
    let ip = "127.0.0.5".to_string();
    let cache = Cache::new(CacheConfig {
        persistent: false,
        ..Default::default()
    })
    .unwrap();
    let mut service = ThrottlesService::new(ip, 3, Duration::from_secs(60), "test_");

    let expire = service.get_expire(&cache);
    assert_eq!(expire, Duration::from_secs(60));
    Cache::drop()
}

fn test_expire_returns_custom_when_set() {
    let ip = "127.0.0.6".to_string();
    let cache = Cache::new(CacheConfig {
        persistent: false,
        ..Default::default()
    })
    .unwrap();
    cache.clear().unwrap();
    let mut service = ThrottlesService::new(ip, 2, Duration::from_secs(1), "test_");
    service.hit(&cache);
    service.hit(&cache);
    sleep(Duration::from_secs(1));
    assert!(service.can_go(&cache));
    Cache::drop()
}