Add PartSet test.

consensus/part_set.go

@@ -4,7 +4,9 @@ import (
 	"bytes"
 	"crypto/sha256"
 	"errors"
+	"fmt"
 	"io"
+	"strings"
 	"sync"
 
 	. "github.com/tendermint/tendermint/binary"
@@ -59,6 +61,26 @@ func (pt *Part) Hash() []byte {
 	}
 }
 
+func (pt *Part) String() string {
+	return pt.StringWithIndent("")
+}
+
+func (pt *Part) StringWithIndent(indent string) string {
+	trailStrings := make([]string, len(pt.Trail))
+	for i, hash := range pt.Trail {
+		trailStrings[i] = fmt.Sprintf("%X", hash)
+	}
+	return fmt.Sprintf(`Part{
+%s  Index: %v
+%s  Trail:
+%s    %v
+%s}`,
+		indent, pt.Index,
+		indent,
+		indent, strings.Join(trailStrings, "\n"+indent+"    "),
+		indent)
+}
+
 //-------------------------------------
 
 type PartSet struct {

consensus/part_set_test.go

@@ -1,9 +1,86 @@
 package consensus
 
 import (
+	"bytes"
+	"io/ioutil"
 	"testing"
+
+	. "github.com/tendermint/tendermint/common"
 )
 
 func TestBasicPartSet(t *testing.T) {
-	// XXX this is fun!
+
+	// Construct random data of size partSize * 100
+	data := RandBytes(partSize * 100)
+
+	partSet := NewPartSetFromData(data)
+	if len(partSet.RootHash()) == 0 {
+		t.Error("Expected to get rootHash")
+	}
+	if partSet.Total() != 100 {
+		t.Errorf("Expected to get 100 parts, but got %v", partSet.Total())
+	}
+	if !partSet.IsComplete() {
+		t.Errorf("PartSet should be complete")
+	}
+
+	// Test adding parts to a new partSet.
+	partSet2 := NewPartSetFromMetadata(partSet.Total(), partSet.RootHash())
+
+	for i := uint16(0); i < partSet.Total(); i++ {
+		part := partSet.GetPart(i)
+		//t.Logf("\n%v", part)
+		added, err := partSet2.AddPart(part)
+		if !added || err != nil {
+			t.Errorf("Failed to add part %v, error: %v", i, err)
+		}
+	}
+
+	if !bytes.Equal(partSet.RootHash(), partSet2.RootHash()) {
+		t.Error("Expected to get same rootHash")
+	}
+	if partSet2.Total() != 100 {
+		t.Errorf("Expected to get 100 parts, but got %v", partSet2.Total())
+	}
+	if !partSet2.IsComplete() {
+		t.Errorf("Reconstructed PartSet should be complete")
+	}
+
+	// Reconstruct data, assert that they are equal.
+	data2Reader := partSet2.GetReader()
+	data2, err := ioutil.ReadAll(data2Reader)
+	if err != nil {
+		t.Errorf("Error reading data2Reader: %v", err)
+	}
+	if !bytes.Equal(data, data2) {
+		t.Errorf("Got wrong data.")
+	}
+
+}
+
+func TestWrongTrail(t *testing.T) {
+
+	// Construct random data of size partSize * 100
+	data := RandBytes(partSize * 100)
+	partSet := NewPartSetFromData(data)
+
+	// Test adding a part with wrong data.
+	partSet2 := NewPartSetFromMetadata(partSet.Total(), partSet.RootHash())
+
+	// Test adding a part with wrong trail.
+	part := partSet.GetPart(0)
+	part.Trail[0][0] += byte(0x01)
+	added, err := partSet2.AddPart(part)
+	if added || err == nil {
+		t.Errorf("Expected to fail adding a part with bad trail.")
+	}
+
+	// Test adding a part with wrong bytes.
+	part = partSet.GetPart(1)
+	part.Bytes[0] += byte(0x01)
+	added, err = partSet2.AddPart(part)
+	if added || err == nil {
+		t.Errorf("Expected to fail adding a part with bad bytes.")
+	}
+
 }