Initial 智念AIGC platform

tests/prompt-assembler.test.ts

@@ -0,0 +1,44 @@
+import { describe, expect, it } from "vitest";
+import { assemblePrompt, extractMaterialRequirements } from "@/lib/prompt/assembler";
+
+describe("prompt assembler", () => {
+  it("preserves storyboard cards and @material references for video", () => {
+    const result = assemblePrompt({
+      mode: "video",
+      projectName: "咖啡门店",
+      audience: "周边上班族",
+      offer: "新品拿铁",
+      storyboard: [
+        { id: "s1", title: "开场", visual: "门店外立面，参考@图片1", camera: "推进" },
+        { id: "s2", title: "新品", visual: "新品拿铁特写，参考@图片2", caption: "今日新品" }
+      ],
+      materials: [
+        { type: "image", url: "https://example.com/a.png", label: "@图片1" },
+        { type: "image", url: "https://example.com/b.png", label: "@图片2" }
+      ]
+    });
+    expect(result.prompt).toContain("咖啡门店");
+    expect(result.prompt).toContain("@图片1");
+    expect(result.prompt).toContain("@图片2");
+    expect(result.prompt).toContain("镜头=推进");
+    expect(result.warnings).toEqual([]);
+  });
+
+  it("warns when prompt references unbound materials", () => {
+    const result = assemblePrompt({
+      mode: "image",
+      projectName: "品牌海报",
+      manualPrompt: "使用@图片2生成主视觉",
+      materials: [{ type: "image", url: "https://example.com/a.png", label: "@图片1" }]
+    });
+    expect(result.warnings[0]).toContain("@图片2");
+  });
+
+  it("extracts image, video, and audio requirements", () => {
+    expect(extractMaterialRequirements("参考@图片3、@视频2和@音频1")).toEqual({
+      image: 3,
+      video: 2,
+      audio: 1
+    });
+  });
+});