From a0333f3e9d27823cf0bc6bdb4d434b8a4ce9d469 Mon Sep 17 00:00:00 2001 From: dotta Date: Tue, 7 Apr 2026 09:38:08 -0500 Subject: [PATCH] Stabilize heartbeat comment batching assertion --- .../src/__tests__/heartbeat-comment-wake-batching.test.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/server/src/__tests__/heartbeat-comment-wake-batching.test.ts b/server/src/__tests__/heartbeat-comment-wake-batching.test.ts index 8f5c250b..7ac77ec4 100644 --- a/server/src/__tests__/heartbeat-comment-wake-batching.test.ts +++ b/server/src/__tests__/heartbeat-comment-wake-batching.test.ts @@ -495,7 +495,12 @@ describe("heartbeat comment wake batching", () => { .from(heartbeatRuns) .where(eq(heartbeatRuns.agentId, agentId)) .orderBy(asc(heartbeatRuns.createdAt)); - return runs.length === 2 && runs.every((run) => run.status === "succeeded"); + return ( + runs.length === 2 && + runs.every((run) => run.status === "succeeded") && + runs[0]?.issueCommentStatus === "retry_queued" && + runs[1]?.issueCommentStatus === "retry_exhausted" + ); }); const runs = await db