test(agent): split stuck recovery registry coverage

mushuiyu886 · mushuiyu886 · commit 8c5ed1a5a9bf · 2026-07-17T10:55:37.000+08:00
diff --git a/src/agents/embedded-agent-runner/runs.stuck-recovery.test.ts b/src/agents/embedded-agent-runner/runs.stuck-recovery.test.ts
@@ -0,0 +1,33 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { createReplyOperation } from "../../auto-reply/reply/reply-run-registry.js";
+import { testing as replyRunTesting } from "../../auto-reply/reply/reply-run-registry.test-support.js";
+import { abortEmbeddedAgentRun } from "./runs.js";
+import { testing } from "./runs.test-support.js";
+
+describe("embedded-agent runner stuck recovery", () => {
+  afterEach(() => {
+    testing.resetActiveEmbeddedRuns();
+    replyRunTesting.resetReplyRunRegistry();
+    vi.restoreAllMocks();
+  });
+
+  it("passes stuck recovery abort reasons through reply-run fallback", () => {
+    const operation = createReplyOperation({
+      sessionKey: "agent:main:main",
+      sessionId: "session-reply-run",
+      resetTriggered: false,
+    });
+    const cancel = vi.fn();
+    operation.attachBackend({
+      kind: "embedded",
+      cancel,
+      isStreaming: () => true,
+    });
+    operation.setPhase("running");
+
+    expect(abortEmbeddedAgentRun("session-reply-run", { reason: "stuck_recovery" })).toBe(true);
+
+    expect(cancel).toHaveBeenCalledWith("stuck_recovery");
+    expect(operation.result).toEqual({ kind: "aborted", code: "aborted_for_stuck_recovery" });
+  });
+});
diff --git a/src/agents/embedded-agent-runner/runs.test.ts b/src/agents/embedded-agent-runner/runs.test.ts
@@ -898,26 +898,6 @@ describe("embedded-agent runner run registry", () => {
     expect(queueMessage).not.toHaveBeenCalled();
   });
 
-  it("passes stuck recovery abort reasons through reply-run fallback", () => {
-    const operation = createReplyOperation({
-      sessionKey: "agent:main:main",
-      sessionId: "session-reply-run",
-      resetTriggered: false,
-    });
-    const cancel = vi.fn();
-    operation.attachBackend({
-      kind: "embedded",
-      cancel,
-      isStreaming: () => true,
-    });
-    operation.setPhase("running");
-
-    expect(abortEmbeddedAgentRun("session-reply-run", { reason: "stuck_recovery" })).toBe(true);
-
-    expect(cancel).toHaveBeenCalledWith("stuck_recovery");
-    expect(operation.result).toEqual({ kind: "aborted", code: "aborted_for_stuck_recovery" });
-  });
-
   it("force-clears an aborted run that does not drain", async () => {
     vi.useFakeTimers();
     try {
