test flake bft deliverer

Signed-off-by: Yoav Tock <tock@il.ibm.com>
hyperledger · May 6, 2024 · f074697 · f074697
1 parent 906acba
commit f074697
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/common/deliverclient/blocksprovider/bft_deliverer_test.go b/common/deliverclient/blocksprovider/bft_deliverer_test.go
@@ -1113,18 +1113,17 @@ func TestBFTDeliverer_CensorshipMonitorEvents(t *testing.T) {
 				setup.monErrC <- &blocksprovider.ErrCensorship{Message: fmt.Sprintf("censorship %d", n)}
 			}()
 
-			setup.gWithT.Eventually(
-				func() int {
-					setup.mutex.Lock()
-					defer setup.mutex.Unlock()
+			numMon := func() int {
+				setup.mutex.Lock()
+				defer setup.mutex.Unlock()
 
-					return len(setup.monitorSet)
-				}).Should(Equal(n + 1))
+				return len(setup.monitorSet)
+			}
+			setup.gWithT.Eventually(numMon, 10*time.Second, 10*time.Millisecond).Should(Equal(n + 1))
 
 			setup.gWithT.Eventually(setup.fakeDialer.DialCallCount).Should(Equal(n + 1))
 			setup.gWithT.Expect(setup.fakeSleeper.SleepCallCount()).To(Equal(n))
 			setup.gWithT.Eventually(setup.fakeCensorshipMonFactory.CreateCallCount).Should(Equal(n + 1))
-
 		}
 
 		t.Log("Exponential backoff after every round, with saturation")