diff --git a/src/agents/subagent-registry.nested.test.ts b/src/agents/subagent-registry.nested.test.ts index 399917d177..2ff207a79b 100644 --- a/src/agents/subagent-registry.nested.test.ts +++ b/src/agents/subagent-registry.nested.test.ts @@ -14,15 +14,26 @@ vi.mock("../infra/agent-events.js", () => ({ onAgentEvent: vi.fn(() => noop), })); +vi.mock("../config/config.js", () => ({ + loadConfig: vi.fn(() => ({ + agents: { defaults: { subagents: { archiveAfterMinutes: 0 } } }, + })), +})); + vi.mock("./subagent-announce.js", () => ({ runSubagentAnnounceFlow: vi.fn(async () => true), buildSubagentSystemPrompt: vi.fn(() => "test prompt"), })); +vi.mock("./subagent-registry.store.js", () => ({ + loadSubagentRegistryFromDisk: vi.fn(() => new Map()), + saveSubagentRegistryToDisk: vi.fn(() => {}), +})); + describe("subagent registry nested agent tracking", () => { afterEach(async () => { const mod = await import("./subagent-registry.js"); - mod.resetSubagentRegistryForTests(); + mod.resetSubagentRegistryForTests({ persist: false }); }); it("listSubagentRunsForRequester returns children of the requesting session", async () => { diff --git a/src/agents/subagent-registry.steer-restart.test.ts b/src/agents/subagent-registry.steer-restart.test.ts index a4a0a70109..0d60a16a69 100644 --- a/src/agents/subagent-registry.steer-restart.test.ts +++ b/src/agents/subagent-registry.steer-restart.test.ts @@ -22,17 +22,28 @@ vi.mock("../infra/agent-events.js", () => ({ }), })); +vi.mock("../config/config.js", () => ({ + loadConfig: vi.fn(() => ({ + agents: { defaults: { subagents: { archiveAfterMinutes: 0 } } }, + })), +})); + const announceSpy = vi.fn(async () => true); vi.mock("./subagent-announce.js", () => ({ runSubagentAnnounceFlow: (...args: unknown[]) => announceSpy(...args), })); +vi.mock("./subagent-registry.store.js", () => ({ + loadSubagentRegistryFromDisk: vi.fn(() => new Map()), + saveSubagentRegistryToDisk: vi.fn(() => {}), +})); + describe("subagent registry steer restarts", () => { afterEach(async () => { announceSpy.mockClear(); lifecycleHandler = undefined; const mod = await import("./subagent-registry.js"); - mod.resetSubagentRegistryForTests(); + mod.resetSubagentRegistryForTests({ persist: false }); }); it("suppresses announce for interrupted runs and only announces the replacement run", async () => {