@zhushanwen/pi-subagent-workflow 7.4.0 → 8.0.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 (148) hide show
  1. package/README.md +15 -0
  2. package/package.json +1 -1
  3. package/skills/workflow-script-format/SKILL.md +1 -1
  4. package/src/execution/__tests__/__fixtures__/truncline.snapshot.json +1 -0
  5. package/src/execution/__tests__/ask-user-transit-e2e.test.ts +10 -4
  6. package/src/execution/__tests__/before-agent-start-injection.test.ts +132 -0
  7. package/src/execution/__tests__/bg-notify-render.test.ts +15 -15
  8. package/src/execution/__tests__/chatmode-first-round-closure-service.test.ts +365 -0
  9. package/src/execution/__tests__/chatmode-first-round-closure-spawn.test.ts +190 -0
  10. package/src/execution/__tests__/chatmode-round-notify-real-chain.test.ts +215 -0
  11. package/src/execution/__tests__/conversation-wiring.test.ts +198 -0
  12. package/src/execution/__tests__/crash-recovery.test.ts +8 -2
  13. package/src/execution/__tests__/delivery-methods.test.ts +385 -0
  14. package/src/execution/__tests__/epipe-fallback.test.ts +241 -0
  15. package/src/execution/__tests__/execute-and-await-worktree.test.ts +49 -2
  16. package/src/execution/__tests__/execute-nesting.test.ts +20 -72
  17. package/src/execution/__tests__/execution-record.test.ts +199 -0
  18. package/src/execution/__tests__/finalize-record.test.ts +170 -14
  19. package/src/execution/__tests__/format.test.ts +131 -7
  20. package/src/execution/__tests__/gc-timer.test.ts +184 -0
  21. package/src/execution/__tests__/get-record-for-action-restart.test.ts +254 -0
  22. package/src/execution/__tests__/helpers/spawn-mock.ts +25 -7
  23. package/src/execution/__tests__/index-session-start-identity.test.ts +371 -0
  24. package/src/execution/__tests__/index-session-start.test.ts +257 -5
  25. package/src/execution/__tests__/lifecycle-manager-lock.test.ts +211 -0
  26. package/src/execution/__tests__/lifecycle-manager.test.ts +337 -0
  27. package/src/execution/__tests__/lifecycle-predicates.test.ts +116 -0
  28. package/src/execution/__tests__/list-component.test.ts +59 -5
  29. package/src/execution/__tests__/list-fields.test.ts +109 -0
  30. package/src/execution/__tests__/model-resolver.test.ts +38 -1
  31. package/src/execution/__tests__/nested-visibility-env-propagation.test.ts +287 -0
  32. package/src/execution/__tests__/nested-visibility.test.ts +325 -0
  33. package/src/execution/__tests__/notifier-flush.test.ts +209 -7
  34. package/src/execution/__tests__/one-shot-upgrade.test.ts +205 -0
  35. package/src/execution/__tests__/parent-child-matrix.test.ts +336 -0
  36. package/src/execution/__tests__/record-store.test.ts +158 -52
  37. package/src/execution/__tests__/recursive-visibility-baseline.test.ts +11 -12
  38. package/src/execution/__tests__/recursive-visibility-env.test.ts +18 -20
  39. package/src/execution/__tests__/resource-policy.test.ts +109 -0
  40. package/src/execution/__tests__/run-and-finalize-chatmode.test.ts +267 -0
  41. package/src/execution/__tests__/run-spawn-chatmode-settled.test.ts +253 -0
  42. package/src/execution/__tests__/run-spawn-edges.test.ts +18 -25
  43. package/src/execution/__tests__/run-spawn-integration.test.ts +29 -25
  44. package/src/execution/__tests__/run-spawn-resume.test.ts +322 -0
  45. package/src/execution/__tests__/run-spawn-rpc-mode.test.ts +14 -11
  46. package/src/execution/__tests__/session-pending.test.ts +61 -2
  47. package/src/execution/__tests__/session-reconstructor.test.ts +4 -4
  48. package/src/execution/__tests__/session-runner-epipe.test.ts +178 -0
  49. package/src/execution/__tests__/session-runner-schema-env.test.ts +15 -21
  50. package/src/execution/__tests__/session-start-reaper.test.ts +10 -8
  51. package/src/execution/__tests__/spawn-args.test.ts +127 -49
  52. package/src/execution/__tests__/spawn-worktree-guidance.test.ts +1 -0
  53. package/src/execution/__tests__/spawned-children.test.ts +92 -0
  54. package/src/execution/__tests__/status-refactor.test.ts +345 -0
  55. package/src/execution/__tests__/stdin-writer.test.ts +97 -0
  56. package/src/execution/__tests__/subagent-service-message-close.test.ts +598 -0
  57. package/src/execution/__tests__/subagent-service-parent-guard.test.ts +180 -0
  58. package/src/execution/__tests__/subagent-service.test.ts +49 -11
  59. package/src/execution/__tests__/timeout-integration.test.ts +27 -13
  60. package/src/execution/__tests__/tool-action.test.ts +12 -10
  61. package/src/execution/__tests__/truncline-snapshot.test.ts +81 -0
  62. package/src/execution/__tests__/turn-limiter-semantics.test.ts +194 -0
  63. package/src/execution/__tests__/worktree-manager.test.ts +292 -89
  64. package/src/execution/__tests__/worktree-pid-registration.integration.test.ts +13 -12
  65. package/src/execution/argv-mirror.ts +21 -2
  66. package/src/execution/execution-record.ts +126 -9
  67. package/src/execution/finalize-record.ts +90 -13
  68. package/src/execution/lifecycle-manager.ts +484 -0
  69. package/src/execution/lifecycle-predicates.ts +65 -0
  70. package/src/execution/manifest-store.ts +61 -16
  71. package/src/execution/model-resolver.ts +26 -5
  72. package/src/execution/notifier.ts +69 -12
  73. package/src/execution/pi-invocation.ts +21 -1
  74. package/src/execution/record-store.ts +554 -107
  75. package/src/execution/session-pending.ts +116 -45
  76. package/src/execution/session-reconstructor.ts +224 -7
  77. package/src/execution/session-runner.ts +270 -68
  78. package/src/execution/sessions-index.ts +304 -0
  79. package/src/execution/stdin-writer.ts +93 -7
  80. package/src/execution/stream-sink.ts +20 -3
  81. package/src/execution/subagent-service.ts +867 -138
  82. package/src/execution/turn-limiter.ts +14 -0
  83. package/src/execution/types.ts +200 -19
  84. package/src/execution/worktree-manager.ts +128 -49
  85. package/src/execution/worktree-registry.ts +13 -2
  86. package/src/index.ts +277 -19
  87. package/src/injectors/subagent-list-injector.ts +26 -8
  88. package/src/injectors/workflow-list-injector.ts +25 -8
  89. package/src/interface/__tests__/subagent-tool-prompt.test.ts +17 -4
  90. package/src/interface/__tests__/tool-render.test.ts +10 -8
  91. package/src/interface/bg-notify-render.ts +32 -8
  92. package/src/interface/command-actions.ts +26 -7
  93. package/src/interface/commands.ts +21 -22
  94. package/src/interface/format.ts +43 -16
  95. package/src/interface/gui-mappers.ts +6 -8
  96. package/src/interface/helpers.ts +170 -10
  97. package/src/interface/list-component.ts +53 -14
  98. package/src/interface/subagent-actions.ts +235 -17
  99. package/src/interface/subagent-tool.ts +81 -16
  100. package/src/interface/subagents.ts +2 -1
  101. package/src/interface/tool-render.ts +11 -24
  102. package/src/interface/tool-workflow.ts +20 -35
  103. package/src/interface/views/WorkflowsView.ts +89 -32
  104. package/src/interface/views/__tests__/WorkflowsView-signature.test.ts +264 -0
  105. package/src/interface/views/format.ts +3 -3
  106. package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +325 -0
  107. package/src/orchestration/__tests__/args-validator.test.ts +1 -1
  108. package/src/orchestration/__tests__/config-loader.test.ts +38 -0
  109. package/src/orchestration/__tests__/error-recovery-handlers.test.ts +394 -4
  110. package/src/orchestration/__tests__/error-recovery-workflow-call.test.ts +4 -4
  111. package/src/orchestration/__tests__/execute-agent-call.test.ts +95 -0
  112. package/src/orchestration/__tests__/jsonl-run-store-session-file.test.ts +657 -18
  113. package/src/orchestration/__tests__/launcher-nested-workflow.test.ts +0 -2
  114. package/src/orchestration/__tests__/lifecycle.test.ts +332 -149
  115. package/src/orchestration/__tests__/skill-discovery.test.ts +130 -61
  116. package/src/orchestration/__tests__/test-mocks.ts +191 -0
  117. package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +98 -0
  118. package/src/orchestration/__tests__/workflow-nesting-e2e.test.ts +0 -2
  119. package/src/orchestration/__tests__/workflow-script-lint-memo.test.ts +110 -0
  120. package/src/orchestration/__tests__/workflows-e2e.test.ts +1 -1
  121. package/src/orchestration/agent-opts-resolver.ts +4 -1
  122. package/src/orchestration/args-validator.ts +2 -2
  123. package/src/orchestration/config-loader.ts +30 -1
  124. package/src/orchestration/error-recovery.ts +133 -29
  125. package/src/orchestration/execute-agent-call.ts +31 -7
  126. package/src/orchestration/jsonl-run-store.ts +287 -40
  127. package/src/orchestration/launcher.ts +7 -1
  128. package/src/orchestration/lifecycle.ts +135 -132
  129. package/src/orchestration/models/__tests__/trace.test.ts +408 -0
  130. package/src/orchestration/models/budget.ts +1 -1
  131. package/src/orchestration/models/run-runtime.ts +15 -17
  132. package/src/orchestration/models/run-spec.ts +2 -2
  133. package/src/orchestration/models/run-state.ts +3 -3
  134. package/src/orchestration/models/trace.ts +95 -15
  135. package/src/orchestration/models/types.ts +8 -9
  136. package/src/orchestration/models/workflow-run.ts +50 -71
  137. package/src/orchestration/models/workflow-script.ts +32 -1
  138. package/src/orchestration/skill-discovery.ts +30 -0
  139. package/src/orchestration/worker-handle.ts +1 -1
  140. package/src/orchestration/worker-host.ts +1 -1
  141. package/src/orchestration/worker-script-builder.ts +29 -10
  142. package/src/shared/__tests__/agent-ref.test.ts +34 -0
  143. package/src/shared/__tests__/resource-discovery-manifest-cache.test.ts +280 -0
  144. package/src/shared/__tests__/resource-discovery.test.ts +55 -0
  145. package/src/shared/__tests__/schema-jsonify.test.ts +81 -0
  146. package/src/shared/agent-ref.ts +16 -0
  147. package/src/shared/resource-discovery.ts +147 -58
  148. package/src/shared/schema-jsonify.ts +53 -0
@@ -0,0 +1,365 @@
1
+ // src/execution/__tests__/chatmode-first-round-closure-service.test.ts
2
+ //
3
+ // [V2 决策 2/3] chatMode 首轮闭环——onRoundSettled 注入 + runAndFinalize early return
4
+ // + double-notify 防护(Step 4a 改动 2/3)。
5
+ //
6
+ // 改动 2:buildSessionRunnerContext 注入 onRoundSettled 回调——设 record.status=idle +
7
+ // round+=1 + notifyComplete(轻量 idle 化,不调 doFinalizeRoundToIdle)。
8
+ // 改动 3:runAndFinalize 检测 chatMode + status===idle(onRoundSettled 已设)→ early return,
9
+ // 不进现有 chatMode 分流(那是 close 后 done/failed/cancelled 终态化的)。
10
+ // double-notify 防护:onRoundSettled notify + kickOffBackground.then notify 同 id:round,
11
+ // notifier dedup(id:round key,60s TTL)吞第二次——notifier.ts L122 是天然防线。
12
+ //
13
+ // mock 结构与 run-and-finalize-chatmode.test.ts 一致(mock session-runner + logger,
14
+ // 走真实 SubagentService 的 buildSessionRunnerContext / runAndFinalize 分流逻辑)。
15
+
16
+ import * as fs from "node:fs";
17
+ import * as os from "node:os";
18
+ import * as path from "node:path";
19
+
20
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
21
+
22
+ // mock logger(doFinalizeRecord 的 manifest 写入降级路径用 logger.error)
23
+ const { loggerMock } = vi.hoisted(() => ({
24
+ loggerMock: { debug: vi.fn(), warn: vi.fn(), error: vi.fn(), info: vi.fn() },
25
+ }));
26
+ vi.mock("@zhushanwen/pi-extension-logger", () => ({ getLogger: () => loggerMock }));
27
+
28
+ // mock session-runner:runSpawn 受控返回 result,killAllSpawnedChildren 空实现。
29
+ // [M3] getChildByRecord 返回活进程假句柄:让 piAdapter.hasRunningBackground 的判定只能靠
30
+ // 「!hasIdleTimer 排除等待续聊 record」通过(还原生产 Path A:轮次完成、进程保活)。
31
+ vi.mock("../session-runner.ts", () => ({
32
+ runSpawn: vi.fn(),
33
+ killAllSpawnedChildren: vi.fn(),
34
+ getChildByRecord: vi.fn(() => ({ killed: false, kill: () => true })),
35
+ }));
36
+
37
+ import { runSpawn, getChildByRecord } from "../session-runner.ts";
38
+ import type { SessionRunnerContext } from "../session-runner.ts";
39
+ import { armIdleTimer, disarmIdleTimer } from "../lifecycle-manager.ts";
40
+ import { createRecord, updateFromEvent } from "../execution-record.ts";
41
+ import { ModelConfigService } from "../model-config-service.ts";
42
+ import type { ModelInfo, ModelRegistryLike } from "../model-resolver.ts";
43
+ import { RecordStore } from "../record-store.ts";
44
+ import { SubagentService } from "../subagent-service.ts";
45
+ import type { PiLike } from "../subagent-service.ts";
46
+ import type {
47
+ AgentResult,
48
+ ExecutionRecord,
49
+ ExecuteOptions,
50
+ } from "../types.ts";
51
+
52
+ const mockRunSpawn = vi.mocked(runSpawn);
53
+ const mockGetChildByRecord = vi.mocked(getChildByRecord);
54
+
55
+ const STUB_MODEL: ModelInfo = {
56
+ id: "test-model",
57
+ name: "Test",
58
+ provider: "test",
59
+ reasoning: false,
60
+ };
61
+
62
+ /** 最小合法 registry(initModel fail-fast 需要;resolveModel 第三层直接透传 ctxModel)。 */
63
+ function makeEmptyRegistry(): ModelRegistryLike {
64
+ return { getAvailable: () => [], find: () => undefined, hasConfiguredAuth: () => true };
65
+ }
66
+
67
+ function makeTmpAgentDir(): string {
68
+ return fs.mkdtempSync(path.join(os.tmpdir(), "chatmode-closure-"));
69
+ }
70
+
71
+ function makePi(): PiLike & {
72
+ appendEntry: ReturnType<typeof vi.fn>;
73
+ events: { emit: ReturnType<typeof vi.fn> };
74
+ sendMessage: ReturnType<typeof vi.fn>;
75
+ } {
76
+ return { appendEntry: vi.fn(), events: { emit: vi.fn() }, sendMessage: vi.fn() };
77
+ }
78
+
79
+ function makeResult(success: boolean): AgentResult {
80
+ return {
81
+ text: success ? "done" : "err",
82
+ turns: 1,
83
+ durationMs: 100,
84
+ success,
85
+ error: success ? undefined : "boom",
86
+ sessionId: "sess-1",
87
+ toolCalls: [],
88
+ };
89
+ }
90
+
91
+ function makeRecord(chatMode: boolean, id = "sa-test"): ExecutionRecord {
92
+ return createRecord(id, {
93
+ agent: "general-purpose",
94
+ model: "test-model",
95
+ mode: "background",
96
+ task: "do something",
97
+ slug: "test",
98
+ startedAt: 1000,
99
+ rootSessionId: "root-session",
100
+ chatMode,
101
+ });
102
+ }
103
+
104
+ /** 暴露 runAndFinalize / store / buildSessionRunnerContext / notifyComplete 私有访问。 */
105
+ interface ServiceInternals {
106
+ store: RecordStore;
107
+ buildSessionRunnerContext(): SessionRunnerContext;
108
+ notifyComplete(record: ExecutionRecord): void;
109
+ runAndFinalize: (
110
+ record: ExecutionRecord,
111
+ opts: ExecuteOptions,
112
+ ctx: SessionRunnerContext,
113
+ identity: {
114
+ agent: string;
115
+ agentConfig: unknown;
116
+ resolved: { model: ModelInfo; thinkingLevel: string | undefined };
117
+ },
118
+ signal: undefined,
119
+ priority: number,
120
+ ) => Promise<AgentResult>;
121
+ }
122
+
123
+ describe("[V2 决策 2/3] chatMode 首轮闭环:onRoundSettled 注入 + early return(改动 2/3)", () => {
124
+ let agentDir: string;
125
+ let modelService: ModelConfigService;
126
+ let service: SubagentService;
127
+ let internals: ServiceInternals;
128
+
129
+ beforeEach(() => {
130
+ agentDir = makeTmpAgentDir();
131
+ modelService = new ModelConfigService({ agentDir });
132
+ service = new SubagentService({ cwd: agentDir, modelService });
133
+ service.initSession({ pi: makePi(), sessionId: "root-session" });
134
+ internals = service as unknown as ServiceInternals;
135
+ mockRunSpawn.mockReset();
136
+ });
137
+
138
+ afterEach(() => {
139
+ service.dispose();
140
+ fs.rmSync(agentDir, { recursive: true, force: true });
141
+ });
142
+
143
+ // ── 改动 2:onRoundSettled 回调注入 ──────────────────────────────────
144
+
145
+ it("buildSessionRunnerContext 注入 onRoundSettled:设 running(idle 折入) + round+=1 + notifyComplete", () => {
146
+ const ctx = internals.buildSessionRunnerContext();
147
+ expect(typeof ctx.onRoundSettled).toBe("function");
148
+
149
+ const record = makeRecord(true); // chatMode, status=running, round=undefined
150
+ const spy = vi.spyOn(internals, "notifyComplete");
151
+
152
+ ctx.onRoundSettled!(record);
153
+
154
+ // [改动 2] 轻量 idle 化(v4 B-1:idle 折入 running):status=running(notify 守卫放行)+ round 0→1(dedup key 递增)
155
+ expect(record.status).toBe("running");
156
+ expect(record.round).toBe(1);
157
+ // notifyComplete 被调 1 次,入参是当前 record(此时 running+isIdle,toNotifyRecord 守卫放行)
158
+ expect(spy).toHaveBeenCalledTimes(1);
159
+ expect(spy).toHaveBeenCalledWith(record);
160
+ });
161
+
162
+ it("onRoundSettled 连续两轮:round 累加(1→2,dedup key 区分每轮)", () => {
163
+ const ctx = internals.buildSessionRunnerContext();
164
+ const record = makeRecord(true);
165
+ record.round = 1; // 第一轮已完成
166
+ vi.spyOn(internals, "notifyComplete");
167
+
168
+ ctx.onRoundSettled!(record);
169
+
170
+ expect(record.status).toBe("running");
171
+ expect(record.round).toBe(2); // 第二轮 round 累加
172
+ });
173
+
174
+ // ── 改动 3:runAndFinalize chatMode 首轮 early return ────────────────
175
+
176
+ /** 直接调私有 runAndFinalize(mock runSpawn 后走完整分流逻辑)。 */
177
+ async function callRunAndFinalize(record: ExecutionRecord, success: boolean): Promise<void> {
178
+ mockRunSpawn.mockResolvedValueOnce(makeResult(success));
179
+ const opts: ExecuteOptions = { task: "do something", slug: "test" };
180
+ const ctx: SessionRunnerContext = {
181
+ cwd: agentDir,
182
+ agentDir,
183
+ skillDirs: [],
184
+ mainCwd: agentDir,
185
+ };
186
+ const identity = {
187
+ agent: "general-purpose",
188
+ agentConfig: undefined,
189
+ resolved: { model: STUB_MODEL, thinkingLevel: undefined },
190
+ };
191
+ await internals.runAndFinalize(record, opts, ctx, identity, undefined, 0);
192
+ }
193
+
194
+ it("runAndFinalize chatMode 首轮(isIdle:timer armed)early return:不进 finalize 分流,record 保持 running-resumable", async () => {
195
+ // [N3] 真实 isIdle 路径:守卫判据是 `record.chatMode && isIdle(record)`(isIdle = hasIdleTimer,
196
+ // 非 status 值)。旧用例手工赋 status="idle" 且未 arm timer——守卫求值 false 不 early return,
197
+ // 通过路径是「非法状态值被 tryTransition 拒绝」,与守卫无关(假覆盖)。此处构造生产可达状态:
198
+ // agent_settled 已执行(armIdleTimer + onRoundSettled round+1),status 保持 running。
199
+ const record = makeRecord(true);
200
+ record.status = "running";
201
+ record.round = 1; // onRoundSettled 已 +1
202
+ internals.store.register(record);
203
+ armIdleTimer(record.id, () => {}); // 真实 arm(isIdle=true),模拟 agent_settled
204
+ try {
205
+ await callRunAndFinalize(record, true);
206
+
207
+ // [改动 3] early return 守卫正分支生效:不进 finalize 分流——round 不被
208
+ // finalizeRoundToIdle 二次 +1(仍 1)、record.result 不被覆写(MF-2 写点会写
209
+ // makeResult.text="done",early return 后保持 undefined)、未 archive、store 仍持有。
210
+ expect(record.status).toBe("running");
211
+ expect(record.round).toBe(1);
212
+ expect(record.result).toBeUndefined();
213
+ expect(internals.store.getMutable(record.id)).toBe(record);
214
+ } finally {
215
+ disarmIdleTimer(record.id);
216
+ }
217
+ });
218
+
219
+ it("runAndFinalize chatMode 首轮 early return 不依赖 tryTransition:close 后 done 分流仍对 running record 生效", async () => {
220
+ // 对照组:chatMode + status=running(未经 onRoundSettled,如旧路径或测试直调)→
221
+ // tryTransition 成功 → 进现有 chatMode done 分流(finalizeRoundToIdle)。证明 early return
222
+ // 守卫只挡 onRoundSettled 设的 idle,不误伤 running record 的正常分流。
223
+ const record = makeRecord(true);
224
+ record.status = "running";
225
+ internals.store.register(record);
226
+ await callRunAndFinalize(record, true);
227
+
228
+ expect(record.status).toBe("running"); // finalizeRoundToIdle 设的(v4 B-1 idle 折入 running)
229
+ expect(record.round).toBe(1); // finalizeRoundToIdle +1
230
+ expect(internals.store.getMutable(record.id)).toBe(record);
231
+ });
232
+
233
+ // ── double-notify 防护 + [M3] 轮次完成通知立即送达 ───────────────────
234
+
235
+ it("[M3] chatMode 轮次完成通知立即送达(不挂 60s 合并窗口)+ 同 id:round dedup 吞第二次", () => {
236
+ // 生产场景:chatMode 轮次完成(agent_settled arm idle timer → onRoundSettled →
237
+ // notifyComplete),record 留 store、status=running、进程保活(Path A)。
238
+ // 旧 piAdapter.hasRunningBackground 按 mode==="background" 计数 → 对该 record 恒 true
239
+ // → notify 恒挂 60s 合并窗口,主 agent 的续聊回复固定延迟 60s(G1 失效)。
240
+ // 修复后排除 isIdle(timer armed)record → 立即 flush。
241
+ //
242
+ // 走真实 SubagentService.notifyComplete → BgNotifier → piAdapter(真实 host adapter),
243
+ // 不再手搓 NotifierHost mock 绕过生产行为(旧用例 `hasRunningBackground: () => false`
244
+ // 是对生产行为的 mock 绕过)。
245
+ //
246
+ // [N2] 禁止手工预置 record.result(这正是掩盖「通知正文恒 (empty)」断链的方式)——
247
+ // 轮次文本经真实 updateFromEvent 累积进 turns,真实 onRoundSettled 回调派生写入
248
+ // record.result(对齐 collectResult 的 getFullText)后 notify。
249
+ const pi = makePi();
250
+ service.initSession({ pi, sessionId: "root-session" }); // 换上带 spy 的 pi
251
+ const record = makeRecord(true);
252
+ record.status = "running";
253
+ updateFromEvent(record, { type: "text_delta", delta: "first-round-done" });
254
+ updateFromEvent(record, { type: "turn_end" });
255
+ internals.store.register(record);
256
+ armIdleTimer(record.id, () => {}); // 模拟 agent_settled:轮次完成、等待续聊(Path A)
257
+ try {
258
+ // 真实回调链:round 0→1 + record.result 从 turns 派生 + notifyComplete
259
+ internals.buildSessionRunnerContext().onRoundSettled!(record);
260
+ expect(record.round).toBe(1);
261
+ expect(record.result).toBe("first-round-done"); // 真实派生(非手工预置)
262
+
263
+ // [M3] 立即 flush——同步断言 sendMessage 已发出(旧实现此处挂 60s timer,0 次调用)
264
+ expect(pi.sendMessage).toHaveBeenCalledTimes(1);
265
+ const sentMsg = pi.sendMessage.mock.calls[0]![0] as { customType: string; content: string };
266
+ expect(sentMsg.customType).toBe("subagent-bg-notify");
267
+ expect(sentMsg.content).toContain("finished a round");
268
+ expect(sentMsg.content).toContain("first-round-done");
269
+
270
+ // double-notify 防护(原用例语义保留):onRoundSettled notify + kickOffBackground.then
271
+ // notifyComplete 同 id:round → notifier dedup key=`${id}:${round}` 60s 内吞第二次。
272
+ internals.notifyComplete(record);
273
+ expect(pi.sendMessage).toHaveBeenCalledTimes(1);
274
+ } finally {
275
+ disarmIdleTimer(record.id);
276
+ }
277
+
278
+ // 对照:真在跑的 background 工作(活进程 + 无 timer)仍计入合并窗口——closed 通知
279
+ // 挂 60s 不立即发送(合并窗口语义对真正的并发完成保留)。
280
+ const busy = makeRecord(false, "sa-busy");
281
+ busy.status = "running";
282
+ internals.store.register(busy); // 活进程(mock 恒返回)+ 无 timer = busy,计入
283
+ const done = makeRecord(false, "sa-done");
284
+ done.status = "closed"; // 终态 notify(toNotifyRecord 放行 closed)
285
+ try {
286
+ internals.notifyComplete(done);
287
+ expect(pi.sendMessage).toHaveBeenCalledTimes(1); // 未新增——busy 挂起合并窗口
288
+ } finally {
289
+ // busy/done 无 timer,无需 disarm;record 由 afterEach dispose 清理
290
+ }
291
+ });
292
+ });
293
+
294
+ // ── [N1] one-shot 成功完成通知(SP-5 回退 resumable 后仍送达)──────────────
295
+
296
+ describe("[N1] one-shot 成功完成通知:SP-5 回退 resumable 后仍送达", () => {
297
+ let agentDir: string;
298
+ let modelService: ModelConfigService;
299
+ let service: SubagentService;
300
+ let internals: ServiceInternals;
301
+
302
+ beforeEach(() => {
303
+ agentDir = makeTmpAgentDir();
304
+ modelService = new ModelConfigService({ agentDir });
305
+ service = new SubagentService({ cwd: agentDir, modelService });
306
+ service.initSession({ pi: makePi(), sessionId: "root-session" });
307
+ internals = service as unknown as ServiceInternals;
308
+ mockRunSpawn.mockReset();
309
+ });
310
+
311
+ afterEach(() => {
312
+ service.dispose();
313
+ fs.rmSync(agentDir, { recursive: true, force: true });
314
+ // 还原默认活进程句柄实现(其他用例的 busy 判定依赖)
315
+ mockGetChildByRecord.mockImplementation(() => ({ killed: false, kill: () => true }));
316
+ });
317
+
318
+ it("真实 execute + mock runSpawn(success) → 恰 1 条 subagent-bg-notify(status=closed、正文含真实结果),record 保持可升级", async () => {
319
+ // 真实链路:execute → kickOffBackground → runAndFinalize(SP-5 分支 → finalizeRoundToIdle
320
+ // 回退 running-resumable)→ .then notifyComplete → BgNotifier → pi.sendMessage。
321
+ // round2 审查实证:旧守卫(closed/isIdle only)对 SP-5 完成态恒拒绝 → 发送数 0。
322
+ modelService.initModel({
323
+ modelRegistry: makeEmptyRegistry(),
324
+ sessionId: "root-session",
325
+ ctxModel: STUB_MODEL,
326
+ });
327
+ const pi = makePi();
328
+ service.initSession({ pi, sessionId: "root-session" }); // 换上带 spy 的 pi
329
+
330
+ // one-shot 成功完成态:进程已死(close 后 spawnedChildren 已清理)——getChildByRecord
331
+ // 返回 undefined,SP-5 回退的 record 因此 isResumable(running + 无活进程)。
332
+ mockRunSpawn.mockResolvedValueOnce(makeResult(true));
333
+ mockGetChildByRecord.mockImplementation(() => undefined);
334
+ try {
335
+ const handle = await service.execute({ task: "one shot task", slug: "oneshot-n1" });
336
+
337
+ await vi.waitFor(() => {
338
+ expect(pi.sendMessage).toHaveBeenCalledTimes(1);
339
+ });
340
+ const sentMsg = pi.sendMessage.mock.calls[0]![0] as {
341
+ customType: string;
342
+ content: string;
343
+ details?: { status?: string };
344
+ };
345
+ expect(sentMsg.customType).toBe("subagent-bg-notify");
346
+ // 完成语义:status=closed(对齐 tool 契约 "runs once, notifies on completion";
347
+ // running 分支文案不含 worktree patchFile 提示,one-shot 需要 closed 分支)
348
+ expect(sentMsg.details?.status).toBe("closed");
349
+ expect(sentMsg.content).toContain("completed");
350
+ expect(sentMsg.content).toContain("done"); // makeResult(true).text,经 MF-2 写入 record.result
351
+
352
+ // 恰好 1 条:无第二个通知点(onRoundSettled 仅 chatMode;.then 后无再触发)
353
+ await new Promise((r) => setTimeout(r, 20));
354
+ expect(pi.sendMessage).toHaveBeenCalledTimes(1);
355
+
356
+ // SP-5 语义不破坏:record 回退 running-resumable(可 message 升级续聊),未终态化
357
+ const record = internals.store.getMutable(handle.subagentId);
358
+ expect(record).toBeDefined();
359
+ expect(record!.status).toBe("running");
360
+ expect(record!.result).toBe("done");
361
+ } finally {
362
+ mockGetChildByRecord.mockImplementation(() => ({ killed: false, kill: () => true }));
363
+ }
364
+ });
365
+ });
@@ -0,0 +1,190 @@
1
+ // src/execution/__tests__/chatmode-first-round-closure-spawn.test.ts
2
+ //
3
+ // [V2 决策 2] chatMode 首轮闭环——runSpawn resolve 时机(Step 4a 改动 1)。
4
+ //
5
+ // 改动 1:chatMode agent_settled 时提前 resolve runSpawn 的 exitCode promise(exit code 0),
6
+ // 让 runAndFinalize 拿到首轮 result,进程仍保活(idle timer armed)。非 chatMode 不变
7
+ //(agent_end kill → close resolve)。
8
+ //
9
+ // 本文件验证 resolve 时机(run-spawn-chatmode-settled.test.ts 验证 idle timer + 回调挂载点,
10
+ // 收尾靠 emit close 让 resolve——隐含 agent_settled 不 resolve;本文件显式断言 agent_settled
11
+ // 即 resolve,且进程未 close)。
12
+ //
13
+ // mock 结构与 run-spawn-chatmode-settled.test.ts 一致(FakeChild + session-pending count=0)。
14
+
15
+ import { spawn } from "node:child_process";
16
+ import * as fs from "node:fs";
17
+
18
+ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
19
+
20
+ vi.mock("node:child_process", async () => {
21
+ const { FakeChild } = await import("./helpers/spawn-mock.ts");
22
+ return {
23
+ spawn: vi.fn(() => new FakeChild()),
24
+ // buildEnvBlock 的 git branch 调用(execFile 异步):默认 err-first 兜底 → catch → branch=""
25
+ execFile: vi.fn(
26
+ (
27
+ _cmd: string,
28
+ _args: readonly string[],
29
+ _opts: unknown,
30
+ cb: (err: Error | null, stdout?: string, stderr?: string) => void,
31
+ ) => cb(new Error("execFile not configured in this test")),
32
+ ),
33
+ };
34
+ });
35
+
36
+ vi.mock("node:fs", async () => {
37
+ const actual = await import("node:fs");
38
+ return {
39
+ default: {
40
+ ...actual,
41
+ mkdirSync: vi.fn(),
42
+ existsSync: vi.fn(() => false),
43
+ appendFileSync: vi.fn(),
44
+ writeFileSync: vi.fn(),
45
+ readdirSync: vi.fn(() => []),
46
+ },
47
+ mkdirSync: vi.fn(),
48
+ existsSync: vi.fn(() => false),
49
+ appendFileSync: vi.fn(),
50
+ writeFileSync: vi.fn(),
51
+ readdirSync: vi.fn(() => []),
52
+ promises: actual.promises,
53
+ };
54
+ });
55
+
56
+ vi.mock("../alive-store.ts", () => ({ writeAliveMarker: vi.fn() }));
57
+
58
+ // chatMode agent_end 早返回(continue),不读 session-pending;统一 count=0 防干扰。
59
+ vi.mock("../session-pending.ts", () => ({
60
+ readActivePendingFromSessionFile: vi.fn(() => ({ count: 0 })),
61
+ }));
62
+
63
+ vi.mock("../temp-prompt.ts", () => ({
64
+ writePromptToTempFile: vi.fn(async (agent: string) => {
65
+ const safeName = agent.replace(/[^\w.-]+/g, "_");
66
+ return { dir: `/tmp/fake-${safeName}`, filePath: `/tmp/fake-${safeName}/prompt-${safeName}.md` };
67
+ }),
68
+ cleanupTempPrompt: vi.fn(async () => {}),
69
+ }));
70
+
71
+ import { runSpawn } from "../session-runner.ts";
72
+ import type { SessionRunnerContext } from "../session-runner.ts";
73
+ import { _resetLifecycleState } from "../lifecycle-manager.ts";
74
+ import { createRecord } from "../execution-record.ts";
75
+ import type { ExecutionRecord } from "../types.ts";
76
+ import {
77
+ emitStdoutLine,
78
+ type FakeChild,
79
+ lastSpawnedChild as lastSpawnedChildOf,
80
+ makeCtx,
81
+ makeOpts,
82
+ makeRecord,
83
+ sessionHeader,
84
+ waitForSpawn as waitForSpawnOf,
85
+ } from "./helpers/spawn-mock.ts";
86
+
87
+ const mockSpawn = vi.mocked(spawn);
88
+
89
+ const lastSpawnedChild = (): FakeChild => lastSpawnedChildOf(mockSpawn);
90
+ const waitForSpawn = (timeoutMs = 1000): Promise<void> => waitForSpawnOf(mockSpawn, timeoutMs);
91
+
92
+ /** 构造 chatMode record(spawn-mock.makeRecord 默认非 chatMode,此处显式开启)。 */
93
+ function makeChatModeRecord(id = "sa-chat"): ExecutionRecord {
94
+ return createRecord(id, {
95
+ agent: "general-purpose",
96
+ model: "test-model",
97
+ mode: "sync",
98
+ task: "chat task",
99
+ slug: "chat",
100
+ startedAt: 1_000_000,
101
+ rootSessionId: "root-session",
102
+ parentRecordId: undefined,
103
+ depth: 0,
104
+ chatMode: true,
105
+ });
106
+ }
107
+
108
+ /** 用 race 探测 promise 是否在指定时间内 settle(resolve 或 reject)。 */
109
+ async function settlesWithin(p: Promise<unknown>, ms: number): Promise<boolean> {
110
+ return Promise.race([
111
+ p.then(
112
+ () => true,
113
+ () => true,
114
+ ),
115
+ new Promise<boolean>((r) => setTimeout(() => r(false), ms)),
116
+ ]);
117
+ }
118
+
119
+ describe("[V2 决策 2] chatMode 首轮闭环:runSpawn resolve 时机(改动 1)", () => {
120
+ beforeEach(() => {
121
+ vi.clearAllMocks();
122
+ _resetLifecycleState();
123
+ });
124
+
125
+ afterEach(() => {
126
+ vi.restoreAllMocks();
127
+ vi.useRealTimers();
128
+ _resetLifecycleState();
129
+ });
130
+
131
+ it("chatMode agent_settled → runSpawn 提前 resolve(不等 close),进程仍保活", async () => {
132
+ const record = makeChatModeRecord("sa-first-resolve");
133
+ const promise = runSpawn(record, "Task: first-round", makeOpts(), makeCtx());
134
+
135
+ await waitForSpawn();
136
+ const child = lastSpawnedChild();
137
+
138
+ emitStdoutLine(child, sessionHeader());
139
+ emitStdoutLine(child, { type: "agent_end", willRetry: false });
140
+ // 让 stdout flush 处理 agent_end(chatMode 分支 continue,不 kill)
141
+ await new Promise((r) => setTimeout(r, 10));
142
+ emitStdoutLine(child, { type: "agent_settled" });
143
+
144
+ // [改动 1 核心] agent_settled 后 runSpawn 立即 resolve(resolveRun(0)),不等 close。
145
+ // 若改动 1 未生效,promise 会 hang 等 close,settlesWithin 返回 false。
146
+ expect(await settlesWithin(promise, 500)).toBe(true);
147
+ const result = await promise;
148
+ expect(result.success).toBe(true);
149
+ expect(result.sessionId).toBe("sess-abc");
150
+
151
+ // 进程仍保活:agent_settled 只 resolve promise,未 kill、未 emit close。
152
+ // idle timer(5min 真实 setTimeout)armed 但未到期。
153
+ expect(child.killed).toBe(false);
154
+
155
+ // 收尾:手动 close(模拟后续 idle timer 超时 kill / 外部终止)。
156
+ // close handler 的 cleanup 仍执行(resolve(code) 是 no-op,Promise 已 resolve)。
157
+ child.stdout.end();
158
+ child.stderr.end();
159
+ child.emit("close", 0);
160
+ });
161
+
162
+ it("非 chatMode runSpawn 仍在 close resolve(agent_settled 不触发 resolve)", async () => {
163
+ // makeRecord 默认非 chatMode
164
+ const record = makeRecord("sa-nonchat-resolve");
165
+ const promise = runSpawn(record, "Task: nonchat", makeOpts(), makeCtx());
166
+
167
+ await waitForSpawn();
168
+ const child = lastSpawnedChild();
169
+
170
+ emitStdoutLine(child, sessionHeader());
171
+ emitStdoutLine(child, { type: "agent_end", willRetry: false });
172
+ // 非 chatMode agent_end → count=0 → SIGTERM kill
173
+ await new Promise((r) => setTimeout(r, 10));
174
+ expect(child.killed).toBe(true);
175
+
176
+ // agent_settled 在非 chatMode 下被忽略(handler 直接 return,不调 resolveRun)
177
+ emitStdoutLine(child, { type: "agent_settled" });
178
+ await new Promise((r) => setTimeout(r, 20));
179
+
180
+ // [回归防护] agent_settled 不 resolve 非 chatMode runSpawn——promise 仍 pending(等 close)。
181
+ expect(await settlesWithin(promise, 200)).toBe(false);
182
+
183
+ // emit close 才 resolve(现有行为,改动 1 不影响非 chatMode)
184
+ child.stdout.end();
185
+ child.stderr.end();
186
+ child.emit("close", 143);
187
+ const result = await promise;
188
+ expect(result.success).toBe(true); // 信号终止(143 ≥ threshold)视为正常完成
189
+ });
190
+ });