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(partSet.Header()) 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(partSet.Header()) // 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.") } }