add blocklist tests

1 files changed, 3 insertions, 1 deletions

diff --git a/lib/torrent.js b/lib/torrent.js
index 9f6edde..bf110f9 100644
--- a/lib/torrent.js
+++ b/lib/torrent.js
@@ -326,19 +326,21 @@ Torrent.prototype.destroy = function (cb) {
 /**
  * Add a peer to the swarm
  * @param {string|SimplePeer} peer
+ * @return {boolean} true if peer was added, false if peer was blocked
  */
 Torrent.prototype.addPeer = function (peer) {
   var self = this
-
   // TODO: extract IP address from peer object and check blocklist
   if (typeof peer === 'string'
       && self.client.blocked
       && self.client.blocked.contains(addrToIPPort(peer)[0])) {
     self.numBlockedPeers += 1
     self.emit('blocked-peer', peer)
+    return false
   } else {
     self.emit('peer', peer)
     self.swarm.addPeer(peer)
+    return true
   }
 }