You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
3.0 KiB
101 lines
3.0 KiB
package types
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type voteData struct {
|
|
vote1 *Vote
|
|
vote2 *Vote
|
|
valid bool
|
|
}
|
|
|
|
func makeVote(val PrivValidator, chainID string, valIndex int, height int64, round, step int, blockID BlockID) *Vote {
|
|
v := &Vote{
|
|
ValidatorAddress: val.GetAddress(),
|
|
ValidatorIndex: valIndex,
|
|
Height: height,
|
|
Round: round,
|
|
Type: byte(step),
|
|
BlockID: blockID,
|
|
}
|
|
err := val.SignVote(chainID, v)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return v
|
|
}
|
|
|
|
func TestEvidence(t *testing.T) {
|
|
val := NewMockPV()
|
|
val2 := NewMockPV()
|
|
blockID := makeBlockID("blockhash", 1000, "partshash")
|
|
blockID2 := makeBlockID("blockhash2", 1000, "partshash")
|
|
blockID3 := makeBlockID("blockhash", 10000, "partshash")
|
|
blockID4 := makeBlockID("blockhash", 10000, "partshash2")
|
|
|
|
const chainID = "mychain"
|
|
|
|
vote1 := makeVote(val, chainID, 0, 10, 2, 1, blockID)
|
|
badVote := makeVote(val, chainID, 0, 10, 2, 1, blockID)
|
|
err := val2.SignVote(chainID, badVote)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
cases := []voteData{
|
|
{vote1, makeVote(val, chainID, 0, 10, 2, 1, blockID2), true}, // different block ids
|
|
{vote1, makeVote(val, chainID, 0, 10, 2, 1, blockID3), true},
|
|
{vote1, makeVote(val, chainID, 0, 10, 2, 1, blockID4), true},
|
|
{vote1, makeVote(val, chainID, 0, 10, 2, 1, blockID), false}, // wrong block id
|
|
{vote1, makeVote(val, "mychain2", 0, 10, 2, 1, blockID2), false}, // wrong chain id
|
|
{vote1, makeVote(val, chainID, 1, 10, 2, 1, blockID2), false}, // wrong val index
|
|
{vote1, makeVote(val, chainID, 0, 11, 2, 1, blockID2), false}, // wrong height
|
|
{vote1, makeVote(val, chainID, 0, 10, 3, 1, blockID2), false}, // wrong round
|
|
{vote1, makeVote(val, chainID, 0, 10, 2, 2, blockID2), false}, // wrong step
|
|
{vote1, makeVote(val2, chainID, 0, 10, 2, 1, blockID), false}, // wrong validator
|
|
{vote1, badVote, false}, // signed by wrong key
|
|
}
|
|
|
|
pubKey := val.GetPubKey()
|
|
for _, c := range cases {
|
|
ev := &DuplicateVoteEvidence{
|
|
VoteA: c.vote1,
|
|
VoteB: c.vote2,
|
|
}
|
|
if c.valid {
|
|
assert.Nil(t, ev.Verify(chainID, pubKey), "evidence should be valid")
|
|
} else {
|
|
assert.NotNil(t, ev.Verify(chainID, pubKey), "evidence should be invalid")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDuplicatedVoteEvidence(t *testing.T) {
|
|
ev := randomDuplicatedVoteEvidence()
|
|
|
|
assert.True(t, ev.Equal(ev))
|
|
assert.False(t, ev.Equal(&DuplicateVoteEvidence{}))
|
|
}
|
|
|
|
func TestEvidenceList(t *testing.T) {
|
|
ev := randomDuplicatedVoteEvidence()
|
|
evl := EvidenceList([]Evidence{ev})
|
|
|
|
assert.NotNil(t, evl.Hash())
|
|
assert.True(t, evl.Has(ev))
|
|
assert.False(t, evl.Has(&DuplicateVoteEvidence{}))
|
|
}
|
|
|
|
func randomDuplicatedVoteEvidence() *DuplicateVoteEvidence {
|
|
val := NewMockPV()
|
|
blockID := makeBlockID("blockhash", 1000, "partshash")
|
|
blockID2 := makeBlockID("blockhash2", 1000, "partshash")
|
|
const chainID = "mychain"
|
|
return &DuplicateVoteEvidence{
|
|
VoteA: makeVote(val, chainID, 0, 10, 2, 1, blockID),
|
|
VoteB: makeVote(val, chainID, 0, 10, 2, 1, blockID2),
|
|
}
|
|
}
|