package consensus import ( "testing" . "github.com/tendermint/go-common" "github.com/tendermint/tendermint/config/tendermint_test" "github.com/tendermint/tendermint/types" ) func init() { config = tendermint_test.ResetConfig("consensus_height_vote_set_test") } func TestPeerCatchupRounds(t *testing.T) { valSet, privVals := types.RandValidatorSet(10, 1) hvs := NewHeightVoteSet(config.GetString("chain_id"), 1, valSet) vote999_0 := makeVoteHR(t, 1, 999, privVals[0]) added, _, err := hvs.AddByIndex(0, vote999_0, "peer1") if !added || err != nil { t.Error("Expected to successfully add vote from peer", added, err) } vote1000_0 := makeVoteHR(t, 1, 1000, privVals[0]) added, _, err = hvs.AddByIndex(0, vote1000_0, "peer1") if added { t.Error("Expected to *not* add vote from peer, too many catchup rounds.") } added, _, err = hvs.AddByIndex(0, vote1000_0, "peer2") if !added || err != nil { t.Error("Expected to successfully add vote from another peer") } } func makeVoteHR(t *testing.T, height, round int, privVal *types.PrivValidator) *types.Vote { vote := &types.Vote{ Height: height, Round: round, Type: types.VoteTypePrecommit, BlockHash: []byte("fakehash"), } chainID := config.GetString("chain_id") err := privVal.SignVote(chainID, vote) if err != nil { panic(Fmt("Error signing vote: %v", err)) return nil } return vote }