Files
ttstd-infinite-canvas/web/test/agent-approval-presentation.test.ts
T

72 lines
4.4 KiB
TypeScript

import { describe, expect, it } from "bun:test";
import { agentApprovalPresentation } from "@/lib/canvas/agent-approval-presentation";
describe("Agent approval presentation", () => {
const args = JSON.stringify({ snapshotHash: "private-hash", ops: [
{ type: "add_node", id: "image-node", nodeType: "image", title: "开场分镜", content: "private script" },
{ type: "update_node", id: "video-node-1789310237935", patch: { title: "结尾台词", content: "private dialogue" } },
] });
it("prefers the server-authored preview and exposes the target and changed fields", () => {
const view = agentApprovalPresentation({
call: { function: { name: "canvas_apply_ops", arguments: args } },
preview: {
kind: "canvas_mutation",
title: "确认画布修改",
description: "请确认目标节点和修改字段;批准后才会写入画布。",
items: [{ operation: "update_node", nodeId: "video-node-1789310237935", nodeTitle: "满月庆祝视频草稿", nodeType: "video", nodeTypeLabel: "视频", fields: ["节点名称", "下一版提示词"], resultTitle: "满月庆祝视频草稿(舒缓呼吸感)", summary: "修改视频《满月庆祝视频草稿》的节点名称、下一版提示词" }],
},
});
expect(view.source).toBe("server");
expect(view.items[0]).toMatchObject({ nodeTitle: "满月庆祝视频草稿", nodeTypeLabel: "视频", fields: ["节点名称", "下一版提示词"] });
expect(view.items[0].summary).toContain("满月庆祝视频草稿");
expect(JSON.stringify(view)).not.toContain("private");
});
it("summarizes a legacy persisted approval without exposing content or tool names", () => {
const view = agentApprovalPresentation({ call: { function: { name: "canvas_apply_ops", arguments: args } } });
expect(view.title).toBe("确认画布修改");
expect(view.description).toContain("新增 1 个节点,修改 1 个节点");
expect(view.items.map((item) => item.summary).join(" ")).toContain("目标节点");
expect(view.items.find((item) => item.operation === "update_node")?.fields).toEqual(["节点名称", "正文"]);
expect(JSON.stringify(view)).not.toContain("private");
expect(JSON.stringify(view)).not.toContain("canvas_apply_ops");
expect(JSON.stringify(view)).not.toContain("video-node-1789310237935");
});
it("shows both endpoints for legacy reference connections", () => {
const view = agentApprovalPresentation({ toolName: "canvas_apply_ops", arguments: { ops: [
{ type: "connect_nodes", fromNodeId: "reference-image-node", toNodeId: "video-node" },
] } });
expect(view.items[0].summary).toContain("来源");
expect(view.items[0].summary).toContain("目标");
expect(view.description).toContain("建立 1 条引用连线");
});
it("summarizes an event approval and media request", () => {
expect(agentApprovalPresentation({ toolName: "generate_media", arguments: { mode: "video" } }).title).toBe("确认生成视频");
});
it("uses an explicit safe state when arguments are malformed", () => {
const view = agentApprovalPresentation({ toolName: "canvas_apply_ops", arguments: "{" });
expect(view.description).toContain("无法确认具体修改目标");
expect(view.description).not.toContain("修改当前画布");
});
it("shows approved media specifications and references without exposing prompts or IDs", () => {
const view = agentApprovalPresentation({ toolName: "generate_media", arguments: {
mode: "video", title: "镜头1", durationSeconds: 12, size: "9:16", quality: "720p",
videoGenerateAudio: false, referenceNodeIds: ["private-cat", "private-hero"], prompt: "private prompt",
} });
expect(view.description).toContain("结果自动回写画布");
expect(view.items[0].details).toEqual(["引用 2 个画布资产,并建立连线", "时长:12 秒", "画幅:9:16", "质量:720p", "音频:关闭"]);
expect(JSON.stringify(view)).not.toContain("private");
});
it("states when an unknown approval cannot identify a target", () => {
const view = agentApprovalPresentation({ toolName: "unknown_tool", arguments: {} });
expect(view.items).toHaveLength(0);
expect(view.description).toContain("无法识别");
});
});