Merge pull request #3966 from overleaf/jpa-acceptance-tests-rate-limiter
[misc] replace RateLimiter unit tests with acceptance tests on /metrics GitOrigin-RevId: c4ef502b38bb4b300c6b9812e1c97858e93a38fe
This commit is contained in:
@@ -1,155 +0,0 @@
|
||||
/* eslint-disable
|
||||
node/handle-callback-err,
|
||||
max-len,
|
||||
no-return-assign,
|
||||
no-unused-vars,
|
||||
*/
|
||||
// TODO: This file was created by bulk-decaffeinate.
|
||||
// Fix any style issues and re-enable lint.
|
||||
/*
|
||||
* decaffeinate suggestions:
|
||||
* DS102: Remove unnecessary code created because of implicit returns
|
||||
* Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
|
||||
*/
|
||||
const { assert, expect } = require('chai')
|
||||
const sinon = require('sinon')
|
||||
const modulePath = '../../../../app/src/infrastructure/RateLimiter.js'
|
||||
const SandboxedModule = require('sandboxed-module')
|
||||
|
||||
describe('RateLimiter', function () {
|
||||
beforeEach(function () {
|
||||
this.settings = {
|
||||
redis: {
|
||||
web: {
|
||||
port: '1234',
|
||||
host: 'somewhere',
|
||||
password: 'password',
|
||||
},
|
||||
},
|
||||
}
|
||||
this.rclient = {
|
||||
incr: sinon.stub(),
|
||||
get: sinon.stub(),
|
||||
expire: sinon.stub(),
|
||||
exec: sinon.stub(),
|
||||
}
|
||||
this.rclient.multi = sinon.stub().returns(this.rclient)
|
||||
this.RedisWrapper = { client: sinon.stub().returns(this.rclient) }
|
||||
|
||||
this.endpointName = 'compiles'
|
||||
this.subject = 'some-project-id'
|
||||
this.timeInterval = 20
|
||||
this.throttleLimit = 5
|
||||
|
||||
this.requires = {
|
||||
'settings-sharelatex': this.settings,
|
||||
'@overleaf/metrics': (this.Metrics = { inc: sinon.stub() }),
|
||||
'./RedisWrapper': this.RedisWrapper,
|
||||
}
|
||||
|
||||
this.details = {
|
||||
endpointName: this.endpointName,
|
||||
subjectName: this.subject,
|
||||
throttle: this.throttleLimit,
|
||||
timeInterval: this.timeInterval,
|
||||
}
|
||||
return (this.key = `RateLimiter:${this.endpointName}:{${this.subject}}`)
|
||||
})
|
||||
|
||||
describe('when action is permitted', function () {
|
||||
beforeEach(function () {
|
||||
this.requires['rolling-rate-limiter'] = opts => {
|
||||
return sinon.stub().callsArgWith(1, null, 0, 22)
|
||||
}
|
||||
return (this.limiter = SandboxedModule.require(modulePath, {
|
||||
requires: this.requires,
|
||||
}))
|
||||
})
|
||||
|
||||
it('should not produce and error', function (done) {
|
||||
return this.limiter.addCount({}, (err, should) => {
|
||||
expect(err).to.equal(null)
|
||||
return done()
|
||||
})
|
||||
})
|
||||
|
||||
it('should callback with true', function (done) {
|
||||
return this.limiter.addCount({}, (err, should) => {
|
||||
expect(should).to.equal(true)
|
||||
return done()
|
||||
})
|
||||
})
|
||||
|
||||
it('should not increment the metric', function (done) {
|
||||
return this.limiter.addCount(
|
||||
{ endpointName: this.endpointName },
|
||||
(err, should) => {
|
||||
sinon.assert.notCalled(this.Metrics.inc)
|
||||
return done()
|
||||
}
|
||||
)
|
||||
})
|
||||
})
|
||||
|
||||
describe('when action is not permitted', function () {
|
||||
beforeEach(function () {
|
||||
this.requires['rolling-rate-limiter'] = opts => {
|
||||
return sinon.stub().callsArgWith(1, null, 4000, 0)
|
||||
}
|
||||
return (this.limiter = SandboxedModule.require(modulePath, {
|
||||
globals: {
|
||||
console: console,
|
||||
},
|
||||
requires: this.requires,
|
||||
}))
|
||||
})
|
||||
|
||||
it('should not produce and error', function (done) {
|
||||
return this.limiter.addCount({}, (err, should) => {
|
||||
expect(err).to.equal(null)
|
||||
return done()
|
||||
})
|
||||
})
|
||||
|
||||
it('should callback with false', function (done) {
|
||||
return this.limiter.addCount({}, (err, should) => {
|
||||
expect(should).to.equal(false)
|
||||
return done()
|
||||
})
|
||||
})
|
||||
|
||||
it('should increment the metric', function (done) {
|
||||
return this.limiter.addCount(
|
||||
{ endpointName: this.endpointName },
|
||||
(err, should) => {
|
||||
sinon.assert.calledWith(this.Metrics.inc, `rate-limit-hit`, 1, {
|
||||
path: this.endpointName,
|
||||
})
|
||||
return done()
|
||||
}
|
||||
)
|
||||
})
|
||||
})
|
||||
|
||||
describe('when limiter produces an error', function () {
|
||||
beforeEach(function () {
|
||||
this.requires['rolling-rate-limiter'] = opts => {
|
||||
return sinon.stub().callsArgWith(1, new Error('woops'))
|
||||
}
|
||||
return (this.limiter = SandboxedModule.require(modulePath, {
|
||||
globals: {
|
||||
console: console,
|
||||
},
|
||||
requires: this.requires,
|
||||
}))
|
||||
})
|
||||
|
||||
it('should produce and error', function (done) {
|
||||
return this.limiter.addCount({}, (err, should) => {
|
||||
expect(err).to.not.equal(null)
|
||||
expect(err).to.be.instanceof(Error)
|
||||
return done()
|
||||
})
|
||||
})
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user