66 lines
2.3 KiB
TypeScript
66 lines
2.3 KiB
TypeScript
import { expect, test } from "@playwright/test";
|
||
import { defaultVideoCanvasRolePresets } from "../../lib/video-canvas/config";
|
||
import {
|
||
composeVideoCanvasRolePrompt,
|
||
getVideoCanvasAttachedRole,
|
||
setVideoCanvasAttachedRole,
|
||
toVideoCanvasAttachedRole,
|
||
} from "../../lib/video-canvas/video-roles";
|
||
|
||
test.describe("video role attachments", () => {
|
||
test("stores one selected role material in both node and generation params", () => {
|
||
const preset = defaultVideoCanvasRolePresets[0];
|
||
const next = setVideoCanvasAttachedRole(
|
||
{ params: { prompt: "原提示词" } },
|
||
preset,
|
||
"face",
|
||
);
|
||
|
||
expect((next as Record<string, unknown>).videoRole).toEqual(
|
||
toVideoCanvasAttachedRole(preset, "face"),
|
||
);
|
||
expect(next.params).toMatchObject({
|
||
prompt: "原提示词",
|
||
roles: [
|
||
{
|
||
id: "role-003",
|
||
imageUrl: expect.stringContaining("003_face_closeup_1.png"),
|
||
materialKind: "face",
|
||
materialLabel: "近景",
|
||
},
|
||
],
|
||
});
|
||
expect(getVideoCanvasAttachedRole(next)).toMatchObject({
|
||
id: "role-003",
|
||
materialKind: "face",
|
||
name: "甜妹/清新少女",
|
||
});
|
||
});
|
||
|
||
test("removes only the role attachment and preserves unrelated params", () => {
|
||
const attached = setVideoCanvasAttachedRole(
|
||
{ params: { model: "seedance", prompt: "镜头提示" }, untouched: true },
|
||
defaultVideoCanvasRolePresets[0],
|
||
"full",
|
||
);
|
||
const removed = setVideoCanvasAttachedRole(attached, null);
|
||
|
||
expect(getVideoCanvasAttachedRole(removed)).toBeNull();
|
||
expect(removed).toMatchObject({
|
||
params: { model: "seedance", prompt: "镜头提示" },
|
||
untouched: true,
|
||
});
|
||
expect(removed.params).not.toHaveProperty("roles");
|
||
expect(removed).not.toHaveProperty("videoRole");
|
||
});
|
||
|
||
test("merges the role identity, compliant asset and consistency prompt", () => {
|
||
const role = toVideoCanvasAttachedRole(defaultVideoCanvasRolePresets[0], "full");
|
||
const prompt = composeVideoCanvasRolePrompt("女孩转身看向镜头。", [role]);
|
||
|
||
expect(prompt).toContain("女孩转身看向镜头。");
|
||
expect(prompt).toContain("角色引用「甜妹/清新少女」（合规资产 public-role:003）。");
|
||
expect(prompt).toContain("避免换脸、换装、年龄漂移、性别漂移和肢体畸变");
|
||
});
|
||
});
|