package vm import ( "math/big" "testing" "github.com/CityOfZion/neo-go/pkg/vm/stack" "github.com/stretchr/testify/assert" ) func TestInvertOp(t *testing.T) { v := VM{} // 0000 00110 = 5 a, err := stack.NewInt(big.NewInt(5)) assert.Nil(t, err) ctx := stack.NewContext([]byte{}) ctx.Estack.Push(a) // 1111 11001 = -6 (two complement representation) v.executeOp(stack.INVERT, 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(-6), item.Value().Int64()) } func TestAndOp(t *testing.T) { v := VM{} // 110001 = 49 a, err := stack.NewInt(big.NewInt(49)) assert.Nil(t, err) // 100011 = 35 b, err := stack.NewInt(big.NewInt(35)) assert.Nil(t, err) ctx := stack.NewContext([]byte{}) ctx.Estack.Push(a).Push(b) // 100001 = 33 v.executeOp(stack.AND, 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(33), item.Value().Int64()) } func TestOrOp(t *testing.T) { v := VM{} // 110001 = 49 a, err := stack.NewInt(big.NewInt(49)) assert.Nil(t, err) // 100011 = 35 b, err := stack.NewInt(big.NewInt(35)) assert.Nil(t, err) ctx := stack.NewContext([]byte{}) ctx.Estack.Push(a).Push(b) // 110011 = 51 (49 OR 35) v.executeOp(stack.OR, 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(51), item.Value().Int64()) } func TestXorOp(t *testing.T) { v := VM{} // 110001 = 49 a, err := stack.NewInt(big.NewInt(49)) assert.Nil(t, err) // 100011 = 35 b, err := stack.NewInt(big.NewInt(35)) assert.Nil(t, err) ctx := stack.NewContext([]byte{}) ctx.Estack.Push(a).Push(b) // 010010 = 18 (49 XOR 35) v.executeOp(stack.XOR, 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(18), item.Value().Int64()) } func TestEqualOp(t *testing.T) { v := VM{} a, err := stack.NewInt(big.NewInt(10)) assert.Nil(t, err) b, err := stack.NewInt(big.NewInt(10)) assert.Nil(t, err) ctx := stack.NewContext([]byte{}) ctx.Estack.Push(a).Push(b) v.executeOp(stack.EQUAL, ctx) // Stack should have one item assert.Equal(t, 1, ctx.Estack.Len()) item, err := ctx.Estack.PopBoolean() assert.Nil(t, err) assert.Equal(t, true, item.Value()) }