46 lines
2.1 KiB
TypeScript
46 lines
2.1 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
|
|
import { buildEmotionPrompt, emotionProviderMask, normalizeEmotionPromptForProvider, resolveEmotionEditPlan } from "../src/lib/canvas/canvas-emotion";
|
|
|
|
const faceBox = { id: "face-1", x: 120, y: 80, width: 160, height: 180, source: "detected" as const };
|
|
const editRegion = { x: 40, y: 20, width: 320, height: 360 };
|
|
|
|
describe("emotion edit provider plan", () => {
|
|
test("keeps provider mask when the selected image model supports it", () => {
|
|
expect(resolveEmotionEditPlan(true)).toEqual({ mode: "provider-mask", includeMask: true });
|
|
});
|
|
|
|
test("falls back visibly to local compositing when the provider does not support masks", () => {
|
|
const plan = resolveEmotionEditPlan(false);
|
|
expect(plan).toEqual({
|
|
mode: "local-composite",
|
|
includeMask: false,
|
|
notice: "当前渠道不支持蒙版，使用脸部裁切与本地羽化融合",
|
|
});
|
|
expect(emotionProviderMask(plan, { dataUrl: "data:image/png;base64,bWFzaw==" })).toBeUndefined();
|
|
});
|
|
|
|
test("does not promise a provider mask in the shared emotion prompt", () => {
|
|
const prompt = buildEmotionPrompt(
|
|
{ presetId: "emotion-0-0", intimacy: 0, arousal: 0, characterName: "角色1", faceBox },
|
|
editRegion,
|
|
);
|
|
|
|
expect(prompt).not.toContain("透明蒙版");
|
|
expect(prompt).not.toContain("白色蒙版");
|
|
expect(prompt).toContain("目标人脸框");
|
|
});
|
|
|
|
test("removes legacy mask promises before retrying an existing emotion node", () => {
|
|
const legacyPrompt = [
|
|
"只允许修改目标框及其透明蒙版对应的这一张脸。",
|
|
"输入图已裁切到目标人物头部区域；透明蒙版内允许编辑，白色蒙版区域必须保持不变，蒙版外绝对不要生成或修改任何内容。",
|
|
].join("\n");
|
|
|
|
const prompt = normalizeEmotionPromptForProvider(legacyPrompt);
|
|
expect(prompt).not.toContain("透明蒙版");
|
|
expect(prompt).not.toContain("白色蒙版");
|
|
expect(prompt).toContain("目标人脸框及其邻近表情区域");
|
|
});
|
|
});
|