From b55713eda14f4f00c7e5c472ac384c26ac57565a Mon Sep 17 00:00:00 2001 From: Dotta Date: Tue, 5 May 2026 13:31:59 -0500 Subject: [PATCH] Stabilize successful handoff liveness assertion --- server/src/__tests__/heartbeat-process-recovery.test.ts | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/server/src/__tests__/heartbeat-process-recovery.test.ts b/server/src/__tests__/heartbeat-process-recovery.test.ts index a8d8e213a5a..b8278fdcb40 100644 --- a/server/src/__tests__/heartbeat-process-recovery.test.ts +++ b/server/src/__tests__/heartbeat-process-recovery.test.ts @@ -1416,9 +1416,14 @@ describeEmbeddedPostgres("heartbeat orphaned process recovery", () => { return matches.length > 0 ? matches : null; }, 5_000); await waitForHeartbeatIdle(db, 5_000); + const classifiedRun = await db + .select() + .from(heartbeatRuns) + .where(eq(heartbeatRuns.id, runId)) + .then((rows) => rows[0] ?? null); - expect(settledRun?.status).toBe("succeeded"); - expect(settledRun?.livenessState).toBe("advanced"); + expect(classifiedRun?.status ?? settledRun?.status).toBe("succeeded"); + expect(classifiedRun?.livenessState).toBe("advanced"); expect(handoffWakeups).toHaveLength(1); expect(handoffWakeups[0]?.idempotencyKey).toBe(`finish_successful_run_handoff:${issueId}:${runId}:1`);