1import assert from "node:assert/strict";
2import test from "node:test";
3 
4import {
5  extractImageFromResponse,
6  getMimeType,
7  getOpenAISize,
8  parseAspectRatio,
9} from "./openai.ts";
10 
11test("OpenAI aspect-ratio parsing and size selection match model families", () => {
12  assert.deepEqual(parseAspectRatio("16:9"), { width: 16, height: 9 });
13  assert.equal(parseAspectRatio("wide"), null);
14  assert.equal(parseAspectRatio("0:1"), null);
15 
16  assert.equal(getOpenAISize("dall-e-3", "16:9", "2k"), "1792x1024");
17  assert.equal(getOpenAISize("dall-e-3", "9:16", "normal"), "1024x1792");
18  assert.equal(getOpenAISize("dall-e-2", "16:9", "2k"), "1024x1024");
19  assert.equal(getOpenAISize("gpt-image-1.5", "16:9", "2k"), "1536x1024");
20  assert.equal(getOpenAISize("gpt-image-1.5", "4:3", "2k"), "1024x1024");
21});
22 
23test("OpenAI mime-type detection covers supported reference image extensions", () => {
24  assert.equal(getMimeType("frame.png"), "image/png");
25  assert.equal(getMimeType("frame.jpg"), "image/jpeg");
26  assert.equal(getMimeType("frame.webp"), "image/webp");
27  assert.equal(getMimeType("frame.gif"), "image/gif");
28});
29 
30test("OpenAI response extraction supports base64 and URL download flows", async (t) => {
31  const originalFetch = globalThis.fetch;
32  t.after(() => {
33    globalThis.fetch = originalFetch;
34  });
35 
36  const fromBase64 = await extractImageFromResponse({
37    data: [{ b64_json: Buffer.from("hello").toString("base64") }],
38  });
39  assert.equal(Buffer.from(fromBase64).toString("utf8"), "hello");
40 
41  globalThis.fetch = async () =>
42    new Response(Uint8Array.from([1, 2, 3]), {
43      status: 200,
44      headers: { "Content-Type": "application/octet-stream" },
45    });
46 
47  const fromUrl = await extractImageFromResponse({
48    data: [{ url: "https://example.com/image.png" }],
49  });
50  assert.deepEqual([...fromUrl], [1, 2, 3]);
51 
52  await assert.rejects(
53    () => extractImageFromResponse({ data: [{}] }),
54    /No image in response/,
55  );
56});
57