forked from TrueCloudLab/neoneo-go
48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
package iterator
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Iterator is thoroughly tested in VM package, these are smoke tests.
|
|
func TestIterator(t *testing.T) {
|
|
ic := &interop.Context{VM: vm.New()}
|
|
full := []byte{4, 8, 15}
|
|
ic.VM.Estack().PushVal(full[2:])
|
|
require.NoError(t, Create(ic))
|
|
ic.VM.Estack().PushVal(full[:2])
|
|
require.NoError(t, Create(ic))
|
|
require.NoError(t, Concat(ic))
|
|
|
|
res := ic.VM.Estack().Pop().Item()
|
|
ic.VM.Estack().PushVal(res)
|
|
require.NoError(t, vm.EnumeratorNext(ic.VM))
|
|
require.True(t, ic.VM.Estack().Pop().Bool())
|
|
|
|
ic.VM.Estack().PushVal(res)
|
|
require.NoError(t, Key(ic))
|
|
require.Equal(t, big.NewInt(0), ic.VM.Estack().Pop().BigInt())
|
|
|
|
ic.VM.Estack().PushVal(res)
|
|
require.NoError(t, vm.EnumeratorValue(ic.VM))
|
|
require.Equal(t, big.NewInt(int64(full[0])), ic.VM.Estack().Pop().BigInt())
|
|
|
|
ic.VM.Estack().PushVal(res)
|
|
require.NoError(t, vm.EnumeratorNext(ic.VM))
|
|
require.True(t, ic.VM.Estack().Pop().Bool())
|
|
|
|
ic.VM.Estack().PushVal(res)
|
|
require.NoError(t, Keys(ic))
|
|
require.NoError(t, vm.EnumeratorValue(ic.VM))
|
|
require.Equal(t, big.NewInt(1), ic.VM.Estack().Pop().BigInt())
|
|
|
|
ic.VM.Estack().PushVal(res)
|
|
require.NoError(t, Values(ic))
|
|
require.NoError(t, vm.EnumeratorValue(ic.VM))
|
|
require.Equal(t, big.NewInt(int64(full[1])), ic.VM.Estack().Pop().BigInt())
|
|
}
|