express-limit/test/in-memory-store.test.js

A rate-limiter for Express JS.

"use strict";

const InMemoryStore = require("../src/in-memory-store");

test("InMemoryStore - Test increment", function (done) {
  const key = "test-1";
  const reset = Date.now() + 1000;
  const rateLimitStore = new InMemoryStore();

  rateLimitStore.increment(key, reset, (limits) => {
    expect(limits.current).toEqual(1);
    expect(limits.reset).toEqual(reset);

    rateLimitStore.increment(key, reset, (limits) => {
      expect(limits.current).toEqual(2);
      expect(limits.reset).toEqual(reset);

      done();
    });
  });
});

test("InMemoryStore - Test increment expiration", function (done) {
  const key = "test-2";
  const reset = Date.now() + 1000;
  const rateLimitStore = new InMemoryStore();

  rateLimitStore.increment(key, reset, (limits) => {
    expect(limits.current).toEqual(1);
    expect(limits.reset).toEqual(reset);

    rateLimitStore.increment(key, reset, (limits) => {
      expect(limits.current).toEqual(2);
      expect(limits.reset).toEqual(reset);

      setTimeout(() => {
        rateLimitStore.increment(key, reset + 1000, (limits) => {
          expect(limits.current).toEqual(1);
          expect(limits.reset).toEqual(reset + 1000);

          done();
        });
      }, 1500);
    });
  });
});