test for partner removal

whyrusleeping · whyrusleeping · commit 331e60ba345e · 2017-05-19T19:05:12.000-07:00
License: MIT
Signed-off-by: Jeromy &lt;jeromyj@gmail.com&gt;
diff --git a/exchange/bitswap/decision/engine_test.go b/exchange/bitswap/decision/engine_test.go
@@ -89,6 +89,11 @@ func TestPeerIsAddedToPeersWhenMessageReceivedOrSent(t *testing.T) {
 	if !peerIsPartner(sanfrancisco.Peer, seattle.Engine) {
 		t.Fatal("Peer wasn't added as a Partner")
 	}
+
+	seattle.Engine.PeerDisconnected(sanfrancisco.Peer)
+	if peerIsPartner(sanfrancisco.Peer, seattle.Engine) {
+		t.Fatal("expected peer to be removed")
+	}
 }
 
 func peerIsPartner(p peer.ID, e *Engine) bool {

Original file line number	Diff line number	Diff line change
`@@ -89,6 +89,11 @@ func TestPeerIsAddedToPeersWhenMessageReceivedOrSent(t *testing.T) {`
`89`	`89`	`if !peerIsPartner(sanfrancisco.Peer, seattle.Engine) {`
`90`	`90`	`t.Fatal("Peer wasn't added as a Partner")`
`91`	`91`	`}`
	`92`	`+`
	`93`	`+ seattle.Engine.PeerDisconnected(sanfrancisco.Peer)`
	`94`	`+ if peerIsPartner(sanfrancisco.Peer, seattle.Engine) {`
	`95`	`+ t.Fatal("expected peer to be removed")`
	`96`	`+ }`
`92`	`97`	`}`
`93`	`98`
`94`	`99`	`func peerIsPartner(p peer.ID, e *Engine) bool {`