@zhushanwen/pi-subagent-workflow 8.3.0 → 8.5.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) hide show
  1. package/package.json +18 -4
  2. package/relay/relay.mjs +390 -0
  3. package/skills/subagent-ext-config/SKILL.md +80 -0
  4. package/src/execution/__tests__/agent-registry.test.ts +110 -0
  5. package/src/execution/__tests__/chat-engine-routing.test.ts +597 -0
  6. package/src/execution/__tests__/execution-record.test.ts +127 -1
  7. package/src/execution/__tests__/pi-invocation.test.ts +62 -1
  8. package/src/execution/__tests__/relay-agent.test.ts +448 -0
  9. package/src/execution/__tests__/relay-env.test.ts +42 -0
  10. package/src/execution/__tests__/startup-config-declaration.test.ts +35 -0
  11. package/src/execution/__tests__/stream-sink-retirement.test.ts +261 -0
  12. package/src/execution/__tests__/subprocess-agent-runner-routing.test.ts +310 -0
  13. package/src/execution/__tests__/subprocess-agent-runner.test.ts +53 -5
  14. package/src/execution/agent-registry.ts +10 -0
  15. package/src/execution/config.ts +25 -2
  16. package/src/execution/engine/__tests__/common/data-dir.test.ts +53 -0
  17. package/src/execution/engine/__tests__/common/errors.test.ts +132 -0
  18. package/src/execution/engine/__tests__/common/event-journal.test.ts +177 -0
  19. package/src/execution/engine/__tests__/common/kill-chain.test.ts +192 -0
  20. package/src/execution/engine/__tests__/common/nesting-guard.test.ts +81 -0
  21. package/src/execution/engine/__tests__/common/persona-router.test.ts +123 -0
  22. package/src/execution/engine/__tests__/common/pool-manager.test.ts +154 -0
  23. package/src/execution/engine/__tests__/common/schema-emulation.test.ts +128 -0
  24. package/src/execution/engine/__tests__/conformance/__fixtures__/pi-golden-events.json +28 -0
  25. package/src/execution/engine/__tests__/conformance/agent-event-invariants.ts +141 -0
  26. package/src/execution/engine/__tests__/conformance/contract.abort.test.ts +109 -0
  27. package/src/execution/engine/__tests__/conformance/contract.agent-events.test.ts +101 -0
  28. package/src/execution/engine/__tests__/conformance/contract.probe.test.ts +77 -0
  29. package/src/execution/engine/__tests__/conformance/contract.read-degradation.test.ts +104 -0
  30. package/src/execution/engine/__tests__/conformance/contract.relay.test.ts +342 -0
  31. package/src/execution/engine/__tests__/conformance/engine-conformance.live.test.ts +201 -0
  32. package/src/execution/engine/__tests__/conformance/golden-replay.pi.test.ts +76 -0
  33. package/src/execution/engine/__tests__/conformance/golden-replay.zcode.test.ts +79 -0
  34. package/src/execution/engine/__tests__/engine-discovery.test.ts +87 -0
  35. package/src/execution/engine/__tests__/engines-declaration.test.ts +36 -0
  36. package/src/execution/engine/__tests__/model-prompt.test.ts +85 -0
  37. package/src/execution/engine/__tests__/paths.test.ts +39 -0
  38. package/src/execution/engine/__tests__/registry.test.ts +120 -0
  39. package/src/execution/engine/__tests__/routing.test.ts +231 -0
  40. package/src/execution/engine/common/data-dir.ts +62 -0
  41. package/src/execution/engine/common/errors.ts +183 -0
  42. package/src/execution/engine/common/event-journal.ts +254 -0
  43. package/src/execution/engine/common/journal-replay.ts +62 -0
  44. package/src/execution/engine/common/kill-chain.ts +221 -0
  45. package/src/execution/engine/common/nesting-guard.ts +50 -0
  46. package/src/execution/engine/common/persona-router.ts +108 -0
  47. package/src/execution/engine/common/pool-manager.ts +226 -0
  48. package/src/execution/engine/common/schema-emulation.ts +189 -0
  49. package/src/execution/engine/common/session-view-projection.ts +51 -0
  50. package/src/execution/engine/engine-discovery.ts +65 -0
  51. package/src/execution/engine/engines/pi/__tests__/pi-engine.test.ts +469 -0
  52. package/src/execution/engine/engines/pi/__tests__/reader.test.ts +155 -0
  53. package/src/execution/engine/engines/pi/__tests__/task-spec-mapper.test.ts +164 -0
  54. package/src/execution/engine/engines/pi/pi-engine.ts +415 -0
  55. package/src/execution/engine/engines/pi/reader.ts +48 -0
  56. package/src/execution/engine/engines/pi/registration.ts +35 -0
  57. package/src/execution/engine/engines/pi/task-spec-mapper.ts +100 -0
  58. package/src/execution/engine/engines/zcode/__tests__/__fixtures__/zcode-golden-spawn.json +39 -0
  59. package/src/execution/engine/engines/zcode/__tests__/launcher.test.ts +150 -0
  60. package/src/execution/engine/engines/zcode/__tests__/parser.test.ts +246 -0
  61. package/src/execution/engine/engines/zcode/__tests__/preparer.test.ts +228 -0
  62. package/src/execution/engine/engines/zcode/__tests__/reader.test.ts +210 -0
  63. package/src/execution/engine/engines/zcode/__tests__/registration.test.ts +64 -0
  64. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.live.test.ts +127 -0
  65. package/src/execution/engine/engines/zcode/__tests__/zcode-engine.test.ts +567 -0
  66. package/src/execution/engine/engines/zcode/constants.ts +43 -0
  67. package/src/execution/engine/engines/zcode/golden-sample.ts +39 -0
  68. package/src/execution/engine/engines/zcode/launcher.ts +161 -0
  69. package/src/execution/engine/engines/zcode/parser.ts +436 -0
  70. package/src/execution/engine/engines/zcode/preparer.ts +363 -0
  71. package/src/execution/engine/engines/zcode/reader.ts +381 -0
  72. package/src/execution/engine/engines/zcode/registration.ts +37 -0
  73. package/src/execution/engine/engines/zcode/zcode-engine.ts +648 -0
  74. package/src/execution/engine/host-task-spec.ts +47 -0
  75. package/src/execution/engine/model-prompt.ts +59 -0
  76. package/src/execution/engine/paths.ts +42 -0
  77. package/src/execution/engine/port.ts +153 -0
  78. package/src/execution/engine/registry.ts +123 -0
  79. package/src/execution/engine/routing.ts +218 -0
  80. package/src/execution/engine/types.ts +304 -0
  81. package/src/execution/execute-options-mapper.ts +5 -1
  82. package/src/execution/execution-record.ts +6 -0
  83. package/src/execution/model-resolver.ts +6 -0
  84. package/src/execution/pi-invocation.ts +32 -2
  85. package/src/execution/record-entry.ts +14 -0
  86. package/src/execution/record-store.ts +34 -0
  87. package/src/execution/relay-env.ts +37 -0
  88. package/src/execution/session-runner.ts +24 -0
  89. package/src/execution/stream-sink.ts +26 -0
  90. package/src/execution/subagent-service.ts +249 -11
  91. package/src/execution/subprocess-agent-runner.ts +196 -14
  92. package/src/execution/types.ts +56 -0
  93. package/src/index.ts +46 -1
  94. package/src/interface/command-actions.ts +72 -8
  95. package/src/interface/subagent-actions.ts +8 -2
  96. package/src/interface/subagent-tool.ts +6 -0
  97. package/src/interface/subagents.ts +198 -30
  98. package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +5 -2
  99. package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +3 -3
  100. package/src/orchestration/models/types.ts +7 -0
  101. package/src/orchestration/worker-script-builder.ts +5 -2
  102. package/src/shared/meta-parser.ts +5 -1
  103. package/src/shared/resource-meta.ts +5 -0
@@ -0,0 +1,42 @@
1
+ import { describe, expect, it } from 'vitest';
2
+
3
+ import {
4
+ RELAY_ENV_NODE,
5
+ RELAY_ENV_RECORD_ID,
6
+ RELAY_ENV_SCRIPT,
7
+ RELAY_ENV_SESSION_ID,
8
+ RELAY_ENV_SOCKET,
9
+ RELAY_EXIT_CODES,
10
+ RELAY_PROTOCOL_VERSION,
11
+ isRelayActive,
12
+ } from '../relay-env.js';
13
+
14
+ describe('relay env SSOT', () => {
15
+ it('激活判定:三 env 同时非空', () => {
16
+ expect(isRelayActive({ [RELAY_ENV_SOCKET]: '/x.sock', [RELAY_ENV_NODE]: '/node', [RELAY_ENV_SCRIPT]: '/relay.mjs' })).toBe(true);
17
+ });
18
+
19
+ it('任一缺失即不激活(全有或全无)', () => {
20
+ const full = {
21
+ [RELAY_ENV_SOCKET]: '/x.sock',
22
+ [RELAY_ENV_NODE]: '/node',
23
+ [RELAY_ENV_SCRIPT]: '/relay.mjs',
24
+ };
25
+ expect(isRelayActive({ ...full, [RELAY_ENV_SOCKET]: '' })).toBe(false);
26
+ expect(isRelayActive({ ...full, [RELAY_ENV_NODE]: undefined })).toBe(false);
27
+ expect(isRelayActive({})).toBe(false);
28
+ expect(isRelayActive(process.env)).toBe(false);
29
+ });
30
+
31
+ it('env 名与退出码稳定(relay.mjs 镜像一致性由 conformance relay 断言锁定)', () => {
32
+ expect(RELAY_ENV_SESSION_ID).toBe('XYZ_SUBAGENT_RELAY_SESSION_ID');
33
+ expect(RELAY_ENV_RECORD_ID).toBe('XYZ_SUBAGENT_RELAY_RECORD_ID');
34
+ expect(RELAY_PROTOCOL_VERSION).toBe(1);
35
+ expect(RELAY_EXIT_CODES).toEqual({
36
+ VERSION_MISMATCH: 10,
37
+ SOCKET_UNREACHABLE: 11,
38
+ SOCKET_CLOSED: 12,
39
+ MISSING_IDENTITY: 13,
40
+ });
41
+ });
42
+ });
@@ -0,0 +1,35 @@
1
+ // src/execution/__tests__/startup-config-declaration.test.ts
2
+ //
3
+ // startupConfig 声明守护测试:断言 package.json `xyz-agent.startupConfig` 声明的
4
+ // content 与代码 DEFAULT_CONFIG 常量深相等,防止两侧任一改动未同步(漂移)。
5
+ // 机制:runtime 启动序列统一 ensure(extension-startup-config.ts),声明即首建内容。
6
+ //
7
+ // 运行:cd extensions/universal/subagent-workflow && npx vitest run src/execution/__tests__/startup-config-declaration.test.ts
8
+
9
+ import { readFileSync } from "node:fs";
10
+ import { join } from "node:path";
11
+ import { fileURLToPath } from "node:url";
12
+
13
+ import { describe, expect, it } from "vitest";
14
+
15
+ import { DEFAULT_CONFIG } from "../config.ts";
16
+
17
+ /** package.json `xyz-agent.startupConfig` 声明条目的最小形状(守护断言用)。 */
18
+ interface StartupConfigEntry {
19
+ path: string;
20
+ content: unknown;
21
+ }
22
+
23
+ const pkg = JSON.parse(
24
+ readFileSync(join(fileURLToPath(import.meta.url), "../../../../package.json"), "utf-8"),
25
+ ) as { "xyz-agent"?: { startupConfig?: StartupConfigEntry[] } };
26
+
27
+ describe("startupConfig 声明守护", () => {
28
+ it("声明 content 与代码 DEFAULT_CONFIG 深相等", () => {
29
+ const entry = pkg["xyz-agent"]?.startupConfig?.find(
30
+ (e: StartupConfigEntry) => e.path === "subagents/config.json",
31
+ );
32
+ expect(entry).toBeDefined();
33
+ expect(entry?.content).toEqual(DEFAULT_CONFIG);
34
+ });
35
+ });
@@ -0,0 +1,261 @@
1
+ // src/execution/__tests__/stream-sink-retirement.test.ts
2
+ //
3
+ // widget 私货通道退役步骤 2(E 方案 §6.3)测试,两层:
4
+ // 判定层:createBackgroundStream 真实函数(vi.importActual 绕过文件级 mock)——
5
+ // gui+relay 激活不创建;tui / relay 未激活 / headless 原样创建;sink=null 恒降级。
6
+ // 接线层:SubagentService.execute(background → kickOffBackground 同步段)透传
7
+ // mode/streamSink 给工厂——mode 传错 = 判定失效,故接线面必须锁定。
8
+ //
9
+ // spawn 层 mock(FakeChild),与 nested-visibility-env-propagation.test.ts 同模式。
10
+
11
+ import type { PassThrough } from "node:stream";
12
+
13
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
14
+
15
+ // ── mock modules ──
16
+
17
+ vi.mock("node:child_process", async () => {
18
+ const { EventEmitter } = await import("node:events");
19
+ const { PassThrough } = await import("node:stream");
20
+
21
+ class FakeChild extends EventEmitter {
22
+ pid = 12345;
23
+ stdout = new PassThrough();
24
+ stdin = new PassThrough();
25
+ stderr = new PassThrough();
26
+ killed = false;
27
+ killSignal: string | undefined;
28
+ kill(sig?: string): boolean {
29
+ this.killed = true;
30
+ this.killSignal = sig;
31
+ return true;
32
+ }
33
+ }
34
+
35
+ return {
36
+ spawn: vi.fn(() => new FakeChild()),
37
+ execFile: vi.fn(
38
+ (
39
+ _cmd: string,
40
+ _args: readonly string[],
41
+ _opts: unknown,
42
+ cb: (err: Error | null, stdout?: string, stderr?: string) => void,
43
+ ) => cb(new Error("execFile not configured in this test")),
44
+ ),
45
+ };
46
+ });
47
+
48
+ vi.mock("node:fs", async () => {
49
+ const actual = await import("node:fs");
50
+ return {
51
+ default: {
52
+ ...actual,
53
+ mkdirSync: vi.fn(),
54
+ existsSync: vi.fn(() => false),
55
+ appendFileSync: vi.fn(),
56
+ writeFileSync: vi.fn(),
57
+ readdirSync: vi.fn(() => []),
58
+ },
59
+ mkdirSync: vi.fn(),
60
+ existsSync: vi.fn(() => false),
61
+ appendFileSync: vi.fn(),
62
+ writeFileSync: vi.fn(),
63
+ readdirSync: vi.fn(() => []),
64
+ promises: actual.promises,
65
+ };
66
+ });
67
+
68
+ // 接线层观测点:工厂调用记录(参数透传断言用)。返回 stub 形状满足消费面
69
+ // (session-runner onDelta / subagent-service dispose)。判定层走 vi.importActual。
70
+ vi.mock("../stream-sink.ts", async (importOriginal) => {
71
+ const actual = await importOriginal<typeof import("../stream-sink.ts")>();
72
+ return {
73
+ ...actual,
74
+ createBackgroundStream: vi.fn((recordId: string) => ({
75
+ onDelta: () => {},
76
+ dispose: () => {},
77
+ recordId,
78
+ })),
79
+ };
80
+ });
81
+
82
+ vi.mock("../alive-store.ts", async (importOriginal) => {
83
+ const actual = await importOriginal<typeof import("../alive-store.ts")>();
84
+ return {
85
+ ...actual,
86
+ writeAliveMarker: vi.fn(),
87
+ removeAliveMarker: vi.fn(),
88
+ };
89
+ });
90
+
91
+ vi.mock("../finalized-marker.ts", () => ({
92
+ writeFinalized: vi.fn(),
93
+ readFinalized: vi.fn(() => false),
94
+ }));
95
+
96
+ vi.mock("../manifest-store.ts", () => {
97
+ class FakeManifestStore {
98
+ writeManifest = vi.fn(async () => {});
99
+ readManifest = vi.fn(async () => null);
100
+ listAllSync = vi.fn(() => []);
101
+ recoverTmpFiles = vi.fn(async () => []);
102
+ }
103
+ return { ManifestStore: vi.fn(function (_recordsDir: string) { return new FakeManifestStore(); }) };
104
+ });
105
+
106
+ vi.mock("../temp-prompt.ts", () => ({
107
+ writePromptToTempFile: vi.fn(async (agent: string) => {
108
+ const safeName = agent.replace(/[^\w.-]+/g, "_");
109
+ return { dir: `/tmp/fake-${safeName}`, filePath: `/tmp/fake-${safeName}/prompt-${safeName}.md` };
110
+ }),
111
+ cleanupTempPrompt: vi.fn(async () => {}),
112
+ }));
113
+
114
+ import { spawn } from "node:child_process";
115
+
116
+ import { ModelConfigService } from "../model-config-service.ts";
117
+ import type { ModelInfo, ModelRegistryLike } from "../model-resolver.ts";
118
+ import { RELAY_ENV_NODE, RELAY_ENV_SCRIPT, RELAY_ENV_SOCKET } from "../relay-env.ts";
119
+ import { SubagentService } from "../subagent-service.ts";
120
+ import { createBackgroundStream } from "../stream-sink.ts";
121
+ import type { ExtensionMode } from "../host-mode.ts";
122
+
123
+ const mockSpawn = vi.mocked(spawn);
124
+ const mockCreateStream = vi.mocked(createBackgroundStream);
125
+
126
+ interface FakeChild {
127
+ pid: number;
128
+ stdout: PassThrough;
129
+ emit(event: string, ...args: unknown[]): boolean;
130
+ }
131
+
132
+ const RELAY_ENV_KEYS = [RELAY_ENV_SOCKET, RELAY_ENV_NODE, RELAY_ENV_SCRIPT] as const;
133
+
134
+ /** relay 激活三 env 齐备输入(判定矩阵共用)。 */
135
+ const RELAY_ON: Record<string, string> = {
136
+ [RELAY_ENV_SOCKET]: "/tmp/retire.sock",
137
+ [RELAY_ENV_NODE]: "/usr/bin/node",
138
+ [RELAY_ENV_SCRIPT]: "/opt/relay.mjs",
139
+ };
140
+
141
+ function lastSpawnedChild(): FakeChild {
142
+ const result = mockSpawn.mock.results.at(-1);
143
+ if (!result) throw new Error("spawn was not called yet");
144
+ return result.value as FakeChild;
145
+ }
146
+
147
+ function makeEmptyRegistry(): ModelRegistryLike {
148
+ return { getAvailable: () => [], find: () => undefined, hasConfiguredAuth: () => true };
149
+ }
150
+
151
+ function makePi() {
152
+ return { sendMessage: vi.fn(), appendEntry: vi.fn(), events: { emit: vi.fn() } };
153
+ }
154
+
155
+ const ctxModel: ModelInfo = { id: "m", name: "M", provider: "p", reasoning: false };
156
+
157
+ function setupService(mode: ExtensionMode | undefined, sessionId = "retire-it"): SubagentService {
158
+ const agentDir = "/tmp/stream-retire-it";
159
+ const modelService = new ModelConfigService({ agentDir });
160
+ modelService.initModel({
161
+ modelRegistry: makeEmptyRegistry(),
162
+ sessionId,
163
+ ctxModel,
164
+ });
165
+ const service = new SubagentService({
166
+ cwd: agentDir,
167
+ modelService,
168
+ getMainSessionFile: () => "/mock/main-session.jsonl",
169
+ });
170
+ service.initSession({
171
+ pi: makePi(),
172
+ sessionId,
173
+ mode,
174
+ streamSink: { setWidget: vi.fn() },
175
+ });
176
+ return service;
177
+ }
178
+
179
+ /** 等 kickOffBackground 的 detached runAndFinalize 走到 spawn(stream 创建先于此)。 */
180
+ async function waitForSpawnCall(): Promise<void> {
181
+ const baseline = mockSpawn.mock.calls.length;
182
+ const deadline = Date.now() + 2000;
183
+ while (mockSpawn.mock.calls.length <= baseline) {
184
+ if (Date.now() > deadline) throw new Error("spawn was not called within 2000ms");
185
+ await new Promise((r) => setTimeout(r, 5));
186
+ }
187
+ }
188
+
189
+ async function executeBackground(service: SubagentService): Promise<void> {
190
+ await service.execute({ task: "retire test", ctxModel });
191
+ await waitForSpawnCall();
192
+ // 推进 detached runSpawn 收尾(FakeChild close),防句柄悬挂干扰后续用例
193
+ lastSpawnedChild().emit("close", 0);
194
+ }
195
+
196
+ // ── 判定层:createBackgroundStream 真实函数(vi.importActual 绕过文件级 mock)──
197
+
198
+ describe("退役步骤 2 判定层:createBackgroundStream", () => {
199
+ const sink = { setWidget: vi.fn() };
200
+
201
+ it("gui(mode=rpc)+ relay 激活 → 不创建(tee 供数,widget 私货停发)", async () => {
202
+ const actual = await vi.importActual<typeof import("../stream-sink.ts")>("../stream-sink.ts");
203
+ expect(actual.createBackgroundStream("rec-1", sink, "rpc", RELAY_ON)).toBeUndefined();
204
+ });
205
+
206
+ it("tui(mode=tui)+ relay 激活 → 原样创建(TUI widget 行是终端用户的实时预览)", async () => {
207
+ const actual = await vi.importActual<typeof import("../stream-sink.ts")>("../stream-sink.ts");
208
+ const stream = actual.createBackgroundStream("rec-1", sink, "tui", RELAY_ON);
209
+ expect(stream).toBeInstanceOf(actual.SubagentStream);
210
+ stream?.dispose();
211
+ expect(sink.setWidget).toHaveBeenCalledWith(`subagent-stream-rec-1`, undefined);
212
+ });
213
+
214
+ it("gui + relay 未激活 → 原样创建(现状路径,独立 pi / 无 runtime 环境零回归)", async () => {
215
+ const actual = await vi.importActual<typeof import("../stream-sink.ts")>("../stream-sink.ts");
216
+ const stream = actual.createBackgroundStream("rec-1", sink, "rpc", {});
217
+ expect(stream).toBeInstanceOf(actual.SubagentStream);
218
+ stream?.dispose();
219
+ });
220
+
221
+ it("headless(mode 未穿透)+ relay 激活 → 原样创建(抑制条件精确等于 gui,不扩大面)", async () => {
222
+ const actual = await vi.importActual<typeof import("../stream-sink.ts")>("../stream-sink.ts");
223
+ const stream = actual.createBackgroundStream("rec-1", sink, undefined, RELAY_ON);
224
+ expect(stream).toBeInstanceOf(actual.SubagentStream);
225
+ stream?.dispose();
226
+ });
227
+
228
+ it("sink=null 恒不创建(session_start 未注入降级,与退役判定正交)", async () => {
229
+ const actual = await vi.importActual<typeof import("../stream-sink.ts")>("../stream-sink.ts");
230
+ expect(actual.createBackgroundStream("rec-1", null, "tui", {})).toBeUndefined();
231
+ expect(actual.createBackgroundStream("rec-1", null, "rpc", RELAY_ON)).toBeUndefined();
232
+ });
233
+ });
234
+
235
+ // ── 接线层:service → 工厂的参数透传(mode 传错 = 判定失效)──
236
+
237
+ describe("退役步骤 2 接线层:kickOffBackground 透传 mode/sink 给工厂", () => {
238
+ beforeEach(() => {
239
+ vi.clearAllMocks();
240
+ for (const key of RELAY_ENV_KEYS) delete process.env[key];
241
+ });
242
+
243
+ afterEach(() => {
244
+ for (const key of RELAY_ENV_KEYS) delete process.env[key];
245
+ vi.restoreAllMocks();
246
+ });
247
+
248
+ it.each<[string, ExtensionMode | undefined]>([
249
+ ["rpc", "rpc"],
250
+ ["tui", "tui"],
251
+ ["undefined(headless)", undefined],
252
+ ])("mode=%s:工厂收到 initSession 注入的 mode 与 streamSink", async (_label, mode) => {
253
+ const service = setupService(mode);
254
+ await executeBackground(service);
255
+ expect(mockCreateStream).toHaveBeenCalledTimes(1);
256
+ const [recordId, sinkArg, modeArg] = mockCreateStream.mock.calls[0]!;
257
+ expect(typeof recordId).toBe("string");
258
+ expect(sinkArg).toHaveProperty("setWidget");
259
+ expect(modeArg).toBe(mode);
260
+ });
261
+ });
@@ -0,0 +1,310 @@
1
+ // subprocess-agent-runner-routing.test.ts —— P4 SAR 路由集成测试(验收 1/2/3 的
2
+ // runner 级证据面):三层优先级落到 run、fallback 留痕进 ExecuteOptions(record 投影
3
+ // 的入参面)、守卫/strict 报错入 result.error、journal 池 key 对齐(对齐点③:落盘
4
+ // 路径 = 引擎 onPoolResolved 声明的池 key 派生路径)。
5
+ //
6
+ // fake 注入:pi 侧 mock SubagentService(executeAndAwait 记录入参);zcode 侧向
7
+ // registry 注册 fake EnginePort(probe 结果可注入);frontmatter/全局配置经真实
8
+ // ModelConfigService 单例(agentDir 指向临时目录,真实落盘 .md 与 config.json)。
9
+
10
+ import { spawn } from "node:child_process";
11
+ import * as fs from "node:fs";
12
+ import * as os from "node:os";
13
+ import * as path from "node:path";
14
+
15
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
16
+
17
+ import type { AgentCallOpts, AgentResult } from "../../orchestration/models/types.ts";
18
+ import type { RunContext } from "../engine/port.ts";
19
+ import { clearEngines, registerEngine } from "../engine/registry.ts";
20
+ import type { EnginePort, ProbeReport } from "../engine/types.ts";
21
+ import { ModelConfigService, setModelConfigService } from "../model-config-service.ts";
22
+ import { getChildByRecord } from "../session-runner.ts";
23
+ import { SubprocessAgentRunner } from "../subprocess-agent-runner.ts";
24
+ import type { SubagentService } from "../subagent-service.ts";
25
+ import type { ExecuteOptions } from "../types.ts";
26
+
27
+ // ── fake 引擎(registry 登记,probe/run 可记录;run 模拟 onPoolResolved + 事件 emit)──
28
+
29
+ interface FakeEngineCalls {
30
+ probed: number;
31
+ runs: Array<{ taskSpec: unknown; ctx: RunContext }>;
32
+ }
33
+
34
+ function makeFakeZcodeEngine(probeOk: boolean): { engine: EnginePort; calls: FakeEngineCalls } {
35
+ const calls: FakeEngineCalls = { probed: 0, runs: [] };
36
+ const engine: EnginePort = {
37
+ id: "zcode",
38
+ capabilities: () => ({
39
+ schemaEnforcement: "emulated",
40
+ steer: "unsupported",
41
+ conversation: "unsupported",
42
+ personaInjection: "prompt",
43
+ eventGranularity: "coarse",
44
+ sandbox: "none",
45
+ sessionRead: "outcome-only",
46
+ resume: "cold",
47
+ interrupt: "kill-only",
48
+ permissionMode: "native",
49
+ }),
50
+ probe: () => {
51
+ calls.probed++;
52
+ const report: ProbeReport = probeOk
53
+ ? { ok: true, engineVersion: "0.16.5", checks: [{ name: "stub", ok: true }] }
54
+ : {
55
+ ok: false,
56
+ engineVersion: "",
57
+ checks: [{ name: "binary", ok: false, detail: "missing" }],
58
+ error: { code: "engine_probe_failed", recovery: "reinstall zcode and retry the probe" },
59
+ };
60
+ return Promise.resolve(report);
61
+ },
62
+ run: (taskSpec, ctx) => {
63
+ calls.runs.push({ taskSpec, ctx });
64
+ // 对齐点③模拟:prepare 期声明池 key → 事件 emit(zcode coarse 形态:终态后合成)
65
+ ctx.onPoolResolved?.("home-test-provider-m1");
66
+ ctx.onEvent?.({ type: "message_end", usage: { input: 1, output: 1, cacheRead: 0, cacheWrite: 0 } });
67
+ ctx.onEvent?.({ type: "turn_end" });
68
+ return Promise.resolve({
69
+ handle: { data: { v: 1, engineId: "zcode", sessionRef: {}, poolKey: "home-test-provider-m1", adapterVersion: "t" } },
70
+ outcome: { engineId: "zcode", content: "from-zcode" },
71
+ });
72
+ },
73
+ interact: () => Promise.resolve({ ok: false, code: "engine_capability_unsupported", message: "stub" }),
74
+ read: () => Promise.resolve({ engineId: "zcode", turns: [], source: "outcome-only" }),
75
+ };
76
+ return { engine, calls };
77
+ }
78
+
79
+ // ── mock pi 编排服务(executeAndAwait 入参记录——record 投影的入参面)──
80
+
81
+ function makeMockPiService() {
82
+ const executeOpts: ExecuteOptions[] = [];
83
+ const executeAndAwait = vi.fn(async (opts: ExecuteOptions): Promise<AgentResult> => {
84
+ executeOpts.push(opts);
85
+ return { content: "from-pi", durationMs: 1, toolCalls: [] };
86
+ });
87
+ const service = { executeAndAwait } as unknown as SubagentService;
88
+ return { service, executeOpts, executeAndAwait };
89
+ }
90
+
91
+ // ── 环境 ──
92
+
93
+ let tmpRoot: string;
94
+ let agentDir: string;
95
+ let prevDataDirEnv: string | undefined;
96
+
97
+ function writeGlobalConfig(cfg: Record<string, unknown>): void {
98
+ fs.mkdirSync(path.join(agentDir, "subagents"), { recursive: true });
99
+ fs.writeFileSync(path.join(agentDir, "subagents", "config.json"), JSON.stringify(cfg));
100
+ }
101
+
102
+ function writeAgentMd(name: string, frontmatter: string): string {
103
+ fs.mkdirSync(path.join(agentDir, "agents"), { recursive: true });
104
+ const p = path.join(agentDir, "agents", `${name}.md`);
105
+ fs.writeFileSync(p, `---\n${frontmatter}\n---\nbody`);
106
+ return p;
107
+ }
108
+
109
+ function makeOpts(overrides?: Partial<AgentCallOpts>): AgentCallOpts {
110
+ return {
111
+ prompt: "task",
112
+ description: "routing-test",
113
+ ...overrides,
114
+ };
115
+ }
116
+
117
+ /** 装配真实 ModelConfigService 单例(agentDir 指临时目录——frontmatter/config.json 真实落盘)。 */
118
+ function installModelService(cfg?: Record<string, unknown>): void {
119
+ if (cfg !== undefined) writeGlobalConfig(cfg);
120
+ setModelConfigService(new ModelConfigService({ agentDir, cwd: tmpRoot }));
121
+ }
122
+
123
+ beforeEach(() => {
124
+ tmpRoot = fs.mkdtempSync(path.join(os.tmpdir(), "sar-routing-"));
125
+ agentDir = path.join(tmpRoot, "pi-agent");
126
+ fs.mkdirSync(agentDir, { recursive: true });
127
+ // journal 落盘隔离(getEngineDataDir 的权威通道):tmpRoot 下,不污染真实目录
128
+ prevDataDirEnv = process.env["XYZ_AGENT_DATA_DIR"];
129
+ process.env["XYZ_AGENT_DATA_DIR"] = path.join(tmpRoot, "engine-data");
130
+ clearEngines();
131
+ });
132
+
133
+ afterEach(() => {
134
+ if (prevDataDirEnv === undefined) delete process.env["XYZ_AGENT_DATA_DIR"];
135
+ else process.env["XYZ_AGENT_DATA_DIR"] = prevDataDirEnv;
136
+ clearEngines();
137
+ fs.rmSync(tmpRoot, { recursive: true, force: true });
138
+ });
139
+
140
+ describe("SAR 路由集成(P4 验收 1/2/3)", () => {
141
+ it("缺省路径:无任何指定 → pi(本地 DI 绑定),ExecuteOptions.engine='pi' 留痕", async () => {
142
+ installModelService();
143
+ const pi = makeMockPiService();
144
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
145
+
146
+ const result = await sar.run(makeOpts(), new AbortController().signal);
147
+
148
+ expect(result.content).toBe("from-pi");
149
+ expect(pi.executeOpts[0]?.engine).toBe("pi"); // record 投影的入参面(验收 3)
150
+ expect(pi.executeOpts[0]?.engineFallback).toBeUndefined();
151
+ });
152
+
153
+ it("frontmatter 指定 zcode(probe ok)→ registry 引擎执行;journal 落盘路径 = 引擎池 key(对齐点③)", async () => {
154
+ const agentRef = writeAgentMd("reviewer", "name: reviewer\ndescription: d\nengine: zcode");
155
+ installModelService();
156
+ const { engine, calls } = makeFakeZcodeEngine(true);
157
+ registerEngine("zcode", () => engine);
158
+ const pi = makeMockPiService();
159
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
160
+
161
+ const result = await sar.run(makeOpts({ agent: agentRef }), new AbortController().signal);
162
+
163
+ expect(result.content).toBe("from-zcode");
164
+ expect(calls.runs).toHaveLength(1);
165
+ expect(pi.executeAndAwait).not.toHaveBeenCalled();
166
+ // 对齐点③断言:journal 落在引擎声明的池 key 目录(非 pi 'shared' 占位路径)
167
+ const poolDir = path.join(tmpRoot, "engine-data", "engines", "zcode", "home-test-provider-m1");
168
+ const journals = fs.existsSync(poolDir) ? fs.readdirSync(poolDir) : [];
169
+ expect(journals).toHaveLength(1);
170
+ expect(journals[0]).toMatch(/^journal-sa-.+\.jsonl$/);
171
+ const sharedDir = path.join(tmpRoot, "engine-data", "engines", "zcode", "shared");
172
+ expect(fs.existsSync(sharedDir)).toBe(false); // 占位 poolKey 未产生错位落盘
173
+ });
174
+
175
+ it("调用参数 engine 覆盖 frontmatter(三层优先级落到 run,A7)", async () => {
176
+ const agentRef = writeAgentMd("reviewer", "name: reviewer\ndescription: d\nengine: zcode");
177
+ installModelService();
178
+ const zcode = makeFakeZcodeEngine(true);
179
+ registerEngine("zcode", () => zcode.engine);
180
+ const pi = makeMockPiService();
181
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
182
+
183
+ await sar.run(makeOpts({ agent: agentRef, engine: "pi" }), new AbortController().signal);
184
+
185
+ expect(zcode.calls.runs).toHaveLength(0);
186
+ expect(pi.executeOpts).toHaveLength(1); // 调用参数显式 pi 胜出
187
+ });
188
+
189
+ it("全局默认引擎(config defaultEngine=zcode)生效", async () => {
190
+ installModelService({ version: 1, maxConcurrent: 6, defaultEngine: "zcode" });
191
+ const { engine, calls } = makeFakeZcodeEngine(true);
192
+ registerEngine("zcode", () => engine);
193
+ const pi = makeMockPiService();
194
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
195
+
196
+ await sar.run(makeOpts(), new AbortController().signal);
197
+ expect(calls.runs).toHaveLength(1);
198
+ expect(pi.executeAndAwait).not.toHaveBeenCalled();
199
+ });
200
+
201
+ it("frontmatter zcode probe 失败 → 路由回 pi + engineFallback 进 ExecuteOptions(record 留痕面,A9①)", async () => {
202
+ const agentRef = writeAgentMd("reviewer", "name: reviewer\ndescription: d\nengine: zcode");
203
+ installModelService();
204
+ const zcode = makeFakeZcodeEngine(false);
205
+ registerEngine("zcode", () => zcode.engine);
206
+ const pi = makeMockPiService();
207
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
208
+
209
+ const result = await sar.run(makeOpts({ agent: agentRef }), new AbortController().signal);
210
+
211
+ expect(result.content).toBe("from-pi");
212
+ expect(pi.executeOpts[0]?.engine).toBe("pi");
213
+ expect(pi.executeOpts[0]?.engineFallback).toEqual({ from: "zcode", reason: "engine_probe_failed" });
214
+ expect(zcode.calls.probed).toBe(1);
215
+ });
216
+
217
+ it("调用参数显式 zcode + probe 失败 → 不兜底,result.error 含 engine_probe_failed(守卫 a,A9②)", async () => {
218
+ installModelService();
219
+ const zcode = makeFakeZcodeEngine(false);
220
+ registerEngine("zcode", () => zcode.engine);
221
+ const pi = makeMockPiService();
222
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
223
+
224
+ const result = await sar.run(makeOpts({ engine: "zcode" }), new AbortController().signal);
225
+
226
+ expect(result.content).toBe("");
227
+ expect(result.error).toContain("engine_probe_failed");
228
+ expect(pi.executeAndAwait).not.toHaveBeenCalled();
229
+ });
230
+
231
+ it("frontmatter zcode + 显式 model + probe 失败 → model_not_available(守卫 c)", async () => {
232
+ const agentRef = writeAgentMd("reviewer", "name: reviewer\ndescription: d\nengine: zcode");
233
+ installModelService();
234
+ const zcode = makeFakeZcodeEngine(false);
235
+ registerEngine("zcode", () => zcode.engine);
236
+ const pi = makeMockPiService();
237
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
238
+
239
+ const result = await sar.run(
240
+ makeOpts({ agent: agentRef, model: "builtin:bigmodel-coding-plan/GLM-5.3" }),
241
+ new AbortController().signal,
242
+ );
243
+
244
+ expect(result.error).toContain("model_not_available");
245
+ expect(pi.executeAndAwait).not.toHaveBeenCalled();
246
+ });
247
+
248
+ it("strict=true(config engineRouting.strict)→ frontmatter 层 probe 失败直接报错(A5)", async () => {
249
+ const agentRef = writeAgentMd("reviewer", "name: reviewer\ndescription: d\nengine: zcode");
250
+ installModelService({ version: 1, maxConcurrent: 6, engineRouting: { strict: true } });
251
+ const zcode = makeFakeZcodeEngine(false);
252
+ registerEngine("zcode", () => zcode.engine);
253
+ const pi = makeMockPiService();
254
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
255
+
256
+ const result = await sar.run(makeOpts({ agent: agentRef }), new AbortController().signal);
257
+
258
+ expect(result.error).toContain("engine_probe_failed");
259
+ expect(result.error).toContain("strict");
260
+ expect(pi.executeAndAwait).not.toHaveBeenCalled();
261
+ });
262
+
263
+ it("调用参数未注册 id → engine_not_found 入 result.error(文案含注册清单)", async () => {
264
+ installModelService();
265
+ const pi = makeMockPiService();
266
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
267
+
268
+ const result = await sar.run(makeOpts({ engine: "no-such-engine" }), new AbortController().signal);
269
+
270
+ expect(result.error).toContain("engine_not_found");
271
+ expect(result.error).toContain("Registered engines:");
272
+ });
273
+
274
+ it("frontmatter 未注册 id(解析期拦截)→ SAR 转错误结果(不 crash)", async () => {
275
+ // frontmatter 校验在 agent 解析期(agent-registry 抛 EngineNotFoundError)——经
276
+ // ModelConfigService.getAgentConfig 到达 SAR 路由装配,SAR catch 转 result.error
277
+ const agentRef = writeAgentMd("broken", "name: broken\ndescription: d\nengine: ghost-engine");
278
+ installModelService();
279
+ const pi = makeMockPiService();
280
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
281
+
282
+ const result = await sar.run(makeOpts({ agent: agentRef }), new AbortController().signal);
283
+
284
+ expect(result.error).toContain("engine_not_found");
285
+ expect(pi.executeAndAwait).not.toHaveBeenCalled();
286
+ });
287
+
288
+ it("[D10] RunContext.onChildSpawned 注入:子进程注册进 spawnedChildren,退出后按句移除", async () => {
289
+ // C-ext-17 路径①回归:SAR 构造的 RunContext 必须提供 onChildSpawned(引擎 spawn
290
+ // 后回调 → session-runner spawnedChildren 记账 → dispose killAll 收割兜底可见)
291
+ const agentRef = writeAgentMd("reviewer", "name: reviewer\ndescription: d\nengine: zcode");
292
+ installModelService();
293
+ const { engine, calls } = makeFakeZcodeEngine(true);
294
+ registerEngine("zcode", () => engine);
295
+ const pi = makeMockPiService();
296
+ const sar = new SubprocessAgentRunner({ subagentService: pi.service });
297
+
298
+ await sar.run(makeOpts({ agent: agentRef }), new AbortController().signal);
299
+
300
+ const ctx = calls.runs[0]?.ctx;
301
+ expect(ctx?.onChildSpawned).toBeTypeOf("function");
302
+ // 引擎回调(真实引擎在 spawn 成功后同步调)→ 按 taskId('sa-' 记账 key)注册可见
303
+ const child = spawn(process.execPath, ["-e", ""]);
304
+ ctx?.onChildSpawned?.(child);
305
+ expect(getChildByRecord(ctx.taskId)).toBe(child);
306
+ // 子进程退出(真实 close 事件)后记账按句移除——不残留死句柄
307
+ await new Promise<void>((resolve) => child.once("close", () => resolve()));
308
+ expect(getChildByRecord(ctx.taskId)).toBeUndefined();
309
+ });
310
+ });