1import assert from "node:assert/strict";
2import test, { type TestContext } from "node:test";
3 
4import type { CliArgs } from "../types.ts";
5import { generateImage } from "./jimeng.ts";
6 
7function makeArgs(overrides: Partial<CliArgs> = {}): CliArgs {
8  return {
9    prompt: null,
10    promptFiles: [],
11    imagePath: null,
12    provider: null,
13    model: null,
14    aspectRatio: null,
15    size: null,
16    quality: null,
17    imageSize: null,
18    referenceImages: [],
19    n: 1,
20    batchFile: null,
21    jobs: null,
22    json: false,
23    help: false,
24    ...overrides,
25  };
26}
27 
28function useEnv(
29  t: TestContext,
30  values: Record<string, string | null>,
31): void {
32  const previous = new Map<string, string | undefined>();
33  for (const [key, value] of Object.entries(values)) {
34    previous.set(key, process.env[key]);
35    if (value == null) {
36      delete process.env[key];
37    } else {
38      process.env[key] = value;
39    }
40  }
41 
42  t.after(() => {
43    for (const [key, value] of previous.entries()) {
44      if (value == null) {
45        delete process.env[key];
46      } else {
47        process.env[key] = value;
48      }
49    }
50  });
51}
52 
53test("Jimeng submit request uses prompt field expected by current API", async (t) => {
54  useEnv(t, {
55    JIMENG_ACCESS_KEY_ID: "test-access-key",
56    JIMENG_SECRET_ACCESS_KEY: "test-secret-key",
57    JIMENG_BASE_URL: null,
58    JIMENG_REGION: null,
59  });
60 
61  const originalFetch = globalThis.fetch;
62  t.after(() => {
63    globalThis.fetch = originalFetch;
64  });
65 
66  const calls: Array<{
67    input: string;
68    init?: RequestInit;
69  }> = [];
70 
71  globalThis.fetch = async (input, init) => {
72    calls.push({
73      input: String(input),
74      init,
75    });
76 
77    if (calls.length === 1) {
78      return Response.json({
79        code: 10000,
80        data: {
81          task_id: "task-123",
82        },
83      });
84    }
85 
86    return Response.json({
87      code: 10000,
88      data: {
89        status: "done",
90        binary_data_base64: [Buffer.from("jimeng-image").toString("base64")],
91      },
92    });
93  };
94 
95  const image = await generateImage(
96    "A quiet bamboo forest",
97    "jimeng_t2i_v40",
98    makeArgs({ quality: "normal" }),
99  );
100 
101  assert.equal(Buffer.from(image).toString("utf8"), "jimeng-image");
102  assert.equal(calls.length, 2);
103  assert.equal(
104    calls[0]?.input,
105    "https://visual.volcengineapi.com/?Action=CVSync2AsyncSubmitTask&Version=2022-08-31",
106  );
107 
108  const submitBody = JSON.parse(String(calls[0]?.init?.body)) as Record<string, unknown>;
109  assert.equal(submitBody.req_key, "jimeng_t2i_v40");
110  assert.equal(submitBody.prompt, "A quiet bamboo forest");
111  assert.ok(!("prompt_text" in submitBody));
112  assert.equal(submitBody.width, 1024);
113  assert.equal(submitBody.height, 1024);
114});
115