@reedchan/koa-http-proxy/test/host.js

http proxy middleware for koa

var Koa = require("koa");
var agent = require("supertest").agent;
var proxy = require("../");

describe("host can be a dynamic function", function () {
  "use strict";

  this.timeout(10000);

  var app = new Koa();
  var firstProxyApp = new Koa();
  var secondProxyApp = new Koa();
  var firstPort = 10001;
  var secondPort = 10002;

  app.use(
    proxy(function (ctx) {
      return "localhost:" + ctx.url.replace("/proxy/", "");
    }),
  );

  firstProxyApp.use(function (ctx) {
    ctx.status = 204;
  });
  firstProxyApp.listen(firstPort);

  secondProxyApp.use(function (ctx) {
    ctx.status = 200;
  });
  secondProxyApp.listen(secondPort);

  it("can proxy with session value", function (done) {
    agent(app.callback())
      .get("/proxy/" + firstPort)
      .expect(204)
      .end(function (err) {
        if (err) {
          return done(err);
        }
        agent(app.callback())
          .get("/proxy/" + secondPort)
          .expect(200, done);
      });
  });
});