kopia lustrzana https://github.com/c9/core
104 wiersze
3.4 KiB
JavaScript
104 wiersze
3.4 KiB
JavaScript
"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 with different keys", function (done) {
|
|
var limiter = ratelimit("username", 10, 1);
|
|
limiter({params: {username: "super"}}, null, function (err) {
|
|
assert(!err, err);
|
|
limiter({params: {username: "aloha"}}, null, function (err) {
|
|
assert(!err, err);
|
|
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);
|
|
});
|
|
});
|
|
|
|
it("Should expire keys at the correct times", function (done) {
|
|
var limiter = ratelimit("username", 50, 2);
|
|
async.series([
|
|
function(next) {
|
|
limiter({params: {username: "mario"}}, null, function(err) {
|
|
assert(!err, err);
|
|
setTimeout(next, 25);
|
|
});
|
|
},
|
|
function (next) {
|
|
limiter({params: {username: "mario"}}, null, function(err) {
|
|
assert(!err, err);
|
|
setTimeout(next, 40);
|
|
});
|
|
},
|
|
function (next) {
|
|
limiter({params: {username: "mario"}}, null, function(err) {
|
|
assert(!err, err);
|
|
limiter({params: {username: "mario"}}, null, function(err) {
|
|
assert(err);
|
|
assert.equal(err.code, 429);
|
|
setTimeout(next, 20);
|
|
});
|
|
});
|
|
},
|
|
function (next) {
|
|
limiter({params: {username: "mario"}}, null, function(err) {
|
|
assert(!err, err);
|
|
next();
|
|
});
|
|
}
|
|
], done);
|
|
});
|
|
|
|
}); |