bittorrent-tracker/test/scrape.js

var bencode = require('bencode')
var Client = require('../')
var common = require('../lib/common')
var concat = require('concat-stream')
var fs = require('fs')
var http = require('http')
var parseTorrent = require('parse-torrent')
var portfinder = require('portfinder')
var querystring = require('querystring')
var Server = require('../').Server
var test = require('tape')

var infoHash1 = 'aaa67059ed6bd08362da625b3ae77f6f4a075aaa'
var encodedInfoHash1 = common.bytewiseEncodeURIComponent(
  new Buffer(infoHash1, 'hex')
)
var binaryinfoHash1 = new Buffer(infoHash1, 'hex').toString('binary')
var infoHash2 = 'bbb67059ed6bd08362da625b3ae77f6f4a075bbb'
var encodedInfoHash2 = common.bytewiseEncodeURIComponent(
  new Buffer(infoHash2, 'hex')
)
var binaryinfoHash2 = new Buffer(infoHash2, 'hex').toString('binary')

var bitlove = fs.readFileSync(__dirname + '/torrents/bitlove-intro.torrent')
var parsedBitlove = parseTorrent(bitlove)
var encodedBitlove = common.bytewiseEncodeURIComponent(
  new Buffer(parsedBitlove.infoHash, 'hex')
)
var binaryBitlove = new Buffer(parsedBitlove.infoHash, 'hex').toString('binary')
var peerId = new Buffer('01234567890123456789')

test('server: single info_hash scrape', function (t) {
  var server = new Server({ udp: false })
  server.on('error', function (err) {
    t.error(err)
  })
  server.on('warning', function (err) {
    t.error(err)
  })

  portfinder.getPort(function (err, port) {
    t.error(err)
    server.listen(port)
    var scrapeUrl = 'http://127.0.0.1:' + port + '/scrape'

    server.once('listening', function () {
      var url = scrapeUrl + '?' + querystring.stringify({
        info_hash: encodedInfoHash1
      })
      http.get(url, function (res) {
        t.equal(res.statusCode, 200)
        res.pipe(concat(function (data) {
          data = bencode.decode(data)
          t.ok(data.files)
          t.equal(Object.keys(data.files).length, 1)
          t.ok(data.files[binaryinfoHash1])
          t.equal(typeof data.files[binaryinfoHash1].complete, 'number')
          t.equal(typeof data.files[binaryinfoHash1].incomplete, 'number')
          t.equal(typeof data.files[binaryinfoHash1].downloaded, 'number')

          server.close(function () {
            t.end()
          })
        }))
      }).on('error', function (e) {
        t.error(err)
      })
    })
  })
})

test('server: multiple info_hash scrape', function (t) {
  var server = new Server({ udp: false })
  server.on('error', function (err) {
    t.error(err)
  })
  server.on('warning', function (err) {
    t.error(err)
  })

  portfinder.getPort(function (err, port) {
    t.error(err)
    server.listen(port)
    var scrapeUrl = 'http://127.0.0.1:' + port + '/scrape'

    server.once('listening', function () {
      var url = scrapeUrl + '?' + querystring.stringify({
        info_hash: [ encodedInfoHash1, encodedInfoHash2 ]
      })
      http.get(url, function (res) {
        t.equal(res.statusCode, 200)
        res.pipe(concat(function (data) {
          data = bencode.decode(data)
          t.ok(data.files)
          t.equal(Object.keys(data.files).length, 2)

          t.ok(data.files[binaryinfoHash1])
          t.equal(typeof data.files[binaryinfoHash1].complete, 'number')
          t.equal(typeof data.files[binaryinfoHash1].incomplete, 'number')
          t.equal(typeof data.files[binaryinfoHash1].downloaded, 'number')

          t.ok(data.files[binaryinfoHash2])
          t.equal(typeof data.files[binaryinfoHash2].complete, 'number')
          t.equal(typeof data.files[binaryinfoHash2].incomplete, 'number')
          t.equal(typeof data.files[binaryinfoHash2].downloaded, 'number')

          server.close(function () {
            t.end()
          })
        }))
      }).on('error', function (e) {
        t.error(err)
      })
    })
  })
})

test('server: all info_hash scrape', function (t) {
  var server = new Server({ udp: false })
  server.on('error', function (err) {
    t.error(err)
  })
  server.on('warning', function (err) {
    t.error(err)
  })

  portfinder.getPort(function (err, port) {
    t.error(err)
    server.listen(port)
    var announceUrl = 'http://127.0.0.1:' + port + '/announce'
    var scrapeUrl = 'http://127.0.0.1:' + port + '/scrape'

    parsedBitlove.announce = [ announceUrl ]

    server.once('listening', function () {

      // announce a torrent to the tracker
      var client = new Client(peerId, port, parsedBitlove)
      client.on('error', function (err) {
        t.error(err)
      })
      client.start()

      server.once('start', function (data) {

        // now do a scrape of everything by omitting the info_hash param
        http.get(scrapeUrl, function (res) {

          t.equal(res.statusCode, 200)
          res.pipe(concat(function (data) {
            data = bencode.decode(data)
            t.ok(data.files)
            t.equal(Object.keys(data.files).length, 1)

            t.ok(data.files[binaryBitlove])
            t.equal(typeof data.files[binaryBitlove].complete, 'number')
            t.equal(typeof data.files[binaryBitlove].incomplete, 'number')
            t.equal(typeof data.files[binaryBitlove].downloaded, 'number')

            client.stop()
            server.close(function () {
              t.end()
            })
          }))
        }).on('error', function (e) {
          t.error(err)
        })
      })
    })
  })
})