ddos/test/test-maxcount.js

Configurable Denial-Of-Service prevention for http services

const tape = require("tape");
const Ddos = require("../");
const request = require("supertest-light");
const express = require("express");

tape("maxcount ", function(t) {
  t.plan(1);
  const ddos = new Ddos({ burst: 3, limit: 2 });
  const app = express();
  app.use(ddos.express);
  app.get("/user", (req, res) => {
    res.status(200).json({ name: "john" });
  });

  const doCall = function() {
    return request(app).get("/user");
  };

  doCall()
    .then(() => {
      return doCall();
    })
    .then(() => {
      return doCall();
    })
    .then(() => {
      return doCall();
    })
    .then(() => {
      return doCall();
    })
    .then(res => {
      ddos.end();
      t.pass();
    });
});