fix tests

blocks/part_set_test.go

@@ -0,0 +1,86 @@
+package blocks
+
+import (
+	"bytes"
+	"io/ioutil"
+	"testing"
+
+	. "github.com/tendermint/tendermint/common"
+)
+
+func TestBasicPartSet(t *testing.T) {
+
+	// Construct random data of size partSize * 100
+	data := RandBytes(partSize * 100)
+
+	partSet := NewPartSetFromData(data)
+	if len(partSet.Hash()) == 0 {
+		t.Error("Expected to get hash")
+	}
+	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 := NewPartSetFromHeader(PartSetHeader{partSet.Hash(), partSet.Total()})
+
+	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.Hash(), partSet2.Hash()) {
+		t.Error("Expected to get same hash")
+	}
+	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 := NewPartSetFromHeader(PartSetHeader{partSet.Hash(), partSet.Total()})
+
+	// 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.")
+	}
+
+}