Merge pull request #201 from Bytom/v0.1

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

diff --git a/vendor/github.com/bytom/protocol/vm/splice_test.go b/vendor/github.com/bytom/protocol/vm/splice_test.go
new file mode 100644 (file)
index 0000000..dcc3b1c
--- /dev/null
+++ b/vendor/github.com/bytom/protocol/vm/splice_test.go
@@ -0,0 +1,192 @@
+package vm
+
+import (
+       "testing"
+
+       "github.com/bytom/testutil"
+)
+
+func TestSpliceOps(t *testing.T) {
+       type testStruct struct {
+               op      Op
+               startVM *virtualMachine
+               wantErr error
+               wantVM  *virtualMachine
+       }
+       cases := []testStruct{{
+               op: OP_CAT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("hello"), []byte("world")},
+               },
+               wantVM: &virtualMachine{
+                       runLimit:     49986,
+                       deferredCost: -18,
+                       dataStack:    [][]byte{[]byte("helloworld")},
+               },
+       }, {
+               op: OP_CAT,
+               startVM: &virtualMachine{
+                       runLimit:  4,
+                       dataStack: [][]byte{[]byte("hello"), []byte("world")},
+               },
+               wantErr: ErrRunLimitExceeded,
+       }, {
+               op: OP_SUBSTR,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {3}, {5}},
+               },
+               wantVM: &virtualMachine{
+                       runLimit:     49991,
+                       deferredCost: -28,
+                       dataStack:    [][]byte{[]byte("lowor")},
+               },
+       }, {
+               op: OP_SUBSTR,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {3}, Int64Bytes(-1)},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_SUBSTR,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), Int64Bytes(-1), {5}},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_SUBSTR,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {6}, {5}},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_SUBSTR,
+               startVM: &virtualMachine{
+                       runLimit:  4,
+                       dataStack: [][]byte{[]byte("helloworld"), {3}, {5}},
+               },
+               wantErr: ErrRunLimitExceeded,
+       }, {
+               op: OP_LEFT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {5}},
+               },
+               wantVM: &virtualMachine{
+                       runLimit:     49991,
+                       deferredCost: -19,
+                       dataStack:    [][]byte{[]byte("hello")},
+               },
+       }, {
+               op: OP_LEFT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), Int64Bytes(-1)},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_LEFT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {11}},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_LEFT,
+               startVM: &virtualMachine{
+                       runLimit:  4,
+                       dataStack: [][]byte{[]byte("helloworld"), {5}},
+               },
+               wantErr: ErrRunLimitExceeded,
+       }, {
+               op: OP_RIGHT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {5}},
+               },
+               wantVM: &virtualMachine{
+                       runLimit:     49991,
+                       deferredCost: -19,
+                       dataStack:    [][]byte{[]byte("world")},
+               },
+       }, {
+               op: OP_RIGHT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), Int64Bytes(-1)},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_RIGHT,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld"), {11}},
+               },
+               wantErr: ErrBadValue,
+       }, {
+               op: OP_RIGHT,
+               startVM: &virtualMachine{
+                       runLimit:  4,
+                       dataStack: [][]byte{[]byte("helloworld"), {5}},
+               },
+               wantErr: ErrRunLimitExceeded,
+       }, {
+               op: OP_SIZE,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{[]byte("helloworld")},
+               },
+               wantVM: &virtualMachine{
+                       runLimit:     49999,
+                       deferredCost: 9,
+                       dataStack:    [][]byte{[]byte("helloworld"), {10}},
+               },
+       }, {
+               op: OP_CATPUSHDATA,
+               startVM: &virtualMachine{
+                       runLimit:  50000,
+                       dataStack: [][]byte{{0xff}, {0xab, 0xcd}},
+               },
+               wantVM: &virtualMachine{
+                       runLimit:     49993,
+                       deferredCost: -10,
+                       dataStack:    [][]byte{{0xff, 0x02, 0xab, 0xcd}},
+               },
+       }, {
+               op: OP_CATPUSHDATA,
+               startVM: &virtualMachine{
+                       runLimit:  4,
+                       dataStack: [][]byte{{0xff}, {0xab, 0xcd}},
+               },
+               wantErr: ErrRunLimitExceeded,
+       }}
+
+       spliceops := []Op{OP_CAT, OP_SUBSTR, OP_LEFT, OP_RIGHT, OP_CATPUSHDATA, OP_SIZE}
+       for _, op := range spliceops {
+               cases = append(cases, testStruct{
+                       op:      op,
+                       startVM: &virtualMachine{runLimit: 0},
+                       wantErr: ErrRunLimitExceeded,
+               })
+       }
+
+       for i, c := range cases {
+               err := ops[c.op].fn(c.startVM)
+
+               if err != c.wantErr {
+                       t.Errorf("case %d, op %s: got err = %v want %v", i, ops[c.op].name, err, c.wantErr)
+                       continue
+               }
+               if c.wantErr != nil {
+                       continue
+               }
+
+               if !testutil.DeepEqual(c.startVM, c.wantVM) {
+                       t.Errorf("case %d, op %s: unexpected vm result\n\tgot:  %+v\n\twant: %+v\n", i, ops[c.op].name, c.startVM, c.wantVM)
+               }
+       }
+}