1import assert from "node:assert/strict";
2import test from "node:test";
3 
4import type { CliArgs } from "../types.ts";
5import {
6  getDefaultModel,
7  getDefaultOutputExtension,
8  validateArgs,
9} from "./codex-cli.ts";
10 
11function makeArgs(overrides: Partial<CliArgs> = {}): CliArgs {
12  return {
13    prompt: null,
14    promptFiles: [],
15    imagePath: null,
16    provider: "codex-cli",
17    model: null,
18    aspectRatio: null,
19    aspectRatioSource: null,
20    size: null,
21    quality: "2k",
22    imageSize: null,
23    imageSizeSource: null,
24    imageApiDialect: null,
25    referenceImages: [],
26    n: 1,
27    batchFile: null,
28    jobs: null,
29    json: false,
30    help: false,
31    ...overrides,
32  };
33}
34 
35test("codex-cli defaults to codex-image-gen model and PNG output", () => {
36  assert.equal(getDefaultModel(), "codex-image-gen");
37  assert.equal(getDefaultOutputExtension(), ".png");
38});
39 
40test("codex-cli validateArgs rejects n>1 with a non-retryable message", () => {
41  assert.throws(
42    () => validateArgs("codex-image-gen", makeArgs({ n: 2 })),
43    /supports only n=1/,
44  );
45});
46 
47test("codex-cli validateArgs rejects ratio-metadata dialect", () => {
48  assert.throws(
49    () => validateArgs("codex-image-gen", makeArgs({ imageApiDialect: "ratio-metadata" })),
50    /Invalid imageApiDialect/,
51  );
52});
53 
54test("codex-cli validateArgs accepts default n=1 with no dialect", () => {
55  assert.doesNotThrow(() => validateArgs("codex-image-gen", makeArgs()));
56});
57 
58test("codex-cli validateArgs accepts reference images (Codex image_gen supports refs)", () => {
59  assert.doesNotThrow(() =>
60    validateArgs("codex-image-gen", makeArgs({ referenceImages: ["/tmp/a.png", "/tmp/b.png"] })),
61  );
62});
63