From 3756f6547348de55411789019a90c59627051aeb Mon Sep 17 00:00:00 2001 From: Dotta Date: Mon, 20 Apr 2026 15:43:12 -0500 Subject: [PATCH] fix(test): wait for heartbeat runs to settle --- .../heartbeat-dependency-scheduling.test.ts | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/server/src/__tests__/heartbeat-dependency-scheduling.test.ts b/server/src/__tests__/heartbeat-dependency-scheduling.test.ts index 6327c84c897..58464954d26 100644 --- a/server/src/__tests__/heartbeat-dependency-scheduling.test.ts +++ b/server/src/__tests__/heartbeat-dependency-scheduling.test.ts @@ -97,6 +97,21 @@ describeEmbeddedPostgres("heartbeat dependency-aware queued run selection", () = afterEach(async () => { vi.clearAllMocks(); runningProcesses.clear(); + let idlePolls = 0; + for (let attempt = 0; attempt < 100; attempt += 1) { + const runs = await db + .select({ status: heartbeatRuns.status }) + .from(heartbeatRuns); + const hasActiveRun = runs.some((run) => run.status === "queued" || run.status === "running"); + if (!hasActiveRun) { + idlePolls += 1; + if (idlePolls >= 3) break; + } else { + idlePolls = 0; + } + await new Promise((resolve) => setTimeout(resolve, 50)); + } + await new Promise((resolve) => setTimeout(resolve, 50)); await db.delete(activityLog); await db.delete(companySkills); await db.delete(issueComments);