fix test

Signed-off-by: Hagar Meir <[email protected]>

Author: HagarMeir

internal/bft/view_test.go

@@ -462,7 +462,10 @@ func TestTwoSequences(t *testing.T) {
 	assert.NoError(t, err)
 	log := basicLog.Sugar()
 	comm := &mocks.Comm{}
-	comm.On("Broadcast", mock.Anything)
+	commWG := sync.WaitGroup{}
+	comm.On("Broadcast", mock.Anything).Run(func(args mock.Arguments) {
+		commWG.Done()
+	})
 	decider := &mocks.Decider{}
 	deciderWG := sync.WaitGroup{}
 	decidedProposal := make(chan types.Proposal, 1)
@@ -496,16 +499,20 @@ func TestTwoSequences(t *testing.T) {
 	}
 	end := view.Start()
 
+	commWG.Add(1)
 	view.HandleMessage(1, prePrepare)
+	commWG.Wait()
 
 	prepareNext := proto.Clone(prepare).(*protos.Message)
 	prepareNextGet := prepareNext.GetPrepare()
 	prepareNextGet.Seq = 1
 
+	commWG.Add(1)
 	view.HandleMessage(1, prepare)
 	view.HandleMessage(1, prepareNext)
 	view.HandleMessage(2, prepare)
 	view.HandleMessage(2, prepareNext)
+	commWG.Wait()
 
 	commit1Next := proto.Clone(commit1).(*protos.Message)
 	commit1NextGet := commit1Next.GetCommit()
@@ -525,7 +532,9 @@ func TestTwoSequences(t *testing.T) {
 	prePrepareNextGet := prePrepareNext.GetPrePrepare()
 	prePrepareNextGet.Seq = 1
 
+	commWG.Add(2)
 	view.HandleMessage(1, prePrepareNext)
+	commWG.Wait()
 
 	deciderWG.Wait()
 	dProp := <-decidedProposal