fix(comment-checker): improve pending calls handling and add tests

🤖 Generated with assistance of OhMyOpenCode
This commit is contained in:
YeonGyu-Kim
2026-03-31 17:33:38 -07:00
parent 439c9a6499
commit 3c77c048dd
3 changed files with 57 additions and 1 deletions
+9 -1
View File
@@ -22,7 +22,12 @@ import {
processWithCli,
processApplyPatchEditsWithCli,
} from "./cli-runner"
import { registerPendingCall, startPendingCallCleanup, takePendingCall } from "./pending-calls"
import {
registerPendingCall,
startPendingCallCleanup,
stopPendingCallCleanup,
takePendingCall,
} from "./pending-calls"
import * as fs from "fs"
import { tmpdir } from "os"
@@ -180,5 +185,8 @@ export function createCommentCheckerHooks(config?: CommentCheckerConfig) {
debugLog("tool.execute.after failed:", err)
}
},
dispose: (): void => {
stopPendingCallCleanup()
},
}
}
@@ -35,4 +35,43 @@ describe("pending-calls cleanup interval", () => {
globalThis.setInterval = originalSetInterval
}
})
test("#given cleanup timer already started #when stop cleanup runs #then interval state resets for future reuse", async () => {
//#given
const originalSetInterval = globalThis.setInterval
const originalClearInterval = globalThis.clearInterval
let intervalHandle: ReturnType<typeof setInterval> | undefined
let clearCalls = 0
globalThis.setInterval = ((
_handler: TimerHandler,
_timeout?: number,
..._args: any[]
) => {
intervalHandle = { unref: () => {} } as unknown as ReturnType<typeof setInterval>
return intervalHandle
}) as unknown as typeof setInterval
globalThis.clearInterval = ((handle?: ReturnType<typeof setInterval>) => {
if (handle === intervalHandle) {
clearCalls += 1
}
}) as unknown as typeof clearInterval
try {
const modulePath = new URL("./pending-calls.ts", import.meta.url).pathname
const pendingCallsModule = await import(`${modulePath}?pending-calls-test-stop`)
pendingCallsModule.startPendingCallCleanup()
//#when
pendingCallsModule.stopPendingCallCleanup()
pendingCallsModule.startPendingCallCleanup()
//#then
expect(clearCalls).toBe(1)
} finally {
globalThis.setInterval = originalSetInterval
globalThis.clearInterval = originalClearInterval
}
})
})
@@ -24,6 +24,15 @@ export function startPendingCallCleanup(): void {
}
}
export function stopPendingCallCleanup(): void {
pendingCalls.clear()
if (cleanupInterval) {
clearInterval(cleanupInterval)
cleanupInterval = undefined
}
cleanupIntervalStarted = false
}
export function registerPendingCall(callID: string, pendingCall: PendingCall): void {
pendingCalls.set(callID, pendingCall)
}