diff --git a/src/agents/subagent-control.test.ts b/src/agents/subagent-control.test.ts index e0f8d11f12c..2b65b600c4d 100644 --- a/src/agents/subagent-control.test.ts +++ b/src/agents/subagent-control.test.ts @@ -14,6 +14,7 @@ import { steerControlledSubagentRun, } from "./subagent-control.js"; import { + __testing as subagentRegistryTesting, addSubagentRunForTests, getSubagentRunByChildSessionKey, resetSubagentRegistryForTests, @@ -145,6 +146,21 @@ function writeSessionStoreFixture(label: string, store: Record) beforeEach(() => { setSubagentControlDepsForTest(); + subagentRegistryTesting.setDepsForTest({ + cleanupBrowserSessionsForLifecycleEnd: async () => {}, + ensureContextEnginesInitialized: () => {}, + ensureRuntimePluginsLoaded: () => {}, + resolveContextEngine: async () => ({ + info: { id: "test", name: "Test" }, + assemble: async ({ messages }) => ({ messages, estimatedTokens: 0 }), + compact: async () => ({ ok: true, compacted: false }), + ingest: async () => ({ ingested: false }), + }), + }); +}); + +afterEach(() => { + subagentRegistryTesting.setDepsForTest(); }); describe("sendControlledSubagentMessage", () => {