269 lines
		
	
	
		
			6.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			269 lines
		
	
	
		
			6.6 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
var helpers = require('./helpers'),
 | 
						|
    should  = require('should'),
 | 
						|
    sinon   = require('sinon'),
 | 
						|
    http    = require('http'),
 | 
						|
    needle  = require('./../');
 | 
						|
 | 
						|
var port = 7707;
 | 
						|
var url = 'localhost:' + port;
 | 
						|
var nonexisting_host = 'awepfokawepofawe.com';
 | 
						|
 | 
						|
describe('proxy option', function() {
 | 
						|
 | 
						|
  var spy, opts;
 | 
						|
 | 
						|
  function send_request(opts, done) {
 | 
						|
    if (spy) spy.restore();
 | 
						|
    spy = sinon.spy(http, 'request');
 | 
						|
    needle.get(url, opts, done);
 | 
						|
  }
 | 
						|
 | 
						|
  //////////////////////
 | 
						|
  // proxy opts helpers
 | 
						|
 | 
						|
  function not_proxied(done) {
 | 
						|
    return function(err, resp) {
 | 
						|
      var path = spy.args[0][0].path;
 | 
						|
      path.should.eql('/'); // not the full original URI
 | 
						|
      spy.restore();
 | 
						|
      done();
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  function proxied(host, port, done) {
 | 
						|
    return function(err, resp) {
 | 
						|
      var path = spy.args[0][0].path;
 | 
						|
      path.should.eql('http://' + url); // the full original URI
 | 
						|
 | 
						|
      var http_host = spy.args[0][0].host;
 | 
						|
      if (http_host) http_host.should.eql(host);
 | 
						|
 | 
						|
      var http_port = spy.args[0][0].port;
 | 
						|
      if (http_port) http_port.should.eql(port);
 | 
						|
 | 
						|
      spy.restore();
 | 
						|
      done();
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  //////////////////////
 | 
						|
  // auth helpers
 | 
						|
 | 
						|
  function get_auth(header) {
 | 
						|
    var token  = header.split(/\s+/).pop();
 | 
						|
    return token && Buffer.from(token, 'base64').toString().split(':');
 | 
						|
  }
 | 
						|
 | 
						|
  function no_proxy_auth(done) {
 | 
						|
    return function(err, resp) {
 | 
						|
      var headers = spy.args[0][0].headers;
 | 
						|
      Object.keys(headers).should.not.containEql('proxy-authorization');
 | 
						|
      done();
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  function header_set(name, user, pass, done) {
 | 
						|
    return function(err, resp) {
 | 
						|
      var headers = spy.args[0][0].headers;
 | 
						|
      var auth = get_auth(headers[name]);
 | 
						|
      auth[0].should.eql(user);
 | 
						|
      auth[1].should.eql(pass);
 | 
						|
      done();
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  function proxy_auth_set(user, pass, done) {
 | 
						|
    return header_set('proxy-authorization', user, pass, done);
 | 
						|
  }
 | 
						|
 | 
						|
  function basic_auth_set(user, pass, done) {
 | 
						|
    return header_set('authorization', user, pass, done);
 | 
						|
  }
 | 
						|
 | 
						|
  after(function() {
 | 
						|
    spy.restore();
 | 
						|
  })
 | 
						|
 | 
						|
  describe('when null proxy is passed', function() {
 | 
						|
 | 
						|
    it('does not proxy', function(done) {
 | 
						|
      send_request({ proxy: null }, not_proxied(done))
 | 
						|
    })
 | 
						|
 | 
						|
    describe('but defaults has been set', function() {
 | 
						|
 | 
						|
      before(function() {
 | 
						|
        needle.defaults({ proxy: 'foobar' });
 | 
						|
      })
 | 
						|
 | 
						|
      after(function() {
 | 
						|
        needle.defaults({ proxy: null });
 | 
						|
      })
 | 
						|
 | 
						|
      it('tries to proxy anyway', function(done) {
 | 
						|
        send_request({}, proxied('foobar', 80, done))
 | 
						|
      })
 | 
						|
 | 
						|
    })
 | 
						|
 | 
						|
  })
 | 
						|
 | 
						|
  describe('when weird string is passed', function() {
 | 
						|
 | 
						|
    it('tries to proxy anyway', function(done) {
 | 
						|
      send_request({ proxy: 'alfalfa' }, proxied('alfalfa', 80, done))
 | 
						|
    })
 | 
						|
  })
 | 
						|
 | 
						|
  describe('when valid url is passed', function() {
 | 
						|
 | 
						|
    describe('without NO_PROXY env var set', function() {
 | 
						|
      it('proxies request', function(done) {
 | 
						|
        send_request({ proxy: nonexisting_host + ':123/done' }, proxied(nonexisting_host, '123', done))
 | 
						|
      })
 | 
						|
 | 
						|
      it('does not set a Proxy-Authorization header', function(done) {
 | 
						|
        send_request({ proxy: nonexisting_host + ':123/done' }, no_proxy_auth(done));
 | 
						|
      })
 | 
						|
    })
 | 
						|
 | 
						|
    describe('with NO_PROXY env var set', function() {
 | 
						|
 | 
						|
      it('proxies request if matching host not found in list', function(done) {
 | 
						|
        process.env.NO_PROXY = 'foo';
 | 
						|
        send_request({ proxy: nonexisting_host + ':123/done' }, proxied(nonexisting_host, '123', function() {
 | 
						|
          delete process.env.NO_PROXY;
 | 
						|
          done();
 | 
						|
        }))
 | 
						|
      })
 | 
						|
 | 
						|
      it('does not proxy request if matching host in list and just has a different port', function(done) {
 | 
						|
        process.env.NO_PROXY = 'localhost';
 | 
						|
        send_request({ proxy: nonexisting_host + ':123/done' }, not_proxied(function() {
 | 
						|
          delete process.env.NO_PROXY;
 | 
						|
          done();
 | 
						|
        }))
 | 
						|
      })
 | 
						|
 | 
						|
      it('does not proxy if matching host found in list', function(done) {
 | 
						|
        process.env.NO_PROXY = 'foo,' + url;
 | 
						|
        send_request({ proxy: nonexisting_host + ':123/done' }, not_proxied(function() {
 | 
						|
          delete process.env.NO_PROXY;
 | 
						|
          done();
 | 
						|
        }))
 | 
						|
      })
 | 
						|
    })
 | 
						|
 | 
						|
    describe('and proxy url contains user:pass', function() {
 | 
						|
 | 
						|
      before(function() {
 | 
						|
        opts = {
 | 
						|
          proxy: 'http://mj:x@' + nonexisting_host + ':123/done'
 | 
						|
        }
 | 
						|
      })
 | 
						|
 | 
						|
      it('proxies request', function(done) {
 | 
						|
        send_request(opts, proxied(nonexisting_host, '123', done))
 | 
						|
      })
 | 
						|
 | 
						|
      it('sets Proxy-Authorization header', function(done) {
 | 
						|
        send_request(opts, proxy_auth_set('mj', 'x', done));
 | 
						|
      })
 | 
						|
 | 
						|
    })
 | 
						|
 | 
						|
    describe('and a proxy_user is passed', function() {
 | 
						|
 | 
						|
      before(function() {
 | 
						|
        opts = {
 | 
						|
          proxy: nonexisting_host + ':123',
 | 
						|
          proxy_user: 'someone',
 | 
						|
          proxy_pass: 'else'
 | 
						|
        }
 | 
						|
      })
 | 
						|
 | 
						|
      it('proxies request', function(done) {
 | 
						|
        send_request(opts, proxied(nonexisting_host, '123', done))
 | 
						|
      })
 | 
						|
 | 
						|
      it('sets Proxy-Authorization header', function(done) {
 | 
						|
        send_request(opts, proxy_auth_set('someone', 'else', done));
 | 
						|
      })
 | 
						|
 | 
						|
      describe('and url also contains user:pass', function() {
 | 
						|
 | 
						|
        it('url user:pass wins', function(done) {
 | 
						|
          var opts = {
 | 
						|
            proxy: 'http://xxx:yyy@' + nonexisting_host + ':123',
 | 
						|
            proxy_user: 'someone',
 | 
						|
            proxy_pass: 'else'
 | 
						|
          }
 | 
						|
 | 
						|
          send_request(opts, proxy_auth_set('xxx', 'yyy', done));
 | 
						|
        })
 | 
						|
 | 
						|
      })
 | 
						|
 | 
						|
      describe('and options.username is also present', function() {
 | 
						|
 | 
						|
        before(function() {
 | 
						|
          opts = { proxy_user: 'foobar', username: 'someone' };
 | 
						|
        })
 | 
						|
 | 
						|
        it('a separate Authorization header is set', function(done) {
 | 
						|
          var opts = {
 | 
						|
            proxy: nonexisting_host + ':123',
 | 
						|
            proxy_user: 'someone',
 | 
						|
            proxy_pass: 'else',
 | 
						|
            username: 'test',
 | 
						|
            password: 'X'
 | 
						|
          }
 | 
						|
 | 
						|
          send_request(opts, basic_auth_set('test', 'X', done));
 | 
						|
        })
 | 
						|
 | 
						|
      })
 | 
						|
 | 
						|
    })
 | 
						|
 | 
						|
  })
 | 
						|
 | 
						|
  describe('when environment variable is set', function() {
 | 
						|
 | 
						|
    describe('and default is unchanged', function() {
 | 
						|
 | 
						|
      before(function() {
 | 
						|
        process.env.HTTP_PROXY = 'foobar';
 | 
						|
      })
 | 
						|
 | 
						|
      after(function() {
 | 
						|
        delete process.env.HTTP_PROXY;
 | 
						|
      })
 | 
						|
 | 
						|
      it('tries to proxy', function(done) {
 | 
						|
        send_request({}, proxied('foobar', 80, done))
 | 
						|
      })
 | 
						|
 | 
						|
    })
 | 
						|
 | 
						|
    describe('and functionality is disabled', function() {
 | 
						|
 | 
						|
      before(function() {
 | 
						|
        process.env.HTTP_PROXY = 'foobar';
 | 
						|
      })
 | 
						|
 | 
						|
      after(function() {
 | 
						|
        delete process.env.HTTP_PROXY;
 | 
						|
      })
 | 
						|
 | 
						|
      it('ignores proxy', function(done) {
 | 
						|
        send_request({
 | 
						|
          use_proxy_from_env_var: false
 | 
						|
        }, not_proxied(done))
 | 
						|
      })
 | 
						|
 | 
						|
    })
 | 
						|
  })
 | 
						|
 | 
						|
})
 |