node-http-proxy/test/lib-http-proxy-passes-web-outgoing-test.js

var httpProxy = require('../lib/http-proxy/passes/web-outgoing'),
    expect = require('expect.js');

describe('lib/http-proxy/passes/web-outgoing.js', function () {
  describe('#setRedirectHostRewrite', function () {
    context('rewrites location host to option', function() {
      beforeEach(function() {
        this.proxyRes = {
          statusCode: 301,
          headers: {
            location: "http://f.com/"
          }
        };

        this.options = {
          hostRewrite: "x.com"
        };
      });

      it('on 301', function() {
        this.proxyRes.statusCode = 301;
        httpProxy.setRedirectHostRewrite({}, {}, this.proxyRes, this.options);
        expect(this.proxyRes.headers.location).to.eql('http://'+this.options.hostRewrite+'/');
      });

      it('on 302', function() {
        this.proxyRes.statusCode = 302;
        httpProxy.setRedirectHostRewrite({}, {}, this.proxyRes, this.options);
        expect(this.proxyRes.headers.location).to.eql('http://'+this.options.hostRewrite+'/');
      });

      it('on 307', function() {
        this.proxyRes.statusCode = 307;
        httpProxy.setRedirectHostRewrite({}, {}, this.proxyRes, this.options);
        expect(this.proxyRes.headers.location).to.eql('http://'+this.options.hostRewrite+'/');
      });

      it('on 308', function() {
        this.proxyRes.statusCode = 308;
        httpProxy.setRedirectHostRewrite({}, {}, this.proxyRes, this.options);
        expect(this.proxyRes.headers.location).to.eql('http://'+this.options.hostRewrite+'/');
      });

      it('not on 200', function() {
        this.proxyRes.statusCode = 200;
        httpProxy.setRedirectHostRewrite({}, {}, this.proxyRes, this.options);
        expect(this.proxyRes.headers.location).to.eql('http://f.com/');
      });

      it('not when hostRewrite is unset', function() {
        httpProxy.setRedirectHostRewrite({}, {}, this.proxyRes, {});
        expect(this.proxyRes.headers.location).to.eql('http://f.com/');
      });
    });
  });

  describe('#setConnection', function () {
    it('set the right connection with 1.0 - `close`', function() {
      var proxyRes = { headers: {} };
      httpProxy.setConnection({
        httpVersion: '1.0',
        headers: {
          connection: null
        }
      }, {}, proxyRes);

      expect(proxyRes.headers.connection).to.eql('close'); 
    });

    it('set the right connection with 1.0 - req.connection', function() {
      var proxyRes = { headers: {} };
      httpProxy.setConnection({
        httpVersion: '1.0',
        headers: {
          connection: 'hey'
        }
      }, {}, proxyRes);

      expect(proxyRes.headers.connection).to.eql('hey'); 
    });

    it('set the right connection - req.connection', function() {
      var proxyRes = { headers: {} };
      httpProxy.setConnection({
        httpVersion: null,
        headers: {
          connection: 'hola'
        }
      }, {}, proxyRes);

      expect(proxyRes.headers.connection).to.eql('hola'); 
    });

    it('set the right connection - `keep-alive`', function() {
      var proxyRes = { headers: {} };
      httpProxy.setConnection({
        httpVersion: null,
        headers: {
          connection: null
        }
      }, {}, proxyRes);

      expect(proxyRes.headers.connection).to.eql('keep-alive'); 
    });

  });

  describe('#writeStatusCode', function () {
    it('should write status code', function() {
      var res = {
        writeHead: function(n) {
          expect(n).to.eql(200);
        }
      }

      httpProxy.writeStatusCode({}, res, { statusCode: 200 });
    });
  });

  describe('#writeHeaders', function() {
    var proxyRes = {
      headers: {
        hey: 'hello',
        how: 'are you?'
      }
    };

    var res = {
      setHeader: function(k, v) {
        this.headers[k] = v;
      },
      headers: {}
    };

    httpProxy.writeHeaders({}, res, proxyRes);

    expect(res.headers.hey).to.eql('hello');
    expect(res.headers.how).to.eql('are you?');
  });


  describe('#removeChunked', function() {
    var proxyRes = {
      headers: {
        'transfer-encoding': 'hello'
      }
    };


    httpProxy.removeChunked({ httpVersion: '1.0' }, {}, proxyRes);

    expect(proxyRes.headers['transfer-encoding']).to.eql(undefined);
  });

});