| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- assert = require("chai").assert
- sinon = require('sinon')
- chai = require('chai')
- should = chai.should()
- expect = chai.expect
- modulePath = "../../../../app/js/infrastructure/RateLimiter.js"
- SandboxedModule = require('sandboxed-module')
- describe "RateLimiter", ->
- beforeEach ->
- @settings =
- redis:
- web:
- port:"1234"
- host:"somewhere"
- password: "password"
- @rclient =
- incr: sinon.stub()
- get: sinon.stub()
- expire: sinon.stub()
- exec: sinon.stub()
- @rclient.multi = sinon.stub().returns(@rclient)
- @RedisWrapper =
- client: sinon.stub().returns(@rclient)
- @endpointName = "compiles"
- @subject = "some-project-id"
- @timeInterval = 20
- @throttleLimit = 5
- @requires =
- "settings-sharelatex":@settings
- "logger-sharelatex" : @logger = {log:sinon.stub(), err:sinon.stub()}
- "./RedisWrapper": @RedisWrapper
- @details =
- endpointName: @endpointName
- subjectName: @subject
- throttle: @throttleLimit
- timeInterval: @timeInterval
- @key = "RateLimiter:#{@endpointName}:{#{@subject}}"
- describe 'when action is permitted', ->
- beforeEach ->
- @requires["rolling-rate-limiter"] = (opts) =>
- return sinon.stub().callsArgWith(1, null, 0, 22)
- @limiter = SandboxedModule.require modulePath, requires: @requires
- it 'should not produce and error', (done) ->
- @limiter.addCount {}, (err, should) ->
- expect(err).to.equal null
- done()
- it 'should callback with true', (done) ->
- @limiter.addCount {}, (err, should) ->
- expect(should).to.equal true
- done()
- describe 'when action is not permitted', ->
- beforeEach ->
- @requires["rolling-rate-limiter"] = (opts) =>
- return sinon.stub().callsArgWith(1, null, 4000, 0)
- @limiter = SandboxedModule.require modulePath, requires: @requires
- it 'should not produce and error', (done) ->
- @limiter.addCount {}, (err, should) ->
- expect(err).to.equal null
- done()
- it 'should callback with false', (done) ->
- @limiter.addCount {}, (err, should) ->
- expect(should).to.equal false
- done()
- describe 'when limiter produces an error', ->
- beforeEach ->
- @requires["rolling-rate-limiter"] = (opts) =>
- return sinon.stub().callsArgWith(1, new Error('woops'))
- @limiter = SandboxedModule.require modulePath, requires: @requires
- it 'should produce and error', (done) ->
- @limiter.addCount {}, (err, should) ->
- expect(err).to.not.equal null
- expect(err).to.be.instanceof Error
- done()
|
