diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/blocklist-dht.js | 93 | ||||
-rw-r--r-- | test/blocklist-tracker.js (renamed from test/blocklist.js) | 15 |
2 files changed, 103 insertions, 5 deletions
diff --git a/test/blocklist-dht.js b/test/blocklist-dht.js new file mode 100644 index 0000000..b48d200 --- /dev/null +++ b/test/blocklist-dht.js @@ -0,0 +1,93 @@ +var auto = require('run-auto') +var DHT = require('bittorrent-dht/server') +var fs = require('fs') +var parseTorrent = require('parse-torrent') +var test = require('tape') +var WebTorrent = require('../') + +var leavesTorrent = fs.readFileSync(__dirname + '/torrents/leaves.torrent') +var leavesParsed = parseTorrent(leavesTorrent) + +test('blocklist blocks peers discovered via DHT', function (t) { + t.plan(6) + + // remove trackers from .torrent file + leavesParsed.announce = [] + leavesParsed.announceList = [] + + var dhtServer = new DHT({ bootstrap: false }) + + dhtServer.on('error', function (err) { + t.fail(err) + }) + + auto({ + dhtPort: function (cb) { + dhtServer.listen(function (port) { + cb(null, port) + }) + }, + + client1: ['dhtPort', function (cb, r) { + var client1 = new WebTorrent({ + tracker: false, + dht: { bootstrap: '127.0.0.1:' + r.dhtPort } + }) + client1.on('error', function (err) { t.fail(err) }) + + var torrent1 = client1.add(leavesParsed) + + client1.on('torrent', function () { + torrent1.on('dhtAnnounce', function () { + t.pass('client1 announced to dht') + cb(null, client1) + }) + }) + + torrent1.on('peer', function () { + t.fail('client1 should not find any peers') + }) + + torrent1.on('blocked-peer', function () { + t.fail('client1 should not block any peers') + }) + }], + + client2: ['client1', function (cb, r) { + var client2 = new WebTorrent({ + tracker: false, + dht: { bootstrap: '127.0.0.1:' + r.dhtPort }, + blocklist: [ '127.0.0.1' ] + }) + client2.on('error', function (err) { t.fail(err) }) + + var torrent2 = client2.add(leavesParsed) + + torrent2.on('blocked-peer', function () { + t.pass('client2 blocked connection to client1') + }) + + torrent2.on('dhtAnnounce', function () { + t.pass('client2 announced to dht') + cb(null, client2) + }) + + torrent2.on('peer', function () { + t.fail('client2 should not find any peers') + }) + }] + + }, function (err, r) { + if (err) throw err + + dhtServer.destroy(function () { + t.pass('dht server destroyed') + }) + r.client1.destroy(function () { + t.pass('client1 destroyed') + }) + r.client2.destroy(function () { + t.pass('client2 destroyed') + }) + }) +}) diff --git a/test/blocklist.js b/test/blocklist-tracker.js index 1c7199d..2212c32 100644 --- a/test/blocklist.js +++ b/test/blocklist-tracker.js @@ -8,10 +8,7 @@ var WebTorrent = require('../') var leavesTorrent = fs.readFileSync(__dirname + '/torrents/leaves.torrent') var leavesParsed = parseTorrent(leavesTorrent) -// TODO: test blocklist blocking messages to DHT nodes -// TODO: test all ways to specify blocklists - -test('blocklist blocks connections to peers', function (t) { +test('blocklist blocks peers discovered via tracker', function (t) { t.plan(8) auto({ @@ -43,6 +40,10 @@ test('blocklist blocks connections to peers', function (t) { t.pass('client1 found itself') cb(null, client1) }) + + torrent1.on('blocked-peer', function () { + t.fail('client1 should not block any peers') + }) }], client2: ['client1', function (cb) { @@ -55,9 +56,13 @@ test('blocklist blocks connections to peers', function (t) { var torrent2 = client2.add(leavesParsed) torrent2.on('blocked-peer', function () { - t.pass('client2 blocked connection to client1 and client2') + t.pass('client2 blocked connection') // 2x, once for each client cb(null, client2) }) + + torrent2.on('peer', function () { + t.fail('client2 should not find any peers') + }) }] }, function (err, r) { |