openhim-core/test/unit/proxyTest.js

The OpenHIM core application that provides logging and routing of http requests

/* eslint-env mocha */

import * as proxy from '../../src/middleware/proxy'

describe('Proxy', () => {
  const ctx = {}
  ctx.header = {}
  ctx.request = {}
  ctx.request.ip = '192.168.1.42'
  ctx.request.host = 'localhost:5000'
  ctx.request.protocol = 'https'

  describe('.setupProxyHeaders', () => {
    it('should set the X-Forwarded-* headers if not present', (done) => {
      delete ctx.header['X-Forwarded-For']
      delete ctx.header['X-Forwarded-Host']
      proxy.setupProxyHeaders(ctx)
      ctx.header['X-Forwarded-For'].should.equal('192.168.1.42')
      ctx.header['X-Forwarded-Host'].should.equal('localhost:5000')
      return done()
    })

    it('should append values to the X-Forwarded-* headers if already present', (done) => {
      ctx.header['X-Forwarded-For'] = '192.168.2.34'
      ctx.header['X-Forwarded-Host'] = 'someserver.com'
      proxy.setupProxyHeaders(ctx)
      ctx.header['X-Forwarded-For'].should.equal('192.168.2.34, 192.168.1.42')
      ctx.header['X-Forwarded-Host'].should.equal('someserver.com, localhost:5000')
      return done()
    })
  })
})