lesbian.skin
this repo has no description
🧪 Update unit tests
+39
-32
+1
-4
test/glimit_rate_limiter_test.gleam
+1
-4
test/glimit_rate_limiter_test.gleam
···
2
2
import glimit/rate_limiter
3
3
4
4
pub fn rate_limiter_test() {
5
-
let limiter = case rate_limiter.new(2, 2) {
6
-
Ok(limiter) -> limiter
7
-
Error(_) -> panic as "Should be able to create rate limiter"
8
-
}
5
+
let assert Ok(limiter) = rate_limiter.new(2, 2)
9
6
10
7
limiter
11
8
|> rate_limiter.hit
+24
-28
test/glimit_registry_test.gleam
+24
-28
test/glimit_registry_test.gleam
···
4
4
import glimit/registry
5
5
6
6
pub fn same_id_same_actor_test() {
7
-
let registry = case registry.new(2, 2) {
8
-
Ok(registry) -> registry
9
-
Error(_) -> {
10
-
panic as "Should be able to create a new registry"
11
-
}
12
-
}
13
-
7
+
let assert Ok(registry) = registry.new(2, 2)
14
8
let assert Ok(rate_limiter) = registry |> registry.get_or_create("🚀")
15
9
let assert Ok(same_rate_limiter) = registry |> registry.get_or_create("🚀")
16
10
···
19
13
}
20
14
21
15
pub fn other_id_other_actor_test() {
22
-
let registry = case registry.new(2, 2) {
23
-
Ok(registry) -> registry
24
-
Error(_) -> {
25
-
panic as "Should be able to create a new registry"
26
-
}
27
-
}
28
-
16
+
let assert Ok(registry) = registry.new(2, 2)
29
17
let assert Ok(rate_limiter) = registry |> registry.get_or_create("🚀")
30
18
let assert Ok(same_rate_limiter) = registry |> registry.get_or_create("💫")
31
19
···
34
22
}
35
23
36
24
pub fn sweep_full_bucket_test() {
37
-
let registry = case registry.new(2, 2) {
38
-
Ok(registry) -> registry
39
-
Error(_) -> {
40
-
panic as "Should be able to create a new registry"
41
-
}
42
-
}
25
+
let assert Ok(registry) = registry.new(2, 2)
26
+
let assert Ok(rate_limiter) = registry |> registry.get_or_create("🚀")
43
27
44
-
let assert Ok(rate_limiter) = registry |> registry.get_or_create("🚀")
45
28
registry |> registry.sweep(None)
29
+
46
30
let assert Ok(new_rate_limiter) = registry |> registry.get_or_create("🚀")
47
31
48
32
rate_limiter
···
50
34
}
51
35
52
36
pub fn sweep_not_full_bucket_test() {
53
-
let registry = case registry.new(2, 2) {
54
-
Ok(registry) -> registry
55
-
Error(_) -> {
56
-
panic as "Should be able to create a new registry"
57
-
}
58
-
}
59
-
37
+
let assert Ok(registry) = registry.new(2, 2)
60
38
let assert Ok(rate_limiter) = registry |> registry.get_or_create("🚀")
61
39
62
40
let _ = rate_limiter |> rate_limiter.hit
···
67
45
rate_limiter
68
46
|> should.equal(new_rate_limiter)
69
47
}
48
+
49
+
pub fn sweep_after_long_time_test() {
50
+
let assert Ok(registry) = registry.new(2, 2)
51
+
let assert Ok(rate_limiter) = registry |> registry.get_or_create("🚀")
52
+
53
+
rate_limiter |> rate_limiter.set_now(0)
54
+
let _ = rate_limiter |> rate_limiter.hit
55
+
let _ = rate_limiter |> rate_limiter.hit
56
+
let _ = rate_limiter |> rate_limiter.hit
57
+
rate_limiter |> rate_limiter.set_now(1000)
58
+
59
+
registry |> registry.sweep(None)
60
+
61
+
let assert Ok(new_rate_limiter) = registry |> registry.get_or_create("🚀")
62
+
63
+
rate_limiter
64
+
|> should.not_equal(new_rate_limiter)
65
+
}
+14
test/glimit_test.gleam
+14
test/glimit_test.gleam
···
80
80
func(Nil) |> should.equal("OK")
81
81
func(Nil) |> should.equal("Stop!")
82
82
func(Nil) |> should.equal("Stop!")
83
+
84
+
rate_limiter |> rate_limiter.set_now(6)
85
+
func(Nil) |> should.equal("OK")
86
+
func(Nil) |> should.equal("OK")
87
+
func(Nil) |> should.equal("OK")
88
+
func(Nil) |> should.equal("Stop!")
89
+
func(Nil) |> should.equal("Stop!")
90
+
91
+
rate_limiter |> rate_limiter.set_now(13)
92
+
func(Nil) |> should.equal("OK")
93
+
func(Nil) |> should.equal("OK")
94
+
func(Nil) |> should.equal("OK")
95
+
func(Nil) |> should.equal("Stop!")
96
+
func(Nil) |> should.equal("Stop!")
83
97
}