neoneo-go/pkg/vm/script_builder_test.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)
	}
}