bittorrent-tracker/test/client-ws-socket-pool.js
Feross Aboukhadijeh 2959c2cea6 Use safe-buffer
Use the new Buffer APIs from Node v6 for added security. For example,
`Buffer.from()` will throw if passed a number, unlike `Buffer()` which
allocated UNINITIALIZED memory in that case.

Use the `safe-buffer` package for compatibility with previous versions
of
Node.js, including v4.x, v0.12, and v0.10.

https://github.com/feross/safe-buffer
2016-05-29 23:12:23 -07:00

58 lines
1.7 KiB
JavaScript

var Buffer = require('safe-buffer').Buffer
var Client = require('../')
var common = require('./common')
var fixtures = require('webtorrent-fixtures')
var test = require('tape')
var peerId = Buffer.from('01234567890123456789')
var port = 6681
test('ensure client.destroy() callback is called with re-used websockets in socketPool', function (t) {
t.plan(4)
common.createServer(t, 'ws', function (server, announceUrl) {
var client1 = new Client({
infoHash: fixtures.leaves.parsedTorrent.infoHash,
announce: announceUrl,
peerId: peerId,
port: port,
wrtc: {}
})
common.mockWebsocketTracker(client1)
client1.on('error', function (err) { t.error(err) })
client1.on('warning', function (err) { t.error(err) })
client1.start()
client1.once('update', function () {
t.pass('got client1 update')
// second ws client using same announce url will re-use the same websocket
var client2 = new Client({
infoHash: fixtures.alice.parsedTorrent.infoHash, // different info hash
announce: announceUrl,
peerId: peerId,
port: port,
wrtc: {}
})
common.mockWebsocketTracker(client2)
client2.on('error', function (err) { t.error(err) })
client2.on('warning', function (err) { t.error(err) })
client2.start()
client2.once('update', function () {
t.pass('got client2 update')
client1.destroy(function (err) {
t.error(err, 'got client1 destroy callback')
client2.destroy(function (err) {
t.error(err, 'got client2 destroy callback')
server.close()
})
})
})
})
})
})