Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / vendor / github.com / bytom / protocol / bc / types / output_commitment_test.go

diff --git a/vendor/github.com/bytom/protocol/bc/types/output_commitment_test.go b/vendor/github.com/bytom/protocol/bc/types/output_commitment_test.go
new file mode 100644 (file)
index 0000000..bf26c7d
--- /dev/null
+++ b/vendor/github.com/bytom/protocol/bc/types/output_commitment_test.go
@@ -0,0 +1,59 @@
+package types
+
+import (
+       "bytes"
+       "encoding/hex"
+       "testing"
+
+       "github.com/bytom/encoding/blockchain"
+       "github.com/bytom/protocol/bc"
+       "github.com/bytom/testutil"
+)
+
+func TestReadWriteOutputCommitment(t *testing.T) {
+       btmAssetID := testutil.MustDecodeAsset("ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff")
+       cases := []struct {
+               oc           *OutputCommitment
+               encodeString string
+       }{
+               {
+                       oc: &OutputCommitment{
+                               AssetAmount:    bc.AssetAmount{AssetId: &btmAssetID, Amount: 100},
+                               VMVersion:      1,
+                               ControlProgram: testutil.MustDecodeHexString("00140876db6ca8f4542a836f0edd42b87d095d081182"),
+                       },
+                       encodeString: "39ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff64011600140876db6ca8f4542a836f0edd42b87d095d081182",
+               },
+               {
+                       oc: &OutputCommitment{
+                               AssetAmount:    bc.AssetAmount{AssetId: &btmAssetID, Amount: 50},
+                               VMVersion:      1,
+                               ControlProgram: testutil.MustDecodeHexString("00148bf7800b2333afd8414d6e903d58c4908b9bbcc7"),
+                       },
+                       encodeString: "39ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff32011600148bf7800b2333afd8414d6e903d58c4908b9bbcc7",
+               },
+       }
+
+       for _, c := range cases {
+               buff := []byte{}
+               buffer := bytes.NewBuffer(buff)
+               if err := c.oc.writeExtensibleString(buffer, nil, 1); err != nil {
+                       t.Fatal(err)
+               }
+
+               got := hex.EncodeToString(buffer.Bytes())
+               if got != c.encodeString {
+                       t.Errorf("got:%s, want:%s", got, c.encodeString)
+               }
+
+               oc := &OutputCommitment{}
+               _, err := oc.readFrom(blockchain.NewReader(buffer.Bytes()), 1)
+               if err != nil {
+                       t.Fatal(err)
+               }
+
+               if !testutil.DeepEqual(*oc, *c.oc) {
+                       t.Errorf("got:%v, want:%v", *oc, *c.oc)
+               }
+       }
+}