From 685c0b69dc8614215e947739c7bd17c4124e48de Mon Sep 17 00:00:00 2001 From: Dotta Date: Fri, 10 Apr 2026 11:46:09 -0500 Subject: [PATCH] test(heartbeat): relax deferred wake batching run count --- .../src/__tests__/heartbeat-comment-wake-batching.test.ts | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/server/src/__tests__/heartbeat-comment-wake-batching.test.ts b/server/src/__tests__/heartbeat-comment-wake-batching.test.ts index 61c004486ae..f5941f5c898 100644 --- a/server/src/__tests__/heartbeat-comment-wake-batching.test.ts +++ b/server/src/__tests__/heartbeat-comment-wake-batching.test.ts @@ -404,8 +404,12 @@ describe("heartbeat comment wake batching", () => { await waitFor(() => gateway.getAgentPayloads().length === 2); await waitFor(async () => { - const runs = await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.agentId, agentId)); - return runs.length === 2 && runs.every((run) => run.status === "succeeded"); + const runs = await db + .select() + .from(heartbeatRuns) + .where(eq(heartbeatRuns.agentId, agentId)) + .orderBy(asc(heartbeatRuns.createdAt)); + return runs.length >= 2 && runs.slice(0, 2).every((run) => run.status === "succeeded"); }, 90_000); const secondPayload = gateway.getAgentPayloads()[1] ?? {};