87 lines
4.2 KiB
TypeScript
87 lines
4.2 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
||
|
||
import { isolateCopiedNodeMetadata, nextCopiedNodeTitle } from "@/lib/canvas/canvas-node-copy";
|
||
import { buildNodeGenerationContext } from "@/components/canvas/canvas-node-generation";
|
||
import { CanvasNodeType, type CanvasNodeData } from "@/types/canvas";
|
||
|
||
describe("canvas node copy title", () => {
|
||
test("首次复制使用 copy1，后续按已有最大序号递增", () => {
|
||
expect(nextCopiedNodeTitle("女明星角色三视图", ["女明星角色三视图"])).toBe("女明星角色三视图_copy1");
|
||
expect(nextCopiedNodeTitle("女明星角色三视图", ["女明星角色三视图", "女明星角色三视图_copy1"])).toBe("女明星角色三视图_copy2");
|
||
expect(nextCopiedNodeTitle("女明星角色三视图", ["女明星角色三视图_copy1", "女明星角色三视图_copy3"])).toBe("女明星角色三视图_copy4");
|
||
});
|
||
|
||
test("复制已有副本时仍使用原始名称计算下一序号", () => {
|
||
expect(nextCopiedNodeTitle("女明星角色三视图_copy1", ["女明星角色三视图", "女明星角色三视图_copy1"])).toBe("女明星角色三视图_copy2");
|
||
});
|
||
|
||
test("兼容旧版空格 Copy 后缀", () => {
|
||
expect(nextCopiedNodeTitle("女明星角色三视图 Copy", ["女明星角色三视图", "女明星角色三视图 Copy"])).toBe("女明星角色三视图_copy1");
|
||
});
|
||
|
||
test("同一批粘贴可以通过预留标题连续分配序号", () => {
|
||
const titles = new Set(["节点", "节点_copy1"]);
|
||
const first = nextCopiedNodeTitle("节点", titles);
|
||
titles.add(first);
|
||
const second = nextCopiedNodeTitle("节点", titles);
|
||
expect([first, second]).toEqual(["节点_copy2", "节点_copy3"]);
|
||
});
|
||
});
|
||
|
||
describe("canvas generation copy metadata", () => {
|
||
test("媒体副本保留提示词与参考字段，并明确原地回填生成结果", () => {
|
||
const source: CanvasNodeData = {
|
||
id: "source",
|
||
type: CanvasNodeType.Video,
|
||
title: "镜头",
|
||
position: { x: 0, y: 0 },
|
||
width: 340,
|
||
height: 240,
|
||
metadata: {
|
||
content: "video-url",
|
||
prompt: "相同提示词",
|
||
videoStartFrameNodeId: "reference-start",
|
||
videoEndFrameNodeId: "reference-end",
|
||
generationResultPlacement: "new-version",
|
||
generationEffectKeys: ["attach-node:source-task:source:0"],
|
||
},
|
||
};
|
||
|
||
const metadata = isolateCopiedNodeMetadata(source, new Map([[source.id, "copy"]]));
|
||
|
||
expect(metadata.prompt).toBe("相同提示词");
|
||
expect(metadata.videoStartFrameNodeId).toBe("reference-start");
|
||
expect(metadata.videoEndFrameNodeId).toBe("reference-end");
|
||
expect(metadata.copiedFromNodeId).toBe("source");
|
||
expect(metadata.generationResultPlacement).toBe("replace-node");
|
||
expect(metadata.generationEffectKeys).toBeUndefined();
|
||
expect(source.metadata?.generationEffectKeys).toEqual(["attach-node:source-task:source:0"]);
|
||
});
|
||
|
||
test("复制到副本的入边继续解析为同一张参考图", () => {
|
||
const reference: CanvasNodeData = {
|
||
id: "reference",
|
||
type: CanvasNodeType.Image,
|
||
title: "角色参考",
|
||
position: { x: 0, y: 0 },
|
||
width: 340,
|
||
height: 240,
|
||
metadata: { content: "reference-url", mimeType: "image/png" },
|
||
};
|
||
const copy: CanvasNodeData = {
|
||
id: "copy",
|
||
type: CanvasNodeType.Image,
|
||
title: "结果_copy1",
|
||
position: { x: 400, y: 0 },
|
||
width: 340,
|
||
height: 240,
|
||
metadata: { content: "old-result", prompt: "保持角色一致", copiedFromNodeId: "source", generationResultPlacement: "replace-node" },
|
||
};
|
||
|
||
const context = buildNodeGenerationContext(copy.id, [reference, copy], [{ id: "copy-reference", fromNodeId: reference.id, toNodeId: copy.id }], copy.metadata!.prompt!, []);
|
||
|
||
expect(context.prompt).toBe("保持角色一致");
|
||
expect(context.referenceImages.map((image) => image.id)).toEqual([reference.id]);
|
||
});
|
||
});
|