X-Git-Url: http://git.osdn.net/view?p=bytom%2Fvapor.git;a=blobdiff_plain;f=netsync%2Fchainmgr%2Ftool_test.go;h=db17b4ff62e3af3495922a05183559995e0e2668;hp=dba3c89c4b5261bc97a13814107c5a498b5aec62;hb=54373c1a3efe0e373ec1605840a4363e4b246c46;hpb=c071c7f9648e8e8f39c1826ccb4e5d2ba5e8efd7

diff --git a/netsync/chainmgr/tool_test.go b/netsync/chainmgr/tool_test.go
index dba3c89c..db17b4ff 100644
--- a/netsync/chainmgr/tool_test.go
+++ b/netsync/chainmgr/tool_test.go
@@ -89,8 +89,11 @@ func NewPeerSet() *PeerSet {
 	return &PeerSet{}
 }
 
-func (ps *PeerSet) AddBannedPeer(string) error { return nil }
-func (ps *PeerSet) StopPeerGracefully(string)  {}
+func (ps *PeerSet) IsBanned(peerID string, level byte, reason string) bool {
+	return false
+}
+
+func (ps *PeerSet) StopPeerGracefully(string) {}
 
 type NetWork struct {
 	nodes map[*Manager]P2PPeer
@@ -150,8 +153,8 @@ func mockBlocks(startBlock *types.Block, height uint64) []*types.Block {
 	return blocks
 }
 
-func mockSync(blocks []*types.Block) *Manager {
-	chain := mock.NewChain()
+func mockSync(blocks []*types.Block, mempool *mock.Mempool) *Manager {
+	chain := mock.NewChain(mempool)
 	peers := peers.NewPeerSet(NewPeerSet())
 	chain.SetBestBlockHeader(&blocks[len(blocks)-1].BlockHeader)
 	for _, block := range blocks {
@@ -162,6 +165,8 @@ func mockSync(blocks []*types.Block) *Manager {
 		chain:       chain,
 		blockKeeper: newBlockKeeper(chain, peers),
 		peers:       peers,
+		mempool:     mempool,
+		txSyncCh:    make(chan *txSyncMsg),
 	}
 }