modelstudioai__cli/packages/cli/tests/e2e/image-generate.e2e.test.ts

import { describe, expect, test } from "vite-plus/test";
import {
  e2eLabelFromMetaUrl,
  isBailianE2EMediaEnabled,
  isDashScopeE2EReady,
  makeE2eOutputDir,
  parseStdoutJson,
  runCli,
} from "./helpers.ts";

/**
 * Image generate：先做 help / 分组等常规检测（不依赖密钥、不调生成接口）。
 * 需 DashScope + 媒体 E2E 的缺参、dry-run 与真实生成放在 skip 块内；
 * 真实生成用例保持原逻辑与顺序，放在块内最后。
 */

describe("e2e: image generate", () => {
  test("image 分组展示子命令帮助且成功退出", async () => {
    const { stdout, stderr, exitCode } = await runCli(["image"]);
    expect(exitCode, stderr).toBe(0);
    const out = `${stdout}\n${stderr}`;
    expect(out).toMatch(/image|generate|edit/i);
  });

  test("image generate --help 正常退出", async () => {
    const { stderr, exitCode } = await runCli(["image", "generate", "--help"]);
    expect(exitCode, stderr).toBe(0);
    expect(stderr).toMatch(/generate|--prompt|--model/i);
  });
});

describe.skipIf(!isBailianE2EMediaEnabled() || !isDashScopeE2EReady())(
  "e2e: image generate",
  () => {
    test("image generate 缺少 --prompt 时打印子命令帮助并退出 (0)", async () => {
      const { stderr, exitCode } = await runCli([
        "image",
        "generate",
        "--model",
        "qwen-image-2.0",
        "--non-interactive",
      ]);
      expect(exitCode).toBe(0);
      expect(stderr).toMatch(/--prompt|Usage:/i);
    });

    test("【qwen-image-2.0】图片生成", async () => {
      const outDir = makeE2eOutputDir(e2eLabelFromMetaUrl(import.meta.url));
      const { stdout, stderr, exitCode } = await runCli([
        "image",
        "generate",
        "--model",
        "qwen-image-2.0",
        "--prompt",
        "一只简笔画小猫，白底",
        "--out-dir",
        outDir,
        "--out-prefix",
        "e2e-gen",
        "--non-interactive",
        "--output",
        "json",
      ]);
      expect(exitCode, stderr).toBe(0);
      const data = parseStdoutJson<{ saved?: string[] }>(stdout);
      expect(data.saved?.length ?? 0).toBeGreaterThan(0);
      expect(data.saved?.[0]).toContain("e2e-gen");
    }, 300_000);
  },
);