168 lines
9.1 KiB
TypeScript
168 lines
9.1 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
||
|
||
import { getNodeDefinition, getNodeGenerationMode, getNodeInputKind, getNodeListLabel, getNodeMinSize, getNodeResourceKind, listCreatableNodeDefinitions, listNodeDefinitions, shouldKeepAspectRatio } from "../src/lib/canvas/node-registry";
|
||
import { connectionInputSummary } from "../src/lib/canvas/canvas-connection-policy";
|
||
import { CanvasNodeType, type CanvasConnection, type CanvasNodeData, type CanvasNodeMetadata } from "../src/types/canvas";
|
||
|
||
function node(type: CanvasNodeType, metadata?: CanvasNodeMetadata, id = type): CanvasNodeData {
|
||
return { id, type, title: id, position: { x: 0, y: 0 }, width: 100, height: 100, metadata };
|
||
}
|
||
|
||
const ALL_TYPES = Object.values(CanvasNodeType);
|
||
|
||
describe("节点注册表——覆盖完整性", () => {
|
||
test("每种节点类型都有定义", () => {
|
||
expect(ALL_TYPES.every((type) => Boolean(getNodeDefinition(type)))).toBe(true);
|
||
for (const type of ALL_TYPES) expect(getNodeMinSize(type)).toBeDefined();
|
||
});
|
||
|
||
test("仅技能与生成配置不进创建菜单", () => {
|
||
const hidden = ALL_TYPES.filter((type) => !listCreatableNodeDefinitions().some((def) => def.type === type));
|
||
expect(hidden.sort()).toEqual([CanvasNodeType.Config, CanvasNodeType.Skill].sort());
|
||
});
|
||
});
|
||
|
||
describe("节点注册表——几何", () => {
|
||
test("最小尺寸按类型区分", () => {
|
||
expect(getNodeMinSize(CanvasNodeType.Image)).toEqual({ width: 420, height: 236 });
|
||
expect(getNodeMinSize(CanvasNodeType.Video)).toEqual({ width: 420, height: 236 });
|
||
expect(getNodeMinSize(CanvasNodeType.Script).width).toBe(800);
|
||
expect(getNodeMinSize(CanvasNodeType.Text)).toEqual({ width: 220, height: 160 });
|
||
});
|
||
|
||
test("仅图片（未开自由比例）与视频锁定宽高比", () => {
|
||
expect(shouldKeepAspectRatio(node(CanvasNodeType.Image))).toBe(true);
|
||
expect(shouldKeepAspectRatio(node(CanvasNodeType.Image, { freeResize: true }))).toBe(false);
|
||
expect(shouldKeepAspectRatio(node(CanvasNodeType.Video))).toBe(true);
|
||
expect(shouldKeepAspectRatio(node(CanvasNodeType.Video, { freeResize: true }))).toBe(true);
|
||
for (const type of [CanvasNodeType.Text, CanvasNodeType.Drawing, CanvasNodeType.Script, CanvasNodeType.Skill, CanvasNodeType.Config, CanvasNodeType.Audio, CanvasNodeType.Frame]) {
|
||
expect(shouldKeepAspectRatio(node(type))).toBe(false);
|
||
}
|
||
});
|
||
});
|
||
|
||
describe("节点注册表——素材类型（resourceKind）", () => {
|
||
test("有内容才算素材，空节点不算", () => {
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Image, { content: "x" }))).toBe("image");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Image))).toBeNull();
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Video, { content: "x" }))).toBe("video");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Video))).toBeNull();
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Audio, { content: "x" }))).toBe("audio");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Audio))).toBeNull();
|
||
});
|
||
|
||
test("绘图按图片计，且认的是 drawingId 而非 content", () => {
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Drawing, { drawingId: "d1" }))).toBe("image");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Drawing, { content: "x" }))).toBeNull();
|
||
});
|
||
|
||
test("文本认 content 或 prompt", () => {
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Text, { content: "x" }))).toBe("text");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Text, { prompt: "p" }))).toBe("text");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Text))).toBeNull();
|
||
});
|
||
|
||
test("技能按文本计，认 skillSnapshot 或 content", () => {
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Skill, { content: "x" }))).toBe("text");
|
||
expect(getNodeResourceKind(node(CanvasNodeType.Skill))).toBeNull();
|
||
});
|
||
|
||
test("分镜脚本、生成配置、背板都不是素材", () => {
|
||
for (const type of [CanvasNodeType.Script, CanvasNodeType.Config, CanvasNodeType.Frame]) {
|
||
expect(getNodeResourceKind(node(type, { content: "x" }))).toBeNull();
|
||
}
|
||
});
|
||
});
|
||
|
||
describe("节点注册表——生成模式（generationMode）", () => {
|
||
test("按类型映射", () => {
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Image))).toBe("image");
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Video))).toBe("video");
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Audio))).toBe("audio");
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Text))).toBe("text");
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Script))).toBe("text");
|
||
});
|
||
|
||
test("生成配置由 metadata 决定，缺省按图片", () => {
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Config, { generationMode: "video" }))).toBe("video");
|
||
expect(getNodeGenerationMode(node(CanvasNodeType.Config))).toBe("image");
|
||
});
|
||
|
||
test("绘图、技能、背板不产生生成行为", () => {
|
||
for (const type of [CanvasNodeType.Drawing, CanvasNodeType.Skill, CanvasNodeType.Frame]) {
|
||
expect(getNodeGenerationMode(node(type))).toBeNull();
|
||
}
|
||
});
|
||
});
|
||
|
||
describe("节点注册表——输入计数类别（inputKind）", () => {
|
||
test("生成配置与背板不参与计数，其余各归其类", () => {
|
||
expect(getNodeInputKind(CanvasNodeType.Config)).toBeUndefined();
|
||
expect(getNodeInputKind(CanvasNodeType.Frame)).toBeUndefined();
|
||
expect(getNodeInputKind(CanvasNodeType.Image)).toBe("image");
|
||
expect(getNodeInputKind(CanvasNodeType.Drawing)).toBe("image");
|
||
expect(getNodeInputKind(CanvasNodeType.Video)).toBe("video");
|
||
expect(getNodeInputKind(CanvasNodeType.Audio)).toBe("audio");
|
||
for (const type of [CanvasNodeType.Text, CanvasNodeType.Script, CanvasNodeType.Skill]) {
|
||
expect(getNodeInputKind(type)).toBe("text");
|
||
}
|
||
});
|
||
});
|
||
|
||
describe("connectionInputSummary——计数与跨类型覆盖", () => {
|
||
const conn = (fromNodeId: string, toNodeId: string): CanvasConnection => ({ id: `c-${fromNodeId}`, fromNodeId, toNodeId });
|
||
|
||
test("与内容无关：空图片节点仍计入 imageCount", () => {
|
||
const nodes = [node(CanvasNodeType.Image, undefined, "img"), node(CanvasNodeType.Text, undefined, "target")];
|
||
expect(connectionInputSummary("target", nodes, [conn("img", "target")]).imageCount).toBe(1);
|
||
});
|
||
|
||
test("绘图计入 imageCount，脚本与技能计入 textCount", () => {
|
||
const nodes = [
|
||
node(CanvasNodeType.Drawing, undefined, "draw"),
|
||
node(CanvasNodeType.Script, undefined, "script"),
|
||
node(CanvasNodeType.Skill, undefined, "skill"),
|
||
node(CanvasNodeType.Text, undefined, "target"),
|
||
];
|
||
const summary = connectionInputSummary("target", nodes, [conn("draw", "target"), conn("script", "target"), conn("skill", "target")]);
|
||
expect(summary.imageCount).toBe(1);
|
||
expect(summary.textCount).toBe(2);
|
||
});
|
||
|
||
test("生成配置与背板作为上游一律不计数", () => {
|
||
const nodes = [
|
||
node(CanvasNodeType.Config, undefined, "config"),
|
||
node(CanvasNodeType.Frame, undefined, "frame"),
|
||
node(CanvasNodeType.Text, undefined, "target"),
|
||
];
|
||
const summary = connectionInputSummary("target", nodes, [conn("config", "target"), conn("frame", "target")]);
|
||
expect(summary).toEqual({ textCount: 0, imageCount: 0, videoCount: 0, audioCount: 0, characterCount: 0 });
|
||
});
|
||
|
||
test("角色卡覆盖类型归类：图片节点带角色元数据记为角色而非图片", () => {
|
||
const nodes = [node(CanvasNodeType.Image, { workflowKind: "character" }, "char"), node(CanvasNodeType.Text, undefined, "target")];
|
||
const summary = connectionInputSummary("target", nodes, [conn("char", "target")]);
|
||
expect(summary.characterCount).toBe(1);
|
||
expect(summary.imageCount).toBe(0);
|
||
});
|
||
|
||
test("不可计数类型即使带角色元数据也不记为角色（排除早于覆盖）", () => {
|
||
const nodes = [node(CanvasNodeType.Frame, { workflowKind: "character" }, "frame"), node(CanvasNodeType.Text, undefined, "target")];
|
||
expect(connectionInputSummary("target", nodes, [conn("frame", "target")]).characterCount).toBe(0);
|
||
});
|
||
});
|
||
|
||
describe("节点注册表——列表标签", () => {
|
||
test("派生自 label，背板显式钉住不带「节点」后缀", () => {
|
||
expect(getNodeListLabel(CanvasNodeType.Image)).toBe("图片节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Text)).toBe("文本节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Drawing)).toBe("绘图节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Script)).toBe("分镜脚本节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Skill)).toBe("技能节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Config)).toBe("生成配置节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Video)).toBe("视频节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Audio)).toBe("音频节点");
|
||
expect(getNodeListLabel(CanvasNodeType.Frame)).toBe("背板");
|
||
});
|
||
});
|