55 lines
3.3 KiB
TypeScript
55 lines
3.3 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", title: "开场分镜", content: "private script" },
|
||
{ type: "update_node", title: "结尾台词", content: "private dialogue" },
|
||
] });
|
||
|
||
it("summarizes a persisted approval without exposing tool names or content", () => {
|
||
const view = agentApprovalPresentation({ call: { function: { name: "canvas_apply_ops", arguments: args } } });
|
||
expect(view.title).toBe("确认画布修改");
|
||
expect(view.description).toContain("新增 1 个节点，修改 1 个节点");
|
||
expect(view.items).toEqual(["开场分镜", "结尾台词"]);
|
||
expect(JSON.stringify(view)).not.toContain("private");
|
||
expect(JSON.stringify(view)).not.toContain("canvas_apply_ops");
|
||
});
|
||
|
||
it("summarizes an event approval and media request", () => {
|
||
expect(agentApprovalPresentation({ toolName: "canvas_apply_ops", arguments: JSON.parse(args) }).items).toEqual(["开场分镜", "结尾台词"]);
|
||
expect(agentApprovalPresentation({ toolName: "generate_media", arguments: { mode: "video" } }).title).toBe("确认生成视频");
|
||
});
|
||
|
||
it("uses a safe fallback for malformed arguments", () => {
|
||
expect(agentApprovalPresentation({ toolName: "canvas_apply_ops", arguments: "{" }).description).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).toEqual(["节点：镜头1", "引用 2 个画布资产，并建立连线", "时长：12 秒", "画幅：9:16", "质量：720p", "音频：关闭"]);
|
||
expect(JSON.stringify(view)).not.toContain("private");
|
||
});
|
||
|
||
it("summarizes reference connections as canvas edits without generation", () => {
|
||
const view = agentApprovalPresentation({ toolName: "canvas_apply_ops", arguments: { ops: [
|
||
{ type: "add_node", title: "视频节点" }, { type: "connect_nodes" }, { type: "connect_nodes" },
|
||
] } });
|
||
expect(view.description).toContain("新增 1 个节点，建立 2 条引用连线");
|
||
expect(view.description).not.toContain("提交任务");
|
||
});
|
||
|
||
it("shows the real catalog model and states that a draft is not a submitted task", () => {
|
||
const view = agentApprovalPresentation({ modelName: "用户选择的模型", toolName: "generate_media", arguments: { mode: "video", channelModelKey: "internal-key", size: "16:9" } });
|
||
expect(view.items).toContain("模型：用户选择的模型");
|
||
expect(view.items).toContain("画幅：16:9");
|
||
expect(view.description).toContain("尚未提交生成");
|
||
expect(view.description).toContain("拒绝则保留草稿");
|
||
});
|
||
});
|