1 files changed, 56 insertions, 0 deletions
diff --git a/test/node/extensions.js b/test/node/extensions.js
new file mode 100644
index 0000000..399c1e3
--- /dev/null
+++ b/test/node/extensions.js
@@ -0,0 +1,56 @@
+var common = require('../common')
+var test = require('tape')
+var WebTorrent = require('../../')
+
+test('extension support', function (t) {
+  t.plan(6)
+  var extendedHandshakes = 0
+
+  function Extension (wire) {
+    wire.extendedHandshake.test = 'Hello, World!'
+  }
+
+  Extension.prototype.name = 'wt_test'
+  Extension.prototype.onExtendedHandshake = function (extendedHandshake) {
+    extendedHandshakes += 1
+
+    t.equal(
+      extendedHandshake.test.toString(), 'Hello, World!',
+      'handshake.test === Hello, World!'
+    )
+
+    if (extendedHandshakes === 2) {
+      client1.destroy(function (err) {
+        t.error(err, 'client1 destroyed')
+      })
+      client2.destroy(function (err) {
+        t.error(err, 'client2 destroyed')
+      })
+    }
+  }
+
+  var client1 = new WebTorrent({ dht: false, tracker: false })
+
+  client1.on('error', function (err) { t.fail(err) })
+  client1.on('warning', function (err) { t.fail(err) })
+
+  var client2 = new WebTorrent({ dht: false, tracker: false })
+
+  client2.on('error', function (err) { t.fail(err) })
+  client2.on('warning', function (err) { t.fail(err) })
+
+  client1.add(common.leaves.parsedTorrent, function (torrent1) {
+    torrent1.on('wire', function (wire) {
+      t.pass('client1 onWire')
+      wire.use(Extension)
+    })
+    var torrent2 = client2.add(common.leaves.parsedTorrent.infoHash)
+    torrent2.on('wire', function (wire) {
+      t.pass('client2 onWire')
+      wire.use(Extension)
+    })
+    client2.on('listening', function () {
+      torrent2.addPeer('127.0.0.1:' + client1.address().port)
+    })
+  })
+})