toxy/test/rules/response-headers.js

Hackable HTTP proxy to simulate server failure scenarios and network conditions

const expect = require('chai').expect
const headers = require('../..').rules.responseHeaders

suite('rules#responseHeaders', function () {
  test('header present', function (done) {
    const matchHeaders = { 'content-type': true }
    const res = { headers: { 'content-type': 'application/json' } }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.false
      done()
    }
  })

  test('match by string', function (done) {
    const matchHeaders = { 'content-type': 'application/json' }
    const res = { headers: { 'content-type': 'application/json' } }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.false
      done()
    }
  })

  test('match by regexp', function (done) {
    const matchHeaders = { 'content-type': /^application/i }
    const res = { headers: { 'content-type': 'application/json' } }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.false
      done()
    }
  })

  test('match by function', function (done) {
    const matchHeaders = { 'content-type': match }
    const res = { headers: { 'content-type': 'application/json' } }

    function match (value, key) {
      return key === 'content-type' && value === 'application/json'
    }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.false
      done()
    }
  })

  test('multiple headers', function (done) {
    const matchHeaders = { 'content-type': /^application/i, 'server': true }
    const res = { headers: { 'content-type': 'application/json', 'server': 'rocky' } }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.false
      done()
    }
  })

  test('not present', function (done) {
    const matchHeaders = { 'content-type': false }
    const res = { headers: {} }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.false
      done()
    }
  })

  test('cannot match', function (done) {
    const matchHeaders = { 'empty': true }
    const res = { headers: {} }

    headers(matchHeaders)(null, res, next)

    function next (err, ignore) {
      expect(err).to.be.null
      expect(ignore).to.be.true
      done()
    }
  })
})