mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-26 19:42:23 +00:00
433275265f
This is just a much better way to do the same thing. Inspired by nspcc-dev/neofs-sdk-go#407. Signed-off-by: Roman Khimov <roman@nspcc.ru>
70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package payload
|
|
|
|
import (
|
|
"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.Header {
|
|
return &block.Header{
|
|
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{
|
|
Header: 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{
|
|
Header: b,
|
|
TxCount: block.MaxTransactionsPerBlock + 1,
|
|
Hashes: make([]util.Uint256, block.MaxTransactionsPerBlock),
|
|
Flags: []byte{},
|
|
}
|
|
data, err := testserdes.EncodeBinary(expected)
|
|
require.NoError(t, err)
|
|
require.ErrorIs(t, testserdes.DecodeBinary(data, new(MerkleBlock)), block.ErrMaxContentsPerBlock)
|
|
})
|
|
|
|
t.Run("bad flags size", func(t *testing.T) {
|
|
b := newDumbBlock()
|
|
_ = b.Hash()
|
|
expected := &MerkleBlock{
|
|
Header: 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)))
|
|
})
|
|
}
|