YeonGyu-Kim
2b5887aca3
Guarding polling re-entry avoids stacked async polls under slow responses, and unref on pending-call cleanup timer reduces idle wakeups.
57 lines
1.6 KiB
TypeScript
57 lines
1.6 KiB
TypeScript
import { describe, test, expect } from "bun:test"
|
|
import { TmuxPollingManager } from "./polling-manager"
|
|
import type { TrackedSession } from "./types"
|
|
|
|
describe("TmuxPollingManager overlap", () => {
|
|
test("skips overlapping pollSessions executions", async () => {
|
|
//#given
|
|
const sessions = new Map<string, TrackedSession>()
|
|
sessions.set("ses-1", {
|
|
sessionId: "ses-1",
|
|
paneId: "%1",
|
|
description: "test",
|
|
createdAt: new Date(),
|
|
lastSeenAt: new Date(),
|
|
})
|
|
|
|
let activeCalls = 0
|
|
let maxActiveCalls = 0
|
|
let statusCallCount = 0
|
|
let releaseStatus: (() => void) | undefined
|
|
const statusGate = new Promise<void>((resolve) => {
|
|
releaseStatus = resolve
|
|
})
|
|
|
|
const client = {
|
|
session: {
|
|
status: async () => {
|
|
statusCallCount += 1
|
|
activeCalls += 1
|
|
maxActiveCalls = Math.max(maxActiveCalls, activeCalls)
|
|
await statusGate
|
|
activeCalls -= 1
|
|
return { data: { "ses-1": { type: "running" } } }
|
|
},
|
|
messages: async () => ({ data: [] }),
|
|
},
|
|
}
|
|
|
|
const manager = new TmuxPollingManager(
|
|
client as unknown as import("../../tools/delegate-task/types").OpencodeClient,
|
|
sessions,
|
|
async () => {},
|
|
)
|
|
|
|
//#when
|
|
const firstPoll = (manager as unknown as { pollSessions: () => Promise<void> }).pollSessions()
|
|
await Promise.resolve()
|
|
const secondPoll = (manager as unknown as { pollSessions: () => Promise<void> }).pollSessions()
|
|
releaseStatus?.()
|
|
await Promise.all([firstPoll, secondPoll])
|
|
|
|
//#then
|
|
expect(maxActiveCalls).toBe(1)
|
|
expect(statusCallCount).toBe(1)
|
|
})
|
|
})
|