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

package payload

import (
	"testing"

	"github.com/nspcc-dev/neo-go/internal/testserdes"
	"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
	"github.com/stretchr/testify/require"
)

func TestGetBlockEncodeDecode(t *testing.T) {
	start := hash.Sha256([]byte("a"))

	p := NewGetBlocks(start, 124)
	testserdes.EncodeDecodeBinary(t, p, new(GetBlocks))

	// invalid count
	p = NewGetBlocks(start, -2)
	data, err := testserdes.EncodeBinary(p)
	require.NoError(t, err)
	require.Error(t, testserdes.DecodeBinary(data, new(GetBlocks)))

	// invalid count
	p = NewGetBlocks(start, 0)
	data, err = testserdes.EncodeBinary(p)
	require.NoError(t, err)
	require.Error(t, testserdes.DecodeBinary(data, new(GetBlocks)))
}