YeonGyu-Kim
83 lines
2.4 KiB
TypeScript
83 lines
2.4 KiB
TypeScript
import { afterEach, describe, expect, it, mock } from "bun:test"
|
|
|
|
async function* createEmptyEventStream(): AsyncIterable<unknown> {}
|
|
|
|
describe("run telemetry isolation", () => {
|
|
afterEach(() => {
|
|
mock.restore()
|
|
})
|
|
|
|
it("does not crash CLI run when telemetry throws", async () => {
|
|
// given
|
|
mock.module("../../plugin-config", () => ({
|
|
loadPluginConfig: mock(() => ({})),
|
|
}))
|
|
mock.module("./agent-resolver", () => ({
|
|
resolveRunAgent: mock(() => "Sisyphus - Ultraworker"),
|
|
}))
|
|
mock.module("./server-connection", () => ({
|
|
createServerConnection: mock(async () => ({
|
|
client: {
|
|
event: {
|
|
subscribe: mock(async () => ({ stream: createEmptyEventStream() })),
|
|
},
|
|
session: {
|
|
promptAsync: mock(async () => undefined),
|
|
},
|
|
},
|
|
cleanup: mock(() => {}),
|
|
})),
|
|
}))
|
|
mock.module("./session-resolver", () => ({
|
|
resolveSession: mock(async () => "ses_test"),
|
|
}))
|
|
mock.module("./json-output", () => ({
|
|
createJsonOutputManager: mock(() => ({
|
|
redirectToStderr: mock(() => {}),
|
|
restore: mock(() => {}),
|
|
emitResult: mock(() => {}),
|
|
})),
|
|
}))
|
|
mock.module("./on-complete-hook", () => ({
|
|
executeOnCompleteHook: mock(async () => {}),
|
|
}))
|
|
mock.module("./model-resolver", () => ({
|
|
resolveRunModel: mock(() => null),
|
|
}))
|
|
mock.module("./poll-for-completion", () => ({
|
|
pollForCompletion: mock(async () => 0),
|
|
}))
|
|
mock.module("./agent-profile-colors", () => ({
|
|
loadAgentProfileColors: mock(async () => ({})),
|
|
}))
|
|
mock.module("./stdin-suppression", () => ({
|
|
suppressRunInput: mock(() => mock(() => {})),
|
|
}))
|
|
mock.module("./timestamp-output", () => ({
|
|
createTimestampedStdoutController: mock(() => ({
|
|
enable: mock(() => {}),
|
|
restore: mock(() => {}),
|
|
})),
|
|
}))
|
|
mock.module("../../shared/posthog", () => ({
|
|
createCliPostHog: mock(() => ({
|
|
trackActive: () => {
|
|
throw new Error("telemetry failed")
|
|
},
|
|
shutdown: mock(async () => {
|
|
throw new Error("shutdown failed")
|
|
}),
|
|
})),
|
|
getPostHogDistinctId: mock(() => "run-distinct-id"),
|
|
}))
|
|
|
|
const { run } = await import(`./runner?telemetry=${Date.now()}-${Math.random()}`)
|
|
|
|
// when
|
|
const result = await run({ message: "test" })
|
|
|
|
// then
|
|
expect(result).toBe(0)
|
|
})
|
|
})
|