c9-core/node_modules/c9/ratelimit_test.js

"use server";

require("c9/inline-mocha")(module);

var ratelimit = require("./ratelimit");
var assert = require("assert");
var async = require("async");

describe("ratelimit", function() {
    it("Should limit based on key", function (done) {
        var limiter = ratelimit("username", 10, 1);
        limiter({params: {username: "super"}}, null, function (err) {
            assert(!err, err);
            limiter({params: {username: "super"}}, null, function (err) {
                assert(err);
                assert.equal(err.code, 429);
                done();
            });
        });
    });
    
    it("Should work again after a delay", function (done) {
        var limiter = ratelimit("username", 10, 1);
        limiter({params: {username: "super"}}, null, function (err) {
            assert(!err, err);
            setTimeout(function() {
                limiter({params: {username: "super"}}, null, function (err) {
                    assert(!err, err);
                    done();
                });
            }, 25);
        });
    });
    
    it("Should work with many requests", function (done) {
        var MAX_REQUESTS = 5;
        var limiter = ratelimit("username", 10, MAX_REQUESTS);
        var successfulRequests = 0;
        async.times(10, function(n, next) {
            limiter({params: {username: "super"}}, null, function (err) {
                if (err) return next(err);
                successfulRequests++;
                next();
            });
        }, function (err) {
            assert.equal(successfulRequests, MAX_REQUESTS);
            setTimeout(function() {
                limiter({params: {username: "super"}}, null, function (err) {
                    assert(!err, err);
                    done();
                });
            }, 25);
        });
    });
    
    
});