phase 1 complete i guessHEADmain

1 files changed, 106 insertions, 0 deletions

diff --git a/tests/rateLimiter.test.ts b/tests/rateLimiter.test.ts
new file mode 100644
index 0000000..2645deb
--- /dev/null
+++ b/tests/rateLimiter.test.ts
@@ -0,0 +1,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);
+});