var Buffer = require('safe-buffer').Buffer var Client = require('../') var common = require('./common') var test = require('tape') var wrtc = require('electron-webrtc')() var wrtcReady = false wrtc.electronDaemon.once('ready', function () { wrtcReady = true }) var infoHash = '4cb67059ed6bd08362da625b3ae77f6f4a075705' var peerId = Buffer.from('01234567890123456789') var peerId2 = Buffer.from('12345678901234567890') var peerId3 = Buffer.from('23456789012345678901') function serverTest (t, serverType, serverFamily) { t.plan(36) var hostname = serverFamily === 'inet6' ? '[::1]' : '127.0.0.1' var clientIp = serverFamily === 'inet6' ? '::1' : '127.0.0.1' var opts = { serverType: serverType, peersCacheLength: 2 } common.createServer(t, opts, function (server) { // Not using announceUrl param from `common.createServer()` since we // want to control IPv4 vs IPv6. var port = server[serverType].address().port var announceUrl = serverType + '://' + hostname + ':' + port + '/announce' var client1 = new Client({ infoHash: infoHash, announce: [ announceUrl ], peerId: peerId, port: 6881, wrtc: wrtc }) if (serverType === 'ws') common.mockWebsocketTracker(client1) client1.start() server.once('start', function () { t.pass('got start message from client1') }) client1.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 0) t.equal(data.incomplete, 1) server.getSwarm(infoHash, function (err, swarm) { t.error(err) t.equal(Object.keys(server.torrents).length, 1) t.equal(swarm.complete, 0) t.equal(swarm.incomplete, 1) t.equal(swarm.peers.length, 1) var id = serverType === 'ws' ? peerId.toString('hex') : hostname + ':6881' var peer = swarm.peers.peek(id) t.equal(peer.type, serverType) t.equal(peer.ip, clientIp) t.equal(peer.peerId, peerId.toString('hex')) t.equal(peer.complete, false) if (serverType === 'ws') { t.equal(typeof peer.port, 'number') t.ok(peer.socket) } else { t.equal(peer.port, 6881) t.notOk(peer.socket) } client1.complete() client1.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 1) t.equal(data.incomplete, 0) client1.scrape() client1.once('scrape', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 1) t.equal(data.incomplete, 0) t.equal(typeof data.downloaded, 'number') var client2 = new Client({ infoHash: infoHash, announce: [ announceUrl ], peerId: peerId2, port: 6882, wrtc: wrtc }) if (serverType === 'ws') common.mockWebsocketTracker(client2) client2.start() server.once('start', function () { t.pass('got start message from client2') }) client2.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 1) t.equal(data.incomplete, 1) var client3 = new Client({ infoHash: infoHash, announce: [ announceUrl ], peerId: peerId3, port: 6880, wrtc: wrtc }) if (serverType === 'ws') common.mockWebsocketTracker(client3) client3.start() server.once('start', function () { t.pass('got start message from client3') }) client3.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 1) t.equal(data.incomplete, 2) client2.stop() client2.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 1) t.equal(data.incomplete, 1) client2.destroy(function () { client3.stop() client3.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(data.complete, 1) t.equal(data.incomplete, 0) client3.destroy(function () { client1.destroy(function () { server.close() }) }) }) }) }) }) }) }) }) }) }) }) } test('http ipv4 server', function (t) { serverTest(t, 'http', 'inet') }) test('http ipv6 server', function (t) { serverTest(t, 'http', 'inet6') }) test('udp server', function (t) { serverTest(t, 'udp', 'inet') }) test('ws server', function (t) { if (wrtcReady) { runTest() } else { wrtc.electronDaemon.once('ready', runTest) } function runTest () { t.once('end', function () { wrtc.close() }) serverTest(t, 'ws', 'inet') } })