mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-30 15:04:58 -05:00
82 lines
2.1 KiB
CoffeeScript
82 lines
2.1 KiB
CoffeeScript
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 "FileStoreHandler", ->
|
|
|
|
beforeEach ->
|
|
@settings =
|
|
redis:
|
|
web:
|
|
port:"1234"
|
|
host:"somewhere"
|
|
password: "password"
|
|
@redbackInstance =
|
|
addCount: sinon.stub()
|
|
|
|
@redback =
|
|
createRateLimit: sinon.stub().returns(@redbackInstance)
|
|
@redis =
|
|
createClient: ->
|
|
return auth:->
|
|
|
|
@limiter = SandboxedModule.require modulePath, requires:
|
|
"settings-sharelatex":@settings
|
|
"logger-sharelatex" : @logger = {log:sinon.stub(), err:sinon.stub()}
|
|
"redis-sharelatex": @redis
|
|
"redback": use: => @redback
|
|
|
|
@endpointName = "compiles"
|
|
@subject = "some project id"
|
|
@timeInterval = 20
|
|
@throttleLimit = 5
|
|
|
|
@details =
|
|
endpointName: @endpointName
|
|
subjectName: @subject
|
|
throttle: @throttleLimit
|
|
timeInterval: @timeInterval
|
|
|
|
|
|
describe "addCount", ->
|
|
|
|
beforeEach ->
|
|
@redbackInstance.addCount.callsArgWith(2, null, 10)
|
|
|
|
it "should use correct namespace", (done)->
|
|
@limiter.addCount @details, =>
|
|
@redback.createRateLimit.calledWith(@endpointName).should.equal true
|
|
done()
|
|
|
|
it "should only call it once", (done)->
|
|
@limiter.addCount @details, =>
|
|
@redbackInstance.addCount.callCount.should.equal 1
|
|
done()
|
|
|
|
it "should use the subjectName", (done)->
|
|
@limiter.addCount @details, =>
|
|
@redbackInstance.addCount.calledWith(@details.subjectName, @details.timeInterval).should.equal true
|
|
done()
|
|
|
|
it "should return true if the count is less than throttle", (done)->
|
|
@details.throttle = 100
|
|
@limiter.addCount @details, (err, canProcess)=>
|
|
canProcess.should.equal true
|
|
done()
|
|
|
|
it "should return true if the count is less than throttle", (done)->
|
|
@details.throttle = 1
|
|
@limiter.addCount @details, (err, canProcess)=>
|
|
canProcess.should.equal false
|
|
done()
|
|
|
|
it "should return false if the limit is matched", (done)->
|
|
@details.throttle = 10
|
|
@limiter.addCount @details, (err, canProcess)=>
|
|
canProcess.should.equal false
|
|
done()
|
|
|