routing-proxy-test.js

/*
 * proxy-table-test.js: Tests for the ProxyTable object.
 *
 * (C) 2010, Charlie Robbins
 *
 */

var assert = require('assert'),
    fs = require('fs'),
    path = require('path'),
    util = require('util'),
    argv = require('optimist').argv,
    request = require('request'),
    vows = require('vows'),
    helpers = require('../helpers');

var options = helpers.parseProtocol(),
    testName = [options.source.protocols.http, options.target.protocols.http].join('-to-'),
    runner = new helpers.TestRunner(options),
    routeFile = path.join(__dirname, 'config.json');

var fileOptions = {
  router: {
    "foo.com": "127.0.0.1:8101",
    "bar.com": "127.0.0.1:8102"
  }
};

var defaultOptions = {
  router: {
    "foo.com": "127.0.0.1:8091",
    "bar.com": "127.0.0.1:8092",
    "baz.com/taco": "127.0.0.1:8098",
    "pizza.com/taco/muffins": "127.0.0.1:8099",
  }
};

var hostnameOptions = {
  hostnameOnly: true,
  router: {
    "foo.com": "127.0.0.1:8091",
    "bar.com": "127.0.0.1:8092"
  }
};

var staticOptions = {
  hostnameOnly: true,
  router: {
    "static.com": "127.0.0.1:8121",
    "removed.com": "127.0.0.1:8122",
    "change.com": "127.0.0.1:8123"
  }
};

var dynamicHost = "dynamic1.com";
var dynamicTarget = "127.0.0.1:8125";

vows.describe('node-http-proxy/routing-proxy/' + testName).addBatch({
  "When using server created by httpProxy.createServer()": {
    "when passed a routing table": {
      "and routing by RegExp": {
        topic: function () {
          this.server = runner.startProxyServerWithTable(8090, defaultOptions, this.callback);
        },
        "an incoming request to foo.com": runner.assertProxied('foo.com', 8090, 8091),
        "an incoming request to bar.com": runner.assertProxied('bar.com', 8090, 8092),
        "an incoming request to baz.com/taco": runner.assertProxied('baz.com', 8090, 8098, "/taco", "/"),
        "an incoming request to pizza.com/taco/muffins": runner.assertProxied('pizza.com', 8090, 8099, "/taco/muffins", "/taco"),
        "an incoming request to unknown.com": runner.assertResponseCode(8090, 404)
      },
      "and routing by Hostname": {
        topic: function () {
          this.server = runner.startProxyServerWithTable(8093, hostnameOptions, this.callback);
        },
        "an incoming request to foo.com": runner.assertProxied('foo.com', 8093, 8094),
        "an incoming request to bar.com": runner.assertProxied('bar.com', 8093, 8095),
        "an incoming request to unknown.com": runner.assertResponseCode(8093, 404)
      },
      "and adding a route by Hostname": {
        topic: function () {
          this.server = runner.startProxyServerWithTable(8120, staticOptions, this.callback);
          this.server.removeHost('removed.com');
          this.server.addHost(dynamicHost, dynamicTarget);
          this.server.changeHost('change.com', '127.0.0.1:8124');
        },
        "an incoming request to static.com": runner.assertProxied('static.com', 8120, 8121),
        "an incoming request to removed.com": !runner.assertProxied('removed.com', 8120, 8122),
        "an incoming request to change.com": runner.assertProxied('change.com', 8120, 8124),
        "an incoming request to dynamic1.com": runner.assertProxied('dynamic1.com', 8120, 8125)
      }
    },
    "when passed a routing file": {
      topic: function () {
        fs.writeFileSync(routeFile, JSON.stringify(fileOptions));
        this.server = runner.startProxyServerWithTable(8100, {
          router: routeFile
        }, this.callback);
      },
      "an incoming request to foo.com": runner.assertProxied('foo.com', 8100, 8101),
      "an incoming request to bar.com": runner.assertProxied('bar.com', 8100, 8102),
      "an incoming request to unknown.com": runner.assertResponseCode(8100, 404),
      "an incoming request to dynamic.com": {
        "after the file has been modified": {
          topic: function () {
            var that = this,
                data = fs.readFileSync(routeFile),
                config = JSON.parse(data);

            config.router['dynamic.com'] = "127.0.0.1:8103";
            fs.writeFileSync(routeFile, JSON.stringify(config));

            this.server.on('routes', function () {
              runner.startTargetServer(8103, 'hello dynamic.com', function () {
                request({
                  method: 'GET',
                  uri: options.source.protocols.http + '://localhost:8100',
                  headers: {
                    host: 'dynamic.com'
                  }
                }, that.callback);
              });
            });
          },
          "should receive 'hello dynamic.com'": function (err, res, body) {
            assert.equal(body, 'hello dynamic.com');
          }
        }
      }
    }
  }
}).addBatch({
  "When using an instance of ProxyTable combined with HttpProxy directly": {
    topic: function () {
      this.server = runner.startProxyServerWithTableAndLatency(8110, 100, {
        router: {
          'foo.com': 'localhost:8111',
          'bar.com': 'localhost:8112'
        }
      }, this.callback);
    },
    "an incoming request to foo.com": runner.assertProxied('foo.com', 8110, 8111),
    "an incoming request to bar.com": runner.assertProxied('bar.com', 8110, 8112),
    "an incoming request to unknown.com": runner.assertResponseCode(8110, 404)
  }
}).addBatch({
  "When the tests are over": {
    topic: function () {
      //fs.unlinkSync(routeFile);
      return runner.closeServers();
    },
    "the servers should clean up": function () {
      assert.isTrue(true);
    }
  }
}).export(module);