WIP

Author: infrmtcs

.github/workflows/juno-test.yml

@@ -18,6 +18,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, macos-latest, ubuntu-arm64-4-core]
+        iteration: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
     runs-on: ${{ matrix.os }}
     env:
       VM_DEBUG: true

consensus/integtest/integ_test.go

@@ -2,10 +2,12 @@ package integtest
 
 import (
 	"fmt"
+	goitre "iter"
 	"maps"
 	"slices"
 	gosync "sync"
 	"testing"
+	"time"
 
 	"github.com/NethermindEth/juno/blockchain"
 	"github.com/NethermindEth/juno/consensus"
@@ -28,8 +30,8 @@ import (
 
 const (
 	commitBufferSize = 1024
-	maxLag           = 10
 	hostAddress      = "/ip4/0.0.0.0/tcp/0"
+	maxCommitWait    = 30 * time.Second
 )
 
 var (
@@ -169,6 +171,21 @@ func writeBlock(
 	}
 }
 
+func commitStream(t *testing.T, commits chan commit) goitre.Seq[commit] {
+	return func(yield func(commit) bool) {
+		for {
+			select {
+			case commit := <-commits:
+				if !yield(commit) {
+					return
+				}
+			case <-time.After(maxCommitWait):
+				require.FailNow(t, "timed out waiting for commit")
+			}
+		}
+	}
+}
+
 func assertCommits(t *testing.T, commits chan commit, cfg testConfig, logger *utils.ZapLogger) {
 	t.Helper()
 
@@ -182,22 +199,23 @@ func assertCommits(t *testing.T, commits chan commit, cfg testConfig, logger *ut
 	commitCount := make(map[uint64]int)
 
 	nextHeight := uint64(1)
-	for commit := range commits {
+	for commit := range commitStream(t, commits) {
 		blockNumber := commit.committedBlock.Block.Number
 		blockHash := commit.committedBlock.Block.Hash
-		// Ignore commits after the target height
-		if blockNumber > cfg.targetHeight {
-			continue
-		}
 
 		require.Falsef(
 			t,
-			blockNumber > nextHeight+maxLag,
+			blockNumber > nextHeight+cfg.targetHeight,
 			"finished height %d is too far behind committed height %d",
 			nextHeight,
 			blockNumber,
 		)
 
+		// Ignore commits after the target height
+		if blockNumber > cfg.targetHeight {
+			continue
+		}
+
 		// Ignore faulty nodes
 		if commit.nodeIndex >= honestNodeCount {
 			continue
@@ -218,12 +236,12 @@ func assertCommits(t *testing.T, commits chan commit, cfg testConfig, logger *ut
 		commitCount[blockNumber]++
 
 		// If all honest nodes committed at this height, increment the finished counter
-		for commitCount[nextHeight] == honestNodeCount && nextHeight < cfg.targetHeight {
+		for commitCount[nextHeight] == honestNodeCount && nextHeight <= cfg.targetHeight {
 			logger.Infow("all honest nodes committed", "height", nextHeight)
 			nextHeight++
 		}
 
-		if nextHeight == cfg.targetHeight {
+		if nextHeight > cfg.targetHeight {
 			break
 		}
 	}