diff --git a/src/cli/run/runner.telemetry.test.ts b/src/cli/run/runner.telemetry.test.ts index 0117f74a8..3d4b5193b 100644 --- a/src/cli/run/runner.telemetry.test.ts +++ b/src/cli/run/runner.telemetry.test.ts @@ -1,5 +1,7 @@ import { afterEach, describe, expect, it, mock } from "bun:test" +async function* createEmptyEventStream(): AsyncIterable {} + describe("run telemetry isolation", () => { afterEach(() => { mock.restore() @@ -13,20 +15,11 @@ describe("run telemetry isolation", () => { mock.module("./agent-resolver", () => ({ resolveRunAgent: mock(() => "Sisyphus - Ultraworker"), })) - mock.module("./events", () => ({ - createEventState: mock(() => ({ - messageCount: 0, - lastPartText: "Run completed", - agentColorsByName: {}, - })), - processEvents: mock(async () => {}), - serializeError: (error: unknown) => (error instanceof Error ? error.message : String(error)), - })) mock.module("./server-connection", () => ({ createServerConnection: mock(async () => ({ client: { event: { - subscribe: mock(async () => ({ stream: {} })), + subscribe: mock(async () => ({ stream: createEmptyEventStream() })), }, session: { promptAsync: mock(async () => undefined), diff --git a/src/index.telemetry.test.ts b/src/index.telemetry.test.ts index 2d554626f..f3026a1cb 100644 --- a/src/index.telemetry.test.ts +++ b/src/index.telemetry.test.ts @@ -95,7 +95,14 @@ function installModuleMocks(): void { startBackgroundCheck: mock(() => {}), })) mock.module("./tools/lsp/client", () => ({ - lspManager: {}, + lspManager: { + getClient: mock(async () => ({ + diagnostics: mock(async () => ({ items: [] })), + })), + stopAll: mock(async () => {}), + releaseClient: mock(() => {}), + cleanupTempDirectoryClients: mock(async () => {}), + }, })) mock.module("./shared/posthog", () => ({ createPluginPostHog: mockCreatePluginPostHog,