ddos/test/test-localhost.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");

// https://github.com/rook2pawn/node-ddos/issues/31

tape("localhost ", function(t) {
  t.plan(4);

  const ddos = new Ddos({ burst: 3, limit: 2 });
  t.equals("::ffff:127.0.0.1".match(ddos.ipv4re)[2], "127.0.0.1");
  t.equals("127.0.0.1".match(ddos.ipv4re)[2], "127.0.0.1");
  t.equals("32.45.32.65:12568".match(ddos.ipv4re)[2], "32.45.32.65:12568");

  const app = express();
  app.use(ddos.express);
  app.get("/user", (req, res) => {
    res.status(200).json({ name: "john" });
  });

  const doCall = function() {
    return request(app)
      .set("x-forwarded-for", "::1")
      .get("/user");
  };

  doCall()
    .then(() => {
      t.equals(
        Object.keys(ddos.table)[0].match(/127\.0\.0\.1/)[0],
        "127.0.0.1"
      );
    })
    .then(() => {
      ddos.end();
    });
});