bittorrent-tracker/test/client.js

415 lines
10 KiB
JavaScript
Raw Normal View History

var Buffer = require('safe-buffer').Buffer
var Client = require('../')
var common = require('./common')
2016-03-16 19:22:33 +00:00
var fixtures = require('webtorrent-fixtures')
2014-03-26 08:17:59 +00:00
var test = require('tape')
var peerId1 = Buffer.from('01234567890123456789')
var peerId2 = Buffer.from('12345678901234567890')
var peerId3 = Buffer.from('23456789012345678901')
2014-03-26 08:17:59 +00:00
var port = 6881
2014-07-11 10:50:35 +00:00
function testClientStart (t, serverType) {
2016-02-29 22:48:23 +00:00
t.plan(4)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
wrtc: {}
})
2014-03-26 08:17:59 +00:00
if (serverType === 'ws') common.mockWebsocketTracker(client)
2016-02-29 22:48:23 +00:00
client.on('error', function (err) { t.error(err) })
client.on('warning', function (err) { t.error(err) })
2014-07-11 04:58:19 +00:00
2014-03-26 08:17:59 +00:00
client.once('update', function (data) {
t.equal(data.announce, announceUrl)
2014-03-26 08:17:59 +00:00
t.equal(typeof data.complete, 'number')
t.equal(typeof data.incomplete, 'number')
2014-05-24 04:14:19 +00:00
client.stop()
client.once('update', function () {
t.pass('got response to stop')
server.close()
client.destroy()
2014-05-24 04:14:19 +00:00
})
})
2014-03-26 08:17:59 +00:00
2014-05-24 04:14:19 +00:00
client.start()
})
2014-07-11 10:50:35 +00:00
}
test('http: client.start()', function (t) {
testClientStart(t, 'http')
2014-05-24 04:14:19 +00:00
})
2014-03-26 08:17:59 +00:00
2014-07-11 10:50:35 +00:00
test('udp: client.start()', function (t) {
testClientStart(t, 'udp')
})
2014-03-26 08:17:59 +00:00
2016-02-29 22:48:23 +00:00
test('ws: client.start()', function (t) {
testClientStart(t, 'ws')
})
2014-07-11 10:50:35 +00:00
function testClientStop (t, serverType) {
t.plan(3)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
wrtc: {}
})
2014-03-26 08:17:59 +00:00
if (serverType === 'ws') common.mockWebsocketTracker(client)
2016-02-29 22:48:23 +00:00
client.on('error', function (err) { t.error(err) })
client.on('warning', function (err) { t.error(err) })
2014-07-11 04:58:19 +00:00
2014-05-24 04:14:19 +00:00
client.start()
2014-03-26 08:17:59 +00:00
2014-05-24 04:14:19 +00:00
setTimeout(function () {
2014-03-26 08:17:59 +00:00
client.stop()
2014-05-24 04:14:19 +00:00
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()
client.destroy()
2014-05-24 04:14:19 +00:00
})
}, 1000)
})
2014-07-11 10:50:35 +00:00
}
test('http: client.stop()', function (t) {
testClientStop(t, 'http')
2014-05-24 04:14:19 +00:00
})
2014-07-11 10:50:35 +00:00
test('udp: client.stop()', function (t) {
testClientStop(t, 'udp')
})
2014-05-24 04:14:19 +00:00
2016-02-29 22:48:23 +00:00
test('ws: client.stop()', function (t) {
testClientStop(t, 'ws')
})
2014-07-11 10:50:35 +00:00
function testClientUpdate (t, serverType) {
t.plan(4)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
wrtc: {}
})
2014-05-24 04:14:19 +00:00
if (serverType === 'ws') common.mockWebsocketTracker(client)
2016-02-29 22:48:23 +00:00
client.on('error', function (err) { t.error(err) })
client.on('warning', function (err) { t.error(err) })
2014-03-26 08:17:59 +00:00
2016-02-29 22:48:23 +00:00
client.setInterval(2000)
2014-07-11 04:58:19 +00:00
2014-05-24 04:14:19 +00:00
client.start()
client.once('update', function () {
2015-12-05 08:06:23 +00:00
client.setInterval(2000)
2015-03-24 08:02:10 +00:00
// after interval (2s), we should get another update
2014-05-24 04:14:19 +00:00
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 () {
t.pass('got response to stop')
server.close()
client.destroy()
2014-05-24 04:14:19 +00:00
})
})
})
2014-03-26 08:17:59 +00:00
})
2014-07-11 10:50:35 +00:00
}
test('http: client.update()', function (t) {
testClientUpdate(t, 'http')
2014-03-26 08:17:59 +00:00
})
2014-07-11 10:50:35 +00:00
test('udp: client.update()', function (t) {
testClientUpdate(t, 'udp')
})
2016-02-29 22:48:23 +00:00
test('ws: client.update()', function (t) {
testClientUpdate(t, 'ws')
})
2014-07-11 10:50:35 +00:00
function testClientScrape (t, serverType) {
t.plan(4)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
wrtc: {}
})
if (serverType === 'ws') common.mockWebsocketTracker(client)
2016-02-29 22:48:23 +00:00
client.on('error', function (err) { t.error(err) })
client.on('warning', function (err) { t.error(err) })
2014-07-11 04:58:19 +00:00
2014-05-24 04:14:19 +00:00
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')
2014-05-12 00:32:57 +00:00
server.close()
client.destroy()
2014-05-24 04:14:19 +00:00
})
client.scrape()
})
2014-07-11 10:50:35 +00:00
}
test('http: client.scrape()', function (t) {
testClientScrape(t, 'http')
})
test('udp: client.scrape()', function (t) {
testClientScrape(t, 'udp')
})
test('ws: client.scrape()', function (t) {
testClientScrape(t, 'ws')
})
2016-02-29 22:48:23 +00:00
function testClientAnnounceWithParams (t, serverType) {
t.plan(5)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
wrtc: {}
})
server.on('start', function (peer, params) {
t.equal(params.testParam, 'this is a test')
})
if (serverType === 'ws') common.mockWebsocketTracker(client)
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.stop()
client.once('update', function () {
t.pass('got response to stop')
server.close()
client.destroy()
})
})
client.start({
testParam: 'this is a test'
})
})
}
test('http: client.announce() with params', function (t) {
testClientAnnounceWithParams(t, 'http')
})
test('ws: client.announce() with params', function (t) {
testClientAnnounceWithParams(t, 'ws')
})
function testClientGetAnnounceOpts (t, serverType) {
t.plan(5)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
getAnnounceOpts: function () {
return {
testParam: 'this is a test'
}
},
wrtc: {}
})
server.on('start', function (peer, params) {
t.equal(params.testParam, 'this is a test')
})
if (serverType === 'ws') common.mockWebsocketTracker(client)
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.stop()
client.once('update', function () {
t.pass('got response to stop')
server.close()
client.destroy()
})
})
client.start()
})
}
test('http: client `opts.getAnnounceOpts`', function (t) {
testClientGetAnnounceOpts(t, 'http')
})
test('ws: client `opts.getAnnounceOpts`', function (t) {
testClientGetAnnounceOpts(t, 'ws')
})
2014-07-11 10:50:48 +00:00
function testClientAnnounceWithNumWant (t, serverType) {
t.plan(4)
2016-03-16 19:22:33 +00:00
common.createServer(t, serverType, function (server, announceUrl) {
var client1 = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: [ announceUrl ],
peerId: peerId1,
port: port,
wrtc: {}
})
2016-02-29 22:48:23 +00:00
if (serverType === 'ws') common.mockWebsocketTracker(client1)
2016-02-29 22:48:23 +00:00
client1.on('error', function (err) { t.error(err) })
client1.on('warning', function (err) { t.error(err) })
2014-07-11 10:50:48 +00:00
client1.start()
2014-11-26 12:11:49 +00:00
client1.once('update', function () {
var client2 = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId2,
port: port + 1,
wrtc: {}
})
2016-02-29 22:48:23 +00:00
if (serverType === 'ws') common.mockWebsocketTracker(client2)
2016-02-29 22:48:23 +00:00
client2.on('error', function (err) { t.error(err) })
client2.on('warning', function (err) { t.error(err) })
2014-07-11 10:50:48 +00:00
client2.start()
client2.once('update', function () {
var client3 = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId3,
port: port + 2,
wrtc: {}
})
2016-02-29 22:48:23 +00:00
if (serverType === 'ws') common.mockWebsocketTracker(client3)
2016-02-29 22:48:23 +00:00
client3.on('error', function (err) { t.error(err) })
client3.on('warning', function (err) { t.error(err) })
2015-07-29 08:47:09 +00:00
client3.start({ numwant: 1 })
2014-07-11 10:50:48 +00:00
client3.on('peer', function () {
t.pass('got one peer (this should only fire once)')
var num = 3
function tryCloseServer () {
num -= 1
if (num === 0) server.close()
}
2014-07-11 10:50:48 +00:00
client1.stop()
client1.once('update', function () {
t.pass('got response to stop (client1)')
client1.destroy()
tryCloseServer()
})
2014-07-11 10:50:48 +00:00
client2.stop()
client2.once('update', function () {
t.pass('got response to stop (client2)')
client2.destroy()
tryCloseServer()
})
2014-07-11 10:50:48 +00:00
client3.stop()
client3.once('update', function () {
t.pass('got response to stop (client3)')
client3.destroy()
tryCloseServer()
})
2014-07-11 10:50:48 +00:00
})
})
})
})
}
test('http: client announce with numwant', function (t) {
2014-07-11 10:50:48 +00:00
testClientAnnounceWithNumWant(t, 'http')
})
test('udp: client announce with numwant', function (t) {
2014-07-11 10:50:48 +00:00
testClientAnnounceWithNumWant(t, 'udp')
})
2017-01-16 23:43:47 +00:00
test('http: userAgent', function (t) {
t.plan(2)
common.createServer(t, 'http', function (server, announceUrl) {
// Confirm that user-agent header is set
server.http.on('request', function (req, res) {
t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)
})
var client = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId1,
port: port,
userAgent: 'WebTorrent/0.98.0 (https://webtorrent.io)',
wrtc: {}
})
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)
server.close()
client.destroy()
})
client.start()
})
})