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

package payload

import (
	"bytes"
	"reflect"
	"testing"

	"github.com/CityOfZion/neo-go/pkg/crypto/hash"
	. "github.com/CityOfZion/neo-go/pkg/util"
)

func TestInventoryEncodeDecode(t *testing.T) {
	hashes := []Uint256{
		hash.Sha256([]byte("a")),
		hash.Sha256([]byte("b")),
	}
	inv := NewInventory(BlockType, hashes)

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

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

	if !reflect.DeepEqual(inv, invDecode) {
		t.Fatalf("expected both inventories to be equal %v and %v", inv, invDecode)
	}
}