ddos/test/test-stop.js

Configurable Denial-Of-Service prevention for http services

var http = require("http");
var express = require("express");
var bodyParser = require("body-parser");
var request = require("request");
var QL = require("queuelib");
var Ddos = require("../");
var ddos = new Ddos();
var tape = require("tape");

tape("stop test", function(t) {
  var app = express();
  app.use(ddos.express);
  app.use(bodyParser.json());
  var q = new QL();
  var server = http.createServer(app);
  server.listen(5050);

  var a = function(req, res, next) {
    next();
  };
  var b = function(req, res, next) {
    // some more random middleware
    next();
  };
  var c = function(req, res, next) {
    var num = req.body.num * 2;
    res.end(JSON.stringify({ foo: num }));
  };
  app.post("/article", a, b, c);

  t.plan(6);
  q.series([
    function(lib) {
      request.post(
        { url: "http://localhost:5050/article", json: true, body: { num: 42 } },
        function(err, resp, body) {
          var key = Object.keys(ddos.table)[0];
          t.deepEqual(ddos.table[key], { count: 1, expiry: 1 });
          t.deepEqual(body, { foo: 84 });
          lib.done();
        }
      );
    },
    function(lib) {
      request.post(
        { url: "http://localhost:5050/article", json: true, body: { num: 42 } },
        function(err, resp, body) {
          var key = Object.keys(ddos.table)[0];
          t.deepEqual(ddos.table[key], { count: 2, expiry: 1 });
          t.deepEqual(body, { foo: 84 });
          ddos.stop();
          lib.done();
        }
      );
    },
    function(lib) {
      request.post(
        { url: "http://localhost:5050/article", json: true, body: { num: 42 } },
        function(err, resp, body) {
          var key = Object.keys(ddos.table)[0];
          t.deepEqual(ddos.table[key], { count: 2, expiry: 1 });
          t.deepEqual(body, { foo: 84 });
          lib.done();
        }
      );
    },
    function(lib) {
      lib.done();
      t.end();
      server.close();
      ddos.stop();
    }
  ]);
});