First commit

framework/node_modules/node-rate-limiter-flexible/test/RateLimiterUnion.test.js

@@ -0,0 +1,83 @@
+// eslint-disable no-unused-expressions
+const { describe, it, beforeEach } = require('mocha');
+const { expect } = require('chai');
+const RateLimiterUnion = require('../lib/RateLimiterUnion');
+const RateLimiterMemory = require('../lib/RateLimiterMemory');
+
+describe('RateLimiterUnion with fixed window', () => {
+  const keyPrefix1 = 'limit1';
+  const keyPrefix2 = 'limit2';
+  let rateLimiter;
+
+  beforeEach(() => {
+    const limiter1 = new RateLimiterMemory({
+      keyPrefix: keyPrefix1,
+      points: 1,
+      duration: 1,
+    });
+    const limiter2 = new RateLimiterMemory({
+      keyPrefix: keyPrefix2,
+      points: 2,
+      duration: 5,
+    });
+    rateLimiter = new RateLimiterUnion(limiter1, limiter2);
+  });
+
+  it('does not allow to create union with limiters number less than 2', () => {
+    try {
+      new RateLimiterUnion(new RateLimiterMemory({ // eslint-disable-line no-new
+        keyPrefix: keyPrefix1,
+        points: 1,
+        duration: 1,
+      }));
+    } catch (err) {
+      expect(err instanceof Error).to.equal(true);
+    }
+  });
+
+  it('all limiters have to be instance of RateLimiterAbstract', () => {
+    try {
+      new RateLimiterUnion(new RateLimiterMemory({ // eslint-disable-line no-new
+        keyPrefix: keyPrefix1,
+        points: 1,
+        duration: 1,
+      }), {});
+    } catch (err) {
+      expect(err instanceof Error).to.equal(true);
+    }
+  });
+
+  it('consume from all limiters', (done) => {
+    rateLimiter.consume('test')
+      .then((res) => {
+        expect(res[keyPrefix1].remainingPoints === 0 && res[keyPrefix2].remainingPoints === 1).to.equal(true);
+        done();
+      })
+      .catch(() => {
+        done(Error('must not reject'));
+      });
+  });
+
+  it('reject consume one "limit1", which does not have enough points', (done) => {
+    rateLimiter.consume('test', 2)
+      .then(() => {
+        done(Error('must not resolve'));
+      })
+      .catch((rej) => {
+        expect(rej[keyPrefix1].remainingPoints === 0).to.equal(true);
+        done();
+      });
+  });
+
+  it('reject both do not have enough points', (done) => {
+    rateLimiter.consume('test', 3)
+      .then(() => {
+        done(Error('must not resolve'));
+      })
+      .catch((rej) => {
+        expect(rej[keyPrefix1].remainingPoints === 0 && rej[keyPrefix2].remainingPoints === 0).to.equal(true);
+        done();
+      });
+  });
+});
+