toxy/test/poisons/slow-read.js

Hackable HTTP proxy to simulate server failure scenarios and network conditions

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

suite('poison#slowRead', function () {
  test('read', function (done) {
    const threshold = 5
    const spy = sinon.spy()
    const init = Date.now()

    const res = new http.OutgoingMessage()
    const req = clone.clonePrototype(new http.IncomingMessage())
    req.method = 'POST'

    Object.getPrototypeOf(req).push = function (data) {
      spy(data)
      if (data === null) assert()
    }

    slowRead({ chunk: 1, threshold: threshold })(req, res, spy)

    req.push(new Buffer('Hello World'))
    req.push(null)

    function assert () {
      expect(Date.now() - init).to.be.at.least(threshold * 10)
      expect(spy.args).to.have.length(13)
      expect(spy.args.shift()[0]).to.be.undefined
      expect(spy.args.shift()[0].toString()).to.be.equal('H')
      expect(spy.args.pop()[0]).to.be.null
      expect(spy.args.pop()[0].toString()).to.be.equal('d')
      done()
    }
  })

  test('premature close', function (done) {
    const threshold = 10
    const spy = sinon.spy()
    const init = Date.now()

    const res = new http.OutgoingMessage()
    const req = clone.clonePrototype(new http.IncomingMessage())
    req.method = 'POST'

    Object.getPrototypeOf(req).push = function (data) {
      spy(data)
      if (data === null) assert()
    }

    slowRead({ chunk: 1, threshold: threshold })(req, res, spy)

    req.emit('close')

    req.push(new Buffer('Hello World'))
    req.push(null)

    function assert () {
      expect(Date.now() - init).to.be.within(0, 5)
      expect(spy.args).to.have.length(3)
      done()
    }
  })
})