forked from TrueCloudLab/neoneo-go
f7d57e4e49
* updated readme * added basic cmd. * added seperate folders for cmd packages. * Fix netmodes in test + reverse bigint bytes * glide get deps
74 lines
1.8 KiB
Go
74 lines
1.8 KiB
Go
package vm
|
|
|
|
import (
|
|
"bytes"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/util"
|
|
)
|
|
|
|
func TestEmitPush(t *testing.T) {
|
|
sb := &ScriptBuilder{new(bytes.Buffer)}
|
|
|
|
if err := sb.emitPush(OpPush1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if sb.buf.Len() != 1 {
|
|
t.Fatalf("expect buffer len of 1 got %d", sb.buf.Len())
|
|
}
|
|
}
|
|
|
|
func TestEmitPushInt(t *testing.T) {
|
|
sb := &ScriptBuilder{new(bytes.Buffer)}
|
|
|
|
val := -1
|
|
if err := sb.emitPushInt(int64(val)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want, have := OpPushM1, OpCode(sb.buf.Bytes()[0]); want != have {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
val = 0
|
|
if err := sb.emitPushInt(int64(val)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want, have := OpPushF, OpCode(sb.buf.Bytes()[1]); want != have {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
val = 1
|
|
if err := sb.emitPushInt(int64(val)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want, have := OpPush1, OpCode(sb.buf.Bytes()[2]); want != have {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
val = 1000
|
|
if err := sb.emitPushInt(int64(val)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
bInt := big.NewInt(int64(val))
|
|
if want, have := byte(len(bInt.Bytes())), byte(sb.buf.Bytes()[3]); want != have {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
want := util.ToArrayReverse(bInt.Bytes()) // reverse
|
|
have := sb.buf.Bytes()[4:]
|
|
if bytes.Compare(want, have) != 0 {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
}
|
|
|
|
func TestEmitPushString(t *testing.T) {
|
|
sb := &ScriptBuilder{new(bytes.Buffer)}
|
|
str := "anthdm"
|
|
if err := sb.emitPushString(str); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want, have := byte(len(str)), sb.buf.Bytes()[0]; want != have {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
want, have := []byte(str), sb.buf.Bytes()[1:]
|
|
if bytes.Compare(want, have) != 0 {
|
|
t.Fatalf("expected %v got %v", want, have)
|
|
}
|
|
}
|