fix consensus tests

consensus/reactor.go

@@ -986,7 +986,7 @@ func (conR *Reactor) peerStatsRoutine() {
 				if numVotes := ps.RecordVote(); numVotes%votesToContributeToBecomeGoodPeer == 0 {
 					conR.Switch.MarkPeerAsGood(peer)
 				}
-			case *BlockPartMessage:
+			case *BlockPartMessage, *CompactBlockMessage:
 				if numParts := ps.RecordBlockPart(); numParts%blocksToContributeToBecomeGoodPeer == 0 {
 					conR.Switch.MarkPeerAsGood(peer)
 				}
@@ -1368,14 +1368,6 @@ func (ps *PeerState) BlockPartsSent() int {
 	return ps.Stats.BlockParts
 }
 
-// HasBlock returns true if the peer has received a valid block for the current round
-func (ps *PeerState) HasBlock() bool {
-	ps.mtx.Lock()
-	defer ps.mtx.Unlock()
-
-	return ps.PRS.Block
-}
-
 // SetHasVote sets the given vote as known by the peer
 func (ps *PeerState) SetHasVote(vote *types.Vote) {
 	ps.mtx.Lock()

consensus/reactor_test.go

@@ -367,7 +367,7 @@ func TestReactorRecordsVotesAndBlockParts(t *testing.T) {
 	ps := peer.Get(types.PeerStateKey).(*PeerState)
 
 	assert.Equal(t, true, ps.VotesSent() > 0, "number of votes sent should have increased")
-	assert.Equal(t, true, ps.HasBlock(), "should have block")
+	assert.Equal(t, true, ps.BlockPartsSent(), "should have block")
 }
 
 //-------------------------------------------------------------