Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / vendor / github.com / bytom / protocol / vm / vmutil / builder_test.go

diff --git a/vendor/github.com/bytom/protocol/vm/vmutil/builder_test.go b/vendor/github.com/bytom/protocol/vm/vmutil/builder_test.go
new file mode 100644 (file)
index 0000000..3df7f8d
--- /dev/null
+++ b/vendor/github.com/bytom/protocol/vm/vmutil/builder_test.go
@@ -0,0 +1,122 @@
+package vmutil
+
+import (
+       "bytes"
+       "encoding/hex"
+       "testing"
+
+       "github.com/bytom/protocol/vm"
+)
+
+func TestAddJump(t *testing.T) {
+       cases := []struct {
+               name    string
+               wantHex string
+               fn      func(t *testing.T, b *Builder)
+       }{
+               {
+                       "single jump single target not yet defined",
+                       "630600000061",
+                       func(t *testing.T, b *Builder) {
+                               target := b.NewJumpTarget()
+                               b.AddJump(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.SetJumpTarget(target)
+                       },
+               },
+               {
+                       "single jump single target already defined",
+                       "616300000000",
+                       func(t *testing.T, b *Builder) {
+                               target := b.NewJumpTarget()
+                               b.SetJumpTarget(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target)
+                       },
+               },
+               {
+                       "two jumps single target not yet defined",
+                       "630c00000061630c00000061",
+                       func(t *testing.T, b *Builder) {
+                               target := b.NewJumpTarget()
+                               b.AddJump(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.SetJumpTarget(target)
+                       },
+               },
+               {
+                       "two jumps single target already defined",
+                       "616300000000616300000000",
+                       func(t *testing.T, b *Builder) {
+                               target := b.NewJumpTarget()
+                               b.SetJumpTarget(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target)
+                       },
+               },
+               {
+                       "two jumps single target, one not yet defined, one already defined",
+                       "630600000061616306000000",
+                       func(t *testing.T, b *Builder) {
+                               target := b.NewJumpTarget()
+                               b.AddJump(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.SetJumpTarget(target)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target)
+                       },
+               },
+               {
+                       "two jumps, two targets, not yet defined",
+                       "630c00000061630d0000006161",
+                       func(t *testing.T, b *Builder) {
+                               target1 := b.NewJumpTarget()
+                               b.AddJump(target1)
+                               b.AddOp(vm.OP_NOP)
+                               target2 := b.NewJumpTarget()
+                               b.AddJump(target2)
+                               b.AddOp(vm.OP_NOP)
+                               b.SetJumpTarget(target1)
+                               b.AddOp(vm.OP_NOP)
+                               b.SetJumpTarget(target2)
+                       },
+               },
+               {
+                       "two jumps, two targets, already defined",
+                       "6161616301000000616302000000",
+                       func(t *testing.T, b *Builder) {
+                               b.AddOp(vm.OP_NOP)
+                               target1 := b.NewJumpTarget()
+                               b.SetJumpTarget(target1)
+                               b.AddOp(vm.OP_NOP)
+                               target2 := b.NewJumpTarget()
+                               b.SetJumpTarget(target2)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target1)
+                               b.AddOp(vm.OP_NOP)
+                               b.AddJump(target2)
+                       },
+               },
+       }
+       for _, c := range cases {
+               t.Run(c.name, func(t *testing.T) {
+                       b := NewBuilder()
+                       c.fn(t, b)
+                       prog, err := b.Build()
+                       if err != nil {
+                               t.Fatal(err)
+                       }
+                       want, err := hex.DecodeString(c.wantHex)
+                       if err != nil {
+                               t.Fatal(err)
+                       }
+                       if !bytes.Equal(prog, want) {
+                               t.Errorf("got %x, want %x", prog, want)
+                       }
+               })
+       }
+}