Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / protocol / vm / vmutil / script_test.go

diff --git a/protocol/vm/vmutil/script_test.go b/protocol/vm/vmutil/script_test.go
index 020ac1b..623bccf 100644 (file)
--- a/protocol/vm/vmutil/script_test.go
+++ b/protocol/vm/vmutil/script_test.go
@@ -1,7 +1,11 @@
 package vmutil
 
 import (
+       "encoding/hex"
        "testing"
+
+       "github.com/vapor/crypto/ed25519"
+       "github.com/vapor/errors"
 )
 
 // TestIsUnspendable ensures the IsUnspendable function returns the expected
@@ -35,3 +39,152 @@ func TestIsUnspendable(t *testing.T) {
                }
        }
 }
+
+func TestP2SPMultiSigProgram(t *testing.T) {
+       pub1, _ := hex.DecodeString("988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6")
+       pub2, _ := hex.DecodeString("7192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c62")
+       pub3, _ := hex.DecodeString("8bcd251d9f4e03877130b6e6f1d577eda562375f07c3cdfad8f1d541002fd1a3")
+
+       tests := []struct {
+               pubkeys     []ed25519.PublicKey
+               nrequired   int
+               wantProgram string
+               wantErr     error
+       }{
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1},
+                       nrequired:   1,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb65151ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1, pub2},
+                       nrequired:   2,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6207192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c625252ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1, pub2, pub3},
+                       nrequired:   2,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6207192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c62208bcd251d9f4e03877130b6e6f1d577eda562375f07c3cdfad8f1d541002fd1a35253ad",
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1},
+                       nrequired: -1,
+                       wantErr:   errors.WithDetail(ErrBadValue, "negative quorum"),
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1},
+                       nrequired: 0,
+                       wantErr:   errors.WithDetail(ErrBadValue, "quorum empty with non-empty pubkey list"),
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1, pub2},
+                       nrequired: 3,
+                       wantErr:   errors.WithDetail(ErrBadValue, "quorum too big"),
+               },
+       }
+
+       for i, test := range tests {
+               got, err := P2SPMultiSigProgram(test.pubkeys, test.nrequired)
+               if err != nil {
+                       if test.wantErr != nil && err.Error() != test.wantErr.Error() {
+                               t.Errorf("TestP2SPMultiSigProgram #%d failed: got %v want %v", i, err.Error(), test.wantErr.Error())
+                       } else if test.wantErr == nil {
+                               t.Fatal(err)
+                       }
+               }
+
+               if hex.EncodeToString(got) != test.wantProgram {
+                       t.Errorf("TestP2SPMultiSigProgram #%d failed: got %v want %v", i, hex.EncodeToString(got), test.wantProgram)
+               }
+       }
+}
+
+func TestP2SPMultiSigProgramWithHeight(t *testing.T) {
+       pub1, _ := hex.DecodeString("988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6")
+       pub2, _ := hex.DecodeString("7192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c62")
+       pub3, _ := hex.DecodeString("8bcd251d9f4e03877130b6e6f1d577eda562375f07c3cdfad8f1d541002fd1a3")
+
+       tests := []struct {
+               pubkeys     []ed25519.PublicKey
+               nrequired   int
+               height      int64
+               wantProgram string
+               wantErr     error
+       }{
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1},
+                       nrequired:   1,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb65151ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1, pub2},
+                       nrequired:   2,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6207192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c625252ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1, pub2, pub3},
+                       nrequired:   2,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6207192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c62208bcd251d9f4e03877130b6e6f1d577eda562375f07c3cdfad8f1d541002fd1a35253ad",
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1},
+                       nrequired: 1,
+                       height:    -1,
+                       wantErr:   errors.WithDetail(ErrBadValue, "negative blockHeight"),
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1},
+                       nrequired:   1,
+                       height:      0,
+                       wantProgram: "ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb65151ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1},
+                       nrequired:   1,
+                       height:      200,
+                       wantProgram: "01c8cda069ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb65151ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1, pub2},
+                       nrequired:   2,
+                       height:      200,
+                       wantProgram: "01c8cda069ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6207192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c625252ad",
+               },
+               {
+                       pubkeys:     []ed25519.PublicKey{pub1, pub2, pub3},
+                       nrequired:   2,
+                       height:      200,
+                       wantProgram: "01c8cda069ae20988650ff921c82d47a953527894f792572ba63197c56e5fe79e5df0c444d6bb6207192bf4eac0789ee19c88dfa87861cf59e215820f7bdb7be02761d9ed92e6c62208bcd251d9f4e03877130b6e6f1d577eda562375f07c3cdfad8f1d541002fd1a35253ad",
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1},
+                       nrequired: -1,
+                       wantErr:   errors.WithDetail(ErrBadValue, "negative quorum"),
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1},
+                       nrequired: 0,
+                       wantErr:   errors.WithDetail(ErrBadValue, "quorum empty with non-empty pubkey list"),
+               },
+               {
+                       pubkeys:   []ed25519.PublicKey{pub1, pub2},
+                       nrequired: 3,
+                       wantErr:   errors.WithDetail(ErrBadValue, "quorum too big"),
+               },
+       }
+
+       for i, test := range tests {
+               got, err := P2SPMultiSigProgramWithHeight(test.pubkeys, test.nrequired, test.height)
+               if err != nil {
+                       if test.wantErr != nil && err.Error() != test.wantErr.Error() {
+                               t.Errorf("TestP2SPMultiSigProgram #%d failed: got %v want %v", i, err.Error(), test.wantErr.Error())
+                       } else if test.wantErr == nil {
+                               t.Fatal(err)
+                       }
+               }
+
+               if hex.EncodeToString(got) != test.wantProgram {
+                       t.Errorf("TestP2SPMultiSigProgram #%d failed: got %v want %v", i, hex.EncodeToString(got), test.wantProgram)
+               }
+       }
+}