var Client = require('../') var fs = require('fs') var parseTorrent = require('parse-torrent') var portfinder = require('portfinder') var Server = require('../').Server var test = require('tape') var torrent = fs.readFileSync(__dirname + '/torrents/bitlove-intro.torrent') var parsedTorrent = parseTorrent(torrent) var peerId1 = new Buffer('01234567890123456789') var peerId2 = new Buffer('12345678901234567890') var peerId3 = new Buffer('23456789012345678901') var announceUrl = '' var port = 6881 function createServer (t, serverType, cb) { var opts = serverType === 'http' ? { udp: false } : { http: false } var server = new Server(opts) server.on('error', function (err) { t.error(err) }) server.on('warning', function (err) { t.error(err) }) portfinder.getPort(function (err, port) { if (err) return t.error(err) announceUrl = serverType + '://127.0.0.1:' + port + '/announce' parsedTorrent.announce = [ announceUrl ] server.listen(port) cb(server) }) } function testClientStart (t, serverType) { t.plan(5) createServer(t, serverType, function (server) { var client = new Client(peerId1, port, parsedTorrent) client.on('error', function (err) { t.error(err) }) client.on('warning', function (err) { t.error(err) }) client.once('update', function (data) { t.equal(data.announce, announceUrl) t.equal(typeof data.complete, 'number') t.equal(typeof data.incomplete, 'number') }) client.once('peer', function (addr) { t.pass('there is at least one peer') client.stop() client.once('update', function () { server.close(function () { t.pass('server close') }) }) }) client.start() }) } test('http: client.start()', function (t) { testClientStart(t, 'http') }) test('udp: client.start()', function (t) { testClientStart(t, 'udp') }) function testClientStop (t, serverType) { t.plan(4) createServer(t, serverType, function (server) { var client = new Client(peerId1, port, parsedTorrent) client.on('error', function (err) { t.error(err) }) client.on('warning', function (err) { t.error(err) }) client.start() setTimeout(function () { client.stop() client.once('update', function (data) { // receive one final update after calling stop t.equal(data.announce, announceUrl) t.equal(typeof data.complete, 'number') t.equal(typeof data.incomplete, 'number') server.close(function () { t.pass('server close') }) }) }, 1000) }) } test('http: client.stop()', function (t) { testClientStop(t, 'http') }) test('udp: client.stop()', function (t) { testClientStop(t, 'udp') }) function testClientUpdate (t, serverType) { t.plan(4) createServer(t, serverType, function (server) { var client = new Client(peerId1, port, parsedTorrent, { interval: 5000 }) client.on('error', function (err) { t.error(err) }) client.on('warning', function (err) { t.error(err) }) client.start() client.once('update', function () { client.once('update', function (data) { // received an update! t.equal(data.announce, announceUrl) t.equal(typeof data.complete, 'number') t.equal(typeof data.incomplete, 'number') client.stop() client.once('update', function () { server.close(function () { t.pass('server close') }) }) }) }) }) } test('http: client.update()', function (t) { testClientUpdate(t, 'http') }) test('udp: client.update()', function (t) { testClientUpdate(t, 'udp') }) function testClientScrape (t, serverType) { t.plan(5) createServer(t, serverType, function (server) { var client = new Client(peerId1, port, parsedTorrent) client.on('error', function (err) { t.error(err) }) client.on('warning', function (err) { t.error(err) }) client.once('scrape', function (data) { t.equal(data.announce, announceUrl) t.equal(typeof data.complete, 'number') t.equal(typeof data.incomplete, 'number') t.equal(typeof data.downloaded, 'number') server.close(function () { t.pass('server close') }) }) client.scrape() }) } test('http: client.scrape()', function (t) { testClientScrape(t, 'http') }) test('udp: client.scrape()', function (t) { testClientScrape(t, 'udp') }) function testClientAnnounceWithNumWant (t, serverType) { t.plan(1) createServer(t, serverType, function (server) { var client1 = new Client(peerId1, port, parsedTorrent) client1.on('error', function (err) { t.error(err) }) client1.start() client1.once('update', function (data) { var client2 = new Client(peerId2, port + 1, parsedTorrent) client2.on('error', function (err) { t.error(err) }) client2.start() client2.once('update', function () { var client3 = new Client(peerId3, port + 2, parsedTorrent, { numWant: 1 }) client3.on('error', function (err) { t.error(err) }) client3.start() client3.on('peer', function () { t.pass('got one peer (this should only fire once)') client1.stop() client2.stop() client3.stop() server.close() }) }) }) }) } test('http: client announce with numWant', function (t) { testClientAnnounceWithNumWant(t, 'http') }) test('udp: client announce with numWant', function (t) { testClientAnnounceWithNumWant(t, 'udp') })