From 8659fd79e54acdb517d145d2afd96ced35a91539 Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Mon, 17 Aug 2020 17:26:50 +0300
Subject: [PATCH] vm: add tests for ByteString enumerator/iterator

---
 pkg/vm/vm_test.go | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/pkg/vm/vm_test.go b/pkg/vm/vm_test.go
index cd2fae269..fbf354f05 100644
--- a/pkg/vm/vm_test.go
+++ b/pkg/vm/vm_test.go
@@ -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"
 	prog := getSyscallProg("System." + typ + ".Create")
 	prog = append(prog, getEnumeratorProg(2, isIter)...)
@@ -526,7 +526,12 @@ func testIterableCreate(t *testing.T, typ string) {
 		stackitem.NewBigInteger(big.NewInt(42)),
 		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)
 	if isIter {
@@ -543,11 +548,13 @@ func testIterableCreate(t *testing.T, typ string) {
 }
 
 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) {
-	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) {