@zhushanwen/subagent-core 0.2.0

package/src/execution/__tests__/ui-request-queue.test.ts

@@ -0,0 +1,133 @@
+// src/__tests__/ui-request-queue.test.ts
+//
+// W3 测试：UI 请求队列机制（L1 per-child createUiRequestQueue）
+// 1. 多个请求按 FIFO 顺序处理
+// 2. 第一个请求未完成时第二个不开始
+//
+// 直接测试 createUiRequestQueue 的队列逻辑（纯函数，不需要 mock runSpawn）。
+// 用 fake child（PassThrough stdin）+ 手动控制 Promise resolve 时序。
+//
+// 协议迁移（W2）：enqueue 第二参从旧 Record params（含 questions/context）改为
+// ExtensionUiRequest（method 平铺）；handler 签名从 (questions,context) 改为
+// UiRequestHandler (req: UiRequest) => Promise<UiResponse>。
+
+import type { ChildProcess } from "node:child_process";
+import { PassThrough } from "node:stream";
+
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+
+import {
+  createUiRequestQueue,
+  type UiRequest,
+  type UiRequestHandler,
+  type UiResponse,
+} from "../ui-request-queue.ts";
+
+beforeEach(() => {
+  vi.useFakeTimers();
+});
+
+afterEach(() => {
+  vi.useRealTimers();
+});
+
+function makeFakeChild(): ChildProcess {
+  const stdin = new PassThrough();
+  return {
+    stdin,
+    killed: false,
+    pid: 10001,
+    on: vi.fn(),
+    removeListener: vi.fn(),
+  } as unknown as ChildProcess;
+}
+
+/** 构造 select method 的 ExtensionUiRequest（ask_user 借道 select dialog 通道）。 */
+function makeSelectReq(question: string): { method: "select"; title: string; options: string[] } {
+  return {
+    method: "select",
+    title: question,
+    options: [JSON.stringify({ question, options: [{ label: "A" }] })],
+  };
+}
+
+describe("UI 请求队列", () => {
+  it("多个 extension_ui_request 按 FIFO 顺序处理", async () => {
+    const callOrder: string[] = [];
+    // 每个请求返回独立的可控 Promise
+    const resolvers: Array<(v: unknown) => void> = [];
+
+    const handler: UiRequestHandler = vi.fn((req: UiRequest) => {
+      callOrder.push(req.title ?? "");
+      return new Promise<UiResponse>((resolve) => {
+        resolvers.push(resolve);
+      });
+    }) as unknown as UiRequestHandler;
+
+    const child = makeFakeChild();
+    const ctx = { uiRequestHandler: handler } as Parameters<
+      typeof createUiRequestQueue
+    >[1];
+    const enqueue = createUiRequestQueue(child, ctx);
+
+    // 快速入队三个请求（handler 被调用但不 resolve）
+    enqueue("r1", makeSelectReq("Q1"));
+    enqueue("r2", makeSelectReq("Q2"));
+    enqueue("r3", makeSelectReq("Q3"));
+
+    // 第一个请求立即开始处理
+    expect(handler).toHaveBeenCalledTimes(1);
+    expect(callOrder).toEqual(["Q1"]);
+
+    // resolve 第一个 → 第二个开始处理
+    resolvers[0]({ value: "a1" });
+    await vi.advanceTimersByTimeAsync(0);
+    expect(handler).toHaveBeenCalledTimes(2);
+    expect(callOrder).toEqual(["Q1", "Q2"]);
+
+    // resolve 第二个 → 第三个开始处理
+    resolvers[1]({ value: "a2" });
+    await vi.advanceTimersByTimeAsync(0);
+    expect(handler).toHaveBeenCalledTimes(3);
+    expect(callOrder).toEqual(["Q1", "Q2", "Q3"]);
+  });
+
+  it("第一个请求未 resolve 时第二个不调用 uiRequestHandler", async () => {
+    const callOrder: string[] = [];
+    let firstResolve: (v: unknown) => void;
+
+    const handler: UiRequestHandler = vi.fn((req: UiRequest) => {
+      callOrder.push(req.title ?? "");
+      if (req.title === "Q1") {
+        return new Promise<UiResponse>((resolve) => {
+          firstResolve = resolve;
+        });
+      }
+      return Promise.resolve<UiResponse>({ value: "done" });
+    }) as unknown as UiRequestHandler;
+
+    const child = makeFakeChild();
+    const ctx = { uiRequestHandler: handler } as Parameters<
+      typeof createUiRequestQueue
+    >[1];
+    const enqueue = createUiRequestQueue(child, ctx);
+
+    enqueue("r1", makeSelectReq("Q1"));
+    enqueue("r2", makeSelectReq("Q2"));
+
+    // 只有 Q1 被调用，Q2 还在队列里
+    expect(handler).toHaveBeenCalledTimes(1);
+    expect(callOrder).toEqual(["Q1"]);
+
+    // 等一下，Q2 仍然不应该被调用
+    await vi.advanceTimersByTimeAsync(100);
+    expect(handler).toHaveBeenCalledTimes(1);
+    expect(callOrder).toEqual(["Q1"]);
+
+    // resolve Q1 → Q2 才开始
+    firstResolve!({ value: "a1" });
+    await vi.advanceTimersByTimeAsync(0);
+    expect(handler).toHaveBeenCalledTimes(2);
+    expect(callOrder).toEqual(["Q1", "Q2"]);
+  });
+});