fix tests to work with new bittorrent-dht

1 files changed, 4 insertions, 3 deletions

diff --git a/test/blocklist-dht.js b/test/blocklist-dht.js
index 9ddb316..bcbf03e 100644
--- a/test/blocklist-dht.js
+++ b/test/blocklist-dht.js
@@ -1,6 +1,7 @@
 var auto = require('run-auto')
 var DHT = require('bittorrent-dht/server')
 var fs = require('fs')
+var networkAddress = require('network-address')
 var parseTorrent = require('parse-torrent')
 var test = require('tape')
 var WebTorrent = require('../')
@@ -13,7 +14,7 @@ leavesParsed.announce = []
 leavesParsed.announceList = []
 
 test('blocklist blocks peers discovered via DHT', function (t) {
-  t.plan(6)
+  t.plan(7)
 
   var dhtServer = new DHT({ bootstrap: false })
 
@@ -58,7 +59,7 @@ test('blocklist blocks peers discovered via DHT', function (t) {
       var client2 = new WebTorrent({
         tracker: false,
         dht: { bootstrap: '127.0.0.1:' + r.dhtPort },
-        blocklist: [ '127.0.0.1' ]
+        blocklist: [ '127.0.0.1', networkAddress.ipv4() ]
       })
       client2.on('error', function (err) { t.fail(err) })
 
@@ -73,7 +74,7 @@ test('blocklist blocks peers discovered via DHT', function (t) {
         cb(null, client2)
       })
 
-      torrent2.on('peer', function () {
+      torrent2.on('peer', function (addr) {
         t.fail('client2 should not find any peers')
       })
     }]