84 lines
5.2 KiB
TypeScript
84 lines
5.2 KiB
TypeScript
import { expect, test } from "bun:test";
|
|
import { applyToolMention, buildToolMentionReference, parseToolMentionTokens, removeToolMentions } from "../src/lib/canvas/canvas-resource-references";
|
|
import { createNineGridNode } from "../src/lib/canvas/canvas-image-source";
|
|
import { CanvasNodeType } from "../src/types/canvas";
|
|
import type { CanvasNodeData } from "../src/types/canvas";
|
|
import "../src/lib/canvas/node-registry/definitions/index";
|
|
import { writeCanvasNodePrompt } from "../src/lib/canvas/canvas-node-prompt";
|
|
import { canonicalGenerationMetadata } from "../src/lib/canvas/generation-contract";
|
|
|
|
for (const type of ["style", "nine_grid"]) {
|
|
test(`${type} selection, replacement, removal and clearing persist across prompt panel reads`, () => {
|
|
for (const content of [undefined, "/generated-image.png"]) {
|
|
let node: CanvasNodeData = {
|
|
id: "image",
|
|
type: CanvasNodeType.Image,
|
|
title: "图片",
|
|
position: { x: 0, y: 0 },
|
|
width: 320,
|
|
height: 180,
|
|
metadata: { content, prompt: "原始生成记录" },
|
|
};
|
|
const editAndRead = (prompt: string) => {
|
|
node = writeCanvasNodePrompt(node, prompt);
|
|
node = JSON.parse(JSON.stringify(node)) as CanvasNodeData;
|
|
const metadata = canonicalGenerationMetadata(node, "image");
|
|
expect(metadata.composerContent).toBe(prompt);
|
|
expect(metadata.generationSpec?.prompt).toBe(prompt);
|
|
if (content) expect(node.metadata?.prompt).toBe("原始生成记录");
|
|
return metadata.composerContent!;
|
|
};
|
|
let prompt = editAndRead(applyToolMention("场景描述", { id: 1, type, label: "初始选择" }));
|
|
prompt = editAndRead(applyToolMention(prompt, { id: 2, type, label: "替换选择" }));
|
|
expect(parseToolMentionTokens(prompt).map((token) => token.toolId)).toEqual([2]);
|
|
prompt = editAndRead(`${prompt}\n连续编辑`);
|
|
prompt = editAndRead(removeToolMentions(prompt, type));
|
|
expect(prompt).toBe("场景描述 \n连续编辑");
|
|
expect(parseToolMentionTokens(prompt)).toEqual([]);
|
|
editAndRead(applyToolMention(prompt, { id: 3, type, label: "再次选择" }));
|
|
expect(parseToolMentionTokens(editAndRead(""))).toEqual([]);
|
|
editAndRead("清空后的新内容");
|
|
}
|
|
});
|
|
}
|
|
|
|
test("tool labels round trip; style replaces, motion deduplicates; clear preserves prose", () => {
|
|
const style = { id: 1, type: "style", label: "风格: [a] 中文" };
|
|
const first = applyToolMention("第一行\n第二行", style);
|
|
expect(parseToolMentionTokens(first)[0].label).toBe(style.label);
|
|
const changed = applyToolMention(first, { ...style, id: 2 });
|
|
expect(parseToolMentionTokens(changed).map((t) => t.toolId)).toEqual([2]);
|
|
expect(removeToolMentions(changed, "style")).toBe("第一行\n第二行");
|
|
const motion = { id: 46, type: "motion", label: "推镜" };
|
|
const combined = applyToolMention(changed, motion);
|
|
expect(applyToolMention(combined, motion)).toBe(combined);
|
|
expect(parseToolMentionTokens(combined)).toHaveLength(2);
|
|
});
|
|
|
|
test("tool references with the same numeric id remain distinct across types", () => {
|
|
let prompt = applyToolMention("", { id: 7, type: "style", label: "水墨" }, "Palette");
|
|
prompt = applyToolMention(prompt, { id: 7, type: "motion", label: "推镜" }, "Camera");
|
|
expect(parseToolMentionTokens(prompt).map(({ type, toolId, label }) => ({ type, toolId, label }))).toEqual([
|
|
{ type: "style", toolId: 7, label: "水墨" },
|
|
{ type: "motion", toolId: 7, label: "推镜" },
|
|
]);
|
|
expect(buildToolMentionReference(7, "水墨", "style", "Palette").id)
|
|
.not.toBe(buildToolMentionReference(7, "推镜", "motion", "Camera").id);
|
|
});
|
|
|
|
test("nine grid creates idle child using stable source token, not an auto generation", () => {
|
|
const child = createNineGridNode({ id: "source", type: CanvasNodeType.Image, title: "图", position: { x: 0, y: 0 }, width: 320, height: 180, metadata: { content: "/image.png" } }, "child", 79, "九宫格", "nine_grid", "Grid3x3");
|
|
expect(child.metadata?.prompt).toContain("@[node:source]");
|
|
expect(parseToolMentionTokens(child.metadata?.prompt ?? "")[0].toolId).toBe(79);
|
|
expect(child.metadata?.content).toBeFalsy();
|
|
});
|
|
|
|
test("node style overrides project style and removing it restores inheritance", async () => {
|
|
const { resolveCanvasStyleExecution } = await import("../src/lib/canvas/canvas-style-execution");
|
|
const { defaultConfig } = await import("../src/stores/use-config-store");
|
|
const styleNode = { id: "style", type: CanvasNodeType.Config, title: "画风", position: { x: 0, y: 0 }, width: 300, height: 200, metadata: { workflowKind: "styleboard" as const, stylePresetId: "test", prompt: "project style" } };
|
|
const prompt = applyToolMention("scene", { id: 1, type: "style", label: "custom" });
|
|
expect(resolveCanvasStyleExecution([styleNode], undefined, prompt, defaultConfig, "image")).toBeNull();
|
|
expect(resolveCanvasStyleExecution([styleNode], undefined, removeToolMentions(prompt, "style"), defaultConfig, "image")).not.toBeNull();
|
|
});
|