bittorrent-tracker/test/server.js

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')
var torrentLength = 50000

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(peerId, 6881, {
      infoHash: infoHash,
      length: torrentLength,
      announce: [ announceUrl ]
    }, { 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(peerId2, 6882, {
              infoHash: infoHash,
              length: torrentLength,
              announce: [ announceUrl ]
            }, { 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')
})