test(background-agent): cover abort timeout handling
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
@@ -0,0 +1,57 @@
|
||||
import { afterAll, describe, expect, mock, test } from "bun:test"
|
||||
|
||||
const logMock = mock(() => {})
|
||||
|
||||
mock.module("../../shared", () => ({
|
||||
log: logMock,
|
||||
}))
|
||||
|
||||
import { abortWithTimeout } from "./abort-with-timeout"
|
||||
import type { OpencodeClient } from "./opencode-client"
|
||||
|
||||
function createClient(abort: (...args: Array<unknown>) => Promise<unknown>): OpencodeClient {
|
||||
return {
|
||||
session: {
|
||||
abort: abort as never,
|
||||
},
|
||||
} as never
|
||||
}
|
||||
|
||||
describe("abortWithTimeout", () => {
|
||||
afterAll(() => {
|
||||
mock.restore()
|
||||
})
|
||||
|
||||
test("#given abort resolves before timeout #when abortWithTimeout runs #then it returns true", async () => {
|
||||
// given
|
||||
const abort = mock(async () => ({}))
|
||||
|
||||
// when
|
||||
const result = await abortWithTimeout(createClient(abort), "session-1", 10)
|
||||
|
||||
// then
|
||||
expect(result).toBe(true)
|
||||
expect(abort).toHaveBeenCalledWith({ path: { id: "session-1" } })
|
||||
expect(logMock).not.toHaveBeenCalled()
|
||||
})
|
||||
|
||||
test("#given abort hangs indefinitely #when abortWithTimeout runs #then it logs warning and continues", async () => {
|
||||
// given
|
||||
const abort = mock(() => new Promise<never>(() => {}))
|
||||
|
||||
// when
|
||||
const result = await Promise.race([
|
||||
abortWithTimeout(createClient(abort), "session-2", 1),
|
||||
new Promise<never>((_, reject) => {
|
||||
setTimeout(() => reject(new Error("abort timeout test exceeded wait budget")), 100)
|
||||
}),
|
||||
])
|
||||
|
||||
// then
|
||||
expect(result).toBe(false)
|
||||
expect(logMock).toHaveBeenCalledWith(
|
||||
"[background-agent] Session abort timed out; continuing cleanup:",
|
||||
{ sessionID: "session-2", timeoutMs: 1 },
|
||||
)
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user