304 lines
9.4 KiB
JavaScript
304 lines
9.4 KiB
JavaScript
|
|
||
|
|
/**
|
||
|
|
* Module dependencies.
|
||
|
|
*/
|
||
|
|
|
||
|
|
var fs = require('fs');
|
||
|
|
var url = require('url');
|
||
|
|
var http = require('http');
|
||
|
|
var https = require('https');
|
||
|
|
var assert = require('assert');
|
||
|
|
var Proxy = require('proxy');
|
||
|
|
var HttpProxyAgent = require('../');
|
||
|
|
|
||
|
|
describe('HttpProxyAgent', function () {
|
||
|
|
|
||
|
|
var server;
|
||
|
|
var serverPort;
|
||
|
|
|
||
|
|
var proxy;
|
||
|
|
var proxyPort;
|
||
|
|
|
||
|
|
var sslProxy;
|
||
|
|
var sslProxyPort;
|
||
|
|
|
||
|
|
before(function (done) {
|
||
|
|
// setup HTTP proxy server
|
||
|
|
proxy = Proxy();
|
||
|
|
proxy.listen(function () {
|
||
|
|
proxyPort = proxy.address().port;
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
before(function (done) {
|
||
|
|
// setup target HTTP server
|
||
|
|
server = http.createServer();
|
||
|
|
server.listen(function () {
|
||
|
|
serverPort = server.address().port;
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
before(function (done) {
|
||
|
|
// setup SSL HTTP proxy server
|
||
|
|
var options = {
|
||
|
|
key: fs.readFileSync(__dirname + '/ssl-cert-snakeoil.key'),
|
||
|
|
cert: fs.readFileSync(__dirname + '/ssl-cert-snakeoil.pem')
|
||
|
|
};
|
||
|
|
sslProxy = Proxy(https.createServer(options));
|
||
|
|
sslProxy.listen(function () {
|
||
|
|
sslProxyPort = sslProxy.address().port;
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
// shut down test HTTP server
|
||
|
|
after(function (done) {
|
||
|
|
proxy.once('close', function () { done(); });
|
||
|
|
proxy.close();
|
||
|
|
});
|
||
|
|
|
||
|
|
after(function (done) {
|
||
|
|
server.once('close', function () { done(); });
|
||
|
|
server.close();
|
||
|
|
});
|
||
|
|
|
||
|
|
after(function (done) {
|
||
|
|
sslProxy.once('close', function () { done(); });
|
||
|
|
sslProxy.close();
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('constructor', function () {
|
||
|
|
it('should throw an Error if no "proxy" argument is given', function () {
|
||
|
|
assert.throws(function () {
|
||
|
|
new HttpProxyAgent();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should accept a "string" proxy argument', function () {
|
||
|
|
var agent = new HttpProxyAgent('http://127.0.0.1:' + proxyPort);
|
||
|
|
assert.equal('127.0.0.1', agent.proxy.host);
|
||
|
|
assert.equal(proxyPort, agent.proxy.port);
|
||
|
|
});
|
||
|
|
it('should accept a `url.parse()` result object argument', function () {
|
||
|
|
var opts = url.parse('http://127.0.0.1:' + proxyPort);
|
||
|
|
var agent = new HttpProxyAgent(opts);
|
||
|
|
assert.equal('127.0.0.1', agent.proxy.host);
|
||
|
|
assert.equal(proxyPort, agent.proxy.port);
|
||
|
|
});
|
||
|
|
describe('secureProxy', function () {
|
||
|
|
it('should default to `false`', function () {
|
||
|
|
var agent = new HttpProxyAgent({ port: proxyPort });
|
||
|
|
assert.equal(false, agent.secureProxy);
|
||
|
|
});
|
||
|
|
it('should be `false` when "http:" protocol is used', function () {
|
||
|
|
var agent = new HttpProxyAgent({ port: proxyPort, protocol: 'http:' });
|
||
|
|
assert.equal(false, agent.secureProxy);
|
||
|
|
});
|
||
|
|
it('should be `true` when "https:" protocol is used', function () {
|
||
|
|
var agent = new HttpProxyAgent({ port: proxyPort, protocol: 'https:' });
|
||
|
|
assert.equal(true, agent.secureProxy);
|
||
|
|
});
|
||
|
|
it('should be `true` when "https" protocol is used', function () {
|
||
|
|
var agent = new HttpProxyAgent({ port: proxyPort, protocol: 'https' });
|
||
|
|
assert.equal(true, agent.secureProxy);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('"http" module', function () {
|
||
|
|
it('should work over an HTTP proxy', function (done) {
|
||
|
|
// set HTTP "request" event handler for this test
|
||
|
|
server.once('request', function (req, res) {
|
||
|
|
res.end(JSON.stringify(req.headers));
|
||
|
|
});
|
||
|
|
|
||
|
|
var proxy = process.env.HTTP_PROXY || process.env.http_proxy || 'http://127.0.0.1:' + proxyPort;
|
||
|
|
var agent = new HttpProxyAgent(proxy);
|
||
|
|
|
||
|
|
var opts = url.parse('http://127.0.0.1:' + serverPort);
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
http.get(opts, function (res) {
|
||
|
|
var data = '';
|
||
|
|
res.setEncoding('utf8');
|
||
|
|
res.on('data', function (b) {
|
||
|
|
data += b;
|
||
|
|
});
|
||
|
|
res.on('end', function () {
|
||
|
|
data = JSON.parse(data);
|
||
|
|
assert.equal('127.0.0.1:' + serverPort, data.host);
|
||
|
|
assert('via' in data);
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should work over an HTTPS proxy', function (done) {
|
||
|
|
// set HTTP "request" event handler for this test
|
||
|
|
server.once('request', function (req, res) {
|
||
|
|
res.end(JSON.stringify(req.headers));
|
||
|
|
});
|
||
|
|
|
||
|
|
var proxy = process.env.HTTPS_PROXY || process.env.https_proxy || 'https://127.0.0.1:' + sslProxyPort;
|
||
|
|
proxy = url.parse(proxy);
|
||
|
|
proxy.rejectUnauthorized = false;
|
||
|
|
var agent = new HttpProxyAgent(proxy);
|
||
|
|
assert.equal(true, agent.secureProxy);
|
||
|
|
|
||
|
|
var opts = url.parse('http://127.0.0.1:' + serverPort);
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
http.get(opts, function (res) {
|
||
|
|
var data = '';
|
||
|
|
res.setEncoding('utf8');
|
||
|
|
res.on('data', function (b) {
|
||
|
|
data += b;
|
||
|
|
});
|
||
|
|
res.on('end', function () {
|
||
|
|
data = JSON.parse(data);
|
||
|
|
assert.equal('127.0.0.1:' + serverPort, data.host);
|
||
|
|
assert('via' in data);
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should proxy the query string of the request path', function (done) {
|
||
|
|
// set HTTP "request" event handler for this test
|
||
|
|
server.once('request', function (req, res) {
|
||
|
|
res.end(JSON.stringify({
|
||
|
|
url: req.url
|
||
|
|
}));
|
||
|
|
});
|
||
|
|
|
||
|
|
var proxy = process.env.HTTP_PROXY || process.env.http_proxy || 'http://127.0.0.1:' + proxyPort;
|
||
|
|
var agent = new HttpProxyAgent(proxy);
|
||
|
|
|
||
|
|
var opts = url.parse('http://127.0.0.1:' + serverPort + '/test?foo=bar&1=2');
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
http.get(opts, function (res) {
|
||
|
|
var data = '';
|
||
|
|
res.setEncoding('utf8');
|
||
|
|
res.on('data', function (b) {
|
||
|
|
data += b;
|
||
|
|
});
|
||
|
|
res.on('end', function () {
|
||
|
|
data = JSON.parse(data);
|
||
|
|
assert.equal('/test?foo=bar&1=2', data.url);
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should receive the 407 authorization code on the `http.ClientResponse`', function (done) {
|
||
|
|
// set a proxy authentication function for this test
|
||
|
|
proxy.authenticate = function (req, fn) {
|
||
|
|
// reject all requests
|
||
|
|
fn(null, false);
|
||
|
|
};
|
||
|
|
|
||
|
|
var proxyUri = process.env.HTTP_PROXY || process.env.http_proxy || 'http://127.0.0.1:' + proxyPort;
|
||
|
|
var agent = new HttpProxyAgent(proxyUri);
|
||
|
|
|
||
|
|
var opts = {};
|
||
|
|
// `host` and `port` don't really matter since the proxy will reject anyways
|
||
|
|
opts.host = '127.0.0.1';
|
||
|
|
opts.port = 80;
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
http.get(opts, function (res) {
|
||
|
|
assert.equal(407, res.statusCode);
|
||
|
|
assert('proxy-authenticate' in res.headers);
|
||
|
|
delete proxy.authenticate;
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should send the "Proxy-Authorization" request header', function (done) {
|
||
|
|
// set a proxy authentication function for this test
|
||
|
|
proxy.authenticate = function (req, fn) {
|
||
|
|
// username:password is "foo:bar"
|
||
|
|
fn(null, req.headers['proxy-authorization'] == 'Basic Zm9vOmJhcg==');
|
||
|
|
};
|
||
|
|
|
||
|
|
// set HTTP "request" event handler for this test
|
||
|
|
server.once('request', function (req, res) {
|
||
|
|
res.end(JSON.stringify(req.headers));
|
||
|
|
});
|
||
|
|
|
||
|
|
var proxyUri = process.env.HTTP_PROXY || process.env.http_proxy || 'http://127.0.0.1:' + proxyPort;
|
||
|
|
var proxyOpts = url.parse(proxyUri);
|
||
|
|
proxyOpts.auth = 'foo:bar';
|
||
|
|
var agent = new HttpProxyAgent(proxyOpts);
|
||
|
|
|
||
|
|
var opts = url.parse('http://127.0.0.1:' + serverPort);
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
http.get(opts, function (res) {
|
||
|
|
var data = '';
|
||
|
|
res.setEncoding('utf8');
|
||
|
|
res.on('data', function (b) {
|
||
|
|
data += b;
|
||
|
|
});
|
||
|
|
res.on('end', function () {
|
||
|
|
data = JSON.parse(data);
|
||
|
|
assert.equal('127.0.0.1:' + serverPort, data.host);
|
||
|
|
assert('via' in data);
|
||
|
|
delete proxy.authenticate;
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should emit an "error" event on the `http.ClientRequest` if the proxy does not exist', function (done) {
|
||
|
|
// port 4 is a reserved, but "unassigned" port
|
||
|
|
var proxyUri = 'http://127.0.0.1:4';
|
||
|
|
var agent = new HttpProxyAgent(proxyUri);
|
||
|
|
|
||
|
|
var opts = url.parse('http://nodejs.org');
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
var req = http.get(opts);
|
||
|
|
req.once('error', function (err) {
|
||
|
|
assert.equal('ECONNREFUSED', err.code);
|
||
|
|
req.abort();
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
it('should work after the first tick of the `http.ClientRequest` instance', function (done) {
|
||
|
|
// set HTTP "request" event handler for this test
|
||
|
|
server.once('request', function (req, res) {
|
||
|
|
res.end(JSON.stringify(req.url));
|
||
|
|
});
|
||
|
|
|
||
|
|
var proxy = process.env.HTTP_PROXY || process.env.http_proxy || 'http://127.0.0.1:' + proxyPort;
|
||
|
|
var agent = new HttpProxyAgent(proxy);
|
||
|
|
|
||
|
|
var opts = url.parse('http://127.0.0.1:' + serverPort + '/test');
|
||
|
|
opts.agent = agent;
|
||
|
|
|
||
|
|
// defer the "connect()" function logic, since calling .end() before the
|
||
|
|
// "socket" event can cause weirdness since the HTTP header will have been
|
||
|
|
// cached and the HttpProxyAgent `req.path` patches won't be respected
|
||
|
|
var callback = agent.callback;
|
||
|
|
agent.callback = function (req, opts, fn) {
|
||
|
|
setTimeout(function () {
|
||
|
|
agent.callback = callback;
|
||
|
|
agent.callback(req, opts, fn);
|
||
|
|
}, 10);
|
||
|
|
};
|
||
|
|
|
||
|
|
http.get(opts, function (res) {
|
||
|
|
var data = '';
|
||
|
|
res.setEncoding('utf8');
|
||
|
|
res.on('data', function (b) {
|
||
|
|
data += b;
|
||
|
|
});
|
||
|
|
res.on('end', function () {
|
||
|
|
data = JSON.parse(data);
|
||
|
|
assert.equal('/test', data);
|
||
|
|
done();
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
});
|