should = require "should"
sinon = require "sinon"
http = require "http"
proxy = require "../../lib/middleware/proxy"

describe "Proxy", ->
  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'
      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'
      done()