File size: 3,002 Bytes
20f83d9 | 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 | // @vitest-environment node
import { afterEach, beforeEach, describe, expect, test, vi } from "vitest";
import { summarizeArticle } from "../worldmonitor/news/v1/summarize-article";
const originalFetch = globalThis.fetch;
const originalEnv = { ...process.env };
function restoreEnv() {
for (const key of Object.keys(process.env)) {
if (!(key in originalEnv)) delete process.env[key];
}
Object.assign(process.env, originalEnv);
}
function makeContext(headers: Record<string, string> = {}) {
return {
request: new Request("https://www.worldmonitor.app/api/news/v1/summarize-article", { headers }),
pathParams: {},
headers,
};
}
function request(mode = "brief") {
return {
provider: "groq",
headlines: ["Headline one", "Headline two"],
mode,
geoContext: "",
variant: "full",
lang: "en",
systemAppend: "",
bodies: [],
};
}
beforeEach(() => {
restoreEnv();
process.env.GROQ_API_KEY = "test-groq-key";
globalThis.fetch = vi.fn(async () => {
throw new Error("non-premium summarize should not call providers");
}) as typeof fetch;
});
afterEach(() => {
globalThis.fetch = originalFetch;
restoreEnv();
});
describe("summarizeArticle handler premium mode gate", () => {
test("anonymous article summaries are rejected before provider fetch", async () => {
const result = await summarizeArticle(makeContext(), request("brief"));
expect(result).toMatchObject({
summary: "",
fallback: true,
error: "Pro subscription required",
errorType: "AuthError",
status: "SUMMARIZE_STATUS_ERROR",
statusDetail: "Pro subscription required",
});
expect(globalThis.fetch).not.toHaveBeenCalled();
});
test("anonymous analysis mode is rejected before provider fetch", async () => {
const result = await summarizeArticle(makeContext({ "X-WorldMonitor-Key": "wms_basic_session" }), request("analysis"));
expect(result.error).toBe("Pro subscription required");
expect(globalThis.fetch).not.toHaveBeenCalled();
});
test("translation mode remains outside the premium summary gate", async () => {
delete process.env.GROQ_API_KEY;
const result = await summarizeArticle(makeContext(), request("translate"));
expect(result).toMatchObject({
fallback: true,
status: "SUMMARIZE_STATUS_SKIPPED",
statusDetail: "GROQ_API_KEY not configured",
});
expect(globalThis.fetch).not.toHaveBeenCalled();
});
test("premium callers pass the summary gate", async () => {
delete process.env.GROQ_API_KEY;
process.env.WORLDMONITOR_VALID_KEYS = "enterprise-test-key";
const result = await summarizeArticle(
makeContext({ "X-WorldMonitor-Key": "enterprise-test-key" }),
request("brief"),
);
expect(result).toMatchObject({
fallback: true,
status: "SUMMARIZE_STATUS_SKIPPED",
statusDetail: "GROQ_API_KEY not configured",
});
expect(result.error).not.toBe("Pro subscription required");
});
});
|