test for partner removal
License: MIT Signed-off-by: Jeromy <jeromyj@gmail.com>
Jeromy committed
May 19, 2017 at 19:05 UTC
331e60ba345e079ea3310f29ff21a60339c52879
1 file changed
+5
exchange/bitswap/decision/engine_test.go
+5
@@ -89,6 +89,11 @@ func TestPeerIsAddedToPeersWhenMessageReceivedOrSent(t *testing.T) {
89
if !peerIsPartner(sanfrancisco.Peer, seattle.Engine) {
90
t.Fatal("Peer wasn't added as a Partner")
91
}
92
+
93
+ seattle.Engine.PeerDisconnected(sanfrancisco.Peer)
94
+ if peerIsPartner(sanfrancisco.Peer, seattle.Engine) {
95
+ t.Fatal("expected peer to be removed")
96
+ }
97
}
98
99
func peerIsPartner(p peer.ID, e *Engine) bool {