forked from TrueCloudLab/neoneo-go
4dc11ee48f
1) NOP 2) JMP 3) JMPIF 4) JMPIFNOT
168 lines
3.3 KiB
Go
168 lines
3.3 KiB
Go
package vm
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/vm/stack"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNopOp(t *testing.T) {
|
|
|
|
v := VM{}
|
|
|
|
a, err := stack.NewInt(big.NewInt(10))
|
|
assert.Nil(t, err)
|
|
|
|
ctx := stack.NewContext([]byte{})
|
|
ctx.Estack.Push(a)
|
|
|
|
v.executeOp(stack.NOP, ctx)
|
|
|
|
// Stack should have one item
|
|
assert.Equal(t, 1, ctx.Estack.Len())
|
|
|
|
item, err := ctx.Estack.PopInt()
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, int64(10), item.Value().Int64())
|
|
}
|
|
|
|
func TestJmpOp(t *testing.T) {
|
|
|
|
v := VM{}
|
|
|
|
a, err := stack.NewInt(big.NewInt(10))
|
|
assert.Nil(t, err)
|
|
|
|
ctx := stack.NewContext([]byte{5, 0, 2, 3, 4})
|
|
ctx.Estack.Push(a)
|
|
|
|
// ctx.ip = -1
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
|
|
// ctx.ip will be set to offset.
|
|
// offset = ctx.IP() + int(ctx.ReadInt16()) - 3
|
|
// = 0 + 5 -3 = 2
|
|
v.executeOp(stack.JMP, ctx)
|
|
|
|
// Stack should have one item
|
|
assert.Equal(t, 1, ctx.Estack.Len())
|
|
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 3, ctx.IP())
|
|
}
|
|
|
|
// test JMPIF instruction with true boolean
|
|
// on top of the stack
|
|
func TestJmpIfOp1(t *testing.T) {
|
|
|
|
v := VM{}
|
|
|
|
a := stack.NewBoolean(true)
|
|
|
|
ctx := stack.NewContext([]byte{5, 0, 2, 3, 4})
|
|
ctx.Estack.Push(a)
|
|
|
|
// ctx.ip = -1
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
|
|
// ctx.ip will be set to offset
|
|
// because the there is a true boolean
|
|
// on top of the stack.
|
|
// offset = ctx.IP() + int(ctx.ReadInt16()) - 3
|
|
// = 0 + 5 -3 = 2
|
|
v.executeOp(stack.JMPIF, ctx)
|
|
|
|
// Stack should have 0 item
|
|
assert.Equal(t, 0, ctx.Estack.Len())
|
|
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 3, ctx.IP())
|
|
}
|
|
|
|
// test JMPIF instruction with false boolean
|
|
// on top of the stack
|
|
func TestJmpIfOp2(t *testing.T) {
|
|
|
|
v := VM{}
|
|
|
|
a := stack.NewBoolean(false)
|
|
|
|
ctx := stack.NewContext([]byte{5, 0, 2, 3, 4})
|
|
ctx.Estack.Push(a)
|
|
|
|
// ctx.ip = -1
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
|
|
// nothing will happen because
|
|
// the value of the boolean on top of the stack
|
|
// is false
|
|
v.executeOp(stack.JMPIF, ctx)
|
|
|
|
// Stack should have 0 item
|
|
assert.Equal(t, 0, ctx.Estack.Len())
|
|
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
}
|
|
|
|
// test JMPIFNOT instruction with true boolean
|
|
// on top of the stack
|
|
func TestJmpIfNotOp1(t *testing.T) {
|
|
|
|
v := VM{}
|
|
|
|
a := stack.NewBoolean(true)
|
|
|
|
ctx := stack.NewContext([]byte{5, 0, 2, 3, 4})
|
|
ctx.Estack.Push(a)
|
|
|
|
// ctx.ip = -1
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
|
|
// nothing will happen because
|
|
// the value of the boolean on top of the stack
|
|
// is true
|
|
v.executeOp(stack.JMPIFNOT, ctx)
|
|
|
|
// Stack should have 0 item
|
|
assert.Equal(t, 0, ctx.Estack.Len())
|
|
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
}
|
|
|
|
// test JMPIFNOT instruction with false boolean
|
|
// on top of the stack
|
|
func TestJmpIfNotOp2(t *testing.T) {
|
|
|
|
v := VM{}
|
|
|
|
a := stack.NewBoolean(false)
|
|
|
|
ctx := stack.NewContext([]byte{5, 0, 2, 3, 4})
|
|
ctx.Estack.Push(a)
|
|
|
|
// ctx.ip = -1
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 0, ctx.IP())
|
|
|
|
// ctx.ip will be set to offset
|
|
// because the there is a false boolean
|
|
// on top of the stack.
|
|
// offset = ctx.IP() + int(ctx.ReadInt16()) - 3
|
|
// = 0 + 5 -3 = 2
|
|
v.executeOp(stack.JMPIFNOT, ctx)
|
|
|
|
// Stack should have one item
|
|
assert.Equal(t, 0, ctx.Estack.Len())
|
|
|
|
// ctx.IP() = ctx.ip + 1
|
|
assert.Equal(t, 3, ctx.IP())
|
|
}
|