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

40 lines
893 B
Go
Raw Normal View History

2018-01-30 10:56:36 +00:00
package payload
import (
"testing"
"github.com/CityOfZion/neo-go/pkg/crypto/hash"
"github.com/CityOfZion/neo-go/pkg/io"
2018-02-01 18:54:23 +00:00
. "github.com/CityOfZion/neo-go/pkg/util"
"github.com/stretchr/testify/assert"
2018-01-30 10:56:36 +00:00
)
func TestInventoryEncodeDecode(t *testing.T) {
hashes := []Uint256{
hash.Sha256([]byte("a")),
hash.Sha256([]byte("b")),
2018-01-30 10:56:36 +00:00
}
inv := NewInventory(BlockType, hashes)
buf := io.NewBufBinWriter()
inv.EncodeBinary(buf.BinWriter)
assert.Nil(t, buf.Err)
2018-01-30 10:56:36 +00:00
b := buf.Bytes()
r := io.NewBinReaderFromBuf(b)
2018-01-30 10:56:36 +00:00
invDecode := &Inventory{}
invDecode.DecodeBinary(r)
assert.Nil(t, r.Err)
assert.Equal(t, inv, invDecode)
2018-01-30 10:56:36 +00:00
}
func TestEmptyInv(t *testing.T) {
msgInv := NewInventory(TXType, []Uint256{})
buf := io.NewBufBinWriter()
msgInv.EncodeBinary(buf.BinWriter)
assert.Nil(t, buf.Err)
assert.Equal(t, []byte{byte(TXType), 0}, buf.Bytes())
assert.Equal(t, 0, len(msgInv.Hashes))
}