get uts to compile

internal/consensus/byzantine_test.go

@@ -183,16 +183,19 @@ func TestByzantinePrevoteEquivocation(t *testing.T) {
 		}
 
 		var extCommit *types.ExtendedCommit
+		lc, ok := lazyProposer.LastCommit.(*types.VoteSet)
 		switch {
 		case lazyProposer.Height == lazyProposer.state.InitialHeight:
 			// We're creating a proposal for the first block.
 			// The commit is empty, but not nil.
 			extCommit = &types.ExtendedCommit{}
-		case lazyProposer.LastCommit.HasTwoThirdsMajority():
+		case ok && lc.HasTwoThirdsMajority():
 			// Make the commit from LastCommit
 			// Vote extensions are enabled by default for test units
 			veHeightParam := lazyProposer.state.ConsensusParams.Feature
-			extCommit = lazyProposer.LastCommit.MakeExtendedCommit(veHeightParam)
+			if lastC, ok := lazyProposer.LastCommit.(*types.VoteSet); ok {
+				extCommit = lastC.MakeExtendedCommit(veHeightParam)
+			}
 		default: // This shouldn't happen.
 			lazyProposer.Logger.Error("enterPropose: Cannot propose anything: No commit for the previous block")
 			return

internal/consensus/common_test.go

@@ -331,11 +331,14 @@ func validateLastPrecommit(t *testing.T, cs *State, privVal *validatorStub, bloc
 	require.NoError(t, err)
 	address := pv.Address()
 	var vote *types.Vote
-	if vote = votes.GetByAddress(address); vote == nil {
-		panic("Failed to find precommit from validator")
-	}
-	if !bytes.Equal(vote.BlockID.Hash, blockHash) {
-		panic(fmt.Sprintf("Expected precommit to be for %X, got %X", blockHash, vote.BlockID.Hash))
+	if vs, ok := votes.(*types.VoteSet); ok {
+
+		if vote = vs.GetByAddress(address); vote == nil {
+			panic("Failed to find precommit from validator")
+		}
+		if !bytes.Equal(vote.BlockID.Hash, blockHash) {
+			panic(fmt.Sprintf("Expected precommit to be for %X, got %X", blockHash, vote.BlockID.Hash))
+		}
 	}
 }