mirror of
https://github.com/webtorrent/bittorrent-tracker.git
synced 2024-12-14 19:36:24 +00:00
153 lines
4.2 KiB
JavaScript
153 lines
4.2 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')
|
|
})
|