test-redirect-complex.js

'use strict'

var server = require('./server')
var request = require('../index')
var events = require('events')
var tape = require('tape')
var destroyable = require('server-destroy')

var s = server.createServer()
var ss = server.createSSLServer()
var e = new events.EventEmitter()

destroyable(s)
destroyable(ss)

function bouncy (s, serverUrl) {
  var redirs = {
    a: 'b',
    b: 'c',
    c: 'd',
    d: 'e',
    e: 'f',
    f: 'g',
    g: 'h',
    h: 'end'
  }

  var perm = true
  Object.keys(redirs).forEach(function (p) {
    var t = redirs[p]

    // switch type each time
    var type = perm ? 301 : 302
    perm = !perm
    s.on('/' + p, function (req, res) {
      setTimeout(function () {
        res.writeHead(type, { location: serverUrl + '/' + t })
        res.end()
      }, Math.round(Math.random() * 25))
    })
  })

  s.on('/end', function (req, res) {
    var key = req.headers['x-test-key']
    e.emit('hit-' + key, key)
    res.writeHead(200)
    res.end(key)
  })
}

tape('setup', function (t) {
  s.listen(0, function () {
    ss.listen(0, function () {
      bouncy(s, ss.url)
      bouncy(ss, s.url)
      t.end()
    })
  })
})

tape('lots of redirects', function (t) {
  var n = 10
  t.plan(n * 4)

  function doRedirect (i) {
    var key = 'test_' + i
    request({
      url: (i % 2 ? s.url : ss.url) + '/a',
      headers: { 'x-test-key': key },
      rejectUnauthorized: false
    }, function (err, res, body) {
      t.equal(err, null)
      t.equal(res.statusCode, 200)
      t.equal(body, key)
    })

    e.once('hit-' + key, function (v) {
      t.equal(v, key)
    })
  }

  for (var i = 0; i < n; i++) {
    doRedirect(i)
  }
})

tape('cleanup', function (t) {
  s.destroy(function () {
    ss.destroy(function () {
      t.end()
    })
  })
})