fix unit tests

multiversx · Jan 28, 2025 · 5ca3486 · 5ca3486
1 parent 943c287
commit 5ca3486
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/consensus/spos/bls/v2/subroundEndRound_test.go b/consensus/spos/bls/v2/subroundEndRound_test.go
@@ -1181,8 +1181,13 @@ func TestSubroundEndRound_DoEndRoundJobByNode(t *testing.T) {
 		t.Parallel()
 
 		container := consensusMocks.InitConsensusCore()
+		numCalls := 0
 		container.SetEquivalentProofsPool(&dataRetriever.ProofsPoolMock{
 			HasProofCalled: func(shardID uint32, headerHash []byte) bool {
+				if numCalls <= 1 {
+					numCalls++
+					return false
+				}
 				return true
 			},
 		})