improve check for EnsureNoStaleBallots to ignore Pending Ballots

zeta-chain · Mar 4, 2025 · 2a44a8b · 2a44a8b
1 parent f104014
commit 2a44a8b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/e2e/runner/require.go b/e2e/runner/require.go
@@ -74,6 +74,12 @@ func (r *E2ERunner) EnsureNoStaleBallots() {
 	if len(ballotsRes.Ballots) < 1 {
 		return
 	}
-	firstBallotCreationHeight := ballotsRes.Ballots[0].BallotCreationHeight
+	firstBallotCreationHeight := int64(0)
+
+	for _, ballot := range ballotsRes.Ballots {
+		if ballot.IsFinalized() {
+			firstBallotCreationHeight = ballot.BallotCreationHeight
+		}
+	}
 	require.GreaterOrEqual(r, firstBallotCreationHeight, staleBlockStart, "there should be no stale ballots")
 }