Build API v2

src/lib/server/v2-rate-limit.spec.ts

@@ -0,0 +1,40 @@
+import { describe, it, expect, beforeEach } from 'vitest';
+import { rateLimit, _clearRateLimits } from './v2-rate-limit';
+
+describe('rateLimit', () => {
+	beforeEach(() => {
+		_clearRateLimits();
+	});
+
+	it('allows requests up to the limit', () => {
+		for (let i = 0; i < 3; i++) {
+			const r = rateLimit({ key: 'k', limit: 3, windowMs: 60_000, now: 1000 });
+			expect(r.ok).toBe(true);
+		}
+	});
+
+	it('rejects once the limit is exceeded and reports Retry-After', () => {
+		for (let i = 0; i < 3; i++) {
+			rateLimit({ key: 'k', limit: 3, windowMs: 60_000, now: 1000 });
+		}
+		const r = rateLimit({ key: 'k', limit: 3, windowMs: 60_000, now: 1000 });
+		expect(r.ok).toBe(false);
+		expect(r.retryAfterMs).toBeGreaterThan(0);
+	});
+
+	it('treats keys independently', () => {
+		rateLimit({ key: 'a', limit: 1, windowMs: 60_000, now: 1000 });
+		expect(rateLimit({ key: 'a', limit: 1, windowMs: 60_000, now: 1000 }).ok).toBe(false);
+		expect(rateLimit({ key: 'b', limit: 1, windowMs: 60_000, now: 1000 }).ok).toBe(true);
+	});
+
+	it('lets the window expire before allowing again', () => {
+		for (let i = 0; i < 2; i++) {
+			rateLimit({ key: 'k', limit: 2, windowMs: 10_000, now: 1000 });
+		}
+		expect(rateLimit({ key: 'k', limit: 2, windowMs: 10_000, now: 1000 }).ok).toBe(false);
+		expect(rateLimit({ key: 'k', limit: 2, windowMs: 10_000, now: 1000 + 10_000 + 1 }).ok).toBe(
+			true
+		);
+	});
+});