diff --git a/src/__tests__/body-limits.test.ts b/src/__tests__/body-limits.test.ts new file mode 100644 index 0000000..cbbbdcd --- /dev/null +++ b/src/__tests__/body-limits.test.ts @@ -0,0 +1,103 @@ +import { describe, it, expect, vi, beforeEach } from "vitest"; +import express from "express"; +import request from "supertest"; + +vi.mock("../services/browser.js", () => ({ + renderPdf: vi.fn(), + renderUrlPdf: vi.fn(), + initBrowser: vi.fn(), + closeBrowser: vi.fn(), +})); + +vi.mock("../services/keys.js", () => ({ + loadKeys: vi.fn(), + getAllKeys: vi.fn().mockReturnValue([]), + keyStore: new Map(), +})); + +vi.mock("../services/db.js", () => ({ + initDatabase: vi.fn(), + pool: { query: vi.fn(), end: vi.fn() }, + cleanupStaleData: vi.fn(), +})); + +vi.mock("../services/verification.js", () => ({ + verifyToken: vi.fn(), + loadVerifications: vi.fn(), +})); + +vi.mock("../middleware/usage.js", () => ({ + usageMiddleware: (_req: any, _res: any, next: any) => next(), + loadUsageData: vi.fn(), + getUsageStats: vi.fn().mockReturnValue({}), +})); + +vi.mock("../middleware/pdfRateLimit.js", () => ({ + pdfRateLimitMiddleware: (_req: any, _res: any, next: any) => next(), + getConcurrencyStats: vi.fn().mockReturnValue({}), +})); + +vi.mock("../middleware/auth.js", () => ({ + authMiddleware: (req: any, _res: any, next: any) => { + req.apiKeyInfo = { key: "test-key", tier: "pro" }; + next(); + }, +})); + +describe("Body size limits", () => { + let app: express.Express; + + beforeEach(async () => { + vi.clearAllMocks(); + vi.resetModules(); + + const { renderPdf } = await import("../services/browser.js"); + vi.mocked(renderPdf).mockResolvedValue(Buffer.from("%PDF-1.4 mock")); + + const { demoRouter } = await import("../routes/demo.js"); + const { convertRouter } = await import("../routes/convert.js"); + + app = express(); + + // Simulate the production middleware setup: + // Route-specific parsers BEFORE global parser + app.use("/v1/demo", express.json({ limit: "50kb" }), demoRouter); + app.use("/v1/convert", express.json({ limit: "500kb" }), convertRouter); + // No global express.json() — that's the fix + }); + + it("demo rejects payloads > 50KB with 413", async () => { + const bigHtml = "x".repeat(51 * 1024); // ~51KB + const res = await request(app) + .post("/v1/demo/html") + .set("content-type", "application/json") + .send(JSON.stringify({ html: bigHtml })); + expect(res.status).toBe(413); + }); + + it("demo accepts payloads < 50KB", async () => { + const res = await request(app) + .post("/v1/demo/html") + .set("content-type", "application/json") + .send({ html: "