From 7d98cfd3d6d3788a70df5db7167abc017d45ed6b Mon Sep 17 00:00:00 2001
From: Alexander Simmerl <a.simmerl@gmail.com>
Date: Wed, 23 May 2018 01:24:27 +0200
Subject: [PATCH] Test duplicate IP guard in peer set

---
 p2p/peer_set_test.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/p2p/peer_set_test.go b/p2p/peer_set_test.go
index ff2bdbad5..b043a36b2 100644
--- a/p2p/peer_set_test.go
+++ b/p2p/peer_set_test.go
@@ -146,6 +146,17 @@ func TestPeerSetAddDuplicate(t *testing.T) {
 }
 
 func TestPeerSetAddDuplicateIP(t *testing.T) {
+	t.Parallel()
+
+	peerSet := NewPeerSet()
+
+	if err := peerSet.Add(randPeer(net.IP{172, 0, 0, 1})); err != nil {
+		t.Fatal(err)
+	}
+
+	// Add peer with same IP.
+	err := peerSet.Add(randPeer(net.IP{172, 0, 0, 1}))
+	assert.Equal(t, ErrSwitchDuplicatePeerIP{IP: net.IP{172, 0, 0, 1}}, err)
 }
 
 func TestPeerSetGet(t *testing.T) {