59 lines
3.5 KiB
TypeScript
59 lines
3.5 KiB
TypeScript
import { expect, test } from "bun:test";
|
|
import { experimentalLayerPrompt, imageLayerGenerationConfig } from "@/lib/canvas/canvas-image-layers";
|
|
import { imageLayerPlanningPrompt, parseImageLayerPlan } from "@/lib/canvas/canvas-image-layer-plan";
|
|
import { defaultConfig } from "@/stores/use-config-store";
|
|
|
|
const plan = {
|
|
canPlan: true,
|
|
layers: [
|
|
{ name: "背景", description: "背景场景描述标记", kind: "background", generation: { prompt: "移除蓝色绒毯，补全其后地毯与沙发；保留书架和窗户", background: "opaque" } },
|
|
{ name: "绒毯", description: "保留完整主体与透明背景的前景描述标记", kind: "object", generation: { prompt: "仅保留蓝色花朵绒毯，保持轮廓、纹理和原始位置", background: "transparent" }, bbox: [169, 38, 969, 955] },
|
|
],
|
|
};
|
|
|
|
test("背景只收到本层规划，不混入对象描述或前景生成指令", () => {
|
|
const parsed = parseImageLayerPlan(JSON.stringify(plan));
|
|
const background = experimentalLayerPrompt(parsed.layers[0].generation);
|
|
const foreground = experimentalLayerPrompt(parsed.layers[1].generation, parsed.layers[1].bbox);
|
|
expect(background).toContain("移除蓝色绒毯");
|
|
expect(background).toContain("alpha=255");
|
|
expect(background).not.toContain(plan.layers[1].description);
|
|
expect(background).not.toContain(plan.layers[1].generation.prompt);
|
|
expect(background).not.toContain("alpha=0");
|
|
expect(foreground).toContain("仅保留蓝色花朵绒毯");
|
|
expect(foreground).toContain("<bbox>169 38 969 955</bbox>");
|
|
expect(foreground).toContain("alpha=0");
|
|
expect(foreground).not.toContain(plan.layers[0].generation.prompt);
|
|
});
|
|
|
|
test("按规划的本层透明度生成，保持模型、源图尺寸及默认质量", () => {
|
|
const parsed = parseImageLayerPlan(JSON.stringify(plan));
|
|
const config = { ...defaultConfig, model: "chosen", size: "1672x941", quality: "medium", count: "4" };
|
|
expect(imageLayerGenerationConfig(config, parsed.layers[0].generation, 0)).toMatchObject({ model: "chosen", size: "1672x941", quality: "medium", count: "1", transparentBackground: "false" });
|
|
expect(imageLayerGenerationConfig(config, parsed.layers[1].generation, 1).transparentBackground).toBe("true");
|
|
expect(config.count).toBe("4");
|
|
});
|
|
|
|
test("缺失、空白、过长、角色冲突或额外参数的规划在图像调用前停止", () => {
|
|
for (const generation of [
|
|
undefined,
|
|
{},
|
|
{ prompt: " ", background: "opaque" },
|
|
{ prompt: "x".repeat(2401), background: "opaque" },
|
|
{ prompt: "错误背景", background: "transparent" },
|
|
{ prompt: "合法提示", background: "opaque", model: "other" },
|
|
{ prompt: "合法提示", background: "opaque", size: "1:1" },
|
|
]) {
|
|
const value = { ...plan, layers: [{ ...plan.layers[0], generation }, plan.layers[1]] };
|
|
expect(() => parseImageLayerPlan(JSON.stringify(value))).toThrow("未提交拆图任务");
|
|
}
|
|
expect(() => parseImageLayerPlan(JSON.stringify({ ...plan, layers: [plan.layers[0], { ...plan.layers[1], generation: { prompt: "对象", background: "opaque" } }] }))).toThrow("图层角色冲突");
|
|
});
|
|
|
|
test("规划合同要求每张图有独立参数且不粘贴跨层指令", () => {
|
|
const prompt = imageLayerPlanningPrompt("拆分背景和主体");
|
|
expect(prompt).toContain("每层 generation 必填");
|
|
expect(prompt).toContain("不会收到其他层描述或用户总体要求");
|
|
expect(prompt).toContain("不能粘贴前景");
|
|
});
|