@zhushanwen/subagent-core 0.2.0

package/src/execution/__tests__/session-reconstructor.test.ts

@@ -0,0 +1,379 @@
+// src/__tests__/session-reconstructor.test.ts
+//
+// session-reconstructor 专属测试。
+// 覆盖：从 session.jsonl 重建 turns[]/usage/result/error/eventLog；
+//      identity custom entry 解析；toolCall↔toolResult 配对；
+//      防御性降级（文件缺失/损坏/缺 identity/无 assistant message）。
+//
+// 用 tmpdir + 真实 .jsonl 文件（隔离文件系统）。
+
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+
+import { IDENTITY_CUSTOM_TYPE, reconstructFromFile } from "../session-reconstructor.ts";
+
+/** 写一行到文件（JSON.stringify + 换行）。 */
+function writeLine(file: number | fs.PathOrFileDescriptor, obj: unknown): void {
+  fs.writeSync(file, `${JSON.stringify(obj)}\n`);
+}
+
+/** session header 行。 */
+function headerLine(cwd = "/tmp"): unknown {
+  return { type: "session", version: 3, id: "sess-uuid", timestamp: "2026-01-01T00:00:00.000Z", cwd };
+}
+
+/** identity custom entry。 */
+function identityEntry(identity: object): unknown {
+  return {
+    type: "custom", id: "id-1", parentId: null, timestamp: "2026-01-01T00:00:00.000Z",
+    customType: IDENTITY_CUSTOM_TYPE, data: identity,
+  };
+}
+
+/** assistant message entry（content blocks + usage + stopReason）。 */
+function assistantEntry(
+  blocks: object[],
+  opts: { usage?: object; stopReason?: string; errorMessage?: string; ts?: number; parentId?: string } = {},
+): unknown {
+  return {
+    type: "message", id: `msg-${Math.random().toString(36).slice(2, 8)}`,
+    parentId: opts.parentId ?? "id-1",
+    timestamp: new Date(opts.ts ?? 1000).toISOString(),
+    message: {
+      role: "assistant",
+      content: blocks,
+      usage: opts.usage ?? { input: 10, output: 20, cacheRead: 0, cacheWrite: 0, totalTokens: 30, cost: { total: 0 } },
+      stopReason: opts.stopReason ?? "stop",
+      errorMessage: opts.errorMessage,
+      timestamp: opts.ts ?? 1000,
+    },
+  };
+}
+
+/** toolResult message entry。 */
+function toolResultEntry(toolCallId: string, toolName: string, opts: { isError?: boolean; parentId?: string; text?: string } = {}): unknown {
+  return {
+    type: "message", id: `tr-${Math.random().toString(36).slice(2, 8)}`,
+    parentId: opts.parentId ?? "id-1",
+    timestamp: new Date(2000).toISOString(),
+    message: {
+      role: "toolResult", toolCallId, toolName,
+      content: [{ type: "text", text: opts.text ?? "result" }],
+      isError: opts.isError ?? false,
+      timestamp: 2000,
+    },
+  };
+}
+
+describe("reconstructFromFile", () => {
+  let tmpDir: string;
+  let filePath: string;
+
+  beforeEach(() => {
+    tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "sr-test-"));
+    filePath = path.join(tmpDir, "test.jsonl");
+  });
+  afterEach(() => {
+    fs.rmSync(tmpDir, { recursive: true, force: true });
+  });
+
+  function writeJsonl(lines: unknown[]): void {
+    const fd = fs.openSync(filePath, "w");
+    for (const line of lines) writeLine(fd, line);
+    fs.closeSync(fd);
+  }
+
+  // ============================================================
+  // 基本重建
+  // ============================================================
+  describe("基本重建", () => {
+    it("单 assistant message → 1 turn，text/usage 正确", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "worker", mode: "background", task: "do it", startedAt: 500 }),
+        assistantEntry([{ type: "text", text: "hello world" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec).toBeDefined();
+      expect(rec!.id).toBe("bg-1");
+      expect(rec!.agent).toBe("worker");
+      expect(rec!.mode).toBe("background");
+      expect(rec!.task).toBe("do it");
+      expect(rec!.status).toBe("closed");
+      expect(rec!.turns).toHaveLength(1);
+      expect(rec!.turns[0].text).toBe("hello world");
+      expect(rec!.turnCount).toBe(1);
+      expect(rec!.totalTokens).toBe(30); // 10+20+0+0
+      expect(rec!.result).toBe("hello world");
+    });
+
+    it("thinking block 累积进 turn.thinking", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([
+          { type: "thinking", thinking: "let me think" },
+          { type: "text", text: "answer" },
+        ]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.turns[0].thinking).toBe("let me think");
+      expect(rec!.turns[0].text).toBe("answer");
+    });
+
+    it("多 assistant message → 多 turn，result 用空行拼接", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "first" }], { ts: 1000 }),
+        assistantEntry([{ type: "text", text: "second" }], { ts: 2000, parentId: undefined }),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.turns).toHaveLength(2);
+      expect(rec!.turnCount).toBe(2);
+      expect(rec!.result).toBe("first\n\nsecond");
+    });
+
+    it("读出 identity 里的 rootSessionId", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", startedAt: 100, rootSessionId: "sess-A" }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.rootSessionId).toBe("sess-A");
+    });
+
+    it("旧文件 identity 写 parentSessionId → fallback 读到 rootSessionId（向后兼容）", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", startedAt: 100, parentSessionId: "sess-legacy" }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.rootSessionId).toBe("sess-legacy");
+    });
+
+    it("identity 无 rootSessionId（旧文件）→ rootSessionId 为 undefined", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.rootSessionId).toBeUndefined();
+    });
+
+    it("读出 identity 里的 slug", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", slug: "extract-urls", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.slug).toBe("extract-urls");
+    });
+
+    it("旧文件 identity 无 slug → 兜底空串（向后兼容）", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.slug).toBe("");
+    });
+
+    it("读出 identity 里的 parentRecordId/depth（递归层级）", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "run-2", agent: "w", mode: "sync", task: "t", startedAt: 100, rootSessionId: "sess-A", parentRecordId: "run-1", depth: 2 }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.parentRecordId).toBe("run-1");
+      expect(rec!.depth).toBe(2);
+    });
+
+    it("旧文件无 parentRecordId/depth → 兑底 undefined/0（顶层）", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", startedAt: 100, rootSessionId: "sess-A" }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.parentRecordId).toBeUndefined();
+      expect(rec!.depth).toBe(0);
+    });
+
+    it("endedAt 为最后一条 entry 的时间戳（非 now）", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "bg-1", agent: "w", mode: "background", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "first" }], { ts: 1000 }),
+        assistantEntry([{ type: "text", text: "second" }], { ts: 5000, parentId: undefined }),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.endedAt).toBe(5000);
+    });
+  });
+
+  // ============================================================
+  // toolCall ↔ toolResult 配对
+  // ============================================================
+  describe("toolCall 配对", () => {
+    it("toolCall + toolResult → InternalToolCall done", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([
+          { type: "toolCall", id: "call-1", name: "read", arguments: { path: "/x.ts" } },
+        ]),
+        toolResultEntry("call-1", "read"),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.turns[0].toolCalls).toHaveLength(1);
+      const tc = rec!.turns[0].toolCalls[0];
+      expect(tc.toolName).toBe("read");
+      expect(tc._status).toBe("done");
+      expect(tc.isError).toBe(false);
+    });
+
+    it("toolResult isError → InternalToolCall failed", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([
+          { type: "toolCall", id: "call-1", name: "bash", arguments: { command: "false" } },
+        ]),
+        toolResultEntry("call-1", "bash", { isError: true }),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.turns[0].toolCalls[0]._status).toBe("failed");
+      expect(rec!.turns[0].toolCalls[0].isError).toBe(true);
+    });
+
+    it("孤儿 toolResult（无匹配 toolCall）→ 丢弃，不崩", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "ok" }]),
+        toolResultEntry("nonexistent", "read"),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec).toBeDefined();
+      expect(rec!.turns[0].toolCalls).toHaveLength(0);
+    });
+  });
+
+  // ============================================================
+  // error / stopReason
+  // ============================================================
+  describe("error 处理", () => {
+    it("stopReason=error → lastError + error 字段", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "partial" }], {
+          stopReason: "error", errorMessage: "API timeout",
+        }),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.lastError).toBe("API timeout");
+      expect(rec!.error).toBe("API timeout");
+      expect(rec!.status).toBe("closed"); // error stopReason → failed
+    });
+
+    it("stopReason=aborted 无 errorMessage → lastError = 'aborted'", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "" }], { stopReason: "aborted" }),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.lastError).toBe("aborted");
+      expect(rec!.status).toBe("closed");
+    });
+
+    it("前序 error 但最后 stop → lastError 清除（镜像 turn_end 语义），status=done", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([{ type: "text", text: "oops" }], {
+          stopReason: "error", errorMessage: "transient", ts: 1000,
+        }),
+        assistantEntry([{ type: "text", text: "recovered" }], { stopReason: "stop", ts: 2000 }),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      expect(rec!.lastError).toBeUndefined(); // 后续 stop 清除了 error
+      expect(rec!.status).toBe("closed");
+      expect(rec!.result).toBe("oops\n\nrecovered");
+    });
+  });
+
+  // ============================================================
+  // eventLog 派生
+  // ============================================================
+  describe("eventLog 派生", () => {
+    it("tool_start + tool_end + turn_end 条目", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+        assistantEntry([
+          { type: "toolCall", id: "c1", name: "read", arguments: { path: "/x.ts" } },
+        ]),
+        toolResultEntry("c1", "read"),
+      ]);
+      const rec = reconstructFromFile(filePath);
+      const types = rec!.eventLog.map((e) => e.type);
+      expect(types).toContain("tool_start");
+      expect(types).toContain("tool_end");
+      expect(types).toContain("turn_end");
+    });
+  });
+
+  // ============================================================
+  // 防御性降级
+  // ============================================================
+  describe("防御性降级", () => {
+    it("文件缺失 → undefined", () => {
+      expect(reconstructFromFile(path.join(tmpDir, "nonexistent.jsonl"))).toBeUndefined();
+    });
+
+    it("空文件 → undefined", () => {
+      fs.writeFileSync(filePath, "", "utf-8");
+      expect(reconstructFromFile(filePath)).toBeUndefined();
+    });
+
+    it("缺 identity custom entry → undefined", () => {
+      writeJsonl([
+        headerLine(),
+        assistantEntry([{ type: "text", text: "no identity" }]),
+      ]);
+      expect(reconstructFromFile(filePath)).toBeUndefined();
+    });
+
+    it("有 identity 但无 assistant message → undefined", () => {
+      writeJsonl([
+        headerLine(),
+        identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }),
+      ]);
+      expect(reconstructFromFile(filePath)).toBeUndefined();
+    });
+
+    it("损坏 JSON 行跳过，合法行仍解析", () => {
+      const fd = fs.openSync(filePath, "w");
+      fs.writeSync(fd, `${JSON.stringify(headerLine())}\n`);
+      fs.writeSync(fd, "THIS IS NOT JSON\n");
+      fs.writeSync(fd, `${JSON.stringify(identityEntry({ id: "r1", agent: "w", mode: "sync", task: "t", startedAt: 100 }))}\n`);
+      fs.writeSync(fd, `${JSON.stringify(assistantEntry([{ type: "text", text: "survived" }]))}\n`);
+      fs.closeSync(fd);
+      const rec = reconstructFromFile(filePath);
+      expect(rec).toBeDefined();
+      expect(rec!.result).toBe("survived");
+    });
+  });
+});

package/src/execution/__tests__/session-runner-epipe.test.ts

@@ -0,0 +1,178 @@
+// src/execution/__tests__/session-runner-epipe.test.ts
+//
+// [v4 A-1] session-runner 异步 stdin 'error' listener 测试。
+//
+// 背景：writeStdinLine 的 try/catch 只覆盖同步 write 抛错；子进程退出后内核回写 EPIPE
+// 以异步 stream 'error' event 到达，若无 listener 会让 Node 抛 unhandled 'error' 崩主进程（P1）。
+// runSpawn 在 spawn 后、首次 stdin 写入前注册 child.stdin.on('error') handler，合并
+// 同步/异步 EPIPE 计数（共用 stdin-writer 的 recordEpipeFailure）。
+//
+// 验收：① listener 存在 ② error 触发后 spawnedChildren 不含该 recordId
+//      ③ recordEpipeFailure 计数累加（同步/异步合并）
+
+import { spawn } from "node:child_process";
+
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+
+vi.mock("node:child_process", async () => {
+  const { FakeChild } = await import("./helpers/spawn-mock.ts");
+  return {
+    spawn: vi.fn(() => new FakeChild()),
+    // buildEnvBlock 的 git branch 调用（execFile 异步）：默认 err-first 兜底 → catch → branch=""
+    execFile: vi.fn(
+      (
+        _cmd: string,
+        _args: readonly string[],
+        _opts: unknown,
+        cb: (err: Error | null, stdout?: string, stderr?: string) => void,
+      ) => cb(new Error("execFile not configured in this test")),
+    ),
+  };
+});
+
+vi.mock("node:fs", async () => {
+  const actual = await import("node:fs");
+  return {
+    default: {
+      ...actual,
+      mkdirSync: vi.fn(),
+      existsSync: vi.fn(() => false),
+      appendFileSync: vi.fn(),
+      writeFileSync: vi.fn(),
+      readdirSync: vi.fn(() => []),
+    },
+    mkdirSync: vi.fn(),
+    existsSync: vi.fn(() => false),
+    appendFileSync: vi.fn(),
+    writeFileSync: vi.fn(),
+    readdirSync: vi.fn(() => []),
+    promises: actual.promises,
+  };
+});
+
+vi.mock("../alive-store.ts", () => ({ writeAliveMarker: vi.fn() }));
+
+vi.mock("../session-pending.ts", () => ({
+  readActivePendingFromSessionFile: vi.fn(() => ({ count: 0 })),
+}));
+
+vi.mock("../temp-prompt.ts", () => ({
+  writePromptToTempFile: vi.fn(async (agent: string) => {
+    const safeName = agent.replace(/[^\w.-]+/g, "_");
+    return { dir: `/tmp/fake-${safeName}`, filePath: `/tmp/fake-${safeName}/prompt-${safeName}.md` };
+  }),
+  cleanupTempPrompt: vi.fn(async () => {}),
+}));
+
+import {
+  EPIPE_FAILURE_THRESHOLD,
+  recordEpipeFailure,
+  resetAllEpipeFailures,
+} from "../stdin-writer.ts";
+import { runSpawn, spawnedChildren } from "../session-runner.ts";
+import { _resetLifecycleState } from "../lifecycle-manager.ts";
+import {
+  FakeChild,
+  lastSpawnedChild as lastSpawnedChildOf,
+  makeCtx,
+  makeOpts,
+  makeRecord,
+  waitForSpawn as waitForSpawnOf,
+} from "./helpers/spawn-mock.ts";
+
+const mockSpawn = vi.mocked(spawn);
+const lastSpawnedChild = (): FakeChild => lastSpawnedChildOf(mockSpawn);
+const waitForSpawn = (timeoutMs = 1000): Promise<void> => waitForSpawnOf(mockSpawn, timeoutMs);
+
+/** 构造 EPIPE errno 错误（模拟子进程退出后内核回写 EPIPE）。 */
+function epipeError(message = "write after end"): NodeJS.ErrnoException {
+  const err = new Error(message) as NodeJS.ErrnoException;
+  err.code = "EPIPE";
+  return err;
+}
+
+/** 手动 emit close 让 runSpawn 的 exitCode promise resolve（非 chatMode 默认 close resolve）。 */
+function settleRunSpawn(child: FakeChild): void {
+  child.stdout.end();
+  child.stderr.end();
+  child.emit("close", 0);
+}
+
+describe("[v4 A-1] session-runner 异步 stdin 'error' listener", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    _resetLifecycleState();
+    resetAllEpipeFailures();
+    spawnedChildren.clear();
+  });
+
+  afterEach(() => {
+    _resetLifecycleState();
+    resetAllEpipeFailures();
+    spawnedChildren.clear();
+  });
+
+  it("① spawn 后注册 child.stdin 'error' listener", async () => {
+    const record = makeRecord("sa-listener");
+    const promise = runSpawn(record, "Task: listener-exists", makeOpts(), makeCtx());
+
+    await waitForSpawn();
+    const child = lastSpawnedChild();
+
+    // listener 已注册——无 listener 时异步 EPIPE 会崩主进程（这是 A-1 修复的 P1）
+    expect(child.stdin.listenerCount("error")).toBeGreaterThan(0);
+
+    settleRunSpawn(child);
+    await promise;
+  });
+
+  it("②③ async stdin 'error' → 移出 spawnedChildren + recordEpipeFailure 计数递增", async () => {
+    const record = makeRecord("sa-async-epipe");
+    const promise = runSpawn(record, "Task: async-epipe", makeOpts(), makeCtx());
+
+    await waitForSpawn();
+    const child = lastSpawnedChild();
+
+    // runSpawn 已把 child 注册进 spawnedChildren
+    expect(spawnedChildren.has(record.id)).toBe(true);
+
+    // 发一次异步 stdin 'error'（count 0→1，未达阈值，handler 只 warn 不 throw）
+    child.stdin.emit("error", epipeError());
+
+    // ② spawnedChildren 已移除该 recordId（标记 dead）
+    expect(spawnedChildren.has(record.id)).toBe(false);
+
+    // ③ 计数累加：handler 调 recordEpipeFailure 设为 1；此处再调一次应返回 2
+    //    （若 handler 未运行，counter 仍为 0，本次调用只会返回 1）
+    expect(recordEpipeFailure(record.id)).toBe(2);
+
+    settleRunSpawn(child);
+    await promise;
+  });
+
+  it("③ 同步预置计数 + 异步 stdin 'error' 合并计数（handler 不 throw，防 listener 内 throw 崩进程）", async () => {
+    const record = makeRecord("sa-threshold");
+    const promise = runSpawn(record, "Task: threshold", makeOpts(), makeCtx());
+
+    await waitForSpawn();
+    const child = lastSpawnedChild();
+
+    // 预置 count=1（模拟 deliverMessage 同步路径已计 1 次失败——证明共用同一计数器）
+    expect(recordEpipeFailure(record.id)).toBe(1);
+
+    // [v4 A-1 裁决] async handler 内不 throw（stream 'error' listener 内 throw 会经 emit
+    // 传播为 uncaughtException 崩主进程，违背 A-1 防崩）——达阈值 throw 留同步路径。
+    // emit 不抛错：
+    expect(() => child.stdin.emit("error", epipeError())).not.toThrow();
+
+    // handler 调 recordEpipeFailure → count=2（达阈值但 async 不 throw）；此处再调返回 3
+    expect(recordEpipeFailure(record.id)).toBe(3);
+    expect(EPIPE_FAILURE_THRESHOLD).toBe(2);
+
+    // handler 已移出 spawnedChildren（标记 dead，下次 deliverMessage 检测走冷路径）
+    expect(spawnedChildren.has(record.id)).toBe(false);
+
+    settleRunSpawn(child);
+    await promise;
+  });
+});