v1 softlaunch

test/rate-limit.test.ts

@@ -0,0 +1,50 @@
+import { afterEach, describe, expect, test } from "vitest";
+import { check, reset } from "@/lib/rate-limit";
+
+afterEach(() => reset());
+
+describe("rate limiter", () => {
+  test("allows requests under the limit", () => {
+    for (let i = 0; i < 5; i++) {
+      const r = check("k", 5, 60_000);
+      expect(r.ok).toBe(true);
+    }
+  });
+
+  test("blocks at the limit and reports retry-after", () => {
+    for (let i = 0; i < 5; i++) check("k", 5, 60_000);
+    const r = check("k", 5, 60_000);
+    expect(r.ok).toBe(false);
+    if (!r.ok) {
+      expect(r.retryAfterSec).toBeGreaterThan(0);
+      expect(r.retryAfterSec).toBeLessThanOrEqual(60);
+    }
+  });
+
+  test("different keys are independent", () => {
+    for (let i = 0; i < 5; i++) check("a", 5, 60_000);
+    const r = check("b", 5, 60_000);
+    expect(r.ok).toBe(true);
+  });
+
+  test("window resets when expired", () => {
+    // Use a 10ms window so the test runs fast
+    for (let i = 0; i < 3; i++) check("k", 3, 10);
+    const blocked = check("k", 3, 10);
+    expect(blocked.ok).toBe(false);
+    return new Promise<void>((done) => {
+      setTimeout(() => {
+        const after = check("k", 3, 10);
+        expect(after.ok).toBe(true);
+        done();
+      }, 25);
+    });
+  });
+
+  test("remaining count counts down", () => {
+    const r1 = check("k", 3, 60_000);
+    const r2 = check("k", 3, 60_000);
+    expect(r1.ok && r1.remaining).toBe(2);
+    expect(r2.ok && r2.remaining).toBe(1);
+  });
+});