141 lines
3.1 KiB
Go
141 lines
3.1 KiB
Go
package state
|
|
|
|
import (
|
|
"math/big"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/nspcc-dev/neo-go/internal/random"
|
|
"github.com/nspcc-dev/neo-go/internal/testserdes"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTokenTransferLog_Append17(t *testing.T) {
|
|
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
expected := []*NEP17Transfer{
|
|
random17Transfer(r),
|
|
random17Transfer(r),
|
|
random17Transfer(r),
|
|
random17Transfer(r),
|
|
}
|
|
|
|
lg := new(TokenTransferLog)
|
|
for _, tr := range expected {
|
|
require.NoError(t, lg.Append(tr))
|
|
}
|
|
|
|
require.Equal(t, len(expected), lg.Size())
|
|
|
|
i := len(expected) - 1
|
|
cont, err := lg.ForEachNEP17(func(tr *NEP17Transfer) (bool, error) {
|
|
require.Equal(t, expected[i], tr)
|
|
i--
|
|
return true, nil
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, cont)
|
|
}
|
|
|
|
func TestTokenTransferLog_Append(t *testing.T) {
|
|
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
expected := []*NEP11Transfer{
|
|
random11Transfer(r),
|
|
random11Transfer(r),
|
|
random11Transfer(r),
|
|
random11Transfer(r),
|
|
}
|
|
|
|
lg := new(TokenTransferLog)
|
|
for _, tr := range expected {
|
|
require.NoError(t, lg.Append(tr))
|
|
}
|
|
|
|
require.Equal(t, len(expected), lg.Size())
|
|
|
|
i := len(expected) - 1
|
|
cont, err := lg.ForEachNEP11(func(tr *NEP11Transfer) (bool, error) {
|
|
require.Equal(t, expected[i], tr)
|
|
i--
|
|
return true, nil
|
|
})
|
|
require.NoError(t, err)
|
|
require.True(t, cont)
|
|
}
|
|
|
|
func BenchmarkTokenTransferLog_Append(b *testing.B) {
|
|
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
ts := make([]*NEP17Transfer, TokenTransferBatchSize)
|
|
for i := range ts {
|
|
ts[i] = random17Transfer(r)
|
|
}
|
|
|
|
lg := new(TokenTransferLog)
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
for _, tr := range ts {
|
|
err := lg.Append(tr)
|
|
if err != nil {
|
|
b.FailNow()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNEP17Transfer_DecodeBinary(t *testing.T) {
|
|
expected := &NEP17Transfer{
|
|
Asset: 123,
|
|
From: util.Uint160{5, 6, 7},
|
|
To: util.Uint160{8, 9, 10},
|
|
Amount: *big.NewInt(42),
|
|
Block: 12345,
|
|
Timestamp: 54321,
|
|
Tx: util.Uint256{8, 5, 3},
|
|
}
|
|
|
|
testserdes.EncodeDecodeBinary(t, expected, new(NEP17Transfer))
|
|
}
|
|
|
|
func TestNEP11Transfer_DecodeBinary(t *testing.T) {
|
|
expected := &NEP11Transfer{
|
|
NEP17Transfer: NEP17Transfer{
|
|
Asset: 123,
|
|
From: util.Uint160{5, 6, 7},
|
|
To: util.Uint160{8, 9, 10},
|
|
Amount: *big.NewInt(42),
|
|
Block: 12345,
|
|
Timestamp: 54321,
|
|
Tx: util.Uint256{8, 5, 3},
|
|
},
|
|
ID: []byte{42, 42, 42},
|
|
}
|
|
|
|
testserdes.EncodeDecodeBinary(t, expected, new(NEP11Transfer))
|
|
}
|
|
|
|
func random17Transfer(r *rand.Rand) *NEP17Transfer {
|
|
return &NEP17Transfer{
|
|
Amount: *big.NewInt(int64(r.Uint64())),
|
|
Block: r.Uint32(),
|
|
Asset: int32(random.Int(10, 10000000)),
|
|
From: random.Uint160(),
|
|
To: random.Uint160(),
|
|
Tx: random.Uint256(),
|
|
}
|
|
}
|
|
|
|
func random11Transfer(r *rand.Rand) *NEP11Transfer {
|
|
return &NEP11Transfer{
|
|
NEP17Transfer: NEP17Transfer{
|
|
Amount: *big.NewInt(int64(r.Uint64())),
|
|
Block: r.Uint32(),
|
|
Asset: int32(random.Int(10, 10000000)),
|
|
From: random.Uint160(),
|
|
To: random.Uint160(),
|
|
Tx: random.Uint256(),
|
|
},
|
|
ID: random.Uint256().BytesBE(),
|
|
}
|
|
}
|