redbird/test/test_hostheader.js

"use strict";

var Redbird = require('../');
var Promise = require('bluebird');
var http = require('http');
var expect = require('chai').expect;

var TEST_PORT = 54674
var PROXY_PORT = 53433

var opts = {
	port: PROXY_PORT,
	bunyan: false
}

describe("Target with a hostname", function(){

	it("Should have the host header passed to the target", function(done){
		var redbird = Redbird(opts);

		expect(redbird.routing).to.be.an("object");

		redbird.register('127.0.0.1', '127.0.0.1.xip.io:'+TEST_PORT, {
			useTargetHostHeader: true
		});

		expect(redbird.routing).to.have.property("127.0.0.1");

		testServer().then(function(req){
			expect(req.headers['host']).to.be.eql('127.0.0.1.xip.io:'+TEST_PORT)
		})

		http.get('http://127.0.0.1:'+PROXY_PORT, function(res) {
			redbird.close();
			done();
		});

	})

	it("Should not have the host header passed to the target", function(done){
		var redbird = Redbird(opts);

		expect(redbird.routing).to.be.an("object");

		redbird.register('127.0.0.1', '127.0.0.1.xip.io:'+TEST_PORT);

		expect(redbird.routing).to.have.property("127.0.0.1");

		testServer().then(function(req){
			expect(req.headers['host']).to.be.eql('127.0.0.1:'+PROXY_PORT)
		})

		http.get('http://127.0.0.1:'+PROXY_PORT, function(res) {
			redbird.close();
			done();
		});

	})

	it("Should return 404 after route is unregister", function(done){
		var redbird = Redbird(opts);

		expect(redbird.routing).to.be.an("object");

		redbird.register('127.0.0.1', '127.0.0.1.xip.io:'+TEST_PORT);
		redbird.unregister('127.0.0.1', '127.0.0.1.xip.io:'+TEST_PORT);

		expect(redbird.routing).to.have.property("127.0.0.1");

		testServer().then(function(req){
			expect(req.headers['host']).to.be.eql('127.0.0.1:'+PROXY_PORT)
		})

		http.get('http://127.0.0.1:'+PROXY_PORT, function(res) {
			expect(res.statusCode).to.be.eql(404);

			redbird.close();
			done();
		});

	})

	it("Should return 502 after route with no backend", function(done){
		var redbird = Redbird(opts);

		expect(redbird.routing).to.be.an("object");

		redbird.register('127.0.0.1', '127.0.0.1.xip.io:502');

		expect(redbird.routing).to.have.property("127.0.0.1");

		http.get('http://127.0.0.1:'+PROXY_PORT, function(res) {
			expect(res.statusCode).to.be.eql(502);

			redbird.close();
			done();
		});
	})
})

describe("Request with forwarded host header", function () {
	it("should prefer forwarded hostname if desired", function () {
		var redbird = Redbird({
			bunyan: false,
			preferForwardedHost: true
		});

		expect(redbird.routing).to.be.an("object");
		var req = { headers: {host: '127.0.0.1', "x-forwarded-host": "subdomain.example.com"} }
		
		var source = redbird._getSource(req);
		expect(source).to.be.eql('subdomain.example.com')

		redbird.close();
	});

		it("should use original host if not further specified", function () {
		var redbird = Redbird(opts);

		expect(redbird.routing).to.be.an("object");
		var req = { headers: { host: '127.0.0.1', "x-forwarded-host": "subdomain.example.com"} }
		
		var source = redbird._getSource(req);
		expect(source).to.be.eql('127.0.0.1')

		redbird.close();
	});
});

function testServer(){
	return new Promise(function(resolve, reject){
		var server = http.createServer(function(req, res){
			res.write("");
			res.end();
			resolve(req);
			server.close();
		});

		server.listen(TEST_PORT);
	})
}