package payload import ( "errors" "testing" "github.com/stretchr/testify/require" "github.com/nspcc-dev/neo-go/internal/testserdes" "github.com/nspcc-dev/neo-go/pkg/core/block" "github.com/nspcc-dev/neo-go/pkg/core/transaction" "github.com/nspcc-dev/neo-go/pkg/crypto/hash" "github.com/nspcc-dev/neo-go/pkg/util" ) func newDumbBlock() *block.Base { return &block.Base{ Version: 0, PrevHash: hash.Sha256([]byte("a")), MerkleRoot: hash.Sha256([]byte("b")), Timestamp: 100500, Index: 1, NextConsensus: hash.Hash160([]byte("a")), Script: transaction.Witness{ VerificationScript: []byte{0x51}, // PUSH1 InvocationScript: []byte{0x61}, // NOP }, } } func TestMerkleBlock_EncodeDecodeBinary(t *testing.T) { t.Run("positive", func(t *testing.T) { b := newDumbBlock() _ = b.Hash() expected := &MerkleBlock{ Base: b, TxCount: 0, Hashes: []util.Uint256{}, Flags: []byte{}, } testserdes.EncodeDecodeBinary(t, expected, new(MerkleBlock)) }) t.Run("bad contents count", func(t *testing.T) { b := newDumbBlock() _ = b.Hash() expected := &MerkleBlock{ Base: b, TxCount: block.MaxContentsPerBlock + 1, Hashes: make([]util.Uint256, block.MaxContentsPerBlock), Flags: []byte{}, } data, err := testserdes.EncodeBinary(expected) require.NoError(t, err) require.True(t, errors.Is(block.ErrMaxContentsPerBlock, testserdes.DecodeBinary(data, new(MerkleBlock)))) }) t.Run("bad flags size", func(t *testing.T) { b := newDumbBlock() _ = b.Hash() expected := &MerkleBlock{ Base: b, TxCount: 0, Hashes: []util.Uint256{}, Flags: []byte{1, 2, 3, 4, 5}, } data, err := testserdes.EncodeBinary(expected) require.NoError(t, err) require.Error(t, testserdes.DecodeBinary(data, new(MerkleBlock))) }) }