Merge pull request #201 from Bytom/v0.1

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

diff --git a/vendor/github.com/bytom/protocol/bc/types/spend_test.go b/vendor/github.com/bytom/protocol/bc/types/spend_test.go
new file mode 100644 (file)
index 0000000..f451a5e
--- /dev/null
+++ b/vendor/github.com/bytom/protocol/bc/types/spend_test.go
@@ -0,0 +1,70 @@
+package types
+
+import (
+       "bytes"
+       "encoding/hex"
+       "strings"
+       "testing"
+
+       "github.com/davecgh/go-spew/spew"
+
+       "github.com/bytom/encoding/blockchain"
+       "github.com/bytom/protocol/bc"
+       "github.com/bytom/testutil"
+)
+
+func TestSerializationSpendCommitment(t *testing.T) {
+       assetID := testutil.MustDecodeAsset("81756fdab39a17163b0ce582ee4ee256fb4d1e156c692b997d608a42ecb38d47")
+       sc := &SpendCommitment{
+               AssetAmount: bc.AssetAmount{
+                       AssetId: &assetID,
+                       Amount:  254354,
+               },
+               SourceID:       testutil.MustDecodeHash("bef8ff450b877df84174ac5c279fc97da0f507ffe7beef7badf116ea9e2ff041"),
+               SourcePosition: 3,
+               VMVersion:      1,
+               ControlProgram: []byte("TestSerializationSpendCommitment"),
+       }
+
+       wantHex := strings.Join([]string{
+               "66", // serialization length
+               "bef8ff450b877df84174ac5c279fc97da0f507ffe7beef7badf116ea9e2ff041", // sourceID
+               "81756fdab39a17163b0ce582ee4ee256fb4d1e156c692b997d608a42ecb38d47", // assetID
+               "92c30f", // amount
+               "03",     // position
+               "01",     // version
+               "20",     // control program length
+               "5465737453657269616c697a6174696f6e5370656e64436f6d6d69746d656e74", // control program
+       }, "")
+
+       // Test convert struct to hex
+       var buffer bytes.Buffer
+       suffix := []byte{}
+       if err := sc.writeExtensibleString(&buffer, suffix, 1); err != nil {
+               t.Fatal(err)
+       } else if len(suffix) != 0 {
+               t.Errorf("spend commitment write to got garbage hex left")
+       }
+
+       gotHex := hex.EncodeToString(buffer.Bytes())
+       if gotHex != wantHex {
+               t.Errorf("serialization bytes = %s want %s", gotHex, wantHex)
+       }
+
+       // Test convert hex to struct
+       var gotSC SpendCommitment
+       decodeHex, err := hex.DecodeString(wantHex)
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       if getSuffix, err := gotSC.readFrom(blockchain.NewReader(decodeHex), 1); err != nil {
+               t.Fatal(err)
+       } else if len(getSuffix) != 0 {
+               t.Errorf("spend commitment read from got garbage hex left")
+       }
+
+       if !testutil.DeepEqual(*sc, gotSC) {
+               t.Errorf("expected marshaled/unmarshaled spend commitment to be:\n%sgot:\n%s", spew.Sdump(*sc), spew.Sdump(gotSC))
+       }
+}