TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-12-25 03:56:34 +00:00
Code Issues Projects Releases Packages Wiki Activity

vm: add tests for ByteString enumerator/iterator

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-08-17 17:26:50 +03:00
parent 6bdaefcfa4
commit 8659fd79e5
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
pkg/vm/vm_test.go
View file

@ -516,7 +516,7 @@ func checkEnumeratorStack(t *testing.T, vm *VM, arr []stackitem.Item) {
} }
} }
func testIterableCreate(t *testing.T, typ string) { func testIterableCreate(t *testing.T, typ string, isByteArray bool) {
isIter := typ == "Iterator" isIter := typ == "Iterator"
prog := getSyscallProg("System." + typ + ".Create") prog := getSyscallProg("System." + typ + ".Create")
prog = append(prog, getEnumeratorProg(2, isIter)...) prog = append(prog, getEnumeratorProg(2, isIter)...)
@ -526,7 +526,12 @@ func testIterableCreate(t *testing.T, typ string) {
stackitem.NewBigInteger(big.NewInt(42)), stackitem.NewBigInteger(big.NewInt(42)),
stackitem.NewByteArray([]byte{3, 2, 1}), stackitem.NewByteArray([]byte{3, 2, 1}),
} }
vm.estack.Push(&Element{value: stackitem.NewArray(arr)}) if isByteArray {
arr[1] = stackitem.Make(7)
vm.estack.PushVal([]byte{42, 7})
} else {
vm.estack.Push(&Element{value: stackitem.NewArray(arr)})
}
runVM(t, vm) runVM(t, vm)
if isIter { if isIter {
@ -543,11 +548,13 @@ func testIterableCreate(t *testing.T, typ string) {
} }
func TestEnumeratorCreate(t *testing.T) { func TestEnumeratorCreate(t *testing.T) {
testIterableCreate(t, "Enumerator") t.Run("Array", func(t *testing.T) { testIterableCreate(t, "Enumerator", false) })
t.Run("ByteArray", func(t *testing.T) { testIterableCreate(t, "Enumerator", true) })
} }
func TestIteratorCreate(t *testing.T) { func TestIteratorCreate(t *testing.T) {
testIterableCreate(t, "Iterator") t.Run("Array", func(t *testing.T) { testIterableCreate(t, "Iterator", false) })
t.Run("ByteArray", func(t *testing.T) { testIterableCreate(t, "Iterator", true) })
} }
func testIterableConcat(t *testing.T, typ string) { func testIterableConcat(t *testing.T, typ string) {