authmaker-verify/test/mongo/auditCount.js

AccessToken verification for Authmaker

const chai = require('chai');
const chaiAsPromised = require('chai-as-promised');
const Q = require('q');
const moment = require('moment');

const expect = chai.expect;
chai.use(chaiAsPromised);

const fixture = rootRequire('test/fixtures/users');

const mongoRateLimited = rootRequire('lib/mongo/rateLimited');
const mongoAuditCount = rootRequire('lib/mongo/auditCount');
const commonAccessTokenTests = rootRequire('test/helpers/commonAccessTokenTests');

function manyRequests(params, number) {
  const promises = [];

  for (let i = 0; i < number; i += 1) {
    promises.push(mongoRateLimited(...params));
  }

  return Q.all(promises);
}

describe('mongo auditCount function', function () {
  beforeEach(function () {
    return fixture.init();
  });

  afterEach(function () {
    return fixture.reset();
  });

  commonAccessTokenTests(mongoAuditCount);

  // all following tests assume a valid access_token
  it('should return 0 with no requests', function () {
    return expect(mongoAuditCount('valid_rate_limit_5_second', 'face')).to.eventually.become(0);
  });

  it('should return 1 with one request', function () {
    return mongoRateLimited('valid_rate_limit_5_second', 'face').then(function () {
      return expect(mongoAuditCount('valid_rate_limit_5_second', 'face')).to.eventually.become(1);
    });
  });

  it('should return 100 with 100 requests', function () {
    return manyRequests(['valid_rate_limit_5_second', 'face'], 100).then(function () {
      return expect(mongoAuditCount('valid_rate_limit_5_second', 'face')).to.eventually.become(100);
    });
  });

  it('should return 10 with 2 sets of 5 requests with 1 second between them', function () {
    return manyRequests(['valid_rate_limit_5_second', 'face'], 5)
      .delay(1000)
      .then(function () {
        return manyRequests(['valid_rate_limit_5_second', 'face'], 5);
      })
      .then(function () {
        return expect(mongoAuditCount('valid_rate_limit_5_second', 'face')).to.eventually.become(10);
      });
  });

  it('should return 0 when called with since parameter: 1 second ago after 5 requests and a 1.1 second pause', function () {
    return manyRequests(['valid_rate_limit_5_second', 'face'], 5)
      .delay(1100)
      .then(function () {
        return expect(mongoAuditCount('valid_rate_limit_5_second', 'face', moment().subtract(1, 'seconds'))).to.eventually.become(0);
      });
  });

  it('should return 5 when called with since parameter: 1 second ago after 5 requests, 1.1 second pause, 5 requests', function () {
    return manyRequests(['valid_rate_limit_5_second', 'face'], 5)
      .delay(1100)
      .then(function () {
        return manyRequests(['valid_rate_limit_5_second', 'face'], 5);
      })
      .then(function () {
        return expect(mongoAuditCount('valid_rate_limit_5_second', 'face', moment().subtract(1, 'seconds'))).to.eventually.become(5);
      });
  });

  it('should return 2 after 2 requests are made with different access tokens', function () {
    return mongoRateLimited('valid_rate_limit_5_days_user3_access_token1', 'face')
      .then(function () {
        return mongoRateLimited('valid_rate_limit_5_days_user3_access_token2', 'face');
      })
      .then(function () {
        return expect(mongoAuditCount('valid_rate_limit_5_days_user3_access_token1', 'face', moment().subtract(1, 'seconds'))).to.eventually.become(2);
      });
  });
});