merge master

[bytom/vapor.git] / test / builder_test.go

diff --git a/test/builder_test.go b/test/builder_test.go
new file mode 100644 (file)
index 0000000..4328dbc
--- /dev/null
+++ b/test/builder_test.go
@@ -0,0 +1,130 @@
+package test
+
+import (
+       "testing"
+
+       acc "github.com/vapor/account"
+       "github.com/vapor/blockchain/signers"
+       "github.com/vapor/consensus"
+       "github.com/vapor/crypto/ed25519/chainkd"
+       "github.com/vapor/testutil"
+)
+
+var (
+       //chainTxUtxoNum maximum utxo quantity in a tx
+       chainTxUtxoNum = 5
+       //chainTxMergeGas chain tx gas
+       chainTxMergeGas = uint64(10000000)
+)
+
+func TestBuildBtmTxChain(t *testing.T) {
+       chainTxUtxoNum = 3
+       m := mockAccountManager(t)
+       cases := []struct {
+               inputUtxo  []uint64
+               wantInput  [][]uint64
+               wantOutput [][]uint64
+               wantUtxo   uint64
+       }{
+               {
+                       inputUtxo:  []uint64{5},
+                       wantInput:  [][]uint64{},
+                       wantOutput: [][]uint64{},
+                       wantUtxo:   5 * chainTxMergeGas,
+               },
+               {
+                       inputUtxo: []uint64{5, 4},
+                       wantInput: [][]uint64{
+                               []uint64{5, 4},
+                       },
+                       wantOutput: [][]uint64{
+                               []uint64{8},
+                       },
+                       wantUtxo: 8 * chainTxMergeGas,
+               },
+               {
+                       inputUtxo: []uint64{5, 4, 1, 1},
+                       wantInput: [][]uint64{
+                               []uint64{5, 4, 1, 1},
+                               []uint64{1, 9},
+                       },
+                       wantOutput: [][]uint64{
+                               []uint64{10},
+                               []uint64{9},
+                       },
+                       wantUtxo: 10 * chainTxMergeGas,
+               },
+               {
+                       inputUtxo: []uint64{22, 123, 53, 234, 23, 4, 2423, 24, 23, 43, 34, 234, 234, 24},
+                       wantInput: [][]uint64{
+                               []uint64{22, 123, 53, 234, 23},
+                               []uint64{4, 2423, 24, 23, 43},
+                               []uint64{34, 234, 234, 24, 454},
+                               []uint64{2516, 979},
+                               []uint64{234, 24, 197},
+                               []uint64{260, 2469, 310},
+                               []uint64{454, 3038},
+                       },
+                       wantOutput: [][]uint64{
+                               []uint64{454},
+                               []uint64{2516},
+                               []uint64{979},
+                               []uint64{3494},
+                               []uint64{454},
+                               []uint64{3038},
+                               []uint64{3491},
+                       },
+                       wantUtxo: 3494 * chainTxMergeGas,
+               },
+       }
+
+       acct, err := m.Manager.Create([]chainkd.XPub{testutil.TestXPub}, 1, "testAccount", signers.BIP0044)
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       acp, err := m.Manager.CreateAddress(acct.ID, false)
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       for caseIndex, c := range cases {
+               utxos := []*acc.UTXO{}
+               for _, amount := range c.inputUtxo {
+                       utxos = append(utxos, &acc.UTXO{
+                               Amount:         amount * chainTxMergeGas,
+                               AssetID:        *consensus.BTMAssetID,
+                               Address:        acp.Address,
+                               ControlProgram: acp.ControlProgram,
+                       })
+               }
+
+               tpls, gotUtxo, err := m.Manager.BuildBtmTxChain(utxos, acct.Signer)
+               if err != nil {
+                       t.Fatal(err)
+               }
+
+               for i, tpl := range tpls {
+                       gotInput := []uint64{}
+                       for _, input := range tpl.Transaction.Inputs {
+                               gotInput = append(gotInput, input.Amount()/chainTxMergeGas)
+                       }
+
+                       gotOutput := []uint64{}
+                       for _, output := range tpl.Transaction.Outputs {
+                               gotOutput = append(gotOutput, output.AssetAmount().Amount/chainTxMergeGas)
+                       }
+
+                       if !testutil.DeepEqual(c.wantInput[i], gotInput) {
+                               t.Errorf("case %d tx %d input got %d want %d", caseIndex, i, gotInput, c.wantInput[i])
+                       }
+                       if !testutil.DeepEqual(c.wantOutput[i], gotOutput) {
+                               t.Errorf("case %d tx %d output got %d want %d", caseIndex, i, gotOutput, c.wantOutput[i])
+                       }
+               }
+
+               if c.wantUtxo != gotUtxo.Amount {
+                       t.Errorf("case %d got utxo=%d want utxo=%d", caseIndex, gotUtxo.Amount, c.wantUtxo)
+               }
+       }
+}