Files
lightwalletd/parser/block_test.go
2018-09-17 13:45:21 -04:00

95 lines
2.0 KiB
Go

package parser
import (
"bufio"
"bytes"
"encoding/hex"
"os"
"testing"
"github.com/gtank/ctxd/parser/internal/bytestring"
)
func TestBlockHeader(t *testing.T) {
testBlocks, err := os.Open("testdata/blocks")
if err != nil {
t.Fatal(err)
}
defer testBlocks.Close()
lastBlockTime := uint32(0)
scan := bufio.NewScanner(testBlocks)
for scan.Scan() {
blockDataHex := scan.Text()
decodedBlockData, err := hex.DecodeString(blockDataHex)
if err != nil {
t.Error(err)
continue
}
s := bytestring.String(decodedBlockData)
startLength := len(s)
dec := NewBlockHeaderDecoder(&s)
blockHeader := BlockHeader{}
err = dec.Decode(&blockHeader)
if err != nil {
t.Error(err)
continue
}
if (startLength - len(s)) != SER_BLOCK_HEADER_SIZE {
t.Error("did not advance underlying bytestring")
}
// Some basic sanity checks
if blockHeader.Version != 4 {
t.Error("Read wrong version in a test block.")
break
}
if blockHeader.Time < lastBlockTime {
t.Error("Block times not increasing.")
break
}
lastBlockTime = blockHeader.Time
if blockHeader.SolutionSize.Size != 1344 {
t.Error("Got wrong Equihash solution size.")
break
}
// Re-serialize and check for consistency
serializedHeader, err := blockHeader.MarshalBinary()
if err != nil {
t.Errorf("Error serializing header: %v", err)
break
}
if !bytes.Equal(serializedHeader, decodedBlockData[:SER_BLOCK_HEADER_SIZE]) {
offset := 0
length := 0
for i := 0; i < SER_BLOCK_HEADER_SIZE; i++ {
if serializedHeader[i] != decodedBlockData[i] {
if offset == 0 {
offset = i
}
length++
}
}
t.Errorf("Block header failed round-trip serialization:\nwant\n%x\ngot\n%x\nat %d", serializedHeader[offset:offset+length], decodedBlockData[offset:offset+length], offset)
break
}
hash := blockHeader.GetHash()
// This is not necessarily true for anything but our current test cases.
for _, b := range hash[28:] {
if b != 0 {
t.Errorf("Hash lacked trailing zeros")
}
}
}
}