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

http proxy middleware for koa

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

describe("honors timeout option", function () {
  "use strict";

  var other, http;
  beforeEach(function () {
    http = new Koa();
    other = proxyTarget(8080, 1000, [
      {
        method: "get",
        path: "/",
        fn: function (req, res) {
          res.sendStatus(200);
        },
      },
    ]);
  });

  afterEach(function () {
    other.close();
  });

  function assertSuccess(server, done) {
    agent(http.callback())
      .get("/")
      .expect(200)
      .end(function (err) {
        if (err) {
          return done(err);
        }
        done();
      });
  }

  function assertConnectionTimeout(server, done) {
    agent(http.callback())
      .get("/")
      .expect(504)
      .expect(
        "X-Timout-Reason",
        "koa-http-proxy timed out your request after 100ms.",
      )
      .end(function (err) {
        if (err) {
          return done(err);
        }
        done();
      });
  }

  describe("when timeout option is set lower than server response time", function () {
    it("should fail with CONNECTION TIMEOUT", function (done) {
      http.use(
        proxy("http://localhost:8080", {
          timeout: 100,
        }),
      );

      assertConnectionTimeout(http, done);
    });
  });

  describe("when timeout option is set higher than server response time", function () {
    it("should succeed", function (done) {
      http.use(
        proxy("http://localhost:8080", {
          timeout: 1200,
        }),
      );

      assertSuccess(http, done);
    });
  });
});