142 lines
3.3 KiB
JavaScript
142 lines
3.3 KiB
JavaScript
"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);
|
|
})
|
|
}
|