@zhushanwen/subagent-core 0.2.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.
- package/README.md +114 -0
- package/dist/chunk-3VOERJPJ.js +22 -0
- package/dist/chunk-A4IVZWVX.js +31 -0
- package/dist/execution/engine/engines/zcode/constants.cjs +54 -0
- package/dist/execution/engine/engines/zcode/constants.d.cts +31 -0
- package/dist/execution/engine/engines/zcode/constants.d.ts +31 -0
- package/dist/execution/engine/engines/zcode/constants.js +22 -0
- package/dist/execution/engine/engines/zcode/reader.cjs +276 -0
- package/dist/execution/engine/engines/zcode/reader.d.cts +20 -0
- package/dist/execution/engine/engines/zcode/reader.d.ts +20 -0
- package/dist/execution/engine/engines/zcode/reader.js +239 -0
- package/dist/execution/engine/paths.cjs +55 -0
- package/dist/execution/engine/paths.d.cts +8 -0
- package/dist/execution/engine/paths.d.ts +8 -0
- package/dist/execution/engine/paths.js +26 -0
- package/dist/execution/relay-env.cjs +62 -0
- package/dist/execution/relay-env.d.cts +33 -0
- package/dist/execution/relay-env.d.ts +33 -0
- package/dist/execution/relay-env.js +20 -0
- package/dist/index.cjs +1852 -0
- package/dist/index.d.cts +1184 -0
- package/dist/index.d.ts +1184 -0
- package/dist/index.js +1810 -0
- package/dist/types-BxyAidGf.d.cts +875 -0
- package/dist/types-BxyAidGf.d.ts +875 -0
- package/package.json +99 -0
- package/src/__tests__/agent-opts-resolver-schema-prompt.test.ts +142 -0
- package/src/__tests__/append-system-prompt-assembly.test.ts +242 -0
- package/src/__tests__/builtin-workflows-structure.test.ts +180 -0
- package/src/__tests__/derive-closed-display-parity.test.ts +93 -0
- package/src/__tests__/fr4-get-state-handshake.test.ts +125 -0
- package/src/__tests__/fr8-orphan-recovery.test.ts +259 -0
- package/src/__tests__/m2-append-content-probe.test.ts +73 -0
- package/src/__tests__/manifest-store.test.ts +368 -0
- package/src/__tests__/record-store-cache.test.ts +335 -0
- package/src/__tests__/record-store-index.test.ts +560 -0
- package/src/__tests__/record-store-last-line.test.ts +129 -0
- package/src/__tests__/review-fix-loop-script.test.ts +288 -0
- package/src/__tests__/review-fix-loop-utils.test.ts +2377 -0
- package/src/__tests__/robustness-low-batch1.test.ts +286 -0
- package/src/__tests__/robustness-medium-batch1.test.ts +69 -0
- package/src/__tests__/robustness-medium-batch2.test.ts +65 -0
- package/src/__tests__/robustness-medium-batch3.test.ts +30 -0
- package/src/__tests__/robustness-medium-batch4.test.ts +358 -0
- package/src/__tests__/session-runner.test.ts +69 -0
- package/src/__tests__/smoke.test.ts +9 -0
- package/src/__tests__/stream-sink.test.ts +203 -0
- package/src/core/__tests__/host-services.test.ts +144 -0
- package/src/core/__tests__/logger.test.ts +78 -0
- package/src/core/__tests__/notify-ports.test.ts +104 -0
- package/src/core/host-services.ts +97 -0
- package/src/core/logger.ts +47 -0
- package/src/core/notify-ports.ts +126 -0
- package/src/execution/__tests__/__fixtures__/notifier-golden-snapshots.json +53 -0
- package/src/execution/__tests__/__fixtures__/truncline.snapshot.json +1 -0
- package/src/execution/__tests__/agent-result-mapper.test.ts +150 -0
- package/src/execution/__tests__/alive-store.test.ts +147 -0
- package/src/execution/__tests__/ask-user-transit-e2e.test.ts +490 -0
- package/src/execution/__tests__/channel-registry-handshake.test.ts +242 -0
- package/src/execution/__tests__/chat-engine-routing.test.ts +610 -0
- package/src/execution/__tests__/chatmode-first-round-closure-spawn.test.ts +190 -0
- package/src/execution/__tests__/concurrency-pool.test.ts +250 -0
- package/src/execution/__tests__/config.test.ts +302 -0
- package/src/execution/__tests__/delivery-methods.test.ts +424 -0
- package/src/execution/__tests__/dialog-queue.test.ts +299 -0
- package/src/execution/__tests__/epipe-fallback.test.ts +241 -0
- package/src/execution/__tests__/execute-and-await-worktree.test.ts +243 -0
- package/src/execution/__tests__/execute-nesting.test.ts +323 -0
- package/src/execution/__tests__/execute-options-mapper.test.ts +200 -0
- package/src/execution/__tests__/execution-record.test.ts +1394 -0
- package/src/execution/__tests__/explicit-agent-ref-guard.test.ts +171 -0
- package/src/execution/__tests__/finalize-record.test.ts +367 -0
- package/src/execution/__tests__/finalized-marker.test.ts +104 -0
- package/src/execution/__tests__/format-schema-instruction.test.ts +169 -0
- package/src/execution/__tests__/gc-timer.test.ts +184 -0
- package/src/execution/__tests__/get-record-for-action-restart.test.ts +254 -0
- package/src/execution/__tests__/helpers/mock-extension-api.ts +30 -0
- package/src/execution/__tests__/helpers/spawn-mock.ts +242 -0
- package/src/execution/__tests__/host-mode.test.ts +87 -0
- package/src/execution/__tests__/lifecycle-manager-lock.test.ts +211 -0
- package/src/execution/__tests__/lifecycle-manager.test.ts +383 -0
- package/src/execution/__tests__/lifecycle-predicates.test.ts +116 -0
- package/src/execution/__tests__/manifest-parentid.test.ts +117 -0
- package/src/execution/__tests__/model-resolver.test.ts +465 -0
- package/src/execution/__tests__/nested-visibility-env-propagation.test.ts +287 -0
- package/src/execution/__tests__/nested-visibility.test.ts +325 -0
- package/src/execution/__tests__/output-collector.test.ts +358 -0
- package/src/execution/__tests__/path-encoding.test.ts +104 -0
- package/src/execution/__tests__/pi-invocation.test.ts +134 -0
- package/src/execution/__tests__/record-store.test.ts +1057 -0
- package/src/execution/__tests__/records-cwd-isolation.test.ts +91 -0
- package/src/execution/__tests__/recursive-visibility-baseline.test.ts +339 -0
- package/src/execution/__tests__/recursive-visibility-env.test.ts +273 -0
- package/src/execution/__tests__/relay-env.test.ts +42 -0
- package/src/execution/__tests__/resource-policy.test.ts +109 -0
- package/src/execution/__tests__/rpc-mode.test.ts +89 -0
- package/src/execution/__tests__/run-and-finalize-chatmode.test.ts +267 -0
- package/src/execution/__tests__/run-spawn-chatmode-settled.test.ts +253 -0
- package/src/execution/__tests__/run-spawn-edges.test.ts +687 -0
- package/src/execution/__tests__/run-spawn-integration.test.ts +933 -0
- package/src/execution/__tests__/run-spawn-resume.test.ts +322 -0
- package/src/execution/__tests__/run-spawn-rpc-mode.test.ts +196 -0
- package/src/execution/__tests__/run-spawn-stdout-callback-throw.test.ts +199 -0
- package/src/execution/__tests__/session-context-resolver.test.ts +167 -0
- package/src/execution/__tests__/session-file-gc.test.ts +293 -0
- package/src/execution/__tests__/session-reconstructor.test.ts +379 -0
- package/src/execution/__tests__/session-runner-epipe.test.ts +178 -0
- package/src/execution/__tests__/session-runner-schema-env.test.ts +350 -0
- package/src/execution/__tests__/spawn-args.test.ts +445 -0
- package/src/execution/__tests__/spawn-event-adapter-rpc.test.ts +189 -0
- package/src/execution/__tests__/spawn-event-adapter.test.ts +167 -0
- package/src/execution/__tests__/spawn-worktree-guidance.test.ts +207 -0
- package/src/execution/__tests__/spawned-children.test.ts +92 -0
- package/src/execution/__tests__/start-sync-model-guard.test.ts +150 -0
- package/src/execution/__tests__/stdin-writer.test.ts +462 -0
- package/src/execution/__tests__/stream-sink-retirement.test.ts +261 -0
- package/src/execution/__tests__/subagent-service-abort.test.ts +60 -0
- package/src/execution/__tests__/subagent-service-message-close.test.ts +629 -0
- package/src/execution/__tests__/subagent-service-parent-guard.test.ts +180 -0
- package/src/execution/__tests__/subagent-service.test.ts +788 -0
- package/src/execution/__tests__/subprocess-agent-runner-routing.test.ts +310 -0
- package/src/execution/__tests__/subprocess-agent-runner.test.ts +612 -0
- package/src/execution/__tests__/temp-prompt.test.ts +53 -0
- package/src/execution/__tests__/timeout-integration.test.ts +615 -0
- package/src/execution/__tests__/tombstone-store.test.ts +73 -0
- package/src/execution/__tests__/turn-limiter-semantics.test.ts +194 -0
- package/src/execution/__tests__/turn-limiter.test.ts +65 -0
- package/src/execution/__tests__/ui-channels.test.ts +187 -0
- package/src/execution/__tests__/ui-interaction-model.test.ts +67 -0
- package/src/execution/__tests__/ui-request-handler-factory.test.ts +369 -0
- package/src/execution/__tests__/ui-request-handler.test.ts +204 -0
- package/src/execution/__tests__/ui-request-observability.test.ts +113 -0
- package/src/execution/__tests__/ui-request-queue.test.ts +133 -0
- package/src/execution/__tests__/worktree-manager.test.ts +633 -0
- package/src/execution/__tests__/worktree-pid-registration.integration.test.ts +229 -0
- package/src/execution/__tests__/worktree-reconcile.integration.test.ts +181 -0
- package/src/execution/__tests__/worktree-registry.test.ts +199 -0
- package/src/execution/agent-registry.ts +199 -0
- package/src/execution/agent-result-mapper.ts +90 -0
- package/src/execution/alive-store.ts +119 -0
- package/src/execution/argv-mirror.ts +113 -0
- package/src/execution/best-effort.ts +38 -0
- package/src/execution/channel-registry-access.ts +144 -0
- package/src/execution/concurrency-pool.ts +116 -0
- package/src/execution/config.ts +165 -0
- package/src/execution/dialog-queue.ts +330 -0
- package/src/execution/engine/__tests__/common/data-dir.test.ts +78 -0
- package/src/execution/engine/__tests__/common/errors.test.ts +132 -0
- package/src/execution/engine/__tests__/common/event-journal.test.ts +177 -0
- package/src/execution/engine/__tests__/common/kill-chain.test.ts +192 -0
- package/src/execution/engine/__tests__/common/nesting-guard.test.ts +81 -0
- package/src/execution/engine/__tests__/common/persona-router.test.ts +123 -0
- package/src/execution/engine/__tests__/common/pool-manager.test.ts +154 -0
- package/src/execution/engine/__tests__/common/schema-emulation.test.ts +128 -0
- package/src/execution/engine/__tests__/conformance/__fixtures__/pi-golden-events.json +28 -0
- package/src/execution/engine/__tests__/conformance/agent-event-invariants.ts +141 -0
- package/src/execution/engine/__tests__/conformance/contract.abort.test.ts +109 -0
- package/src/execution/engine/__tests__/conformance/contract.agent-events.test.ts +101 -0
- package/src/execution/engine/__tests__/conformance/contract.probe.test.ts +77 -0
- package/src/execution/engine/__tests__/conformance/contract.read-degradation.test.ts +104 -0
- package/src/execution/engine/__tests__/conformance/engine-conformance.live.test.ts +201 -0
- package/src/execution/engine/__tests__/conformance/golden-replay.pi.test.ts +76 -0
- package/src/execution/engine/__tests__/conformance/golden-replay.zcode.test.ts +79 -0
- package/src/execution/engine/__tests__/engine-discovery.test.ts +87 -0
- package/src/execution/engine/__tests__/model-prompt.test.ts +197 -0
- package/src/execution/engine/__tests__/paths.test.ts +39 -0
- package/src/execution/engine/__tests__/registry.test.ts +120 -0
- package/src/execution/engine/__tests__/routing.test.ts +231 -0
- package/src/execution/engine/common/data-dir.ts +66 -0
- package/src/execution/engine/common/errors.ts +183 -0
- package/src/execution/engine/common/event-journal.ts +254 -0
- package/src/execution/engine/common/journal-replay.ts +62 -0
- package/src/execution/engine/common/kill-chain.ts +221 -0
- package/src/execution/engine/common/nesting-guard.ts +50 -0
- package/src/execution/engine/common/persona-router.ts +108 -0
- package/src/execution/engine/common/pool-manager.ts +226 -0
- package/src/execution/engine/common/schema-emulation.ts +189 -0
- package/src/execution/engine/common/session-view-projection.ts +51 -0
- package/src/execution/engine/engine-discovery.ts +65 -0
- package/src/execution/engine/engines/pi/__tests__/pi-engine.test.ts +469 -0
- package/src/execution/engine/engines/pi/__tests__/reader.test.ts +155 -0
- package/src/execution/engine/engines/pi/__tests__/task-spec-mapper.test.ts +164 -0
- package/src/execution/engine/engines/pi/pi-engine.ts +415 -0
- package/src/execution/engine/engines/pi/reader.ts +48 -0
- package/src/execution/engine/engines/pi/registration.ts +35 -0
- package/src/execution/engine/engines/pi/task-spec-mapper.ts +100 -0
- package/src/execution/engine/engines/zcode/__tests__/__fixtures__/zcode-golden-spawn.json +39 -0
- package/src/execution/engine/engines/zcode/__tests__/launcher.test.ts +150 -0
- package/src/execution/engine/engines/zcode/__tests__/parser.test.ts +246 -0
- package/src/execution/engine/engines/zcode/__tests__/preparer.test.ts +228 -0
- package/src/execution/engine/engines/zcode/__tests__/reader.test.ts +210 -0
- package/src/execution/engine/engines/zcode/__tests__/registration.test.ts +71 -0
- package/src/execution/engine/engines/zcode/__tests__/zcode-engine.live.test.ts +127 -0
- package/src/execution/engine/engines/zcode/__tests__/zcode-engine.test.ts +580 -0
- package/src/execution/engine/engines/zcode/constants.ts +43 -0
- package/src/execution/engine/engines/zcode/golden-sample.ts +39 -0
- package/src/execution/engine/engines/zcode/launcher.ts +161 -0
- package/src/execution/engine/engines/zcode/parser.ts +436 -0
- package/src/execution/engine/engines/zcode/preparer.ts +363 -0
- package/src/execution/engine/engines/zcode/reader.ts +381 -0
- package/src/execution/engine/engines/zcode/registration.ts +37 -0
- package/src/execution/engine/engines/zcode/zcode-engine.ts +658 -0
- package/src/execution/engine/host-task-spec.ts +47 -0
- package/src/execution/engine/model-prompt.ts +158 -0
- package/src/execution/engine/paths.ts +42 -0
- package/src/execution/engine/port.ts +153 -0
- package/src/execution/engine/registry.ts +133 -0
- package/src/execution/engine/routing.ts +218 -0
- package/src/execution/engine/types.ts +309 -0
- package/src/execution/execute-options-mapper.ts +115 -0
- package/src/execution/execution-record.ts +984 -0
- package/src/execution/finalize-record.ts +254 -0
- package/src/execution/finalized-marker.ts +70 -0
- package/src/execution/get-state-handshake.ts +104 -0
- package/src/execution/host-mode.ts +56 -0
- package/src/execution/idle-gc.ts +47 -0
- package/src/execution/lifecycle-manager.ts +513 -0
- package/src/execution/lifecycle-predicates.ts +65 -0
- package/src/execution/manifest-store.ts +256 -0
- package/src/execution/model-config-service.ts +250 -0
- package/src/execution/model-resolver.ts +248 -0
- package/src/execution/notifier.ts +372 -0
- package/src/execution/notify-ledger.ts +580 -0
- package/src/execution/output-collector.ts +228 -0
- package/src/execution/path-encoding.ts +60 -0
- package/src/execution/pi-invocation.ts +120 -0
- package/src/execution/record-entry.ts +132 -0
- package/src/execution/record-store.ts +1265 -0
- package/src/execution/relay-env.ts +37 -0
- package/src/execution/session-context-resolver.ts +64 -0
- package/src/execution/session-file-gc.ts +120 -0
- package/src/execution/session-pending.ts +170 -0
- package/src/execution/session-reconstructor.ts +678 -0
- package/src/execution/session-runner.ts +1789 -0
- package/src/execution/sessions-index.ts +304 -0
- package/src/execution/spawn-event-adapter.ts +363 -0
- package/src/execution/stdin-writer.ts +198 -0
- package/src/execution/stream-sink.ts +126 -0
- package/src/execution/subagent-service.ts +2268 -0
- package/src/execution/subprocess-agent-runner.ts +317 -0
- package/src/execution/temp-prompt.ts +62 -0
- package/src/execution/tombstone-store.ts +72 -0
- package/src/execution/turn-limiter.ts +102 -0
- package/src/execution/types.ts +1023 -0
- package/src/execution/ui-channels.ts +216 -0
- package/src/execution/ui-interaction-model.ts +48 -0
- package/src/execution/ui-request-handler-factory.ts +243 -0
- package/src/execution/ui-request-observability.ts +81 -0
- package/src/execution/ui-request-queue.ts +184 -0
- package/src/execution/worktree-manager.ts +688 -0
- package/src/execution/worktree-registry.ts +279 -0
- package/src/index.ts +87 -0
- package/src/orchestration/__tests__/__fixtures__/worker-template.snapshot.txt +349 -0
- package/src/orchestration/__tests__/agent-call-catch-fallback.test.ts +202 -0
- package/src/orchestration/__tests__/agent-call-stream.test.ts +152 -0
- package/src/orchestration/__tests__/args-validator.test.ts +143 -0
- package/src/orchestration/__tests__/config-loader.test.ts +503 -0
- package/src/orchestration/__tests__/error-recovery-handlers.test.ts +809 -0
- package/src/orchestration/__tests__/error-recovery-postmessage-defense.test.ts +404 -0
- package/src/orchestration/__tests__/error-recovery-serialize-failed-result.test.ts +56 -0
- package/src/orchestration/__tests__/error-recovery-workflow-call.test.ts +166 -0
- package/src/orchestration/__tests__/execute-agent-call.test.ts +451 -0
- package/src/orchestration/__tests__/launcher-nested-workflow.test.ts +600 -0
- package/src/orchestration/__tests__/lifecycle-runid-injection.test.ts +96 -0
- package/src/orchestration/__tests__/lifecycle.test.ts +659 -0
- package/src/orchestration/__tests__/non-cloneable-return-e2e.test.ts +95 -0
- package/src/orchestration/__tests__/review-fix-loop-scriptpath-failfast.test.ts +183 -0
- package/src/orchestration/__tests__/script-lint.test.ts +831 -0
- package/src/orchestration/__tests__/skill-discovery.test.ts +210 -0
- package/src/orchestration/__tests__/test-mocks.ts +197 -0
- package/src/orchestration/__tests__/worker-exit-without-result.test.ts +368 -0
- package/src/orchestration/__tests__/worker-host.test.ts +120 -0
- package/src/orchestration/__tests__/worker-returnmeta-passthrough.test.ts +164 -0
- package/src/orchestration/__tests__/worker-script-builder-runtime.test.ts +563 -0
- package/src/orchestration/__tests__/worker-script-builder.test.ts +309 -0
- package/src/orchestration/__tests__/worker-script-template-snapshot.test.ts +129 -0
- package/src/orchestration/__tests__/workflow-nesting-e2e.test.ts +317 -0
- package/src/orchestration/__tests__/workflow-script-lint-memo.test.ts +110 -0
- package/src/orchestration/agent-opts-resolver.ts +174 -0
- package/src/orchestration/args-validator.ts +127 -0
- package/src/orchestration/config-loader.ts +316 -0
- package/src/orchestration/error-recovery.ts +1018 -0
- package/src/orchestration/execute-agent-call.ts +256 -0
- package/src/orchestration/launcher.ts +455 -0
- package/src/orchestration/lifecycle.ts +399 -0
- package/src/orchestration/models/__tests__/budget.test.ts +307 -0
- package/src/orchestration/models/agent-call.ts +83 -0
- package/src/orchestration/models/budget.ts +118 -0
- package/src/orchestration/models/ports.ts +164 -0
- package/src/orchestration/models/run-runtime.ts +104 -0
- package/src/orchestration/models/run-spec.ts +83 -0
- package/src/orchestration/models/run-state.ts +44 -0
- package/src/orchestration/models/trace.ts +183 -0
- package/src/orchestration/models/types.ts +301 -0
- package/src/orchestration/models/workflow-run.ts +254 -0
- package/src/orchestration/models/workflow-script-registry.ts +35 -0
- package/src/orchestration/models/workflow-script.ts +117 -0
- package/src/orchestration/script-lint.ts +766 -0
- package/src/orchestration/skill-discovery.ts +135 -0
- package/src/orchestration/worker-handle.ts +115 -0
- package/src/orchestration/worker-host.ts +98 -0
- package/src/orchestration/worker-script-builder.ts +421 -0
- package/src/orchestration/workflow-files.ts +85 -0
- package/src/orchestration/workflow-script-registry-impl.ts +133 -0
- package/src/shared/__tests__/agent-ref.test.ts +34 -0
- package/src/shared/__tests__/meta-parser.test.ts +304 -0
- package/src/shared/__tests__/model-ref.test.ts +306 -0
- package/src/shared/__tests__/resource-discovery-manifest-cache.test.ts +288 -0
- package/src/shared/__tests__/resource-discovery.test.ts +749 -0
- package/src/shared/__tests__/resource-meta.test.ts +51 -0
- package/src/shared/__tests__/schema-jsonify.test.ts +81 -0
- package/src/shared/__tests__/timer-delay.test.ts +61 -0
- package/src/shared/agent-event.ts +13 -0
- package/src/shared/agent-ref.ts +57 -0
- package/src/shared/meta-parser.ts +261 -0
- package/src/shared/model-ref.ts +286 -0
- package/src/shared/resource-discovery.ts +835 -0
- package/src/shared/resource-meta.ts +65 -0
- package/src/shared/schema-env.ts +44 -0
- package/src/shared/schema-jsonify.ts +58 -0
- package/src/shared/timer-delay.ts +54 -0
- package/src/shared/xml-injection.ts +35 -0
- package/workflows/README.md +81 -0
- package/workflows/_shared/agent-refs.cjs +40 -0
- package/workflows/chain.js +137 -0
- package/workflows/map-reduce.js +180 -0
- package/workflows/parallel.js +154 -0
- package/workflows/review-fix-loop-utils.cjs +1542 -0
- package/workflows/review-fix-loop.js +1605 -0
- package/workflows/scatter-gather.js +184 -0
|
@@ -0,0 +1,600 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* executeNestedWorkflow — workflow() 嵌套调用实现测试。
|
|
3
|
+
*
|
|
4
|
+
* 覆盖 6 个场景:循环检测、registry 未找到、lint 失败、成功执行、budget 同步、
|
|
5
|
+
* 子 workflow 失败。
|
|
6
|
+
*
|
|
7
|
+
* 通过 vi.mock("../lifecycle.ts") 控制 runWorkflow:返回固定 runId + 把预构造的
|
|
8
|
+
* child run 注入 deps.runs,使 pollRunToResult 首轮即命中 done 返回(无需真实 worker)。
|
|
9
|
+
*/
|
|
10
|
+
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
|
11
|
+
|
|
12
|
+
import { Budget } from "../models/budget.ts";
|
|
13
|
+
import type { RunSpec } from "../models/run-spec.ts";
|
|
14
|
+
import type { LauncherDeps } from "../launcher.ts";
|
|
15
|
+
import type { LintResult } from "../script-lint.ts";
|
|
16
|
+
import type { WorkflowRun } from "../models/workflow-run.ts";
|
|
17
|
+
import type { WorkflowScript } from "../models/workflow-script.ts";
|
|
18
|
+
|
|
19
|
+
// ── module mock:lifecycle.runWorkflow 由各 test 配置 ──────────────────
|
|
20
|
+
|
|
21
|
+
vi.mock("../lifecycle.ts", () => ({
|
|
22
|
+
runWorkflow: vi.fn(),
|
|
23
|
+
abortRun: vi.fn(async () => {}),
|
|
24
|
+
scheduleTimeBudget: vi.fn(() => undefined),
|
|
25
|
+
}));
|
|
26
|
+
|
|
27
|
+
// import 在 vi.mock 之后(hoisting 保证拿到 mock 版本)。runWorkflow 从被 mock 的
|
|
28
|
+
// lifecycle 模块导入,与 launcher.ts 内部引用的是同一 mock 实例。
|
|
29
|
+
import { runWorkflow, abortRun } from "../lifecycle.ts";
|
|
30
|
+
import { ArgsValidationError } from "../args-validator.ts";
|
|
31
|
+
import { RUN_WATCHDOG_ENV, executeNestedWorkflow, runAndWait } from "../launcher.ts";
|
|
32
|
+
|
|
33
|
+
const MOCK_RUN_ID = "wf-test-child";
|
|
34
|
+
|
|
35
|
+
// ── helpers ──────────────────────────────────────────────────
|
|
36
|
+
|
|
37
|
+
/** 构造 mock WorkflowScript(validate / toExecutable 可控)。 */
|
|
38
|
+
function makeScript(opts: { valid?: boolean; lintErrorMsg?: string } = {}): WorkflowScript {
|
|
39
|
+
const valid = opts.valid ?? true;
|
|
40
|
+
return {
|
|
41
|
+
name: "child-wf",
|
|
42
|
+
path: "/fake/child-wf.js",
|
|
43
|
+
meta: { name: "child-wf", description: "child workflow", phases: [] },
|
|
44
|
+
toExecutable: () => "const meta = {}; execute() {}",
|
|
45
|
+
validate: (): LintResult => ({
|
|
46
|
+
valid,
|
|
47
|
+
findings: valid
|
|
48
|
+
? []
|
|
49
|
+
: [
|
|
50
|
+
{
|
|
51
|
+
severity: "error",
|
|
52
|
+
line: 3,
|
|
53
|
+
message: opts.lintErrorMsg ?? "lint boom",
|
|
54
|
+
suggestion: "fix it",
|
|
55
|
+
},
|
|
56
|
+
],
|
|
57
|
+
}),
|
|
58
|
+
} as unknown as WorkflowScript;
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
/**
|
|
62
|
+
* 构造 mock parent WorkflowRun。
|
|
63
|
+
*
|
|
64
|
+
* parentBudget 用真实 Budget 实例(remaining() + 可写 usedTokens/usedCost)。
|
|
65
|
+
* controller 提供 AbortSignal(默认未 abort)。
|
|
66
|
+
*/
|
|
67
|
+
function makeParentRun(opts: {
|
|
68
|
+
scriptName?: string;
|
|
69
|
+
parentWorkflowChain?: readonly string[];
|
|
70
|
+
budget?: Budget;
|
|
71
|
+
aborted?: boolean;
|
|
72
|
+
/** 预算语义对齐:父 run 级时间预算(deadline 传导测试用) */
|
|
73
|
+
budgetTimeMs?: number;
|
|
74
|
+
} = {}): WorkflowRun {
|
|
75
|
+
const controller = new AbortController();
|
|
76
|
+
if (opts.aborted) controller.abort();
|
|
77
|
+
const spec = {
|
|
78
|
+
scriptName: opts.scriptName ?? "parent-wf",
|
|
79
|
+
parentWorkflowChain: opts.parentWorkflowChain,
|
|
80
|
+
budgetTimeMs: opts.budgetTimeMs,
|
|
81
|
+
};
|
|
82
|
+
return {
|
|
83
|
+
spec,
|
|
84
|
+
state: { budget: opts.budget ?? new Budget({ maxTokens: 10000 }) },
|
|
85
|
+
runtime: { controller },
|
|
86
|
+
} as unknown as WorkflowRun;
|
|
87
|
+
}
|
|
88
|
+
|
|
89
|
+
/**
|
|
90
|
+
* 构造 mock child WorkflowRun(done 终态,pollRunToResult 首轮命中)。
|
|
91
|
+
*
|
|
92
|
+
* childBudget 用普通对象(仅需 usedTokens/usedCost 供 budget 同步读取)。
|
|
93
|
+
*/
|
|
94
|
+
function makeDoneChildRun(opts: {
|
|
95
|
+
reason?: "completed" | "failed" | "aborted";
|
|
96
|
+
scriptResult?: unknown;
|
|
97
|
+
error?: string;
|
|
98
|
+
usedTokens?: number;
|
|
99
|
+
usedCost?: number;
|
|
100
|
+
}): WorkflowRun {
|
|
101
|
+
const reason = opts.reason ?? "completed";
|
|
102
|
+
return {
|
|
103
|
+
runId: MOCK_RUN_ID,
|
|
104
|
+
spec: { scriptName: "child-wf" },
|
|
105
|
+
state: {
|
|
106
|
+
status: "done",
|
|
107
|
+
reason,
|
|
108
|
+
scriptResult: opts.scriptResult,
|
|
109
|
+
error: opts.error,
|
|
110
|
+
budget: { usedTokens: opts.usedTokens ?? 0, usedCost: opts.usedCost ?? 0 },
|
|
111
|
+
},
|
|
112
|
+
} as unknown as WorkflowRun;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
/**
|
|
116
|
+
* 构造 mock child WorkflowRun(running 态,pollRunToResult 持续轮询)。
|
|
117
|
+
* state 可变——deadline 测试中手动翻转为 done 模拟子 run 完成。
|
|
118
|
+
*/
|
|
119
|
+
function makeRunningChildRun(): WorkflowRun {
|
|
120
|
+
return {
|
|
121
|
+
runId: MOCK_RUN_ID,
|
|
122
|
+
spec: { scriptName: "child-wf" },
|
|
123
|
+
state: {
|
|
124
|
+
status: "running",
|
|
125
|
+
reason: undefined,
|
|
126
|
+
scriptResult: undefined,
|
|
127
|
+
error: undefined,
|
|
128
|
+
budget: { usedTokens: 0, usedCost: 0 },
|
|
129
|
+
},
|
|
130
|
+
} as unknown as WorkflowRun;
|
|
131
|
+
}
|
|
132
|
+
|
|
133
|
+
/** 构造 LauncherDeps mock:registry + runs(Map)+ 占位 port。 */
|
|
134
|
+
function makeDeps(opts: {
|
|
135
|
+
script?: WorkflowScript;
|
|
136
|
+
childRun?: WorkflowRun;
|
|
137
|
+
registry?: { getPath: ReturnType<typeof vi.fn> };
|
|
138
|
+
} = {}): LauncherDeps {
|
|
139
|
+
const runs = new Map<string, WorkflowRun>();
|
|
140
|
+
if (opts.childRun) runs.set(MOCK_RUN_ID, opts.childRun);
|
|
141
|
+
const registry = opts.registry ?? {
|
|
142
|
+
getPath: vi.fn(async () => opts.script),
|
|
143
|
+
};
|
|
144
|
+
return {
|
|
145
|
+
registry,
|
|
146
|
+
runs,
|
|
147
|
+
store: { save: vi.fn(async () => {}), loadAll: vi.fn(async () => []) },
|
|
148
|
+
workerHost: { start: vi.fn(() => ({ postMessage: vi.fn() })) },
|
|
149
|
+
runner: { run: vi.fn(async () => ({})) },
|
|
150
|
+
log: vi.fn(),
|
|
151
|
+
eventBus: { emit: vi.fn() },
|
|
152
|
+
} as unknown as LauncherDeps;
|
|
153
|
+
}
|
|
154
|
+
|
|
155
|
+
/** 配置 runWorkflow mock:把 childRun 注入 deps.runs 并返回 MOCK_RUN_ID。 */
|
|
156
|
+
function setupRunWorkflow(childRun: WorkflowRun): void {
|
|
157
|
+
vi.mocked(runWorkflow).mockImplementation(async (_spec, deps) => {
|
|
158
|
+
deps.runs.set(MOCK_RUN_ID, childRun);
|
|
159
|
+
return MOCK_RUN_ID;
|
|
160
|
+
});
|
|
161
|
+
}
|
|
162
|
+
|
|
163
|
+
beforeEach(() => {
|
|
164
|
+
vi.clearAllMocks();
|
|
165
|
+
// [F-4 假红源修复] env 隔离对称化:本文件的「不限时」用例(deadline 传导 /
|
|
166
|
+
// runAndWait timeoutMs 语义)依赖「XYZ_SUBAGENT_RUN_WATCHDOG_MS 未设」基线,
|
|
167
|
+
// 宿主 shell export 即假红(U7 describe 内部 stubEnv 不覆盖同文件其他用例)。
|
|
168
|
+
// 空串 = 未设(getEnvRunWatchdogMs 的 raw falsy 判定);SPAWN_WATCHDOG 同款
|
|
169
|
+
// 防御(launcher 不读,防未来接线后回归)。
|
|
170
|
+
vi.stubEnv(RUN_WATCHDOG_ENV, "");
|
|
171
|
+
vi.stubEnv("XYZ_SUBAGENT_SPAWN_WATCHDOG_MS", "");
|
|
172
|
+
});
|
|
173
|
+
|
|
174
|
+
afterEach(() => {
|
|
175
|
+
vi.unstubAllEnvs();
|
|
176
|
+
});
|
|
177
|
+
|
|
178
|
+
// ── tests ────────────────────────────────────────────────────
|
|
179
|
+
|
|
180
|
+
describe("executeNestedWorkflow", () => {
|
|
181
|
+
it("returns error result when workflow not found", async () => {
|
|
182
|
+
const parent = makeParentRun();
|
|
183
|
+
const signal = parent.runtime!.controller.signal;
|
|
184
|
+
const addSpy = vi.spyOn(signal, "addEventListener");
|
|
185
|
+
const removeSpy = vi.spyOn(signal, "removeEventListener");
|
|
186
|
+
const deps = makeDeps({ script: undefined });
|
|
187
|
+
|
|
188
|
+
const result = await executeNestedWorkflow("missing", {}, parent, deps);
|
|
189
|
+
|
|
190
|
+
expect(result.error).toBe("Workflow 'missing' not found");
|
|
191
|
+
expect(result.content).toBe("");
|
|
192
|
+
expect(runWorkflow).not.toHaveBeenCalled();
|
|
193
|
+
// E8:早返回也走 finally——Step 2 注册的 listener 被移除(原实现泄漏)
|
|
194
|
+
expect(addSpy).toHaveBeenCalledTimes(1);
|
|
195
|
+
expect(removeSpy).toHaveBeenCalledTimes(1);
|
|
196
|
+
});
|
|
197
|
+
|
|
198
|
+
it("returns error result on lint failure", async () => {
|
|
199
|
+
const parent = makeParentRun();
|
|
200
|
+
const deps = makeDeps({ script: makeScript({ valid: false, lintErrorMsg: "no entry point" }) });
|
|
201
|
+
|
|
202
|
+
const result = await executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
203
|
+
|
|
204
|
+
expect(result.error).toContain("has lint errors");
|
|
205
|
+
expect(result.error).toContain("no entry point");
|
|
206
|
+
expect(result.content).toBe("");
|
|
207
|
+
expect(runWorkflow).not.toHaveBeenCalled();
|
|
208
|
+
});
|
|
209
|
+
|
|
210
|
+
it("TC11: chokepoint 校验失败 → {error} + parentSignal listener 不泄漏(E8)", async () => {
|
|
211
|
+
// 模块级 mock lifecycle:runWorkflow reject ArgsValidationError(真实实例——
|
|
212
|
+
// launcher.ts 未被 mock,instanceof 检查命中同一 class)。
|
|
213
|
+
vi.mocked(runWorkflow).mockRejectedValueOnce(
|
|
214
|
+
new ArgsValidationError(
|
|
215
|
+
"child-wf",
|
|
216
|
+
"Invalid args for workflow 'child-wf': 1 error(s)\n- /target: is required\nRead the workflow script file (location from <available_workflows>) for the parameter schema and usage.",
|
|
217
|
+
),
|
|
218
|
+
);
|
|
219
|
+
const parent = makeParentRun();
|
|
220
|
+
const signal = parent.runtime!.controller.signal;
|
|
221
|
+
const addSpy = vi.spyOn(signal, "addEventListener");
|
|
222
|
+
const removeSpy = vi.spyOn(signal, "removeEventListener");
|
|
223
|
+
const deps = makeDeps({ script: makeScript() });
|
|
224
|
+
|
|
225
|
+
const result = await executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
226
|
+
|
|
227
|
+
expect(result.content).toBe("");
|
|
228
|
+
expect(result.error).toContain("Invalid args for workflow 'child-wf'");
|
|
229
|
+
// Step 2 注册 1 次 + finally 移除 1 次 → 平衡即无泄漏(E8 修复点)
|
|
230
|
+
expect(addSpy).toHaveBeenCalledTimes(1);
|
|
231
|
+
expect(removeSpy).toHaveBeenCalledTimes(1);
|
|
232
|
+
});
|
|
233
|
+
|
|
234
|
+
it("detects circular call chain", async () => {
|
|
235
|
+
// parent chain ["a"], parent scriptName "b", target name "a" → a→b→a 循环
|
|
236
|
+
const parent = makeParentRun({
|
|
237
|
+
scriptName: "b",
|
|
238
|
+
parentWorkflowChain: ["a"],
|
|
239
|
+
});
|
|
240
|
+
const deps = makeDeps({ script: makeScript() });
|
|
241
|
+
|
|
242
|
+
const result = await executeNestedWorkflow("a", {}, parent, deps);
|
|
243
|
+
|
|
244
|
+
expect(result.error).toContain("Circular workflow call detected");
|
|
245
|
+
expect(result.error).toContain("a → b → a");
|
|
246
|
+
expect(result.content).toBe("");
|
|
247
|
+
expect(runWorkflow).not.toHaveBeenCalled();
|
|
248
|
+
});
|
|
249
|
+
|
|
250
|
+
it("executes child workflow and returns result on success", async () => {
|
|
251
|
+
const parent = makeParentRun();
|
|
252
|
+
const childRun = makeDoneChildRun({
|
|
253
|
+
reason: "completed",
|
|
254
|
+
scriptResult: { summary: "done" },
|
|
255
|
+
});
|
|
256
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
257
|
+
setupRunWorkflow(childRun);
|
|
258
|
+
|
|
259
|
+
const result = await executeNestedWorkflow("child-wf", { k: 1 }, parent, deps);
|
|
260
|
+
|
|
261
|
+
// content = JSON.stringify(scriptResult);parsedOutput 原样回传对象
|
|
262
|
+
expect(result.error).toBeUndefined();
|
|
263
|
+
expect(result.content).toBe(JSON.stringify({ summary: "done" }));
|
|
264
|
+
expect(result.parsedOutput).toEqual({ summary: "done" });
|
|
265
|
+
expect(runWorkflow).toHaveBeenCalledTimes(1);
|
|
266
|
+
});
|
|
267
|
+
|
|
268
|
+
it("shares parent budget reference with child run (no sync-back)", async () => {
|
|
269
|
+
const parentBudget = new Budget({ maxTokens: 10000 });
|
|
270
|
+
parentBudget.usedTokens = 100;
|
|
271
|
+
const parent = makeParentRun({ budget: parentBudget });
|
|
272
|
+
const childRun = makeDoneChildRun({
|
|
273
|
+
reason: "completed",
|
|
274
|
+
scriptResult: "ok",
|
|
275
|
+
});
|
|
276
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
277
|
+
|
|
278
|
+
// 捕获传给 runWorkflow 的 spec——验证 budgetRef 共享父 Budget 引用
|
|
279
|
+
let capturedSpec: RunSpec | undefined;
|
|
280
|
+
vi.mocked(runWorkflow).mockImplementation(async (spec, d) => {
|
|
281
|
+
capturedSpec = spec;
|
|
282
|
+
d.runs.set(MOCK_RUN_ID, childRun);
|
|
283
|
+
return MOCK_RUN_ID;
|
|
284
|
+
});
|
|
285
|
+
|
|
286
|
+
await executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
287
|
+
|
|
288
|
+
// 子 run 直接复用父 Budget 引用(budgetRef),而非独立 Budget + sync-back
|
|
289
|
+
expect(capturedSpec?.budgetRef).toBe(parentBudget);
|
|
290
|
+
expect(capturedSpec?.budgetTokens).toBeUndefined();
|
|
291
|
+
// 无 sync-back:parent budget 不被 launcher 直接修改(mock 未真实 consume)
|
|
292
|
+
expect(parent.state.budget.usedTokens).toBe(100);
|
|
293
|
+
});
|
|
294
|
+
|
|
295
|
+
it("returns error result when child workflow fails", async () => {
|
|
296
|
+
const parent = makeParentRun();
|
|
297
|
+
const childRun = makeDoneChildRun({
|
|
298
|
+
reason: "failed",
|
|
299
|
+
error: "agent exploded",
|
|
300
|
+
});
|
|
301
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
302
|
+
setupRunWorkflow(childRun);
|
|
303
|
+
|
|
304
|
+
const result = await executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
305
|
+
|
|
306
|
+
expect(result.content).toBe("");
|
|
307
|
+
expect(result.error).toBe("agent exploded");
|
|
308
|
+
});
|
|
309
|
+
|
|
310
|
+
// ── deadline 传导(预算语义对齐 2026-08)──
|
|
311
|
+
// 语义:父 spec.budgetTimeMs 显式设定 → 原样作为子 run 轮询 deadline(不 min(DEFAULT)
|
|
312
|
+
// 封顶);父未设 → 无 deadline。旧实现 min(父, 10min) 把父 time:2h 截断到 10min。
|
|
313
|
+
describe("deadline 传导(预算语义对齐)", () => {
|
|
314
|
+
it("父 budgetTimeMs=2h → 子 deadline 传导 2h(不 min(10min) 封顶)", async () => {
|
|
315
|
+
vi.useFakeTimers();
|
|
316
|
+
try {
|
|
317
|
+
const parent = makeParentRun({ budgetTimeMs: 7_200_000 });
|
|
318
|
+
const childRun = makeRunningChildRun();
|
|
319
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
320
|
+
setupRunWorkflow(childRun);
|
|
321
|
+
|
|
322
|
+
const promise = executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
323
|
+
|
|
324
|
+
// 越过旧 10min 默认:不应 abort(若回归 min 封顶,这里会触发 time_limited abort)
|
|
325
|
+
await vi.advanceTimersByTimeAsync(600_000 + 500);
|
|
326
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
327
|
+
|
|
328
|
+
// 2h deadline 到期:abortRun 收到完整 7200000ms 超时 + time_limited
|
|
329
|
+
await vi.advanceTimersByTimeAsync(7_200_000 + 1_000 - 600_500);
|
|
330
|
+
const result = await promise;
|
|
331
|
+
expect(vi.mocked(abortRun)).toHaveBeenCalledTimes(1);
|
|
332
|
+
expect(vi.mocked(abortRun)).toHaveBeenCalledWith(
|
|
333
|
+
MOCK_RUN_ID,
|
|
334
|
+
deps,
|
|
335
|
+
"Workflow timed out after 7200000ms",
|
|
336
|
+
"time_limited",
|
|
337
|
+
);
|
|
338
|
+
void result;
|
|
339
|
+
} finally {
|
|
340
|
+
vi.useRealTimers();
|
|
341
|
+
}
|
|
342
|
+
});
|
|
343
|
+
|
|
344
|
+
it("父未传 budgetTimeMs → 无 deadline(远超旧 10min 默认不 abort,子完成正常返回)", async () => {
|
|
345
|
+
vi.useFakeTimers();
|
|
346
|
+
try {
|
|
347
|
+
const parent = makeParentRun();
|
|
348
|
+
const childRun = makeRunningChildRun();
|
|
349
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
350
|
+
setupRunWorkflow(childRun);
|
|
351
|
+
|
|
352
|
+
const promise = executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
353
|
+
|
|
354
|
+
// 旧实现 DEFAULT_RUNANDWAIT_TIMEOUT_MS=10min 会在此 abort;不限时则不 abort
|
|
355
|
+
await vi.advanceTimersByTimeAsync(10 * 60 * 1000 + 500);
|
|
356
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
357
|
+
|
|
358
|
+
// 子 run 完成 → 正常返回 completed(deadline 仍是 undefined)
|
|
359
|
+
childRun.state.status = "done";
|
|
360
|
+
childRun.state.reason = "completed";
|
|
361
|
+
childRun.state.scriptResult = "ok";
|
|
362
|
+
await vi.advanceTimersByTimeAsync(500);
|
|
363
|
+
const result = await promise;
|
|
364
|
+
// 成功分支返回 { content, parsedOutput? }(无 reason 字段)——string scriptResult 走 content
|
|
365
|
+
expect(result.content).toBe("ok");
|
|
366
|
+
expect(result.error).toBeUndefined();
|
|
367
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
368
|
+
} finally {
|
|
369
|
+
vi.useRealTimers();
|
|
370
|
+
}
|
|
371
|
+
});
|
|
372
|
+
|
|
373
|
+
// [U2] 预算语义统一:budgetTimeMs <=0(含 0/负)与 undefined 同义 = 不限——
|
|
374
|
+
// 旧实现 0 传导给子 run 后 deadline=now 立即超时("timed out after 0ms"),与
|
|
375
|
+
// lifecycle 的 0=不限(>0 才挂 scheduleTimeBudget)语义分裂。
|
|
376
|
+
it("父 budgetTimeMs=0 → 不限(旧实现立即超时;子 run 晚完成正常返回,U2)", async () => {
|
|
377
|
+
vi.useFakeTimers();
|
|
378
|
+
try {
|
|
379
|
+
const parent = makeParentRun({ budgetTimeMs: 0 });
|
|
380
|
+
const childRun = makeRunningChildRun();
|
|
381
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
382
|
+
setupRunWorkflow(childRun);
|
|
383
|
+
|
|
384
|
+
const promise = executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
385
|
+
|
|
386
|
+
// 旧实现 0 落成 deadline=now:首轮轮询即 abort time_limited。不限时则不 abort
|
|
387
|
+
await vi.advanceTimersByTimeAsync(5_000);
|
|
388
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
389
|
+
|
|
390
|
+
childRun.state.status = "done";
|
|
391
|
+
childRun.state.reason = "completed";
|
|
392
|
+
childRun.state.scriptResult = "ok";
|
|
393
|
+
await vi.advanceTimersByTimeAsync(500);
|
|
394
|
+
const result = await promise;
|
|
395
|
+
expect(result.content).toBe("ok");
|
|
396
|
+
expect(result.error).toBeUndefined();
|
|
397
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
398
|
+
} finally {
|
|
399
|
+
vi.useRealTimers();
|
|
400
|
+
}
|
|
401
|
+
});
|
|
402
|
+
|
|
403
|
+
it("父 budgetTimeMs=-5000 → 不限且错误串无 \"-5000ms\"(U2 负数同义不限)", async () => {
|
|
404
|
+
vi.useFakeTimers();
|
|
405
|
+
try {
|
|
406
|
+
const parent = makeParentRun({ budgetTimeMs: -5_000 });
|
|
407
|
+
const childRun = makeRunningChildRun();
|
|
408
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
409
|
+
setupRunWorkflow(childRun);
|
|
410
|
+
|
|
411
|
+
const promise = executeNestedWorkflow("child-wf", {}, parent, deps);
|
|
412
|
+
|
|
413
|
+
await vi.advanceTimersByTimeAsync(5_000);
|
|
414
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
415
|
+
|
|
416
|
+
childRun.state.status = "done";
|
|
417
|
+
childRun.state.reason = "completed";
|
|
418
|
+
childRun.state.scriptResult = "ok";
|
|
419
|
+
await vi.advanceTimersByTimeAsync(500);
|
|
420
|
+
const result = await promise;
|
|
421
|
+
expect(result.content).toBe("ok");
|
|
422
|
+
expect(result.error).toBeUndefined();
|
|
423
|
+
// 防御:任何路径都不产生 "timed out after -5000ms" 类错误串
|
|
424
|
+
const abortCalls = vi.mocked(abortRun).mock.calls.map((c) => String(c[2]));
|
|
425
|
+
expect(abortCalls.join(";")).not.toContain("-5000ms");
|
|
426
|
+
} finally {
|
|
427
|
+
vi.useRealTimers();
|
|
428
|
+
}
|
|
429
|
+
});
|
|
430
|
+
});
|
|
431
|
+
|
|
432
|
+
// ── runAndWait timeoutMs 语义(预算语义对齐 2026-08)──
|
|
433
|
+
describe("runAndWait timeoutMs 语义(预算语义对齐)", () => {
|
|
434
|
+
it("未传 timeoutMs → 不限(远超旧 10min 默认不 abort,完成后正常返回)", async () => {
|
|
435
|
+
vi.useFakeTimers();
|
|
436
|
+
try {
|
|
437
|
+
const childRun = makeRunningChildRun();
|
|
438
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
439
|
+
setupRunWorkflow(childRun);
|
|
440
|
+
|
|
441
|
+
// 只传 3 参:不传 signal / timeoutMs
|
|
442
|
+
const promise = runAndWait("child-wf", {}, deps);
|
|
443
|
+
|
|
444
|
+
await vi.advanceTimersByTimeAsync(10 * 60 * 1000 + 500);
|
|
445
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
446
|
+
|
|
447
|
+
childRun.state.status = "done";
|
|
448
|
+
childRun.state.reason = "completed";
|
|
449
|
+
childRun.state.scriptResult = "ok";
|
|
450
|
+
await vi.advanceTimersByTimeAsync(500);
|
|
451
|
+
const result = await promise;
|
|
452
|
+
expect(result.status).toBe("done");
|
|
453
|
+
expect(result.reason).toBe("completed");
|
|
454
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
455
|
+
} finally {
|
|
456
|
+
vi.useRealTimers();
|
|
457
|
+
}
|
|
458
|
+
});
|
|
459
|
+
|
|
460
|
+
it("显式 timeoutMs=1000 → 到期 abort time_limited(限时能力保留)", async () => {
|
|
461
|
+
vi.useFakeTimers();
|
|
462
|
+
try {
|
|
463
|
+
const childRun = makeRunningChildRun();
|
|
464
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
465
|
+
setupRunWorkflow(childRun);
|
|
466
|
+
|
|
467
|
+
const promise = runAndWait("child-wf", {}, deps, undefined, 1_000);
|
|
468
|
+
|
|
469
|
+
await vi.advanceTimersByTimeAsync(1_000 + 500);
|
|
470
|
+
const result = await promise;
|
|
471
|
+
expect(vi.mocked(abortRun)).toHaveBeenCalledTimes(1);
|
|
472
|
+
expect(vi.mocked(abortRun)).toHaveBeenCalledWith(
|
|
473
|
+
MOCK_RUN_ID,
|
|
474
|
+
deps,
|
|
475
|
+
"Workflow timed out after 1000ms",
|
|
476
|
+
"time_limited",
|
|
477
|
+
);
|
|
478
|
+
void result;
|
|
479
|
+
} finally {
|
|
480
|
+
vi.useRealTimers();
|
|
481
|
+
}
|
|
482
|
+
});
|
|
483
|
+
|
|
484
|
+
// [U2] 0/负 timeoutMs 与 undefined 同义 = 不限(旧实现 0 → 立即超时、
|
|
485
|
+
// 负数 → "timed out after -5000ms" 类错误串)。
|
|
486
|
+
it("timeoutMs=0 → 不限(旧实现立即超时;不产生 \"timed out after 0ms\",U2)", async () => {
|
|
487
|
+
vi.useFakeTimers();
|
|
488
|
+
try {
|
|
489
|
+
const childRun = makeRunningChildRun();
|
|
490
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
491
|
+
setupRunWorkflow(childRun);
|
|
492
|
+
|
|
493
|
+
const promise = runAndWait("child-wf", {}, deps, undefined, 0);
|
|
494
|
+
|
|
495
|
+
await vi.advanceTimersByTimeAsync(5_000);
|
|
496
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
497
|
+
|
|
498
|
+
childRun.state.status = "done";
|
|
499
|
+
childRun.state.reason = "completed";
|
|
500
|
+
childRun.state.scriptResult = "ok";
|
|
501
|
+
await vi.advanceTimersByTimeAsync(500);
|
|
502
|
+
const result = await promise;
|
|
503
|
+
expect(result.reason).toBe("completed");
|
|
504
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
505
|
+
} finally {
|
|
506
|
+
vi.useRealTimers();
|
|
507
|
+
}
|
|
508
|
+
});
|
|
509
|
+
|
|
510
|
+
// [U7] XYZ_SUBAGENT_RUN_WATCHDOG_MS:显式不限时 run 的绝对时限兑底(与
|
|
511
|
+
// spawn watchdog env 同款:未设 = 无兑底,设置 = opt-in 挂载)。
|
|
512
|
+
describe("XYZ_SUBAGENT_RUN_WATCHDOG_MS 兑底(U7)", () => {
|
|
513
|
+
it("env 设置 + 未传 timeoutMs → env 时限到期 abort time_limited,错误串含 env 值", async () => {
|
|
514
|
+
vi.useFakeTimers();
|
|
515
|
+
try {
|
|
516
|
+
vi.stubEnv(RUN_WATCHDOG_ENV, "1000");
|
|
517
|
+
const childRun = makeRunningChildRun();
|
|
518
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
519
|
+
setupRunWorkflow(childRun);
|
|
520
|
+
|
|
521
|
+
const promise = runAndWait("child-wf", {}, deps);
|
|
522
|
+
|
|
523
|
+
await vi.advanceTimersByTimeAsync(1_000 + 500);
|
|
524
|
+
const result = await promise;
|
|
525
|
+
expect(vi.mocked(abortRun)).toHaveBeenCalledWith(
|
|
526
|
+
MOCK_RUN_ID,
|
|
527
|
+
deps,
|
|
528
|
+
"Workflow timed out after 1000ms",
|
|
529
|
+
"time_limited",
|
|
530
|
+
);
|
|
531
|
+
// mock abortRun 不翻转 run 状态(对齐同组 timeoutMs=1000 用例):result.reason
|
|
532
|
+
// 是 toResult 对 running run 的防御 fallback,不在此断言
|
|
533
|
+
void result;
|
|
534
|
+
} finally {
|
|
535
|
+
vi.useRealTimers();
|
|
536
|
+
vi.unstubAllEnvs();
|
|
537
|
+
}
|
|
538
|
+
});
|
|
539
|
+
|
|
540
|
+
it("env 设置 + 显式 timeoutMs → 显式值优先(env 不截断显式限时)", async () => {
|
|
541
|
+
vi.useFakeTimers();
|
|
542
|
+
try {
|
|
543
|
+
vi.stubEnv(RUN_WATCHDOG_ENV, "1000");
|
|
544
|
+
const childRun = makeRunningChildRun();
|
|
545
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
546
|
+
setupRunWorkflow(childRun);
|
|
547
|
+
|
|
548
|
+
const promise = runAndWait("child-wf", {}, deps, undefined, 60_000);
|
|
549
|
+
|
|
550
|
+
// env 1000ms 到期:不 abort(显式 60s 优先)
|
|
551
|
+
await vi.advanceTimersByTimeAsync(1_000 + 500);
|
|
552
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
553
|
+
|
|
554
|
+
// 显式 60s 到期:才 abort,错误串为显式值
|
|
555
|
+
await vi.advanceTimersByTimeAsync(60_000);
|
|
556
|
+
const result = await promise;
|
|
557
|
+
expect(vi.mocked(abortRun)).toHaveBeenCalledWith(
|
|
558
|
+
MOCK_RUN_ID,
|
|
559
|
+
deps,
|
|
560
|
+
"Workflow timed out after 60000ms",
|
|
561
|
+
"time_limited",
|
|
562
|
+
);
|
|
563
|
+
void result;
|
|
564
|
+
} finally {
|
|
565
|
+
vi.useRealTimers();
|
|
566
|
+
vi.unstubAllEnvs();
|
|
567
|
+
}
|
|
568
|
+
});
|
|
569
|
+
|
|
570
|
+
it("env 非法值(abc / -1)→ 视为未设(无兑底,不限)", async () => {
|
|
571
|
+
vi.useFakeTimers();
|
|
572
|
+
try {
|
|
573
|
+
const childRun = makeRunningChildRun();
|
|
574
|
+
const deps = makeDeps({ script: makeScript(), childRun });
|
|
575
|
+
setupRunWorkflow(childRun);
|
|
576
|
+
|
|
577
|
+
vi.stubEnv(RUN_WATCHDOG_ENV, "abc");
|
|
578
|
+
const promiseAbc = runAndWait("child-wf", {}, deps);
|
|
579
|
+
await vi.advanceTimersByTimeAsync(10 * 60 * 1000 + 500);
|
|
580
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
581
|
+
// 收尾完成,防用例间 promise 悬挂
|
|
582
|
+
childRun.state.status = "done";
|
|
583
|
+
childRun.state.reason = "completed";
|
|
584
|
+
childRun.state.scriptResult = "ok";
|
|
585
|
+
await vi.advanceTimersByTimeAsync(500);
|
|
586
|
+
await promiseAbc;
|
|
587
|
+
|
|
588
|
+
vi.mocked(abortRun).mockClear();
|
|
589
|
+
vi.stubEnv(RUN_WATCHDOG_ENV, "-1");
|
|
590
|
+
// childRun 已是 done 态:pollRunToResult 首轮即命中终态返回,无 deadline 可言
|
|
591
|
+
await runAndWait("child-wf", {}, deps);
|
|
592
|
+
expect(vi.mocked(abortRun)).not.toHaveBeenCalled();
|
|
593
|
+
} finally {
|
|
594
|
+
vi.useRealTimers();
|
|
595
|
+
vi.unstubAllEnvs();
|
|
596
|
+
}
|
|
597
|
+
});
|
|
598
|
+
});
|
|
599
|
+
});
|
|
600
|
+
});
|
|
@@ -0,0 +1,96 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* rfl 仪表 T2(tier-1 §7.1):引擎注入稳定 _runId。
|
|
3
|
+
*
|
|
4
|
+
* A3 runWorkflow 在 validateRunArgs 后向 spec.args 注入 _runId(值 = 返回的
|
|
5
|
+
* runId)——runAndWait 与 executeNestedWorkflow 两个 args 入口共用的单一
|
|
6
|
+
* choke point(lifecycle.ts runWorkflow)。
|
|
7
|
+
* A4 rebuildRuntime 复用 run.spec.args 同一对象,_runId 跨 worker rebuild 不
|
|
8
|
+
* 漂移(修复「rebuild 回退 run-<Date.now()> 导致 run 碎裂」)。
|
|
9
|
+
*/
|
|
10
|
+
import { describe, expect, it, vi } from "vitest";
|
|
11
|
+
|
|
12
|
+
import { rebuildRuntime } from "../error-recovery.ts";
|
|
13
|
+
import { runWorkflow } from "../lifecycle.ts";
|
|
14
|
+
import type { RunSpec } from "../models/run-spec.ts";
|
|
15
|
+
import type { LifecycleDeps } from "../models/ports.ts";
|
|
16
|
+
|
|
17
|
+
function makeSpec(args: Record<string, unknown> = {}): RunSpec {
|
|
18
|
+
return {
|
|
19
|
+
scriptSource: "module.exports = { execute: async () => 'ok' };",
|
|
20
|
+
args,
|
|
21
|
+
scriptName: "test-wf",
|
|
22
|
+
scriptPath: "/fake/test.js",
|
|
23
|
+
};
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
/** LifecycleDeps mock:workerHost.start 可观察(记录每次调用收到的 args 引用)。 */
|
|
27
|
+
function makeRecordingDeps(): LifecycleDeps & {
|
|
28
|
+
startCalls: Array<{ spec: RunSpec; args: Record<string, unknown> }>;
|
|
29
|
+
} {
|
|
30
|
+
const startCalls: Array<{ spec: RunSpec; args: Record<string, unknown> }> = [];
|
|
31
|
+
const deps = {
|
|
32
|
+
store: { save: vi.fn(async () => {}), loadAll: vi.fn(async () => []) },
|
|
33
|
+
workerHost: {
|
|
34
|
+
start: vi.fn((spec: RunSpec, args: Record<string, unknown>) => {
|
|
35
|
+
startCalls.push({ spec, args });
|
|
36
|
+
return { postMessage: vi.fn(), terminate: vi.fn(async () => {}) };
|
|
37
|
+
}),
|
|
38
|
+
},
|
|
39
|
+
runner: { run: vi.fn(async () => ({})) },
|
|
40
|
+
runs: new Map(),
|
|
41
|
+
eventBus: { emit: vi.fn() },
|
|
42
|
+
onRunDone: vi.fn(),
|
|
43
|
+
log: vi.fn(),
|
|
44
|
+
} as unknown as LifecycleDeps;
|
|
45
|
+
return Object.assign(deps, { startCalls });
|
|
46
|
+
}
|
|
47
|
+
|
|
48
|
+
describe("A3 runWorkflow 注入 _runId(tier-1 §7.1)", () => {
|
|
49
|
+
it("A3 runWorkflow 后 spec.args._runId === 返回的 runId(字符串),且 workerHost.start 收到同一值", async () => {
|
|
50
|
+
const deps = makeRecordingDeps();
|
|
51
|
+
const spec = makeSpec({ targetType: "file", target: "/tmp/x" });
|
|
52
|
+
const runId = await runWorkflow(spec, deps);
|
|
53
|
+
expect(runId).toBeTruthy();
|
|
54
|
+
expect(spec.args._runId).toBe(runId);
|
|
55
|
+
expect(typeof spec.args._runId).toBe("string");
|
|
56
|
+
expect(deps.startCalls.length).toBe(1);
|
|
57
|
+
expect(deps.startCalls[0].args._runId).toBe(runId);
|
|
58
|
+
});
|
|
59
|
+
|
|
60
|
+
it("A3 引擎注入覆盖用户预传的 _runId(非公开参数,引擎值权威)", async () => {
|
|
61
|
+
const deps = makeRecordingDeps();
|
|
62
|
+
const spec = makeSpec({ _runId: "user-supplied" });
|
|
63
|
+
const runId = await runWorkflow(spec, deps);
|
|
64
|
+
expect(runId).not.toBe("user-supplied");
|
|
65
|
+
expect(spec.args._runId).toBe(runId);
|
|
66
|
+
});
|
|
67
|
+
});
|
|
68
|
+
|
|
69
|
+
describe("A4 rebuild 后 _runId 稳定(tier-1 §7.1)", () => {
|
|
70
|
+
it("A4 rebuildRuntime 第二次 workerHost.start 收到的 args._runId 非 undefined 且与首启一致(同一 args 对象)", async () => {
|
|
71
|
+
const deps = makeRecordingDeps();
|
|
72
|
+
const spec = makeSpec({ targetType: "file" });
|
|
73
|
+
const runId = await runWorkflow(spec, deps);
|
|
74
|
+
expect(deps.startCalls.length).toBe(1);
|
|
75
|
+
|
|
76
|
+
const run = deps.runs.get(runId);
|
|
77
|
+
expect(run).toBeTruthy();
|
|
78
|
+
expect(run.state.status).toBe("running");
|
|
79
|
+
|
|
80
|
+
// 触发 rebuild(对齐 error-recovery handleWorkerError 的恢复路径)
|
|
81
|
+
rebuildRuntime(run, deps, {
|
|
82
|
+
onMessage: vi.fn(),
|
|
83
|
+
onError: vi.fn(),
|
|
84
|
+
onExit: vi.fn(),
|
|
85
|
+
});
|
|
86
|
+
|
|
87
|
+
expect(deps.startCalls.length).toBe(2);
|
|
88
|
+
const first = deps.startCalls[0].args;
|
|
89
|
+
const second = deps.startCalls[1].args;
|
|
90
|
+
expect(second._runId).not.toBeUndefined();
|
|
91
|
+
expect(second._runId).toBe(first._runId);
|
|
92
|
+
expect(second._runId).toBe(runId);
|
|
93
|
+
// 同一对象引用(复用而非拷贝)——这是跨 rebuild 稳定的结构保证
|
|
94
|
+
expect(second).toBe(first);
|
|
95
|
+
});
|
|
96
|
+
});
|