X-Git-Url: http://git.osdn.net/view?p=bytom%2Fvapor.git;a=blobdiff_plain;f=netsync%2Fchainmgr%2Ftool_test.go;h=63816417f6189df4d1c10fb46579cc26b3db9280;hp=e3549841e33695be6a3de268fb91275b5e71cfa6;hb=4607cea3c6d3b1501ae1edb2c078b9ae0ee4bc79;hpb=93dd7ceb9c426fa99362a81bf7e137d410d4ef92

diff --git a/netsync/chainmgr/tool_test.go b/netsync/chainmgr/tool_test.go
index e3549841..63816417 100644
--- a/netsync/chainmgr/tool_test.go
+++ b/netsync/chainmgr/tool_test.go
@@ -5,7 +5,7 @@ import (
 	"math/rand"
 	"net"
 
-	wire "github.com/tendermint/go-wire"
+	"github.com/tendermint/go-wire"
 	"github.com/tendermint/tmlibs/flowrate"
 
 	"github.com/vapor/consensus"
@@ -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
@@ -120,7 +123,8 @@ func (nw *NetWork) HandsShake(nodeA, nodeB *Manager) (*P2PPeer, *P2PPeer, error)
 
 	nodeA.AddPeer(&A2B)
 	nodeB.AddPeer(&B2A)
-
+	nodeA.SendStatus(B2A.srcPeer)
+	nodeB.SendStatus(A2B.srcPeer)
 	A2B.setAsync(true)
 	B2A.setAsync(true)
 	return &B2A, &A2B, nil