90 lines
1.8 KiB
Go
90 lines
1.8 KiB
Go
// stm: #unit
|
|
package vectors
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
)
|
|
|
|
func LoadVector(t *testing.T, f string, out interface{}) {
|
|
p := filepath.Join("../../extern/serialization-vectors", f)
|
|
fi, err := os.Open(p)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer fi.Close() //nolint:errcheck
|
|
|
|
if err := json.NewDecoder(fi).Decode(out); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestBlockHeaderVectors(t *testing.T) {
|
|
//stm: @CHAIN_TYPES_SERIALIZATION_BLOCK_001
|
|
var headers []HeaderVector
|
|
LoadVector(t, "block_headers.json", &headers)
|
|
|
|
for i, hv := range headers {
|
|
if hv.Block.Cid().String() != hv.Cid {
|
|
t.Fatalf("CID mismatch in test vector %d", i)
|
|
}
|
|
|
|
data, err := hv.Block.Serialize()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if fmt.Sprintf("%x", data) != hv.CborHex {
|
|
t.Fatalf("serialized data mismatched for test vector %d", i)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMessageSigningVectors(t *testing.T) {
|
|
//stm: @CHAIN_TYPES_SERIALIZATION_SIGNED_MESSAGE_001
|
|
var msvs []MessageSigningVector
|
|
LoadVector(t, "message_signing.json", &msvs)
|
|
|
|
for i, msv := range msvs {
|
|
smsg := &types.SignedMessage{
|
|
Message: *msv.Unsigned,
|
|
Signature: *msv.Signature,
|
|
}
|
|
|
|
if smsg.Cid().String() != msv.Cid {
|
|
t.Fatalf("cid of message in vector %d mismatches", i)
|
|
}
|
|
|
|
// TODO: check signature
|
|
}
|
|
}
|
|
|
|
func TestUnsignedMessageVectors(t *testing.T) {
|
|
//stm: @CHAIN_TYPES_SERIALIZATION_MESSAGE_001
|
|
var msvs []UnsignedMessageVector
|
|
LoadVector(t, "unsigned_messages.json", &msvs)
|
|
|
|
for i, msv := range msvs {
|
|
b, err := msv.Message.Serialize()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
dec, err := hex.DecodeString(msv.HexCbor)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !bytes.Equal(b, dec) {
|
|
t.Fatalf("serialization vector %d mismatches bytes", i)
|
|
}
|
|
}
|
|
}
|