state: prepare for NEP-11 data tracking, refactor/rename

There is a lot of similarity, so try reusing common code and use more neutral
naming.

pkg/core/state/tokens_test.go

@@ -0,0 +1,141 @@
+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(),
+	}
+}