|
| 1 | +import { afterEach, describe, expect, mock, test } from "bun:test"; |
| 2 | +import { fetchCliMediaPromptItems } from "./prompt-library"; |
| 3 | + |
| 4 | +const originalFetch = globalThis.fetch; |
| 5 | + |
| 6 | +afterEach(() => { |
| 7 | + globalThis.fetch = originalFetch; |
| 8 | +}); |
| 9 | + |
| 10 | +function apiItem(overrides: Record<string, unknown> = {}) { |
| 11 | + return { |
| 12 | + artifact: { alt: "Result", kind: "image", url: "https://cdn.example.com/result.png" }, |
| 13 | + category: "image", |
| 14 | + model: "gpt-image-2", |
| 15 | + output: { label: { en: "Image" }, ratio: "4:5" }, |
| 16 | + prompt: "Create a product portrait.", |
| 17 | + slug: "product-portrait", |
| 18 | + source: { label: "Example", platform: "External", url: "https://example.com/source" }, |
| 19 | + summary: { en: "A product portrait prompt." }, |
| 20 | + tags: ["product"], |
| 21 | + title: { en: "Product portrait" }, |
| 22 | + updatedAt: "2026-09-07", |
| 23 | + ...overrides, |
| 24 | + }; |
| 25 | +} |
| 26 | + |
| 27 | +function mockPromptApi(items: unknown[]) { |
| 28 | + globalThis.fetch = mock(async () => |
| 29 | + new Response(JSON.stringify({ data: { items } }), { |
| 30 | + headers: { "content-type": "application/json" }, |
| 31 | + status: 200, |
| 32 | + })) as typeof fetch; |
| 33 | +} |
| 34 | + |
| 35 | +describe("prompt library API normalization", () => { |
| 36 | + test("preserves explicit upstream output ratios", async () => { |
| 37 | + mockPromptApi([apiItem()]); |
| 38 | + |
| 39 | + const items = await fetchCliMediaPromptItems("image"); |
| 40 | + |
| 41 | + expect(items).toHaveLength(1); |
| 42 | + expect(items[0]?.output.ratio).toBe("4:5"); |
| 43 | + }); |
| 44 | + |
| 45 | + test("rejects non-http source URLs before they can reach links", async () => { |
| 46 | + mockPromptApi([ |
| 47 | + apiItem({ slug: "unsafe-source", source: { label: "Unsafe", platform: "External", url: "javascript:alert(1)" } }), |
| 48 | + apiItem({ slug: "safe-source" }), |
| 49 | + ]); |
| 50 | + |
| 51 | + const items = await fetchCliMediaPromptItems("image"); |
| 52 | + |
| 53 | + expect(items.map((item) => item.slug)).toEqual(["safe-source"]); |
| 54 | + expect(items[0]?.source.url).toBe("https://example.com/source"); |
| 55 | + }); |
| 56 | +}); |
0 commit comments