koa-better-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-better-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);
    });
  });

});