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
|
import { assertEquals } from "@std/assert";
import { config } from "../config.ts";
Deno.test("RateLimiter - allows first request", () => {
// Create a new rate limiter instance for testing
class TestRateLimiter {
private storage = new Map<string, { count: number; resetTime: number }>();
isRateLimited(ip: string): boolean {
const now = Date.now();
const entry = this.storage.get(ip);
if (!entry || now > entry.resetTime) {
this.storage.set(ip, {
count: 1,
resetTime: now + config.rateLimit.windowMs,
});
return false;
}
if (entry.count >= config.rateLimit.maxUploads) {
return true;
}
entry.count++;
return false;
}
}
const limiter = new TestRateLimiter();
assertEquals(limiter.isRateLimited("127.0.0.1"), false);
});
Deno.test("RateLimiter - allows requests within limit", () => {
class TestRateLimiter {
private storage = new Map<string, { count: number; resetTime: number }>();
isRateLimited(ip: string): boolean {
const now = Date.now();
const entry = this.storage.get(ip);
if (!entry || now > entry.resetTime) {
this.storage.set(ip, {
count: 1,
resetTime: now + config.rateLimit.windowMs,
});
return false;
}
if (entry.count >= config.rateLimit.maxUploads) {
return true;
}
entry.count++;
return false;
}
}
const limiter = new TestRateLimiter();
const ip = "192.168.1.1";
// Make requests up to the limit
for (let i = 0; i < config.rateLimit.maxUploads; i++) {
assertEquals(limiter.isRateLimited(ip), false, `Request ${i + 1} should be allowed`);
}
// Next request should be blocked
assertEquals(limiter.isRateLimited(ip), true, "Request beyond limit should be blocked");
});
Deno.test("RateLimiter - different IPs have separate limits", () => {
class TestRateLimiter {
private storage = new Map<string, { count: number; resetTime: number }>();
isRateLimited(ip: string): boolean {
const now = Date.now();
const entry = this.storage.get(ip);
if (!entry || now > entry.resetTime) {
this.storage.set(ip, {
count: 1,
resetTime: now + config.rateLimit.windowMs,
});
return false;
}
if (entry.count >= config.rateLimit.maxUploads) {
return true;
}
entry.count++;
return false;
}
}
const limiter = new TestRateLimiter();
// Max out first IP
for (let i = 0; i < config.rateLimit.maxUploads; i++) {
limiter.isRateLimited("10.0.0.1");
}
assertEquals(limiter.isRateLimited("10.0.0.1"), true);
// Second IP should still be allowed
assertEquals(limiter.isRateLimited("10.0.0.2"), false);
});
|
