authelia/test/unitary/test_authentication_regulator.js

var AuthenticationRegulator = require('../../src/lib/authentication_regulator');
var UserDataStore = require('../../src/lib/user_data_store');
var DataStore = require('nedb');
var exceptions = require('../../src/lib/exceptions');
var MockDate = require('mockdate');

describe('test authentication regulator', function() {
  it('should mark 2 authentication and regulate (resolve)', function() {
    var options = {};
    options.inMemoryOnly = true;
    var data_store = new UserDataStore(DataStore, options);
    var regulator = new AuthenticationRegulator(data_store, 10);
    var 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) {
    var options = {};
    options.inMemoryOnly = true;
    var data_store = new UserDataStore(DataStore, options);
    var regulator = new AuthenticationRegulator(data_store, 10);
    var 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) {
    var options = {};
    options.inMemoryOnly = true;
    var data_store = new UserDataStore(DataStore, options);
    var regulator = new AuthenticationRegulator(data_store, 10);
    var 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();
    })
  });
});