neo-go/pkg/network/payload/headers_test.go

package payload

import (
	"bytes"
	"reflect"
	"testing"

	"github.com/CityOfZion/neo-go/pkg/core"
)

func TestHeadersEncodeDecode(t *testing.T) {
	headers := &Headers{[]*core.Header{
		&core.Header{
			BlockBase: core.BlockBase{
				Version: 0,
				Index:   1,
				Script: &core.Witness{
					InvocationScript:   []byte{0x0},
					VerificationScript: []byte{0x1},
				},
			}},
		&core.Header{
			BlockBase: core.BlockBase{
				Version: 0,
				Index:   2,
				Script: &core.Witness{
					InvocationScript:   []byte{0x0},
					VerificationScript: []byte{0x1},
				},
			}},
		&core.Header{
			BlockBase: core.BlockBase{
				Version: 0,
				Index:   3,
				Script: &core.Witness{
					InvocationScript:   []byte{0x0},
					VerificationScript: []byte{0x1},
				},
			}},
	}}

	buf := new(bytes.Buffer)
	if err := headers.EncodeBinary(buf); err != nil {
		t.Fatal(err)
	}

	headersDecode := &Headers{}
	if err := headersDecode.DecodeBinary(buf); err != nil {
		t.Fatal(err)
	}

	if !reflect.DeepEqual(headers, headersDecode) {
		t.Fatalf("expected both header payload to be equal %+v and %+v", headers, headersDecode)
	}
}