1import assert from "node:assert/strict";
2import fs from "node:fs/promises";
3import os from "node:os";
4import path from "node:path";
5import test, { type TestContext } from "node:test";
6 
7import type { CliArgs } from "../types.ts";
8import {
9  generateImage,
10  getDefaultModel,
11  parseAzureBaseURL,
12  validateArgs,
13} from "./azure.ts";
14 
15function useEnv(
16  t: TestContext,
17  values: Record<string, string | null>,
18): void {
19  const previous = new Map<string, string | undefined>();
20  for (const [key, value] of Object.entries(values)) {
21    previous.set(key, process.env[key]);
22    if (value == null) {
23      delete process.env[key];
24    } else {
25      process.env[key] = value;
26    }
27  }
28 
29  t.after(() => {
30    for (const [key, value] of previous.entries()) {
31      if (value == null) {
32        delete process.env[key];
33      } else {
34        process.env[key] = value;
35      }
36    }
37  });
38}
39 
40function makeArgs(overrides: Partial<CliArgs> = {}): CliArgs {
41  return {
42    prompt: null,
43    promptFiles: [],
44    imagePath: null,
45    provider: null,
46    model: null,
47    aspectRatio: null,
48    size: null,
49    quality: null,
50    imageSize: null,
51    referenceImages: [],
52    n: 1,
53    batchFile: null,
54    jobs: null,
55    json: false,
56    help: false,
57    ...overrides,
58  };
59}
60 
61async function makeTempDir(prefix: string): Promise<string> {
62  return fs.mkdtemp(path.join(os.tmpdir(), prefix));
63}
64 
65test("Azure endpoint parsing and default deployment selection follow env precedence", (t) => {
66  assert.deepEqual(parseAzureBaseURL("https://example.openai.azure.com"), {
67    resourceBaseURL: "https://example.openai.azure.com/openai",
68    deployment: null,
69  });
70  assert.deepEqual(
71    parseAzureBaseURL("https://example.openai.azure.com/openai/deployments/from-url"),
72    {
73      resourceBaseURL: "https://example.openai.azure.com/openai",
74      deployment: "from-url",
75    },
76  );
77 
78  useEnv(t, {
79    AZURE_OPENAI_BASE_URL: "https://example.openai.azure.com/openai/deployments/from-url",
80    AZURE_OPENAI_DEPLOYMENT: "explicit-deploy",
81    AZURE_OPENAI_IMAGE_MODEL: "env-fallback",
82  });
83  assert.equal(getDefaultModel(), "explicit-deploy");
84});
85 
86test("Azure validateArgs rejects unsupported edit input formats before the API call", () => {
87  assert.doesNotThrow(() =>
88    validateArgs("demo-deployment", makeArgs({ referenceImages: ["hero.png", "photo.jpeg"] })),
89  );
90  assert.throws(
91    () => validateArgs("demo-deployment", makeArgs({ referenceImages: ["hero.webp"] })),
92    /PNG or JPG\/JPEG/,
93  );
94});
95 
96test("Azure image generation routes model to deployment and sends mapped quality", async (t) => {
97  useEnv(t, {
98    AZURE_OPENAI_API_KEY: "azure-key",
99    AZURE_OPENAI_BASE_URL: "https://example.openai.azure.com/openai/deployments/default-deploy",
100    AZURE_API_VERSION: null,
101    AZURE_OPENAI_DEPLOYMENT: null,
102    AZURE_OPENAI_IMAGE_MODEL: null,
103  });
104 
105  const originalFetch = globalThis.fetch;
106  t.after(() => {
107    globalThis.fetch = originalFetch;
108  });
109 
110  const calls: Array<{ url: string; body: string }> = [];
111  globalThis.fetch = async (input, init) => {
112    calls.push({
113      url: String(input),
114      body: String(init?.body ?? ""),
115    });
116    return Response.json({
117      data: [{ b64_json: Buffer.from("azure-image").toString("base64") }],
118    });
119  };
120 
121  const bytes = await generateImage(
122    "A calm lake at sunset",
123    "custom-deploy",
124    makeArgs({ quality: "normal" }),
125  );
126 
127  assert.equal(Buffer.from(bytes).toString("utf8"), "azure-image");
128  assert.equal(
129    calls[0]?.url,
130    "https://example.openai.azure.com/openai/deployments/custom-deploy/images/generations?api-version=2025-04-01-preview",
131  );
132 
133  const body = JSON.parse(calls[0]!.body) as Record<string, string>;
134  assert.equal(body.quality, "medium");
135  assert.equal(body.size, "1024x1024");
136});
137 
138test("Azure image edits include quality in multipart requests", async (t) => {
139  const root = await makeTempDir("baoyu-image-gen-azure-");
140  t.after(() => fs.rm(root, { recursive: true, force: true }));
141 
142  const pngPath = path.join(root, "ref.png");
143  const jpgPath = path.join(root, "ref.jpg");
144  await fs.writeFile(pngPath, "png-bytes");
145  await fs.writeFile(jpgPath, "jpg-bytes");
146 
147  useEnv(t, {
148    AZURE_OPENAI_API_KEY: "azure-key",
149    AZURE_OPENAI_BASE_URL: "https://example.openai.azure.com",
150    AZURE_API_VERSION: "2025-04-01-preview",
151    AZURE_OPENAI_DEPLOYMENT: null,
152    AZURE_OPENAI_IMAGE_MODEL: null,
153  });
154 
155  const originalFetch = globalThis.fetch;
156  t.after(() => {
157    globalThis.fetch = originalFetch;
158  });
159 
160  const calls: Array<{ url: string; form: FormData }> = [];
161  globalThis.fetch = async (input, init) => {
162    calls.push({
163      url: String(input),
164      form: init?.body as FormData,
165    });
166    return Response.json({
167      data: [{ b64_json: Buffer.from("edited-image").toString("base64") }],
168    });
169  };
170 
171  const bytes = await generateImage(
172    "Add warm lighting",
173    "edit-deploy",
174    makeArgs({
175      quality: "2k",
176      referenceImages: [pngPath, jpgPath],
177    }),
178  );
179 
180  assert.equal(Buffer.from(bytes).toString("utf8"), "edited-image");
181  assert.equal(
182    calls[0]?.url,
183    "https://example.openai.azure.com/openai/deployments/edit-deploy/images/edits?api-version=2025-04-01-preview",
184  );
185  assert.equal(calls[0]?.form.get("quality"), "high");
186  assert.equal(calls[0]?.form.get("size"), "1024x1024");
187  assert.equal(calls[0]?.form.getAll("image[]").length, 2);
188});
189