X-Git-Url: http://git.osdn.net/view?p=bytom%2Fvapor.git;a=blobdiff_plain;f=netsync%2Fchainmgr%2Ftx_keeper_test.go;fp=netsync%2Fchainmgr%2Ftx_keeper_test.go;h=7401af215dfda27750fbc0786217d9531c22b12a;hp=0000000000000000000000000000000000000000;hb=93dd7ceb9c426fa99362a81bf7e137d410d4ef92;hpb=e0bb6e8ab045e0449a652a9b924b815f59c34e63

diff --git a/netsync/chainmgr/tx_keeper_test.go b/netsync/chainmgr/tx_keeper_test.go
new file mode 100644
index 00000000..7401af21
--- /dev/null
+++ b/netsync/chainmgr/tx_keeper_test.go
@@ -0,0 +1,101 @@
+package chainmgr
+
+import (
+	"reflect"
+	"testing"
+	"time"
+
+	"github.com/davecgh/go-spew/spew"
+
+	"github.com/vapor/consensus"
+	"github.com/vapor/protocol"
+	"github.com/vapor/protocol/bc"
+	"github.com/vapor/protocol/bc/types"
+	"github.com/vapor/test/mock"
+)
+
+const txsNumber = 2000
+
+func getTransactions() []*types.Tx {
+	txs := []*types.Tx{}
+	for i := 0; i < txsNumber; i++ {
+		txInput := types.NewSpendInput(nil, bc.NewHash([32]byte{0x01}), *consensus.BTMAssetID, uint64(i), 1, []byte{0x51})
+		txInput.CommitmentSuffix = []byte{0, 1, 2}
+		txInput.WitnessSuffix = []byte{0, 1, 2}
+
+		tx := &types.Tx{
+
+			TxData: types.TxData{
+				//SerializedSize: uint64(i * 10),
+				Inputs: []*types.TxInput{
+					txInput,
+				},
+				Outputs: []*types.TxOutput{
+					types.NewIntraChainOutput(*consensus.BTMAssetID, uint64(i), []byte{0x6a}),
+				},
+			},
+			Tx: &bc.Tx{
+				ID: bc.Hash{V0: uint64(i), V1: uint64(i), V2: uint64(i), V3: uint64(i)},
+			},
+		}
+		txs = append(txs, tx)
+	}
+	return txs
+}
+
+func TestSyncMempool(t *testing.T) {
+	blocks := mockBlocks(nil, 5)
+	a := mockSync(blocks, &mock.Mempool{})
+	b := mockSync(blocks, &mock.Mempool{})
+
+	netWork := NewNetWork()
+	netWork.Register(a, "192.168.0.1", "test node A", consensus.SFFullNode)
+	netWork.Register(b, "192.168.0.2", "test node B", consensus.SFFullNode)
+	if B2A, A2B, err := netWork.HandsShake(a, b); err != nil {
+		t.Errorf("fail on peer hands shake %v", err)
+	} else {
+		go B2A.postMan()
+		go A2B.postMan()
+	}
+
+	go a.syncMempoolLoop()
+	a.syncMempool("test node B")
+	wantTxs := getTransactions()
+	a.txSyncCh <- &txSyncMsg{"test node B", wantTxs}
+
+	timeout := time.NewTimer(2 * time.Second)
+	defer timeout.Stop()
+	ticker := time.NewTicker(500 * time.Millisecond)
+	defer ticker.Stop()
+
+	gotTxs := []*protocol.TxDesc{}
+	for {
+		select {
+		case <-ticker.C:
+			gotTxs = b.mempool.GetTransactions()
+			if len(gotTxs) >= txsNumber {
+				goto out
+			}
+		case <-timeout.C:
+			t.Fatalf("mempool sync timeout")
+		}
+	}
+
+out:
+	if len(gotTxs) != txsNumber {
+		t.Fatalf("mempool sync txs num err. got:%d want:%d", len(gotTxs), txsNumber)
+	}
+
+	for i, gotTx := range gotTxs {
+		index := gotTx.Tx.Inputs[0].Amount()
+		if !reflect.DeepEqual(gotTx.Tx.Inputs[0].Amount(), wantTxs[index].Inputs[0].Amount()) {
+			t.Fatalf("mempool tx err. index:%d\n,gotTx:%s\n,wantTx:%s", i, spew.Sdump(gotTx.Tx.Inputs), spew.Sdump(wantTxs[0].Inputs))
+		}
+
+		if !reflect.DeepEqual(gotTx.Tx.Outputs[0].AssetAmount(), wantTxs[index].Outputs[0].AssetAmount()) {
+			t.Fatalf("mempool tx err. index:%d\n,gotTx:%s\n,wantTx:%s", i, spew.Sdump(gotTx.Tx.Outputs), spew.Sdump(wantTxs[0].Outputs))
+		}
+
+	}
+
+}