67 lines
3.3 KiB
TypeScript
67 lines
3.3 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
||
|
||
import { agentPlanVisible, latestAgentPlanItems, pendingAgentQuestion } from "@/lib/canvas/cloud-agent-plan";
|
||
|
||
const items = (...statuses: Array<"pending" | "doing" | "done">) => statuses.map((status, index) => ({ id: `${index + 1}`, title: `第 ${index + 1} 项`, status }));
|
||
|
||
/**
|
||
* 回归（2026-09-16）：一个对话会跨多轮运行，每轮各有一条 `plan-${runId}` 清单消息，
|
||
* 都在同一个 messages 数组里。原来取的是**第一条**，于是清单永远钉在第 1 轮的初始状态 ——
|
||
* 后面怎么推进都不刷新、全部做完也不消失（实测反馈：plan 完成了不更新、全完成也不关闭）。
|
||
*/
|
||
describe("latestAgentPlanItems", () => {
|
||
test("跨轮取最新那份，而不是第一条", () => {
|
||
const messages = [
|
||
{ id: "plan-run-1", planItems: items("pending", "pending") },
|
||
{ id: "user-1", text: "..." },
|
||
{ id: "plan-run-2", planItems: items("done", "doing") },
|
||
];
|
||
expect(latestAgentPlanItems(messages).map((entry) => entry.status)).toEqual(["done", "doing"]);
|
||
});
|
||
|
||
test("最新那份全部完成 → 清单收起", () => {
|
||
const messages = [
|
||
{ id: "plan-run-1", planItems: items("pending") },
|
||
{ id: "plan-run-2", planItems: items("done", "done") },
|
||
];
|
||
const plan = latestAgentPlanItems(messages);
|
||
expect(plan.every((entry) => entry.status === "done")).toBe(true);
|
||
expect(agentPlanVisible(plan)).toBe(false);
|
||
});
|
||
|
||
test("最新那份还没做完 → 继续显示", () => {
|
||
expect(agentPlanVisible(latestAgentPlanItems([{ id: "plan-run-1", planItems: items("done", "doing") }]))).toBe(true);
|
||
});
|
||
|
||
test("没有任何清单（包括带空数组的）→ 空且不显示", () => {
|
||
expect(latestAgentPlanItems([])).toEqual([]);
|
||
expect(latestAgentPlanItems([{ id: "user-1" }, { id: "plan-run-1", planItems: [] }])).toEqual([]);
|
||
expect(agentPlanVisible([])).toBe(false);
|
||
});
|
||
});
|
||
|
||
describe("pendingAgentQuestion", () => {
|
||
const question = { question: "选哪个模型？", options: [{ label: "H3文生视频" }, { label: "Seedance 2.5" }] };
|
||
|
||
test("有未作答的提问 → 亮面板", () => {
|
||
const messages = [{ role: "user" }, { role: "assistant", question }];
|
||
expect(pendingAgentQuestion(messages)?.question).toBe("选哪个模型？");
|
||
});
|
||
|
||
test("提问之后用户已回复 → 收起面板", () => {
|
||
const messages = [{ role: "assistant", question }, { role: "user" }];
|
||
expect(pendingAgentQuestion(messages)).toBeUndefined();
|
||
});
|
||
|
||
test("多轮提问取最新那道；旧的那道不算数", () => {
|
||
const later = { question: "确认开始吗？", options: [{ label: "开始" }, { label: "再改改" }] };
|
||
const messages = [{ role: "assistant", question }, { role: "user" }, { role: "assistant", question }, { role: "assistant", question: later }];
|
||
expect(pendingAgentQuestion(messages)?.question).toBe("确认开始吗？");
|
||
});
|
||
|
||
test("没有提问 → 不亮面板", () => {
|
||
expect(pendingAgentQuestion([])).toBeUndefined();
|
||
expect(pendingAgentQuestion([{ role: "assistant" }, { role: "user" }])).toBeUndefined();
|
||
});
|
||
});
|