72 lines
3.3 KiB
TypeScript
72 lines
3.3 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
|
|
import { buildNodeGenerationContext } from "../src/components/canvas/canvas-node-generation";
|
|
import { CanvasNodeType, type CanvasConnection, type CanvasNodeData } from "../src/types/canvas";
|
|
|
|
function node(id: string, type: CanvasNodeType, content: string): CanvasNodeData {
|
|
return {
|
|
id,
|
|
type,
|
|
title: id,
|
|
position: { x: 0, y: 0 },
|
|
width: 100,
|
|
height: 100,
|
|
metadata: { content },
|
|
};
|
|
}
|
|
|
|
function targetNode(): CanvasNodeData {
|
|
return {
|
|
id: "target",
|
|
type: CanvasNodeType.Video,
|
|
title: "target",
|
|
position: { x: 0, y: 0 },
|
|
width: 100,
|
|
height: 100,
|
|
metadata: { composerContent: "让 @图片1 进入画面" },
|
|
};
|
|
}
|
|
|
|
function connection(fromNodeId: string): CanvasConnection {
|
|
return { id: `connection-${fromNodeId}`, fromNodeId, toNodeId: "target" };
|
|
}
|
|
|
|
describe("canvas node generation position mentions", () => {
|
|
test("同一个 @图片1 在换线后自动指向新的第一张图片", () => {
|
|
const target = targetNode();
|
|
const imageA = node("image-a", CanvasNodeType.Image, "data:image/png;base64,a");
|
|
const imageB = node("image-b", CanvasNodeType.Image, "data:image/png;base64,b");
|
|
|
|
const before = buildNodeGenerationContext(target.id, [imageA, target], [connection(imageA.id)], "让 @图片1 进入画面", []);
|
|
const after = buildNodeGenerationContext(target.id, [imageB, target], [connection(imageB.id)], "让 @图片1 进入画面", []);
|
|
|
|
expect(before.referenceImages.map((image) => image.id)).toEqual(["image-a"]);
|
|
expect(after.referenceImages.map((image) => image.id)).toEqual(["image-b"]);
|
|
expect(before.prompt).toBe("让 图片1 进入画面");
|
|
expect(after.prompt).toBe("让 图片1 进入画面");
|
|
});
|
|
|
|
test("按类型位置选择资源，提示词出现顺序不会改变槽位含义", () => {
|
|
const target = targetNode();
|
|
const imageA = node("image-a", CanvasNodeType.Image, "data:image/png;base64,a");
|
|
const audioA = node("audio-a", CanvasNodeType.Audio, "data:audio/mpeg;base64,a");
|
|
const imageB = node("image-b", CanvasNodeType.Image, "data:image/png;base64,b");
|
|
const connections = [connection(imageA.id), connection(audioA.id), connection(imageB.id)];
|
|
const context = buildNodeGenerationContext(target.id, [imageA, audioA, imageB, target], connections, "让 @图片2 配合 @音频1", []);
|
|
|
|
expect(context.referenceImages.map((image) => image.id)).toEqual(["image-b"]);
|
|
expect(context.referenceAudios.map((audio) => audio.id)).toEqual(["audio-a"]);
|
|
expect(context.prompt).toBe("让 图片1 配合 音频1");
|
|
});
|
|
|
|
test("旧节点 token 只做读取迁移，不再进入生成提示词", () => {
|
|
const target = targetNode();
|
|
const image = node("image-a", CanvasNodeType.Image, "data:image/png;base64,a");
|
|
const context = buildNodeGenerationContext(target.id, [image, target], [connection(image.id)], "让 @[node:image-a] 进入画面", []);
|
|
|
|
expect(context.referenceImages.map((item) => item.id)).toEqual(["image-a"]);
|
|
expect(context.prompt).toBe("让 图片1 进入画面");
|
|
expect(context.prompt).not.toContain("@[node:");
|
|
});
|
|
});
|