diff --git a/test/helpers.js b/test/helpers.js index b1bbd22..56ba4aa 100644 --- a/test/helpers.js +++ b/test/helpers.js @@ -128,6 +128,40 @@ TestRunner.prototype.assertResponseCode = function (proxyPort, statusCode, creat return test; }; +// A test helper to check and see if the http headers were set properly. +TestRunner.prototype.assertHeaders = function (proxyPort, headerName, createProxy) { + var assertion = "should receive http header \"" + headerName + "\"", + protocol = this.source.protocols.http; + + var test = { + topic: function () { + var that = this, options = { + method: 'GET', + uri: protocol + '://localhost:' + proxyPort, + headers: { + host: 'unknown.com' + } + }; + + if (createProxy) { + return createProxy(function () { + request(options, that.callback); + }); + } + + request(options, this.callback); + } + }; + + test[assertion] = function (err, res, body) { + assert.isNull(err); + assert.isNotNull(res.headers[headerName]); + }; + + return test; +}; + + // // WebSocketTest // @@ -368,4 +402,4 @@ function merge (target) { }); }); return target; -} \ No newline at end of file +} diff --git a/test/http/http-proxy-test.js b/test/http/http-proxy-test.js index c202f97..e3b08f9 100644 --- a/test/http/http-proxy-test.js +++ b/test/http/http-proxy-test.js @@ -74,8 +74,11 @@ vows.describe('node-http-proxy/http-proxy/' + testName).addBatch({ "and a valid target server": runner.assertProxied('localhost', 8120, 8121, function (callback) { runner.startProxyServerWithForwarding(8120, 8121, 'localhost', forwardOptions, callback); }), - "and without a valid forward server": runner.assertProxied('localhost', 8122, 8123, function (callback) { - runner.startProxyServerWithForwarding(8122, 8123, 'localhost', badForwardOptions, callback); + "and also a valid target server": runner.assertHeaders(8122, "x-forwarded-for", function (callback) { + runner.startProxyServerWithForwarding(8122, 8123, 'localhost', forwardOptions, callback); + }), + "and without a valid forward server": runner.assertProxied('localhost', 8124, 8125, function (callback) { + runner.startProxyServerWithForwarding(8124, 8125, 'localhost', badForwardOptions, callback); }) } }