mirror of
https://github.com/webtorrent/bittorrent-tracker.git
synced 2024-12-15 03:46:25 +00:00
2966165a8f
To use the client, you used to pass in four arguments: `new Client(peerId, port, parsedTorrent, opts)` Now, passing in the torrent is no longer required, just the `announce` and `infoHash` properties. This decouples this package from `parse-torrent`. All options get passed in together now: new Client({ infoHash: '', // hex string or Buffer peerId: '', // hex string or Buffer announce: [], // list of tracker server urls port: 6881 // torrent client port, (in browser, optional) }) All the normal optional arguments (rtcConfig, wrtc, etc.) can still be passed in with the rest of these options. Fixes #118. Fixes #115. Added ws tests for scrape.
153 lines
4.1 KiB
JavaScript
153 lines
4.1 KiB
JavaScript
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 = new Buffer('01234567890123456789')
|
|
var peerId2 = new Buffer('12345678901234567890')
|
|
|
|
function serverTest (t, serverType, serverFamily) {
|
|
t.plan(30)
|
|
|
|
var hostname = serverFamily === 'inet6'
|
|
? '[::1]'
|
|
: '127.0.0.1'
|
|
var clientIp = serverFamily === 'inet6'
|
|
? '::1'
|
|
: '127.0.0.1'
|
|
|
|
common.createServer(t, serverType, function (server) {
|
|
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
|
|
})
|
|
|
|
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(Object.keys(swarm.peers).length, 1)
|
|
|
|
var id = serverType === 'ws'
|
|
? peerId.toString('hex')
|
|
: hostname + ':6881'
|
|
|
|
t.equal(swarm.peers[id].type, serverType)
|
|
t.equal(swarm.peers[id].ip, clientIp)
|
|
t.equal(swarm.peers[id].peerId, peerId.toString('hex'))
|
|
t.equal(swarm.peers[id].complete, false)
|
|
if (serverType === 'ws') {
|
|
t.equal(typeof swarm.peers[id].port, 'number')
|
|
t.ok(swarm.peers[id].socket)
|
|
} else {
|
|
t.equal(swarm.peers[id].port, 6881)
|
|
t.notOk(swarm.peers[id].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(typeof data.complete, 'number')
|
|
t.equal(typeof data.incomplete, 'number')
|
|
t.equal(typeof data.downloaded, 'number')
|
|
|
|
var client2 = new Client({
|
|
infoHash: infoHash,
|
|
announce: [ announceUrl ],
|
|
peerId: peerId2,
|
|
port: 6882,
|
|
wrtc: wrtc
|
|
})
|
|
|
|
client2.start()
|
|
|
|
server.once('start', function () {
|
|
t.pass('got start message from client2')
|
|
})
|
|
|
|
client2.once('peer', function (addr) {
|
|
t.ok(addr === hostname + ':6881' || addr === hostname + ':6882' || addr.id === peerId.toString('hex'))
|
|
|
|
client2.stop()
|
|
client2.once('update', function (data) {
|
|
t.equal(data.announce, announceUrl)
|
|
t.equal(data.complete, 1)
|
|
t.equal(data.incomplete, 0)
|
|
client2.destroy()
|
|
|
|
client1.stop()
|
|
client1.once('update', function (data) {
|
|
t.equal(data.announce, announceUrl)
|
|
t.equal(data.complete, 0)
|
|
t.equal(data.incomplete, 0)
|
|
|
|
client1.destroy(function () {
|
|
server.close()
|
|
if (serverType === 'ws') wrtc.close()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
}
|
|
|
|
test('websocket server', function (t) {
|
|
if (wrtcReady) {
|
|
runTest()
|
|
} else {
|
|
wrtc.electronDaemon.once('ready', runTest)
|
|
}
|
|
function runTest () {
|
|
serverTest(t, 'ws', 'inet')
|
|
}
|
|
})
|
|
|
|
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')
|
|
})
|