POL tests

consensus/pol_test.go

@@ -0,0 +1,196 @@
+package consensus
+
+import (
+	. "github.com/tendermint/tendermint/blocks"
+	. "github.com/tendermint/tendermint/common"
+
+	"bytes"
+	"testing"
+)
+
+// NOTE: see consensus/test.go for common test methods.
+
+func TestVerifyVotes(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 0, 10, 1)
+
+	// Make a POL with -2/3 votes.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 0, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 0, Type: VoteTypeBare, BlockHash: blockHash,
+	}
+	for i := 0; i < 6; i++ {
+		privAccounts[i].Sign(vote)
+		pol.Votes = append(pol.Votes, vote.Signature)
+	}
+
+	// Check that validation fails.
+	if err := pol.Verify(valSet); err == nil {
+		t.Errorf("Expected POL.Verify() to fail, not enough votes.")
+	}
+
+	// Make a POL with +2/3 votes.
+	privAccounts[7].Sign(vote)
+	pol.Votes = append(pol.Votes, vote.Signature)
+
+	// Check that validation succeeds.
+	if err := pol.Verify(valSet); err != nil {
+		t.Errorf("Expected POL.Verify() to succeed")
+	}
+}
+
+func TestVerifyInvalidVote(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 0, 10, 1)
+
+	// Make a POL with +2/3 votes with the wrong signature.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 0, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 0, Type: VoteTypeBare, BlockHash: blockHash,
+	}
+	for i := 0; i < 7; i++ {
+		privAccounts[i].Sign(vote)
+		// Mutate the signature.
+		vote.Signature.Bytes[0] += byte(0x01)
+		pol.Votes = append(pol.Votes, vote.Signature)
+	}
+
+	// Check that validation fails.
+	if err := pol.Verify(valSet); err == nil {
+		t.Errorf("Expected POL.Verify() to fail, wrong signatures.")
+	}
+}
+
+func TestVerifyCommits(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 2, 10, 1)
+
+	// Make a POL with +2/3 votes.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 2, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 1, Type: VoteTypeCommit, BlockHash: blockHash,
+	}
+	for i := 0; i < 7; i++ {
+		privAccounts[i].Sign(vote)
+		pol.Commits = append(pol.Commits, vote.Signature)
+		pol.CommitRounds = append(pol.CommitRounds, 1)
+	}
+
+	// Check that validation succeeds.
+	if err := pol.Verify(valSet); err != nil {
+		t.Errorf("Expected POL.Verify() to succeed")
+	}
+}
+
+func TestVerifyInvalidCommits(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 2, 10, 1)
+
+	// Make a POL with +2/3 votes with the wrong signature.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 2, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 1, Type: VoteTypeCommit, BlockHash: blockHash,
+	}
+	for i := 0; i < 7; i++ {
+		privAccounts[i].Sign(vote)
+		// Mutate the signature.
+		vote.Signature.Bytes[0] += byte(0x01)
+		pol.Commits = append(pol.Commits, vote.Signature)
+		pol.CommitRounds = append(pol.CommitRounds, 1)
+	}
+
+	// Check that validation fails.
+	if err := pol.Verify(valSet); err == nil {
+		t.Errorf("Expected POL.Verify() to fail, wrong signatures.")
+	}
+}
+
+func TestVerifyInvalidCommitRounds(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 2, 10, 1)
+
+	// Make a POL with +2/3 commits for the current round.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 2, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 2, Type: VoteTypeCommit, BlockHash: blockHash,
+	}
+	for i := 0; i < 7; i++ {
+		privAccounts[i].Sign(vote)
+		pol.Commits = append(pol.Commits, vote.Signature)
+		pol.CommitRounds = append(pol.CommitRounds, 2)
+	}
+
+	// Check that validation fails.
+	if err := pol.Verify(valSet); err == nil {
+		t.Errorf("Expected POL.Verify() to fail, same round.")
+	}
+}
+
+func TestVerifyInvalidCommitRounds2(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 2, 10, 1)
+
+	// Make a POL with +2/3 commits for future round.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 2, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 3, Type: VoteTypeCommit, BlockHash: blockHash,
+	}
+	for i := 0; i < 7; i++ {
+		privAccounts[i].Sign(vote)
+		pol.Commits = append(pol.Commits, vote.Signature)
+		pol.CommitRounds = append(pol.CommitRounds, 3)
+	}
+
+	// Check that validation fails.
+	if err := pol.Verify(valSet); err == nil {
+		t.Errorf("Expected POL.Verify() to fail, future round.")
+	}
+}
+
+func TestReadWrite(t *testing.T) {
+	_, valSet, privAccounts := makeVoteSet(0, 0, 10, 1)
+
+	// Make a POL with +2/3 votes.
+	blockHash := RandBytes(32)
+	pol := &POL{
+		Height: 0, Round: 0, BlockHash: blockHash,
+	}
+	vote := &Vote{
+		Height: 0, Round: 0, Type: VoteTypeBare, BlockHash: blockHash,
+	}
+	for i := 0; i < 7; i++ {
+		privAccounts[i].Sign(vote)
+		pol.Votes = append(pol.Votes, vote.Signature)
+	}
+
+	// Write it to a buffer.
+	buf := new(bytes.Buffer)
+	_, err := pol.WriteTo(buf)
+	if err != nil {
+		t.Fatalf("Failed to write POL")
+	}
+
+	// Read from buffer.
+	var n int64
+	pol2 := ReadPOL(buf, &n, &err)
+	if err != nil {
+		t.Fatalf("Failed to read POL")
+	}
+
+	// Check that validation succeeds.
+	if err := pol2.Verify(valSet); err != nil {
+		t.Errorf("Expected POL.Verify() to succeed")
+	}
+}