Stabilize heartbeat comment batching assertion

2026-04-07 09:38:08 -05:00
parent 25d308186d
commit a0333f3e9d
1 changed files with 6 additions and 1 deletions
@@ -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