toxy/test/poisons/slow-close.js

Hackable HTTP proxy to simulate server failure scenarios and network conditions

const sinon = require('sinon')
const clone = require('clone')
const expect = require('chai').expect
const slowClose = require('../..').poisons.slowClose

suite('poison#slowClose', function () {
  test('close', function (done) {
    const delay = 50
    const expected = { body: 'Hello', code: 200, headers: { server: 'rocky' } }
    const spy = sinon.spy()
    const init = Date.now()

    const res = clone.clonePrototype({})
    Object.getPrototypeOf(res).writeHead = spy
    Object.getPrototypeOf(res).end = function (body) {
      spy(body)
      end()
    }

    slowClose({ delay: delay })(null, res, spy)

    res.writeHead(200, { 'content-length': 100, server: 'rocky' })
    res.end(expected.body)

    function end (err) {
      expect(Date.now() - init).to.be.at.least(delay - 1)
      expect(spy.calledThrice).to.be.true
      expect(spy.args[1][0]).to.be.equal(expected.code)
      expect(spy.args[1][1]).to.be.deep.equal(expected.headers)
      expect(spy.args[2][0]).to.be.equal(expected.body)
      done(err)
    }
  })
})