test: add read/write test for cross-chain input/output (#70)

[bytom/vapor.git] / protocol / bc / types / txinput_test.go

diff --git a/protocol/bc/types/txinput_test.go b/protocol/bc/types/txinput_test.go
index a824b54..6ff94a9 100644 (file)
--- a/protocol/bc/types/txinput_test.go
+++ b/protocol/bc/types/txinput_test.go
@@ -12,27 +12,26 @@ import (
        "github.com/vapor/testutil"
 )
 
-func TestSerializationIssuance(t *testing.T) {
+func TestSerializationSpend(t *testing.T) {
        arguments := [][]byte{
                []byte("arguments1"),
                []byte("arguments2"),
        }
-       issuance := NewIssuanceInput([]byte("nonce"), 254354, []byte("issuanceProgram"), arguments, []byte("assetDefinition"))
+       spend := NewSpendInput(arguments, testutil.MustDecodeHash("fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409"), testutil.MustDecodeAsset("fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a"), 254354, 3, []byte("spendProgram"))
 
        wantHex := strings.Join([]string{
-               "01",         // asset version
-               "2a",         // serialization length
-               "00",         // issuance type flag
-               "05",         // nonce length
-               "6e6f6e6365", // nonce
-               "a69849e11add96ac7053aad22ba2349a4abf5feb0475a0afcadff4e128be76cf", // assetID
+               "01", // asset version
+               "54", // input commitment length
+               "01", // spend type flag
+               "52", // spend commitment length
+               "fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409", // source id
+               "fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a", // assetID
                "92c30f", // amount
-               "38",     // input witness length
-               "0f",     // asset definition length
-               "6173736574446566696e6974696f6e", // asset definition
-               "01", // vm version
-               "0f", // issuanceProgram length
-               "69737375616e636550726f6772616d", // issuance program
+               "03",     // source position
+               "01",     // vm version
+               "0c",     // spend program length
+               "7370656e6450726f6772616d", // spend program
+               "17", // witness length
                "02", // argument array length
                "0a", // first argument length
                "617267756d656e747331", // first argument data
@@ -42,7 +41,7 @@ func TestSerializationIssuance(t *testing.T) {
 
        // Test convert struct to hex
        var buffer bytes.Buffer
-       if err := issuance.writeTo(&buffer); err != nil {
+       if err := spend.writeTo(&buffer); err != nil {
                t.Fatal(err)
        }
 
@@ -52,51 +51,53 @@ func TestSerializationIssuance(t *testing.T) {
        }
 
        // Test convert hex to struct
-       var gotIssuance TxInput
+       var gotSpend TxInput
        decodeHex, err := hex.DecodeString(wantHex)
        if err != nil {
                t.Fatal(err)
        }
 
-       if err := gotIssuance.readFrom(blockchain.NewReader(decodeHex)); err != nil {
+       if err := gotSpend.readFrom(blockchain.NewReader(decodeHex)); err != nil {
                t.Fatal(err)
        }
 
-       if !testutil.DeepEqual(*issuance, gotIssuance) {
-               t.Errorf("expected marshaled/unmarshaled txinput to be:\n%sgot:\n%s", spew.Sdump(*issuance), spew.Sdump(gotIssuance))
+       if !testutil.DeepEqual(*spend, gotSpend) {
+               t.Errorf("expected marshaled/unmarshaled txinput to be:\n%sgot:\n%s", spew.Sdump(*spend), spew.Sdump(gotSpend))
        }
 }
 
-func TestSerializationSpend(t *testing.T) {
+func TestSerializationCrossIn(t *testing.T) {
        arguments := [][]byte{
                []byte("arguments1"),
                []byte("arguments2"),
        }
-       spend := NewSpendInput(arguments, testutil.MustDecodeHash("fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409"), testutil.MustDecodeAsset("fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a"), 254354, 3, []byte("spendProgram"))
+       crossIn := NewCrossChainInput(arguments, testutil.MustDecodeHash("fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409"), testutil.MustDecodeAsset("fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a"), 254354, 3, []byte("crossInProgram"), []byte("whatever"))
 
        wantHex := strings.Join([]string{
                "01", // asset version
-               "54", // input commitment length
-               "01", // spend type flag
-               "52", // spend commitment length
+               "56", // input commitment length
+               "00", // cross-chain input type flag
+               "54", // cross-chain input commitment length
                "fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409", // source id
                "fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a", // assetID
                "92c30f", // amount
                "03",     // source position
                "01",     // vm version
-               "0c",     // spend program length
-               "7370656e6450726f6772616d", // spend program
+               "0e",     // spend program length
+               "63726f7373496e50726f6772616d", // spend program
                "17", // witness length
                "02", // argument array length
                "0a", // first argument length
                "617267756d656e747331", // first argument data
                "0a", // second argument length
                "617267756d656e747332", // second argument data
+               "08",               // asset definition length
+               "7768617465766572", // asset definition data
        }, "")
 
        // Test convert struct to hex
        var buffer bytes.Buffer
-       if err := spend.writeTo(&buffer); err != nil {
+       if err := crossIn.writeTo(&buffer); err != nil {
                t.Fatal(err)
        }
 
@@ -106,18 +107,75 @@ func TestSerializationSpend(t *testing.T) {
        }
 
        // Test convert hex to struct
-       var gotSpend TxInput
+       var gotCrossIn TxInput
        decodeHex, err := hex.DecodeString(wantHex)
        if err != nil {
                t.Fatal(err)
        }
 
-       if err := gotSpend.readFrom(blockchain.NewReader(decodeHex)); err != nil {
+       if err := gotCrossIn.readFrom(blockchain.NewReader(decodeHex)); err != nil {
                t.Fatal(err)
        }
 
-       if !testutil.DeepEqual(*spend, gotSpend) {
-               t.Errorf("expected marshaled/unmarshaled txinput to be:\n%sgot:\n%s", spew.Sdump(*spend), spew.Sdump(gotSpend))
+       if !testutil.DeepEqual(*crossIn, gotCrossIn) {
+               t.Errorf("expected marshaled/unmarshaled txinput to be:\n%sgot:\n%s", spew.Sdump(*crossIn), spew.Sdump(gotCrossIn))
+       }
+}
+
+func TestSerializationUnvote(t *testing.T) {
+       arguments := [][]byte{
+               []byte("arguments1"),
+               []byte("arguments2"),
+       }
+
+       unvote := NewUnvoteInput(arguments, testutil.MustDecodeHash("fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409"), testutil.MustDecodeAsset("fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a"), 254354, 3, []byte("spendProgram"), []byte("af594006a40837d9f028daabb6d589df0b9138daefad5683e5233c2646279217294a8d532e60863bcf196625a35fb8ceeffa3c09610eb92dcfb655a947f13269"))
+
+       wantHex := strings.Join([]string{
+               "01", // asset version
+               "54", // input commitment length
+               "03", // unvote type flag
+               "52", // unvote commitment length
+               "fad5195a0c8e3b590b86a3c0a95e7529565888508aecca96e9aeda633002f409", // source id
+               "fe9791d71b67ee62515e08723c061b5ccb952a80d804417c8aeedf7f633c524a", // assetID
+               "92c30f", // amount
+               "03",     // source position
+               "01",     // vm version
+               "0c",     // unvote program length
+               "7370656e6450726f6772616d", // unvote program
+               "9901", // witness length
+               "02",   // argument array length
+               "0a",   // first argument length
+               "617267756d656e747331", // first argument data
+               "0a", // second argument length
+               "617267756d656e747332", // second argument data
+               "8001",                 //xpub length
+               "6166353934303036613430383337643966303238646161626236643538396466306239313338646165666164353638336535323333633236343632373932313732393461386435333265363038363362636631393636323561333566623863656566666133633039363130656239326463666236353561393437663133323639", //voter xpub
+       }, "")
+
+       // Test convert struct to hex
+       var buffer bytes.Buffer
+       if err := unvote.writeTo(&buffer); err != nil {
+               t.Fatal(err)
+       }
+
+       gotHex := hex.EncodeToString(buffer.Bytes())
+       if gotHex != wantHex {
+               t.Errorf("serialization bytes = %s want %s", gotHex, wantHex)
+       }
+
+       // Test convert hex to struct
+       var gotUnvote TxInput
+       decodeHex, err := hex.DecodeString(wantHex)
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       if err := gotUnvote.readFrom(blockchain.NewReader(decodeHex)); err != nil {
+               t.Fatal(err)
+       }
+
+       if !testutil.DeepEqual(*unvote, gotUnvote) {
+               t.Errorf("expected marshaled/unmarshaled txinput to be:\n%sgot:\n%s", spew.Sdump(*unvote), spew.Sdump(gotUnvote))
        }
 }