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