cartodb/CartoDB-SQL-API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactoring rate limit test

Browse Source
This commit is contained in:
Simon Martín 2018-03-02 20:14:10 +01:00
parent 060a43d10d
commit 83c5b81d4e
1 changed files with 25 additions and 103 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

128
test/acceptance/rate-limit.js
View File

@ -39,6 +39,25 @@ function setLimit(count, period, burst) {
});
}
function checkResult (status, limit, remaining, reset, retry, done = null) {
assert.response(
server,
request,
{ status },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], limit);
assert.equal(res.headers['x-rate-limit-remaining'], remaining);
assert.equal(res.headers['x-rate-limit-reset'], reset);
assert.equal(res.headers['x-rate-limit-retry-after'], retry);
if (done) {
setTimeout(done, 1000);
}
}
);
}
describe('rate limit', function() {
before(function() {
global.settings.ratelimits.rateLimitsEnabled = true;
@ -63,109 +82,12 @@ describe('rate limit', function() {
});
it("1 req/sec: 2 req/seg should be limited", function(done) {
assert.response(
server,
request,
{ status: 200 },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], '2');
assert.equal(res.headers['x-rate-limit-remaining'], '1');
assert.equal(res.headers['x-rate-limit-reset'], '1');
assert.equal(res.headers['x-rate-limit-retry-after'], '-1');
}
);
setTimeout(
function() {
assert.response(
server,
request,
{ status: 200 },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], '2');
assert.equal(res.headers['x-rate-limit-remaining'], '0');
assert.equal(res.headers['x-rate-limit-reset'], '1');
assert.equal(res.headers['x-rate-limit-retry-after'], '-1');
}
);
},
250
);
setTimeout(
function() {
assert.response(
server,
request,
{ status: 429 },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], '2');
assert.equal(res.headers['x-rate-limit-remaining'], '0');
assert.equal(res.headers['x-rate-limit-reset'], '1');
assert.equal(res.headers['x-rate-limit-retry-after'], '0');
}
);
},
500
);
setTimeout(
function() {
assert.response(
server,
request,
{ status: 429 },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], '2');
assert.equal(res.headers['x-rate-limit-remaining'], '0');
assert.equal(res.headers['x-rate-limit-reset'], '1');
assert.equal(res.headers['x-rate-limit-retry-after'], '0');
}
);
},
750
);
setTimeout(
function() {
assert.response(
server,
request,
{ status: 429 },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], '2');
assert.equal(res.headers['x-rate-limit-remaining'], '0');
assert.equal(res.headers['x-rate-limit-reset'], '1');
assert.equal(res.headers['x-rate-limit-retry-after'], '0');
}
);
},
950
);
setTimeout(
function() {
assert.response(
server,
request,
{ status: 200 },
function(err, res) {
assert.ifError(err);
assert.equal(res.headers['x-rate-limit-limit'], '2');
assert.equal(res.headers['x-rate-limit-remaining'], '0');
assert.equal(res.headers['x-rate-limit-reset'], '1');
assert.equal(res.headers['x-rate-limit-retry-after'], '-1');
setTimeout(done, 1000);
}
);
},
1050
);
checkResult(200, 2, 1, 1, -1);
setTimeout( () => checkResult(200, 2, 0, 1, -1), 250 );
setTimeout( () => checkResult(429, 2, 0, 1, 0), 500 );
setTimeout( () => checkResult(429, 2, 0, 1, 0), 750 );
setTimeout( () => checkResult(429, 2, 0, 1, 0), 950 );
setTimeout( () => checkResult(200, 2, 0, 1, -1, done), 1050 );
});
});