authelia/test/unit/server/AuthenticationRegulator.test.ts

import { AuthenticationRegulator } from "../../../src/server/lib/AuthenticationRegulator";
import UserDataStore from "../../../src/server/lib/UserDataStore";
import MockDate = require("mockdate");
import exceptions = require("../../../src/server/lib/Exceptions");
import nedb = require("nedb");

describe("test authentication regulator", function() {
  it("should mark 2 authentication and regulate (resolve)", function() {
    const options = {
      inMemoryOnly: true
    };
    const data_store = new UserDataStore(options, nedb);
    const regulator = new AuthenticationRegulator(data_store, 10);
    const user = "user";

    return regulator.mark(user, false)
    .then(function() {
      return regulator.mark(user, true);
    })
    .then(function() {
      return regulator.regulate(user);
    });
  });

  it("should mark 3 authentications and regulate (reject)", function(done) {
    const options = {
      inMemoryOnly: true
    };
    const data_store = new UserDataStore(options, nedb);
    const regulator = new AuthenticationRegulator(data_store, 10);
    const user = "user";

    regulator.mark(user, false)
    .then(function() {
      return regulator.mark(user, false);
    })
    .then(function() {
      return regulator.mark(user, false);
    })
    .then(function() {
      return regulator.regulate(user);
    })
    .catch(exceptions.AuthenticationRegulationError, function() {
      done();
    });
  });

  it("should mark 3 authentications and regulate (resolve)", function(done) {
    const options = {
      inMemoryOnly: true
    };
    const data_store = new UserDataStore(options, nedb);
    const regulator = new AuthenticationRegulator(data_store, 10);
    const user = "user";

    MockDate.set("1/2/2000 00:00:00");
    regulator.mark(user, false)
    .then(function() {
      MockDate.set("1/2/2000 00:00:15");
      return regulator.mark(user, false);
    })
    .then(function() {
      return regulator.mark(user, false);
    })
    .then(function() {
      return regulator.regulate(user);
    })
    .then(function() {
      done();
    });
  });
});