1import assert from "node:assert/strict";
2import test from "node:test";
3 
4import type { CliArgs } from "../types.ts";
5import {
6  buildOpenAIGenerationsBody,
7  extractImageFromResponse,
8  getDefaultModel,
9  getOpenAIAspectRatio,
10  getOpenAIImageApiDialect,
11  getOpenAIResolution,
12  getMimeType,
13  getOpenAISize,
14  getOrientationFromAspectRatio,
15  inferAspectRatioFromSize,
16  inferResolutionFromSize,
17  parseAspectRatio,
18  validateArgs,
19} from "./openai.ts";
20 
21function makeArgs(overrides: Partial<CliArgs> = {}): CliArgs {
22  return {
23    prompt: null,
24    promptFiles: [],
25    imagePath: null,
26    provider: null,
27    model: null,
28    aspectRatio: null,
29    size: null,
30    quality: "2k",
31    imageSize: null,
32    imageApiDialect: null,
33    referenceImages: [],
34    n: 1,
35    batchFile: null,
36    jobs: null,
37    json: false,
38    help: false,
39    ...overrides,
40  };
41}
42 
43test("OpenAI aspect-ratio parsing and size selection match model families", () => {
44  assert.equal(getDefaultModel(), "gpt-image-2");
45  assert.deepEqual(parseAspectRatio("16:9"), { width: 16, height: 9 });
46  assert.equal(parseAspectRatio("wide"), null);
47  assert.equal(parseAspectRatio("0:1"), null);
48 
49  assert.equal(getOpenAISize("dall-e-3", "16:9", "2k"), "1792x1024");
50  assert.equal(getOpenAISize("dall-e-3", "9:16", "normal"), "1024x1792");
51  assert.equal(getOpenAISize("dall-e-2", "16:9", "2k"), "1024x1024");
52  assert.equal(getOpenAISize("gpt-image-1.5", "16:9", "2k"), "1536x1024");
53  assert.equal(getOpenAISize("gpt-image-1.5", "4:3", "2k"), "1024x1024");
54  assert.equal(getOpenAISize("gpt-image-2", "16:9", "2k"), "2048x1152");
55  assert.equal(getOpenAISize("gpt-image-2", "9:16", "2k"), "1152x2048");
56  assert.equal(getOpenAISize("gpt-image-2", "4:3", "2k"), "2048x1536");
57  assert.equal(getOpenAISize("gpt-image-2", "2.35:1", "normal"), "1248x528");
58  assert.equal(inferAspectRatioFromSize("1536x1024"), "3:2");
59  assert.equal(inferResolutionFromSize("1536x1024"), "2K");
60  assert.equal(getOpenAIAspectRatio({ aspectRatio: null, size: "2048x1152" }), "16:9");
61  assert.equal(getOpenAIResolution({ imageSize: null, size: "2048x1152", quality: "normal" }), "2K");
62  assert.equal(getOrientationFromAspectRatio("16:9"), "landscape");
63  assert.equal(getOrientationFromAspectRatio("9:16"), "portrait");
64  assert.equal(getOrientationFromAspectRatio("1:1"), null);
65  assert.equal(getOpenAIImageApiDialect({ imageApiDialect: null }), "openai-native");
66});
67 
68test("OpenAI generations body switches between native and ratio-metadata dialects", () => {
69  assert.deepEqual(
70    buildOpenAIGenerationsBody("Draw a skyline", "gpt-image-2", {
71      aspectRatio: "16:9",
72      size: null,
73      quality: "2k",
74      imageSize: null,
75      imageApiDialect: null,
76    }),
77    {
78      model: "gpt-image-2",
79      prompt: "Draw a skyline",
80      size: "2048x1152",
81      quality: "high",
82    },
83  );
84 
85  assert.deepEqual(
86    buildOpenAIGenerationsBody("Draw a skyline", "gemini-3-pro-image-preview", {
87      aspectRatio: "16:9",
88      size: null,
89      quality: "2k",
90      imageSize: null,
91      imageApiDialect: "ratio-metadata",
92    }),
93    {
94      model: "gemini-3-pro-image-preview",
95      prompt: "Draw a skyline",
96      size: "16:9",
97      metadata: {
98        resolution: "2K",
99        orientation: "landscape",
100      },
101    },
102  );
103 
104  assert.deepEqual(
105    buildOpenAIGenerationsBody("Draw a portrait", "gemini-3-pro-image-preview", {
106      aspectRatio: null,
107      size: "1152x2048",
108      quality: "normal",
109      imageSize: null,
110      imageApiDialect: "ratio-metadata",
111    }),
112    {
113      model: "gemini-3-pro-image-preview",
114      prompt: "Draw a portrait",
115      size: "9:16",
116      metadata: {
117        resolution: "2K",
118        orientation: "portrait",
119      },
120    },
121  );
122});
123 
124test("OpenAI validates gpt-image-2 custom size constraints", () => {
125  assert.doesNotThrow(() =>
126    validateArgs("gpt-image-2", makeArgs({ size: "3840x2160" })),
127  );
128  assert.doesNotThrow(() =>
129    validateArgs("gpt-image-2-2026-04-21", makeArgs({ aspectRatio: "2.35:1" })),
130  );
131 
132  assert.throws(
133    () => validateArgs("gpt-image-2", makeArgs({ size: "1024x576" })),
134    /total pixels/,
135  );
136  assert.throws(
137    () => validateArgs("gpt-image-2", makeArgs({ size: "1025x1024" })),
138    /multiples of 16px/,
139  );
140  assert.throws(
141    () => validateArgs("gpt-image-2", makeArgs({ aspectRatio: "4:1" })),
142    /must not exceed 3:1/,
143  );
144});
145 
146test("OpenAI mime-type detection covers supported reference image extensions", () => {
147  assert.equal(getMimeType("frame.png"), "image/png");
148  assert.equal(getMimeType("frame.jpg"), "image/jpeg");
149  assert.equal(getMimeType("frame.webp"), "image/webp");
150  assert.equal(getMimeType("frame.gif"), "image/gif");
151});
152 
153test("OpenAI response extraction supports base64 and URL download flows", async (t) => {
154  const originalFetch = globalThis.fetch;
155  t.after(() => {
156    globalThis.fetch = originalFetch;
157  });
158 
159  const fromBase64 = await extractImageFromResponse({
160    data: [{ b64_json: Buffer.from("hello").toString("base64") }],
161  });
162  assert.equal(Buffer.from(fromBase64).toString("utf8"), "hello");
163 
164  globalThis.fetch = async () =>
165    new Response(Uint8Array.from([1, 2, 3]), {
166      status: 200,
167      headers: { "Content-Type": "application/octet-stream" },
168    });
169 
170  const fromUrl = await extractImageFromResponse({
171    data: [{ url: "https://example.com/image.png" }],
172  });
173  assert.deepEqual([...fromUrl], [1, 2, 3]);
174 
175  await assert.rejects(
176    () => extractImageFromResponse({ data: [{}] }),
177    /No image in response/,
178  );
179});
180