@zhushanwen/pi-subagent-workflow 8.5.0 → 8.7.0

package/src/execution/__tests__/bg-notify-render.test.ts

@@ -1,256 +0,0 @@
-// src/__tests__/bg-notify-render.test.ts
-//
-// background 完成通知的渲染测试。
-//
-// 核心契约：renderBgNotifyMessage 必须返回带紫色背景 + 圆角边框的组件——
-// Pi 的 CustomMessageComponent 对 customRenderer 返回值是「裸 addChild」，
-// 返回裸 Text 会丢失紫色背景。本测试钉死「施加 customMessageBg + 边框」这一行为。
-//
-// 测试用 mock theme（bg 记录调用色 token），不依赖真实 Pi Theme。
-
-import type { Theme } from "@earendil-works/pi-coding-agent";
-import { describe, expect, it } from "vitest";
-
-import { renderBgNotifyMessage } from "../../interface/bg-notify-render.ts";
-
-/**
- * 构造 mock theme：bg 记录被调用的色 token，fg/bold 透传文本。
- *
- * 真实 theme.fg/bg 会包裹 ANSI 码；这里透传纯文本，让断言可读。
- * 边框/背景 token 通过 bgColors 记录。
- */
-function makeTheme(): { theme: Theme; bgColors: string[]; fgColors: string[] } {
-  const bgColors: string[] = [];
-  const fgColors: string[] = [];
-  const theme = {
-    fg: (tag: string, text: string) => {
-      fgColors.push(tag);
-      return text;
-    },
-    bold: (text: string) => text,
-    bg: (color: string, text: string) => {
-      bgColors.push(color);
-      return text;
-    },
-  };
-  return { theme: theme as Theme, bgColors, fgColors };
-}
-
-describe("renderBgNotifyMessage", () => {
-  it("单条 done → 施加 customMessageBg 背景 + 边框 + 内容可见", () => {
-    const { theme, bgColors } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: "worker", id: "bg-1", result: "All green" } },
-      { expanded: false },
-      theme,
-    );
-
-    expect(comp).toBeDefined();
-    const lines = comp!.render(80);
-    const joined = lines.join("\n");
-    // 紫色背景施加（内容行 ≥ 1 次）
-    expect(bgColors.filter((c) => c === "customMessageBg").length).toBeGreaterThanOrEqual(1);
-    // 圆角边框
-    expect(lines[0]).toContain("╭");
-    expect(lines[0]).toContain("╮");
-    expect(lines[lines.length - 1]).toContain("╰");
-    expect(lines[lines.length - 1]).toContain("╯");
-    for (let i = 1; i < lines.length - 1; i++) {
-      expect(lines[i]).toContain("│");
-    }
-    // 前景内容仍在
-    expect(joined).toContain("worker");
-    expect(joined).toContain("All green");
-  });
-
-  it("id 用 shortId 截断（bg-tag-seq-ts → bg-tag-seq）", () => {
-    const { theme } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: "w", id: "bg-f6f731-10-1719500000000", result: "ok" } },
-      { expanded: false },
-      theme,
-    );
-    const joined = comp!.render(80).join("\n");
-    // shortId 对 background id 取前 3 段（bg/tag/seq）
-    expect(joined).toContain("bg-f6f731-10");
-    // 完整时间戳不应出现
-    expect(joined).not.toContain("1719500000000");
-  });
-
-  it("单条 failed → 内容含 Error + agent", () => {
-    const { theme } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: "scout", id: "bg-2", error: "boom" } },
-      { expanded: false },
-      theme,
-    );
-    expect(comp).toBeDefined();
-    const joined = comp!.render(80).join("\n");
-    expect(joined).toContain("scout");
-    expect(joined).toContain("Error");
-    expect(joined).toContain("boom");
-  });
-
-  it("批量 → 施加 customMessageBg + 边框，每条 agent 可见", () => {
-    const { theme, bgColors } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      {
-        details: {
-          batch: true,
-          items: [
-            { status: "closed", agent: "alpha", id: "1", result: "r1" },
-            { status: "closed", agent: "beta", id: "2", error: "e2" },
-          ],
-        },
-      },
-      { expanded: false },
-      theme,
-    );
-    expect(comp).toBeDefined();
-    const lines = comp!.render(80);
-    const joined = lines.join("\n");
-    expect(bgColors).toContain("customMessageBg");
-    expect(joined).toContain("alpha");
-    expect(joined).toContain("beta");
-    // 边框
-    expect(lines[0]).toContain("╭");
-    expect(lines[lines.length - 1]).toContain("╰");
-  });
-
-  it("closed + closedReason=cancelled → 内容含 cancelled（v4 B-1 cancelled 折入 closed）", () => {
-    const { theme } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", closedReason: "cancelled", agent: "w", id: "bg-3" } },
-      { expanded: false },
-      theme,
-    );
-    expect(comp).toBeDefined();
-    expect(comp!.render(80).join("\n")).toContain("cancelled");
-  });
-
-  it("details 缺失/结构不全 → undefined（走 Pi 默认渲染兜底）", () => {
-    const { theme } = makeTheme();
-    expect(renderBgNotifyMessage({ details: undefined }, { expanded: false }, theme)).toBeUndefined();
-    expect(renderBgNotifyMessage({ details: { foo: 1 } }, { expanded: false }, theme)).toBeUndefined();
-    // 缺 status / agent
-    expect(
-      renderBgNotifyMessage({ details: { id: "x", agent: "w" } }, { expanded: false }, theme),
-    ).toBeUndefined();
-    expect(
-      renderBgNotifyMessage({ details: { status: "closed", id: "x" } }, { expanded: false }, theme),
-    ).toBeUndefined();
-  });
-
-  it("details 无效时不施加任何背景（兜底路径不应泄漏背景色）", () => {
-    const { theme, bgColors } = makeTheme();
-    renderBgNotifyMessage({ details: undefined }, { expanded: false }, theme);
-    expect(bgColors).not.toContain("customMessageBg");
-  });
-
-  // ── model 显示 ──
-
-  it("record 带 model → 内容含 model 字符串（agent 后、状态描述前）", () => {
-    const { theme } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: "general-purpose", model: "anthropic/sonnet-4-5", id: "bg-1", result: "ok" } },
-      { expanded: false },
-      theme,
-    );
-    const joined = comp!.render(80).join("\n");
-    expect(joined).toContain("anthropic/sonnet-4-5");
-    // agent 和状态仍在
-    expect(joined).toContain("general-purpose");
-    expect(joined).toContain("finished");
-  });
-
-  it("record 无 model → 向后兼容，不渲染 model 段不崩", () => {
-    const { theme } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: "worker", id: "bg-2", result: "r" } },
-      { expanded: false },
-      theme,
-    );
-    const joined = comp!.render(80).join("\n");
-    expect(joined).toContain("worker");
-    expect(joined).toContain("finished");
-  });
-
-  // ── ANSI 背景 safety ──
-
-  it("着色行截断的 \\x1b[0m 被替换为精确 reset（不破坏背景）", () => {
-    // mock theme 产生真实 ANSI（fg 用 italic \x1b[3m..\x1b[23m，bold 用 \x1b[1m..\x1b[22m）
-    const theme = {
-      fg: (_tag: string, text: string) => `\x1b[3m${text}\x1b[23m`,
-      bold: (text: string) => `\x1b[1m${text}\x1b[22m`,
-      bg: (_color: string, text: string) => `\x1b[48;5;95m${text}\x1b[49m`,
-    } as Theme;
-    // 关键：用窄 width + 长 agent 名，让 head 行在 truncLine 内被真正截断
-    // head 行含 ANSI（fg 包裹），截断时 activeStyles 非空 → 会产生 \x1b[0m
-    const longAgent = "a".repeat(60);
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: longAgent, id: "bg-1", result: "ok" } },
-      { expanded: false },
-      theme,
-    );
-    const lines = comp!.render(30);
-    // 内容行（非边框）不应含 \x1b[0m（会破坏背景）
-    const contentLines = lines.slice(1, -1);
-    expect(contentLines.length).toBeGreaterThan(0);
-    for (const line of contentLines) {
-      expect(line).not.toContain("\x1b[0m");
-    }
-    // 正向验证：sanitizeAnsiForBg 把 \x1b[0m 替换成了精确 reset（\x1b[39m）
-    // 如果截断真的发生了，内容行应含 \x1b[39m（sanitize 的替换产物）
-    const hasSanitized = contentLines.some((l) => l.includes("\x1b[39m"));
-    expect(hasSanitized).toBe(true);
-  });
-
-  // ── 窄宽度退化 ──
-
-  it("极窄宽度（width < 5）退化：无边框，不崩溃，有背景", () => {
-    const { theme, bgColors } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      { details: { status: "closed", agent: "w", id: "bg-1", result: "ok" } },
-      { expanded: false },
-      theme,
-    );
-    // width=3 < MIN_BORDER_WIDTH(5)，应退化为无边框模式
-    const lines = comp!.render(3);
-    expect(lines.length).toBeGreaterThan(0);
-    // 不应有边框字符（退化模式）
-    for (const line of lines) {
-      expect(line).not.toContain("╭");
-      expect(line).not.toContain("│");
-    }
-    // 仍施加背景
-    expect(bgColors).toContain("customMessageBg");
-  });
-
-  it("批量场景边框完整：所有中间行含 │", () => {
-    const { theme } = makeTheme();
-    const comp = renderBgNotifyMessage(
-      {
-        details: {
-          batch: true,
-          items: [
-            { status: "closed", agent: "alpha", id: "1", result: "r1" },
-            { status: "closed", agent: "beta", id: "2", error: "e2" },
-            { status: "closed", closedReason: "cancelled", agent: "gamma", id: "3" },
-          ],
-        },
-      },
-      { expanded: false },
-      theme,
-    );
-    const lines = comp!.render(80);
-    // 顶底边框
-    expect(lines[0]).toContain("╭");
-    expect(lines[0]).toContain("╮");
-    expect(lines[lines.length - 1]).toContain("╰");
-    expect(lines[lines.length - 1]).toContain("╯");
-    // 所有中间行都应有左右 │
-    for (let i = 1; i < lines.length - 1; i++) {
-      expect(lines[i]).toContain("│");
-    }
-  });
-});

package/src/execution/__tests__/channel-registry-handshake.test.ts

@@ -1,243 +0,0 @@
-// src/execution/__tests__/channel-registry-handshake.test.ts
-//
-// 决策 D（修复 M4）：channel handshake 协议升级测试。
-//
-// 测试对象：extensions/universal/subagent-workflow/src/execution/channel-registry-access.ts
-//
-// 协议契约（与 ask-user 侧严格对齐）：
-//   key 字面量：     "@zhushanwen/pi-subagents.channelHandshake"
-//   handshake 形状： { version: 1, registry?: UiChannelRegistry, pending: [{channel,handler}] }
-//   canonical 唯一创建点：getOrCreateChannelRegistry（仅 subagent-workflow 侧创建）
-//
-// 关键不变量：
-//   1. canonical registry 永远由 subagent-workflow 创建
-//   2. ask-user 先到时只往 slot.pending 推，subagent-workflow 来时 flush
-//   3. 重复调用返回同一实例（===）
-//   4. version !== 1 时 warn + 丢弃重建
-
-import { afterEach, describe, expect, it, vi } from "vitest";
-
-// Mock 共享 logger，让 logger.warn 可被 spy（源码已从 console.warn 改为 logger.warn）
-const { loggerMock } = vi.hoisted(() => ({
-  loggerMock: {
-    debug: vi.fn(),
-    warn: vi.fn(),
-    error: vi.fn(),
-    info: vi.fn(),
-  },
-}));
-vi.mock("@zhushanwen/pi-extension-logger", () => ({
-  getLogger: () => loggerMock,
-}));
-
-import {
-  CHANNEL_HANDSHAKE_KEY,
-  getOrCreateChannelRegistry,
-  type ChannelRegistryHandshake,
-} from "../channel-registry-access.ts";
-
-// ── 用例隔离：每个用例前清空 globalThis 槽位 ─────────────────
-// handshake 是 globalThis 单例，跨用例污染会破坏测试确定性。
-function clearSlot(): void {
-  Reflect.deleteProperty(globalThis, CHANNEL_HANDSHAKE_KEY);
-}
-
-afterEach(() => {
-  clearSlot();
-  loggerMock.warn.mockClear();
-});
-
-/** 读取当前 slot（绕过本模块的 readHandshakeSlot，测试断言用）。 */
-function readRawSlot(): ChannelRegistryHandshake | undefined {
-  return Reflect.get(globalThis, CHANNEL_HANDSHAKE_KEY) as ChannelRegistryHandshake | undefined;
-}
-
-/** 手动塞一个 slot（模拟 ask-user 先到的场景）。 */
-function injectSlot(slot: ChannelRegistryHandshake): void {
-  Reflect.set(globalThis, CHANNEL_HANDSHAKE_KEY, slot);
-}
-
-describe("getOrCreateChannelRegistry — 空 globalThis 场景", () => {
-  it("空 slot → 创建 canonical + slot，返回的 registry 有 register/resolve/list，slot.version===1，pending 为 []", () => {
-    clearSlot();
-    const registry = getOrCreateChannelRegistry();
-
-    // registry 形状：canonical 三方法
-    expect(typeof registry.register).toBe("function");
-    expect(typeof registry.resolve).toBe("function");
-    expect(typeof registry.list).toBe("function");
-
-    // slot 被填充且形状正确
-    const slot = readRawSlot();
-    expect(slot).toBeDefined();
-    expect(slot!.version).toBe(1);
-    expect(slot!.registry).toBe(registry); // slot.registry 指向返回的 registry
-    expect(slot!.pending).toEqual([]);
-  });
-});
-
-describe("getOrCreateChannelRegistry — ask-user 先到（registry 未就绪 + pending 非空）", () => {
-  it("模拟 ask-user 先到塞 pending → getOrCreate 创建 canonical + flush pending（handler 注册成功 + pending 清空）", () => {
-    clearSlot();
-    // ask-user 先到：塞 version:1 slot，registry 未就绪，pending 有 1 条
-    const handler = vi.fn();
-    injectSlot({
-      version: 1,
-      pending: [{ channel: "ask_user", handler }],
-    });
-
-    const registry = getOrCreateChannelRegistry();
-
-    // canonical registry 被创建，pending 被 flush
-    expect(registry.resolve("ask_user")).toBe(handler);
-    // pending 已清空
-    const slot = readRawSlot();
-    expect(slot!.pending).toEqual([]);
-    // slot.registry 现在指向返回的 registry
-    expect(slot!.registry).toBe(registry);
-  });
-
-  it("flush 后 registry.resolve 能拿到 handler（集成验证 resolve 真的能取到）", () => {
-    clearSlot();
-    const handler = vi.fn();
-    injectSlot({
-      version: 1,
-      pending: [{ channel: "ask_user", handler }],
-    });
-
-    const registry = getOrCreateChannelRegistry();
-    // 调一次 resolve 拿到的 handler，确认是原始 handler 引用
-    const resolved = registry.resolve("ask_user");
-    expect(resolved).toBe(handler);
-  });
-});
-
-describe("getOrCreateChannelRegistry — 二次调用（registry 已就绪）", () => {
-  it("二次调用返回同一实例引用（===），不重建，不重复 flush", () => {
-    clearSlot();
-    const first = getOrCreateChannelRegistry();
-    const second = getOrCreateChannelRegistry();
-    expect(second).toBe(first); // 引用相等
-
-    // slot.registry 也仍指向同一实例
-    const slot = readRawSlot();
-    expect(slot!.registry).toBe(first);
-    expect(slot!.registry).toBe(second);
-  });
-
-  it("二次调用不重复 flush（pending 已清空，再次调用 pending 仍为 []）", () => {
-    clearSlot();
-    const handler = vi.fn();
-    injectSlot({
-      version: 1,
-      pending: [{ channel: "ask_user", handler }],
-    });
-
-    getOrCreateChannelRegistry(); // 第一次：flush + 清空 pending
-    getOrCreateChannelRegistry(); // 第二次：应跳过 flush 分支
-
-    const slot = readRawSlot();
-    expect(slot!.pending).toEqual([]);
-    // handler 仍只被注册一次（resolve 仍是原 handler）
-    const registry = slot!.registry!;
-    expect(registry.resolve("ask_user")).toBe(handler);
-    expect(registry.list()).toEqual(["ask_user"]);
-  });
-});
-
-describe("getOrCreateChannelRegistry — canonical registry 三方法（形状验证）", () => {
-  it("返回的 registry register 后 resolve 拿到 handler，list 列出 channel 名", () => {
-    clearSlot();
-    const registry = getOrCreateChannelRegistry();
-    const handler = vi.fn();
-    registry.register("custom_channel", handler);
-
-    expect(registry.resolve("custom_channel")).toBe(handler);
-    expect(registry.list()).toContain("custom_channel");
-  });
-
-  it("未注册的 channel resolve 返回 undefined", () => {
-    clearSlot();
-    const registry = getOrCreateChannelRegistry();
-    expect(registry.resolve("not_registered")).toBeUndefined();
-  });
-});
-
-describe("getOrCreateChannelRegistry — 多条 pending flush 顺序与覆盖", () => {
-  it("多条 pending（不同 channel）flush 后都被注册，list 顺序保留插入顺序", () => {
-    clearSlot();
-    const h1 = vi.fn();
-    const h2 = vi.fn();
-    const h3 = vi.fn();
-    injectSlot({
-      version: 1,
-      pending: [
-        { channel: "alpha", handler: h1 },
-        { channel: "beta", handler: h2 },
-        { channel: "gamma", handler: h3 },
-      ],
-    });
-
-    const registry = getOrCreateChannelRegistry();
-    expect(registry.resolve("alpha")).toBe(h1);
-    expect(registry.resolve("beta")).toBe(h2);
-    expect(registry.resolve("gamma")).toBe(h3);
-    expect(registry.list()).toEqual(["alpha", "beta", "gamma"]);
-  });
-
-  it("同名 channel 后者覆盖前者（pending 内重复 channel）", () => {
-    clearSlot();
-    const old = vi.fn();
-    const fresh = vi.fn();
-    injectSlot({
-      version: 1,
-      pending: [
-        { channel: "ask_user", handler: old },
-        { channel: "ask_user", handler: fresh }, // 后者覆盖
-      ],
-    });
-
-    const registry = getOrCreateChannelRegistry();
-    expect(registry.resolve("ask_user")).toBe(fresh);
-    // list 不重复（Map key 唯一）
-    expect(registry.list()).toEqual(["ask_user"]);
-  });
-});
-
-describe("getOrCreateChannelRegistry — version 校验（向前兼容）", () => {
-  it("version !== 1（如塞 {version:2,...}）→ warn + 重建为新 slot，旧 pending 数据丢弃", () => {
-    clearSlot();
-    loggerMock.warn.mockClear();
-    // 塞一个 version:2 的 slot，含旧数据
-    const staleHandler = vi.fn();
-    injectSlot({
-      version: 2 as unknown as 1, // 故意写错 version
-      registry: undefined,
-      pending: [{ channel: "ask_user", handler: staleHandler }],
-    });
-
-    const registry = getOrCreateChannelRegistry();
-
-    // warn 被调用
-    expect(loggerMock.warn).toHaveBeenCalled();
-    // 旧 pending 数据被丢弃（staleHandler 没被注册）
-    expect(registry.resolve("ask_user")).toBeUndefined();
-    // slot 被重建为 version:1，pending 为空
-    const slot = readRawSlot();
-    expect(slot!.version).toBe(1);
-    expect(slot!.pending).toEqual([]);
-    expect(slot!.registry).toBe(registry);
-  });
-
-  it("version !== 1 时 warn 消息包含 got/expected 字样", () => {
-    clearSlot();
-    loggerMock.warn.mockClear();
-    injectSlot({ version: 99 as unknown as 1, pending: [] });
-
-    getOrCreateChannelRegistry();
-
-    const msg = loggerMock.warn.mock.calls[0]?.[0] ?? "";
-    expect(String(msg)).toMatch(/got/);
-    expect(String(msg)).toMatch(/expected/);
-  });
-});