From 6c67a5b000c4ffa4c6a02b47101ab5fb649ed73b Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Sat, 11 Jul 2026 01:56:37 +0000
Subject: [PATCH] test: drain heartbeat execution before cleanup

---
 server/src/__tests__/heartbeat-retry-scheduling.test.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/src/__tests__/heartbeat-retry-scheduling.test.ts b/server/src/__tests__/heartbeat-retry-scheduling.test.ts
index 3dada976b57..b6b197f1485 100644
--- a/server/src/__tests__/heartbeat-retry-scheduling.test.ts
+++ b/server/src/__tests__/heartbeat-retry-scheduling.test.ts
@@ -219,6 +219,7 @@ describeEmbeddedPostgres("heartbeat bounded retry scheduling", () => {
     expect(failedRun?.status).toBe("failed");
     expect(failedRun?.errorCode).toBe("provider_quota");
     expect((failedRun?.resultJson as Record<string, unknown> | null)?.errorFamily).toBe("provider_quota");
+    await heartbeat.waitForRunExecutionDrain(run!.id);
 
     await expect
       .poll(