mu-agent-bridge 1.0.71 → 1.0.78

package/dist/tests-ts/workbuddy-plugin.test.js

@@ -59,6 +59,87 @@ const node_test_1 = __importDefault(require("node:test"));
     await strict_1.default.rejects(() => client.createTask(), /尚未登录.*桌面端登录/);
     strict_1.default.equal(calls, 1);
 });
+(0, node_test_1.default)("WorkBuddy 恢复会话等待输入组件切换，不只检查侧栏选中", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    let clicked = false;
+    let checks = 0;
+    const button = { className: "_selected_", matches: () => true, getAttribute: () => "saved", click() { clicked = true; } };
+    const editor = { __reactFiberTest: { memoizedProps: { get viewConversationId() { return checks >= 3 ? "saved" : "previous"; } } } };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: { querySelectorAll: () => [button], querySelector: () => editor } });
+    client.waitFor = async (expression) => {
+        strict_1.default.equal(clicked, true);
+        for (checks = 1; checks <= 3; checks++)
+            if (await client.evaluate(expression))
+                break;
+    };
+    await client.activateConversation("saved");
+    strict_1.default.equal(checks, 3);
+});
+(0, node_test_1.default)("WorkBuddy 旧版输入组件没有会话属性时仍能恢复侧栏任务", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    const button = { className: "_selected_", matches: () => true, getAttribute: () => "saved", click() { } };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: { querySelectorAll: () => [button], querySelector: () => ({}) } });
+    client.waitFor = async (expression) => { strict_1.default.equal(await client.evaluate(expression), true); };
+    await client.activateConversation("saved");
+});
+(0, node_test_1.default)("WorkBuddy 预期会话与侧栏或输入组件不符时拒绝输入", async () => {
+    for (const [expected, selected, input] of [["saved", "other", "saved"], ["saved", "saved", "other"], [null, "other", "other"]]) {
+        const client = new cdp_client_1.WorkbuddyCdpClient();
+        let inputs = 0;
+        const editor = { __reactFiberTest: { memoizedProps: { viewConversationId: input } } };
+        client.evaluate = async (expression) => expression.includes("_selected_")
+            ? (0, node_vm_1.runInNewContext)(expression, { document: { querySelectorAll: () => [{ className: "_selected_", getAttribute: () => selected }], querySelector: () => editor } })
+            : expression.includes("JSON.stringify") ? "[]" : true;
+        client.cdp = async () => { inputs++; };
+        client.readInteraction = async () => undefined;
+        client.waitForReply = async () => ({ text: "不应发送" });
+        await strict_1.default.rejects(client.complete({ task: "在线不？", sessionId: "logical" }, undefined, false, expected), /桌面任务|会话/);
+        strict_1.default.equal(inputs, 0);
+    }
+});
+(0, node_test_1.default)("WorkBuddy 输入后会话切换时不点击发送", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    let selected = "saved";
+    let clicks = 0;
+    const editor = { innerText: "在线不？", focus() { }, __reactFiberTest: { memoizedProps: { get viewConversationId() { return selected; } } } };
+    const button = { disabled: false, click() { clicks++; } };
+    client.evaluate = async (expression) => {
+        if (expression.includes("_selected_") || expression.includes('button[aria-label="发送"]'))
+            return (0, node_vm_1.runInNewContext)(expression, { window: { getSelection() { } }, document: {
+                    querySelectorAll: () => [{ className: "_selected_", getAttribute: () => selected }],
+                    querySelector: (selector) => selector.includes("contenteditable") ? editor : button,
+                    createRange: () => ({ selectNodeContents() { } }),
+                } });
+        return expression.includes("JSON.stringify") ? "[]" : true;
+    };
+    client.cdp = async (method) => { if (method === "Input.insertText")
+        selected = "other"; };
+    client.readInteraction = async () => undefined;
+    client.waitForReply = async () => ({ text: "不应发送" });
+    await strict_1.default.rejects(client.complete({ task: "在线不？", sessionId: "logical" }, undefined, false, "saved"), /桌面任务|会话|未发送/);
+    strict_1.default.equal(clicks, 0);
+});
+(0, node_test_1.default)("WorkBuddy 续接选项点击前再次核对桌面会话", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    let selected = "saved";
+    let clicks = 0;
+    const editor = { __reactFiberTest: { memoizedProps: { get viewConversationId() { return selected; } } } };
+    const option = { disabled: false, getAttribute: () => null, click() { clicks++; } };
+    client.pendingInteraction = { prompt: "确认？", options: ["确认"], sessionId: "logical" };
+    client.evaluate = async (expression) => {
+        if (expression.includes("option.click()"))
+            selected = "other";
+        if (expression.includes("_selected_") || expression.includes("option.click()"))
+            return (0, node_vm_1.runInNewContext)(expression, { document: {
+                    querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => selected }] : [option],
+                    querySelector: () => editor,
+                } });
+        return "[]";
+    };
+    client.waitForReply = async () => ({ text: "不应确认" });
+    await strict_1.default.rejects(client.complete({ task: "1", sessionId: "logical" }, undefined, false, "saved"), /桌面任务|会话|未确认/);
+    strict_1.default.equal(clicks, 0);
+});
 const node_vm_1 = require("node:vm");
 const workbuddy_1 = require("../src-ts/plugins/workbuddy");
 const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
@@ -256,6 +337,25 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
     strict_1.default.deepEqual(calls, [{ request, expert: undefined, newTask: false }]);
     await plugin.deactivate("stop");
 });
+(0, node_test_1.default)("WorkBuddy 模型来源复用独立会话并传递取消信号", async () => {
+    const controller = new AbortController();
+    const calls = [];
+    const plugin = new workbuddy_1.WorkbuddyPlugin({ client: {
+            connect: async () => undefined, close() { }, connected: () => true,
+            createTask: async () => { calls.push("new-task"); },
+            complete: async (request) => { calls.push(request); return { text: "结构化模型回复" }; },
+            direct: async () => "不应使用普通入口",
+        } });
+    await plugin.activate({ config: { transport: "cdp" }, logger: { info() { }, warn() { }, error() { } } });
+    try {
+        strict_1.default.deepEqual(await plugin.modelProvider.complete({ prompt: "模型提示", sessionId: "pa-thread", timeoutMs: 5000, signal: controller.signal }), { text: "结构化模型回复" });
+        strict_1.default.equal(calls[0], "new-task");
+        strict_1.default.deepEqual(calls[1], { task: "模型提示", sessionId: "pa-thread", timeoutMs: 5000, signal: controller.signal });
+    }
+    finally {
+        await plugin.deactivate("stop");
+    }
+});
 (0, node_test_1.default)("workbuddy 结构化输出接口不会隐式声明支持附件输入", async () => {
     let calls = 0;
     const client = {
@@ -366,7 +466,7 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
     strict_1.default.deepEqual(fetched, ['file:///C:/movie.bin', 'file:///C:/image.png']);
     strict_1.default.equal(cancelled, true);
 });
-(0, node_test_1.default)("workbuddy 默认允许长任务等待一小时并遵守显式超时", async () => {
+(0, node_test_1.default)("workbuddy 最多等待二十分钟并遵守较短的显式超时", async () => {
     const timeouts = [];
     const plugin = new workbuddy_1.WorkbuddyPlugin({ client: {
             connect: async () => undefined, close() { }, connected: () => true,
@@ -376,7 +476,8 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
     try {
         await plugin.handle({ task: '生成视频' });
         await plugin.handle({ task: '继续', timeoutMs: 1234 });
-        strict_1.default.deepEqual(timeouts, [3_600_000, 1234]);
+        await plugin.handle({ task: '旧版长等待', timeoutMs: 3_600_000 });
+        strict_1.default.deepEqual(timeouts, [1_200_000, 1234, 1_200_000]);
     }
     finally {
         await plugin.deactivate('stop');
@@ -488,6 +589,118 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
     strict_1.default.deepEqual(await client.complete({ task: "继续", timeoutMs: 2_200 }), { text: "视频已停止" });
     strict_1.default.ok(polls >= 2);
 });
+(0, node_test_1.default)("WorkBuddy 隐藏正文回退读取 textContent 且排除隐藏思考区", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    const content = { innerText: "", textContent: "内部推理已完成 10s在线。有事直接说。", querySelectorAll: () => [{ innerText: "", textContent: "内部推理" }, { innerText: "", textContent: "已完成 10s" }] };
+    const message = { getAttribute: () => "reply", querySelector: (selector) => selector === "[data-assistant-content]" ? content : {} };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: { querySelectorAll: () => [message], querySelector: () => null } });
+    client.readInteraction = async () => undefined;
+    client.readArtifacts = async () => [];
+    strict_1.default.deepEqual(await client.waitForReply(new Set(), 1200), { text: "在线。有事直接说。" });
+});
+(0, node_test_1.default)("WorkBuddy 等待期间会话切换立即拒绝其他任务的正文和产物", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    const forwarded = [];
+    let artifacts = 0;
+    client.conversationId = async () => "other";
+    const message = { getAttribute: () => "foreign", querySelector: (selector) => selector === "[data-assistant-content]" ? { innerText: "其他任务结果" } : {} };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: {
+            querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => "other" }] : [message],
+            querySelector: () => ({}),
+        } });
+    client.readInteraction = async () => undefined;
+    client.readArtifacts = async () => { artifacts++; return []; };
+    await strict_1.default.rejects(client.waitForReply(new Set(), 2000, undefined, undefined, "logical", async (text) => { forwarded.push(text); }, undefined, "saved"), (error) => error.code === "WORKBUDDY_CONVERSATION_CHANGED");
+    strict_1.default.deepEqual(forwarded, []);
+    strict_1.default.equal(artifacts, 0);
+});
+(0, node_test_1.default)("WorkBuddy 缺少新任务标识时不能把首次选中的旧会话当成本次任务", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    client.conversationId = async () => "other";
+    const message = { getAttribute: () => "foreign", querySelector: (selector) => selector === "[data-assistant-content]" ? { innerText: "其他任务结果" } : {} };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: {
+            querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => "other" }] : [message],
+            querySelector: () => ({}),
+        } });
+    client.readInteraction = async () => undefined;
+    client.readArtifacts = async () => [];
+    await strict_1.default.rejects(client.waitForReply(new Set(), 1200, undefined, undefined, "logical", undefined, undefined, null), (error) => error.code === "WORKBUDDY_CONVERSATION_CHANGED");
+});
+for (const [mode, label] of [["created", "等待侧栏及输入组件同步后绑定新 ID"], ["navigation", "提交后立即点击旧任务时拒绝绑定"], ["timeout", "分配 ID 超时后清理监听"], ["changed", "输入组件就绪前再切换时拒绝绑定"]]) {
+    (0, node_test_1.default)(`WorkBuddy 首页${label}`, async () => {
+        const client = new cdp_client_1.WorkbuddyCdpClient({ timeoutMs: 150 });
+        let selected = null;
+        let input = null;
+        const listeners = new Map();
+        const bound = [];
+        const editor = { innerText: "", focus() { }, __reactFiberTest: { memoizedProps: { get viewConversationId() { return input; } } } };
+        const button = { disabled: false, click() {
+                listeners.get("click")?.();
+                if (mode === "created" || mode === "changed")
+                    selected = "created";
+                if (mode === "created")
+                    setTimeout(() => { input = "created"; }, 0);
+                if (mode === "navigation")
+                    setTimeout(() => { listeners.get("click")?.(); selected = input = "other"; }, 0);
+                if (mode === "changed")
+                    setTimeout(() => { selected = input = "other"; }, 75);
+            } };
+        client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { setTimeout, clearTimeout, setInterval, clearInterval, window: { getSelection() { } }, document: {
+                querySelectorAll: (selector) => selector === "[data-conversation-id]" ? ["other", ...(selected === "created" ? ["created"] : [])].map(id => ({ className: selected === id ? "_selected_" : "", getAttribute: () => id })) : [],
+                querySelector: (selector) => selector.includes("contenteditable") ? editor : selector.includes('button[aria-label="发送"]') ? button : null,
+                createRange: () => ({ selectNodeContents() { } }),
+                addEventListener: (name, callback) => { listeners.set(name, callback); },
+                removeEventListener: (name) => { listeners.delete(name); },
+            } });
+        client.cdp = async (method, params) => { if (method === "Input.insertText")
+            editor.innerText = params.text; };
+        client.waitForReply = async (...args) => { bound.push(args[7]); return { text: "已完成" }; };
+        const response = client.complete({ task: "在线不？", sessionId: "logical" }, undefined, false, null);
+        if (mode === "created") {
+            strict_1.default.deepEqual(await response, { text: "已完成" });
+            strict_1.default.deepEqual(bound, ["created"]);
+        }
+        else {
+            await strict_1.default.rejects(response, /会话|任务标识/);
+            strict_1.default.deepEqual(bound, []);
+        }
+        strict_1.default.equal(listeners.size, 0);
+    });
+}
+(0, node_test_1.default)("WorkBuddy 正文快照之后切换会话也不能读取其他任务的确认选项", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    let selected = "saved";
+    const editor = { __reactFiberTest: { memoizedProps: { get viewConversationId() { return selected; } } } };
+    const message = { getAttribute: () => "reply", querySelector: (selector) => selector === "[data-assistant-content]" ? { innerText: "本轮正文" } : {} };
+    const option = { innerText: "其他会话选项", closest: () => ({ innerText: "其他会话确认？", getAttribute: () => "foreign", querySelector: () => ({ innerText: "其他会话正文" }) }) };
+    client.evaluate = async (expression) => {
+        if (expression.includes("conversation-input-area"))
+            selected = "other";
+        return (0, node_vm_1.runInNewContext)(expression, { document: {
+                querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => selected }] : selector === ".cr-agent[data-message-id]" ? [message] : [option],
+                querySelector: (selector) => selector.includes("contenteditable") ? editor : null,
+            } });
+    };
+    client.readArtifacts = async () => [];
+    await strict_1.default.rejects(client.waitForReply(new Set(), 1200, undefined, undefined, "logical", undefined, undefined, "saved"), (error) => error.code === "WORKBUDDY_CONVERSATION_CHANGED");
+    strict_1.default.equal(client.pendingInteraction, undefined);
+});
+(0, node_test_1.default)("WorkBuddy 产物读取前及异步读取期间切换会话均拒绝成功回传", async () => {
+    for (const initial of ["other", "saved"]) {
+        const client = new cdp_client_1.WorkbuddyCdpClient();
+        let selected = initial;
+        const editor = { __reactFiberTest: { memoizedProps: { get viewConversationId() { return selected; } } } };
+        const message = { getAttribute: () => "reply", querySelector: () => null, querySelectorAll: () => [{ getAttribute: () => "file:///result.txt" }] };
+        client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { URL, btoa, fetch: async () => {
+                selected = "other";
+                return new Response("结果", { headers: { "content-type": "text/plain" } });
+            }, document: {
+                querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => selected }] : [message],
+                querySelector: () => editor,
+            } });
+        await strict_1.default.rejects(client.readArtifacts(new Set(["reply"]), "saved"), (error) => error.code === "WORKBUDDY_CONVERSATION_CHANGED");
+    }
+});
 (0, node_test_1.default)("workbuddy CDP 选择后仍在处理时不能把工具状态当成最终回复", async () => {
     const client = new cdp_client_1.WorkbuddyCdpClient();
     let polls = 0;
@@ -827,7 +1040,7 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
 });
 (0, node_test_1.default)("workbuddy 清单使用 CDP 插件能力并保持发布版本一致", async () => {
     strict_1.default.equal(workbuddy_1.workbuddyManifest.id, "workbuddy");
-    strict_1.default.deepEqual(workbuddy_1.workbuddyManifest.capabilities, ["chat", "health"]);
+    strict_1.default.deepEqual(workbuddy_1.workbuddyManifest.capabilities, ["chat", "health", "model-provider"]);
     for (const file of ["packages/workbuddy/package.json", "packages/workbuddy/plugin.json", "plugins/workbuddy/plugin.json"]) {
         const metadata = JSON.parse(await (0, promises_1.readFile)(file, "utf8"));
         strict_1.default.equal(workbuddy_1.workbuddyManifest.version, metadata.version, file);
@@ -1028,7 +1241,7 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
             querySelectorAll: (selector) => selector === '.cr-agent[data-message-id]' ? Array(messages).fill({}) : selector === '[data-conversation-id]' ? conversationId ? [{ className: "_selected_old", getAttribute: () => conversationId }] : [] : [button],
             querySelector: () => editor,
         };
-        const client = new cdp_client_1.WorkbuddyCdpClient({ timeoutMs: 1 });
+        const client = new cdp_client_1.WorkbuddyCdpClient({ timeoutMs: 1000 });
         const internal = client;
         internal.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document, location: { href: "file:///D:/WorkBuddy/index.html?login=private#state" } });
         await client.createTask();
@@ -1050,6 +1263,127 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
     strict_1.default.equal(selected, false);
     strict_1.default.equal(messages, 0);
 });
+(0, node_test_1.default)("workbuddy CDP 新建时立即分配会话 ID 后允许发送首条消息", async () => {
+    let conversationId = "old-conversation";
+    let messages = 1;
+    const editor = { innerText: "", focus() { }, querySelector: () => null, __reactFiberTest: { memoizedProps: { get viewConversationId() { return conversationId; } } }, cloneNode: () => ({ textContent: "", querySelectorAll: () => [{ remove() { } }] }) };
+    const createButton = { textContent: "新建任务", disabled: false, getAttribute: () => null, click: () => { conversationId = "new-conversation"; messages = 0; } };
+    const sendButton = { disabled: false, click() { } };
+    const document = {
+        querySelectorAll: (selector) => selector === '.cr-agent[data-message-id]' ? Array(messages).fill({}) : selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => conversationId }] : selector.includes("conversation-input-area") ? [] : [createButton],
+        querySelector: (selector) => selector.includes("contenteditable") ? editor : selector.includes('button[aria-label="发送"]') ? sendButton : null,
+        createRange: () => ({ selectNodeContents() { } }),
+    };
+    const client = new cdp_client_1.WorkbuddyCdpClient({ timeoutMs: 500 });
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document, location: { href: "app://workbuddy/task/old" }, window: { getSelection() { } } });
+    client.cdp = async (_method, params) => { if (params.text)
+        editor.innerText = params.text; };
+    let bound;
+    client.waitForReply = async (...args) => { bound = args[7]; return { text: "已完成" }; };
+    client.connected = () => true;
+    client.connect = async () => undefined;
+    const plugin = new workbuddy_1.WorkbuddyPlugin({ client });
+    await plugin.activate({ config: { transport: "cdp", dailyPoints: false }, logger: { info() { }, warn() { }, error() { } } });
+    strict_1.default.equal((await plugin.handle({ task: "在线不？", sessionId: "logical" })).text, "已完成");
+    strict_1.default.equal(bound, "new-conversation");
+});
+for (const mode of ["同步", "微任务", "异步", "切换", "超时", "目录中切换", "搜索延迟", "搜索回退", "搜索失败"]) {
+    (0, node_test_1.default)(`WorkBuddy 新建逻辑会话后专家召唤${mode}分配 ID 时校验专家会话`, async () => {
+        const client = new cdp_client_1.WorkbuddyCdpClient({ timeoutMs: 220 });
+        let selected = "old-conversation";
+        let editorId = selected;
+        let messages = 1;
+        let sentTo;
+        let catalogOpen = false;
+        let summoned = false;
+        let searchReady = mode !== "搜索延迟" && mode !== "搜索回退" && mode !== "搜索失败";
+        const listeners = new Map();
+        const editor = { innerText: "", focus() { }, querySelector: () => null, cloneNode: () => ({ textContent: "", querySelectorAll: () => [] }), __reactFiberTest: { memoizedProps: { get viewConversationId() { return editorId; } } } };
+        const create = { textContent: "新建任务", disabled: false, getAttribute: () => null, click() { selected = editorId = "new-conversation"; messages = 0; } };
+        const send = { disabled: false, click() { sentTo = selected; } };
+        const summon = { getAttribute: () => null, click() {
+                strict_1.default.equal(selected, "new-conversation");
+                summoned = true;
+                catalogOpen = false;
+                if (mode === "同步" || mode === "搜索延迟" || mode === "搜索回退")
+                    selected = editorId = "expert-created";
+                if (mode === "微任务")
+                    void Promise.resolve().then(() => { selected = editorId = "expert-created"; });
+                if (mode === "异步")
+                    setTimeout(() => { selected = "expert-created"; setTimeout(() => { editorId = selected; }, 60); }, 20);
+                if (mode === "切换")
+                    setTimeout(() => { listeners.get("click")?.(); selected = editorId = "foreign"; }, 0);
+            } };
+        const card = { querySelector: (selector) => selector.includes("_role_") ? { textContent: searchReady ? "测试专家" : "其他专家" } : summon };
+        const input = { value: "测试专家", dispatchEvent() { }, focus() { } };
+        const entry = { textContent: "专家·技能·连接器", click() { catalogOpen = true; if (mode === "目录中切换")
+                selected = "foreign"; if (mode === "搜索延迟")
+                setTimeout(() => { searchReady = true; }, 20); } };
+        client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { setTimeout, clearTimeout, setInterval, clearInterval, location: { href: "app://workbuddy/task/old" }, HTMLInputElement: { prototype: {} }, Event: class {
+            }, window: { getSelection() { } }, document: {
+                querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => selected }] : selector === '.cr-agent[data-message-id]' ? Array(messages).fill({}) : selector === "article[role=\"button\"]" ? [card] : selector.includes("expert_summon_click") ? [summon] : selector === 'button,[role="button"],a' ? [entry] : selector === 'button,[role="button"],[role="tab"]' ? [create] : [],
+                querySelector: (selector) => selector.includes("contenteditable") ? catalogOpen ? null : editor : selector.includes("搜索专家") ? input : selector.includes('button[aria-label="发送"]') ? send : null,
+                createRange: () => ({ selectNodeContents() { } }),
+                addEventListener: (name, callback) => { listeners.set(name, callback); },
+                removeEventListener: (name) => { listeners.delete(name); },
+            } });
+        client.cdp = async (_method, params) => { if (params.text) {
+            if (catalogOpen) {
+                input.value = params.text;
+                if (mode === "搜索回退" && params.text === "测试")
+                    searchReady = true;
+            }
+            else
+                editor.innerText = params.text;
+        } };
+        let bound;
+        client.waitForReply = async (...args) => { bound = args[7]; return { text: "专家已回复" }; };
+        client.connected = () => true;
+        client.connect = async () => undefined;
+        const plugin = new workbuddy_1.WorkbuddyPlugin({ client });
+        await plugin.activate({ config: { transport: "cdp", dailyPoints: false }, logger: { info() { }, warn() { }, error() { } } });
+        const response = plugin.handle({ task: "只要测试专家，帮我处理", sessionId: "logical" });
+        if (mode === "切换" || mode === "超时" || mode === "目录中切换" || mode === "搜索失败") {
+            await strict_1.default.rejects(response, /会话|未输入|未就绪|未找到/);
+            strict_1.default.equal(sentTo, undefined);
+            strict_1.default.equal(bound, undefined);
+        }
+        else {
+            strict_1.default.equal((await response).text, "专家已回复");
+            strict_1.default.equal(sentTo, "expert-created");
+            strict_1.default.equal(bound, "expert-created");
+            strict_1.default.equal(plugin.activeDesktopConversationId, "expert-created");
+        }
+        strict_1.default.equal(summoned, mode !== "目录中切换" && mode !== "搜索失败");
+        strict_1.default.equal(listeners.size, 0);
+    });
+}
+(0, node_test_1.default)("WorkBuddy 新建状态核验后切换到旧会话时不能输入首条消息", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient({ timeoutMs: 20 });
+    let selected = "old";
+    let switched = false;
+    let inputs = 0;
+    const editor = { innerText: "", focus() { }, querySelector: () => null, cloneNode: () => ({ textContent: "", querySelectorAll: () => [] }), __reactFiberTest: { memoizedProps: { get viewConversationId() { return selected; } } } };
+    const create = { textContent: "新建任务", disabled: false, getAttribute: () => null, click() { selected = "created"; } };
+    const document = {
+        querySelectorAll: (selector) => selector === "[data-conversation-id]" ? [{ className: "_selected_", getAttribute: () => selected }] : selector.includes("cr-agent") || selector.includes("conversation-input-area") ? [] : [create],
+        querySelector: () => editor,
+        createRange: () => ({ selectNodeContents() { } }),
+    };
+    client.evaluate = async (expression) => {
+        const result = (0, node_vm_1.runInNewContext)(expression, { document, location: { href: "app://task" }, window: { getSelection() { } } });
+        if (expression.includes("clearedToUnselected") && result) {
+            selected = "foreign";
+            switched = true;
+        }
+        return result;
+    };
+    client.cdp = async () => { inputs++; };
+    client.waitForReply = async () => ({ text: "不应发送" });
+    await strict_1.default.rejects(async () => { await client.createTask(); await client.complete({ task: "在线不？", sessionId: "logical" }, undefined, false, null); }, /会话|输入|新任务/);
+    strict_1.default.equal(switched, true);
+    strict_1.default.equal(inputs, 0);
+});
 (0, node_test_1.default)("workbuddy CDP 等待新建任务入口完成渲染后再点击", async () => {
     let lookups = 0;
     let clicked = false;
@@ -1385,6 +1719,17 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
     client.readArtifacts = async () => { throw Object.assign(new Error("文件已删除"), { code: "ENOENT" }); };
     await strict_1.default.rejects(client.waitForReply(new Set(), 2000), /WORKBUDDY_ARTIFACT_READ_FAILED.*读取.*ENOENT/);
 });
+(0, node_test_1.default)("WorkBuddy 回复正文排除已完成耗时，保留模型决策 JSON", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    const decision = JSON.stringify({ role: "assistant", content: "测试标记", tool_calls: [] });
+    const content = { innerText: `已完成 13s\n\n${decision}`, querySelectorAll: (selector) => selector.includes(".cr-process-row") ? [{ innerText: "已完成 13s" }] : [] };
+    const message = { getAttribute: () => "reply", querySelector: (selector) => selector === "[data-assistant-content]" ? content : {} };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: { querySelectorAll: () => [message], querySelector: () => null } });
+    client.readInteraction = async () => undefined;
+    client.readArtifacts = async () => [];
+    const response = await client.waitForReply(new Set(), 2000);
+    strict_1.default.equal(response.text, decision);
+});
 (0, node_test_1.default)("WorkBuddy 停止期间完成的积分检查不重新启动定时器", async () => {
     const dataDir = await (0, promises_1.mkdtemp)((0, node_path_1.join)((0, node_os_1.tmpdir)(), "wb-stop-points-"));
     let started;
@@ -1468,3 +1813,153 @@ const cdp_client_1 = require("../src-ts/plugins/workbuddy/cdp-client");
         return true;
     });
 });
+(0, node_test_1.default)("WorkBuddy 模型排队取消不新建任务，断线不恢复桌面", async () => {
+    let connected = true;
+    let creates = 0;
+    let reconnects = 0;
+    let release;
+    const plugin = new workbuddy_1.WorkbuddyPlugin({ client: {
+            connect: async () => undefined, close() { }, connected: () => connected,
+            createTask: async () => { creates++; }, direct: async () => "回复",
+            complete: async (request) => { if (request.task === "用户任务")
+                await new Promise(resolve => { release = resolve; }); return { text: "回复" }; },
+        } });
+    await plugin.activate({ config: { transport: "cdp" }, logger: { info() { }, warn() { }, error() { } } });
+    const running = plugin.handle({ task: "用户任务" });
+    await new Promise(resolve => setImmediate(resolve));
+    const controller = new AbortController();
+    const queued = plugin.modelProvider.complete({ prompt: "取消提示", sessionId: "pa-queued", timeoutMs: 5000, signal: controller.signal }).catch(() => undefined);
+    controller.abort(new Error("排队取消"));
+    release();
+    await Promise.all([running, queued]);
+    await new Promise(resolve => setImmediate(resolve));
+    const afterCancel = creates;
+    plugin.reconnect = async () => { reconnects++; connected = true; };
+    connected = false;
+    const error = await plugin.modelProvider.complete({ prompt: "离线提示", sessionId: "pa-offline", timeoutMs: 5000, signal: new AbortController().signal }).then(() => undefined, (value) => value);
+    await plugin.deactivate("stop");
+    strict_1.default.equal(afterCancel, 0);
+    strict_1.default.ok(error);
+    strict_1.default.equal(reconnects, 0);
+});
+(0, node_test_1.default)("WorkBuddy 自动 CDP 模式提供模型能力，模型与用户任务恢复各自原生会话", async () => {
+    let active = "user-chat";
+    let sequence = 0;
+    const calls = [];
+    const plugin = new workbuddy_1.WorkbuddyPlugin({ client: {
+            connect: async () => undefined, close() { }, connected: () => true,
+            conversationId: async () => active,
+            activateConversation: async (id) => { active = id; },
+            createTask: async () => { active = `model-${++sequence}`; },
+            complete: async (request) => { calls.push(`${active}:${request.task}`); return { text: "回复" }; },
+            direct: async () => "回复",
+        } });
+    await plugin.activate({ config: {}, logger: { info() { }, warn() { }, error() { } } });
+    try {
+        const provider = plugin.modelProvider;
+        strict_1.default.ok(provider);
+        for (const prompt of ["调用视频生成专家", "第二轮"]) {
+            await provider.complete({ prompt, sessionId: "pa-thread", timeoutMs: 5000, signal: new AbortController().signal });
+            strict_1.default.equal(active, "user-chat");
+            await plugin.handle({ task: "用户续聊" });
+        }
+        strict_1.default.deepEqual(calls, ["model-1:调用视频生成专家", "user-chat:用户续聊", "model-1:第二轮", "user-chat:用户续聊"]);
+    }
+    finally {
+        await plugin.deactivate("stop");
+    }
+});
+(0, node_test_1.default)("WorkBuddy 用户桌面忙碌或有草稿时模型请求不切换会话", async () => {
+    for (const state of [{ generating: true }, { draft: true }, { waiting: true }]) {
+        const calls = [];
+        const plugin = new workbuddy_1.WorkbuddyPlugin({ client: {
+                connect: async () => undefined, close() { }, connected: () => true,
+                historyAvailability: async () => ({ waiting: false, ...state }),
+                createTask: async () => { calls.push("create"); },
+                activateConversation: async () => { calls.push("switch"); },
+                complete: async () => { calls.push("send"); return { text: "不应发送" }; },
+                direct: async () => "不应发送",
+            } });
+        await plugin.activate({ config: { transport: "cdp" }, logger: { info() { }, warn() { }, error() { } } });
+        try {
+            await strict_1.default.rejects(plugin.modelProvider.complete({ prompt: "模型提示", sessionId: "pa-busy", timeoutMs: 5000, signal: new AbortController().signal }), /用户草稿|正在生成|待确认/);
+            strict_1.default.deepEqual(calls, []);
+        }
+        finally {
+            await plugin.deactivate("stop");
+        }
+    }
+});
+(0, node_test_1.default)("WorkBuddy 模型在点击切换会话前再次保护新增草稿", async () => {
+    const client = new cdp_client_1.WorkbuddyCdpClient();
+    let clicks = 0;
+    const editor = { cloneNode: () => ({ textContent: "刚输入的草稿", querySelectorAll: () => [] }) };
+    client.evaluate = async (expression) => (0, node_vm_1.runInNewContext)(expression, { document: {
+            querySelector: (selector) => selector.includes("contenteditable") ? editor : null,
+            querySelectorAll: () => [{ matches: () => true, getAttribute: (name) => name === "data-conversation-id" ? "model" : null, click() { clicks++; } }],
+        } });
+    await strict_1.default.rejects(client.activateConversation("model", true), /用户草稿|未发送内容/);
+    strict_1.default.equal(clicks, 0);
+});
+(0, node_test_1.default)("WorkBuddy 运行中 CDP 消失后重新启动桥接实例，普通连接和页面错误不启动", async (t) => {
+    const { mock } = await Promise.resolve().then(() => __importStar(require("node:test")));
+    const desktopModule = await Promise.resolve().then(() => __importStar(require("../src-ts/plugins/workbuddy/desktop-launch")));
+    const desktop = desktopModule.launchWorkbuddyDesktop ? desktopModule : desktopModule.default;
+    if (typeof desktop?.launchWorkbuddyDesktop !== "function") {
+        t.skip("当前测试加载器未暴露可替换的桌面启动函数");
+        return;
+    }
+    let connected = true;
+    let launches = 0;
+    let pageError = false;
+    const plugin = new workbuddy_1.WorkbuddyPlugin();
+    plugin.client = {
+        connect: async () => { if (!connected)
+            throw Object.assign(new Error("connect ECONNREFUSED 127.0.0.1:9229"), { code: "ECONNREFUSED" }); },
+        close() { }, connected: () => connected,
+        direct: async () => { if (pageError)
+            throw new Error("页面入口不可用"); return "恢复成功"; },
+    };
+    let stub;
+    try {
+        stub = mock.method(desktop, "launchWorkbuddyDesktop", (file, args, profile, port) => {
+            launches++;
+            strict_1.default.equal(file, process.execPath);
+            strict_1.default.ok(args.includes("--remote-debugging-port=9229"));
+            strict_1.default.match(profile, /desktop$/);
+            strict_1.default.equal(port, 9229);
+            connected = true;
+            return { unref() { }, exitCode: 0 };
+        });
+    }
+    catch {
+        t.skip("当前测试加载器不支持替换桌面启动模块");
+        return;
+    }
+    try {
+        await plugin.connectOrStart({ executablePath: process.execPath }, { info() { }, warn() { }, error() { } }, 9229);
+        strict_1.default.equal(launches, 0);
+        connected = false;
+        strict_1.default.equal((await plugin.handle({ task: "恢复测试" })).text, "恢复成功");
+        strict_1.default.equal(launches, 1);
+        connected = false;
+        strict_1.default.equal((await plugin.handle({ task: "再次恢复" })).text, "恢复成功");
+        strict_1.default.equal(launches, 2);
+        pageError = true;
+        await strict_1.default.rejects(plugin.handle({ task: "页面失败" }), /页面入口不可用/);
+        strict_1.default.equal(launches, 2);
+        connected = false;
+        await plugin.connectOrStart({ executablePath: process.execPath, autoStart: false }, { info() { }, warn() { }, error() { } }, 9229).catch(() => { });
+        await strict_1.default.rejects(plugin.handle({ task: "禁止自动启动" }), /ECONNREFUSED/);
+        strict_1.default.equal(launches, 2);
+    }
+    finally {
+        stub.mock.restore();
+        await plugin.deactivate("stop");
+    }
+});
+(0, node_test_1.default)("WorkBuddy 升级后 CDP 端口未监听时提示现有桌面恢复操作", async () => {
+    const failure = new Error("fetch failed", { cause: Object.assign(new Error("拒绝连接"), { code: "ECONNREFUSED" }) });
+    const client = new cdp_client_1.WorkbuddyCdpClient({ port: 19229, fetchImpl: async () => { throw failure; } });
+    await strict_1.default.rejects(() => client.connect(), (error) => /19229/.test(error.message) && /重启桌面/.test(error.message) && error.cause === failure);
+});