Add more browser tests

Run more of the tests in the browser. There's now a test/node and test/browser folder for tests that are specific to each environment. Anything in test/ will be run in both environments.

1 files changed, 280 insertions, 0 deletions

diff --git a/test/node/blocklist.js b/test/node/blocklist.js
new file mode 100644
index 0000000..8a7ee9c
--- /dev/null
+++ b/test/node/blocklist.js
@@ -0,0 +1,280 @@
+var common = require('../common')
+var fs = require('fs')
+var http = require('http')
+var path = require('path')
+var test = require('tape')
+var WebTorrent = require('../../')
+var zlib = require('zlib')
+
+var blocklistPath = path.resolve(__dirname, '..', 'fixtures', 'blocklist.txt')
+var blocklistGzipPath = path.resolve(__dirname, '..', 'fixtures', 'blocklist.txt.gz')
+
+function assertBlocked (t, torrent, addr) {
+  torrent.once('blockedPeer', function (_addr) {
+    t.equal(addr, _addr)
+  })
+  t.notOk(torrent.addPeer(addr))
+}
+
+function assertReachable (t, torrent, addr) {
+  torrent.once('peer', function (_addr) {
+    t.equal(addr, _addr)
+  })
+  t.ok(torrent.addPeer(addr))
+}
+
+test('blocklist (single IP)', function (t) {
+  t.plan(9)
+
+  var client = new WebTorrent({
+    dht: false,
+    tracker: false,
+    blocklist: [ '1.2.3.4' ]
+  })
+  client.on('error', function (err) { t.fail(err) })
+  client.on('warning', function (err) { t.fail(err) })
+
+  // blocklist isn't fully loaded until `ready` event
+  client.on('ready', function () {
+    client.add(common.leaves.parsedTorrent, function (torrent) {
+      assertBlocked(t, torrent, '1.2.3.4:1234')
+      assertBlocked(t, torrent, '1.2.3.4:6969')
+      assertReachable(t, torrent, '1.1.1.1:1234')
+      assertReachable(t, torrent, '1.1.1.1:6969')
+
+      client.destroy(function (err) {
+        t.error(err, 'client destroyed')
+      })
+    })
+  })
+})
+
+test('blocklist (array of IPs)', function (t) {
+  t.plan(13)
+
+  var client = new WebTorrent({
+    dht: false,
+    tracker: false,
+    blocklist: [ '1.2.3.4', '5.6.7.8' ]
+  })
+  .on('error', function (err) { t.fail(err) })
+  .on('warning', function (err) { t.fail(err) })
+  .on('ready', function () {
+    client.add(common.leaves.parsedTorrent, function (torrent) {
+      assertBlocked(t, torrent, '1.2.3.4:1234')
+      assertBlocked(t, torrent, '1.2.3.4:6969')
+      assertBlocked(t, torrent, '5.6.7.8:1234')
+      assertBlocked(t, torrent, '5.6.7.8:6969')
+      assertReachable(t, torrent, '1.1.1.1:1234')
+      assertReachable(t, torrent, '1.1.1.1:6969')
+
+      client.destroy(function (err) {
+        t.error(err, 'client destroyed')
+      })
+    })
+  })
+})
+
+// 48 asserts
+function assertList (t, torrent) {
+  assertBlocked(t, torrent, '1.2.3.0:1234')
+  assertBlocked(t, torrent, '1.2.3.0:6969')
+
+  assertBlocked(t, torrent, '1.2.3.1:1234')
+  assertBlocked(t, torrent, '1.2.3.1:6969')
+
+  assertBlocked(t, torrent, '1.2.3.1:1234')
+  assertBlocked(t, torrent, '1.2.3.1:6969')
+
+  assertBlocked(t, torrent, '1.2.3.254:1234')
+  assertBlocked(t, torrent, '1.2.3.254:6969')
+
+  assertBlocked(t, torrent, '1.2.3.255:1234')
+  assertBlocked(t, torrent, '1.2.3.255:6969')
+
+  assertBlocked(t, torrent, '5.6.7.0:1234')
+  assertBlocked(t, torrent, '5.6.7.0:6969')
+
+  assertBlocked(t, torrent, '5.6.7.128:1234')
+  assertBlocked(t, torrent, '5.6.7.128:6969')
+
+  assertBlocked(t, torrent, '5.6.7.255:1234')
+  assertBlocked(t, torrent, '5.6.7.255:6969')
+
+  assertReachable(t, torrent, '1.1.1.1:1234')
+  assertReachable(t, torrent, '1.1.1.1:6969')
+
+  assertReachable(t, torrent, '2.2.2.2:1234')
+  assertReachable(t, torrent, '2.2.2.2:6969')
+
+  assertReachable(t, torrent, '1.2.4.0:1234')
+  assertReachable(t, torrent, '1.2.4.0:6969')
+
+  assertReachable(t, torrent, '1.2.2.0:1234')
+  assertReachable(t, torrent, '1.2.2.0:6969')
+}
+
+test('blocklist (array of IP ranges)', function (t) {
+  t.plan(49)
+  var client = new WebTorrent({
+    dht: false,
+    tracker: false,
+    blocklist: [
+      { start: '1.2.3.0', end: '1.2.3.255' },
+      { start: '5.6.7.0', end: '5.6.7.255' }
+    ]
+  })
+  .on('error', function (err) { t.fail(err) })
+  .on('warning', function (err) { t.fail(err) })
+  .on('ready', function () {
+    client.add(common.leaves.parsedTorrent, function (torrent) {
+      assertList(t, torrent)
+      client.destroy(function (err) {
+        t.error(err, 'client destroyed')
+      })
+    })
+  })
+})
+
+test('blocklist (http url)', function (t) {
+  t.plan(51)
+  var server = http.createServer(function (req, res) {
+    // Check that WebTorrent declares a user agent
+    t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)
+
+    fs.createReadStream(blocklistPath).pipe(res)
+  })
+
+  server.listen(0, function () {
+    var port = server.address().port
+    var url = 'http://127.0.0.1:' + port
+    var client = new WebTorrent({
+      dht: false,
+      tracker: false,
+      blocklist: url
+    })
+    .on('error', function (err) { t.fail(err) })
+    .on('warning', function (err) { t.fail(err) })
+    .on('ready', function () {
+      client.add(common.leaves.parsedTorrent, function (torrent) {
+        assertList(t, torrent)
+        client.destroy(function (err) {
+          t.error(err, 'client destroyed')
+        })
+        server.close(function () {
+          t.pass('server closed')
+        })
+      })
+    })
+  })
+})
+
+test('blocklist (http url with gzip encoding)', function (t) {
+  t.plan(51)
+  var server = http.createServer(function (req, res) {
+    // Check that WebTorrent declares a user agent
+    t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)
+
+    res.setHeader('content-encoding', 'gzip')
+    fs.createReadStream(blocklistPath)
+      .pipe(zlib.createGzip())
+      .pipe(res)
+  })
+
+  server.listen(0, function () {
+    var port = server.address().port
+    var url = 'http://127.0.0.1:' + port
+    var client = new WebTorrent({
+      dht: false,
+      tracker: false,
+      blocklist: url
+    })
+    .on('error', function (err) { t.fail(err) })
+    .on('warning', function (err) { t.fail(err) })
+    .on('ready', function () {
+      client.add(common.leaves.parsedTorrent, function (torrent) {
+        assertList(t, torrent)
+        client.destroy(function (err) {
+          t.error(err, 'client destroyed')
+        })
+        server.close(function () {
+          t.pass('server closed')
+        })
+      })
+    })
+  })
+})
+
+test('blocklist (http url with deflate encoding)', function (t) {
+  t.plan(51)
+  var server = http.createServer(function (req, res) {
+    // Check that WebTorrent declares a user agent
+    t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)
+
+    res.setHeader('content-encoding', 'deflate')
+    fs.createReadStream(blocklistPath)
+      .pipe(zlib.createDeflate())
+      .pipe(res)
+  })
+
+  server.listen(0, function () {
+    var port = server.address().port
+    var url = 'http://127.0.0.1:' + port
+    var client = new WebTorrent({
+      dht: false,
+      tracker: false,
+      blocklist: url
+    })
+    .on('error', function (err) { t.fail(err) })
+    .on('warning', function (err) { t.fail(err) })
+    .on('ready', function () {
+      client.add(common.leaves.parsedTorrent, function (torrent) {
+        assertList(t, torrent)
+        client.destroy(function (err) {
+          t.error(err, 'client destroyed')
+        })
+        server.close(function () {
+          t.pass('server closed')
+        })
+      })
+    })
+  })
+})
+
+test('blocklist (fs path)', function (t) {
+  t.plan(49)
+  var client = new WebTorrent({
+    dht: false,
+    tracker: false,
+    blocklist: blocklistPath
+  })
+  .on('error', function (err) { t.fail(err) })
+  .on('warning', function (err) { t.fail(err) })
+  .on('ready', function () {
+    client.add(common.leaves.parsedTorrent, function (torrent) {
+      assertList(t, torrent)
+      client.destroy(function (err) {
+        t.error(err, 'client destroyed')
+      })
+    })
+  })
+})
+
+test('blocklist (fs path with gzip)', function (t) {
+  t.plan(49)
+  var client = new WebTorrent({
+    dht: false,
+    tracker: false,
+    blocklist: blocklistGzipPath
+  })
+  .on('error', function (err) { t.fail(err) })
+  .on('warning', function (err) { t.fail(err) })
+  .on('ready', function () {
+    client.add(common.leaves.parsedTorrent, function (torrent) {
+      assertList(t, torrent)
+      client.destroy(function (err) {
+        t.error(err, 'client destroyed')
+      })
+    })
+  })
+})