mirror of
https://github.com/overleaf/overleaf.git
synced 2024-12-28 02:51:25 +00:00
132 lines
3.9 KiB
JavaScript
132 lines
3.9 KiB
JavaScript
/* eslint-disable
|
|
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 should = require('chai').should()
|
|
const sinon = require('sinon')
|
|
const SandboxedModule = require('sandboxed-module')
|
|
const path = require('path')
|
|
const modulePath = path.join(__dirname, '../../../app/js/DrainManager')
|
|
|
|
describe('DrainManager', function () {
|
|
beforeEach(function () {
|
|
this.DrainManager = SandboxedModule.require(modulePath, {
|
|
requires: {
|
|
'logger-sharelatex': (this.logger = { log: sinon.stub() })
|
|
}
|
|
})
|
|
return (this.io = {
|
|
sockets: {
|
|
clients: sinon.stub()
|
|
}
|
|
})
|
|
})
|
|
|
|
describe('startDrainTimeWindow', function () {
|
|
beforeEach(function () {
|
|
this.clients = []
|
|
for (let i = 0; i <= 5399; i++) {
|
|
this.clients[i] = {
|
|
id: i,
|
|
emit: sinon.stub()
|
|
}
|
|
}
|
|
this.io.sockets.clients.returns(this.clients)
|
|
return (this.DrainManager.startDrain = sinon.stub())
|
|
})
|
|
|
|
return it('should set a drain rate fast enough', function (done) {
|
|
this.DrainManager.startDrainTimeWindow(this.io, 9)
|
|
this.DrainManager.startDrain.calledWith(this.io, 10).should.equal(true)
|
|
return done()
|
|
})
|
|
})
|
|
|
|
return describe('reconnectNClients', function () {
|
|
beforeEach(function () {
|
|
this.clients = []
|
|
for (let i = 0; i <= 9; i++) {
|
|
this.clients[i] = {
|
|
id: i,
|
|
emit: sinon.stub()
|
|
}
|
|
}
|
|
return this.io.sockets.clients.returns(this.clients)
|
|
})
|
|
|
|
return describe('after first pass', function () {
|
|
beforeEach(function () {
|
|
return this.DrainManager.reconnectNClients(this.io, 3)
|
|
})
|
|
|
|
it('should reconnect the first 3 clients', function () {
|
|
return [0, 1, 2].map((i) =>
|
|
this.clients[i].emit
|
|
.calledWith('reconnectGracefully')
|
|
.should.equal(true)
|
|
)
|
|
})
|
|
|
|
it('should not reconnect any more clients', function () {
|
|
return [3, 4, 5, 6, 7, 8, 9].map((i) =>
|
|
this.clients[i].emit
|
|
.calledWith('reconnectGracefully')
|
|
.should.equal(false)
|
|
)
|
|
})
|
|
|
|
return describe('after second pass', function () {
|
|
beforeEach(function () {
|
|
return this.DrainManager.reconnectNClients(this.io, 3)
|
|
})
|
|
|
|
it('should reconnect the next 3 clients', function () {
|
|
return [3, 4, 5].map((i) =>
|
|
this.clients[i].emit
|
|
.calledWith('reconnectGracefully')
|
|
.should.equal(true)
|
|
)
|
|
})
|
|
|
|
it('should not reconnect any more clients', function () {
|
|
return [6, 7, 8, 9].map((i) =>
|
|
this.clients[i].emit
|
|
.calledWith('reconnectGracefully')
|
|
.should.equal(false)
|
|
)
|
|
})
|
|
|
|
it('should not reconnect the first 3 clients again', function () {
|
|
return [0, 1, 2].map((i) =>
|
|
this.clients[i].emit.calledOnce.should.equal(true)
|
|
)
|
|
})
|
|
|
|
return describe('after final pass', function () {
|
|
beforeEach(function () {
|
|
return this.DrainManager.reconnectNClients(this.io, 100)
|
|
})
|
|
|
|
it('should not reconnect the first 6 clients again', function () {
|
|
return [0, 1, 2, 3, 4, 5].map((i) =>
|
|
this.clients[i].emit.calledOnce.should.equal(true)
|
|
)
|
|
})
|
|
|
|
return it('should log out that it reached the end', function () {
|
|
return this.logger.log
|
|
.calledWith('All clients have been told to reconnectGracefully')
|
|
.should.equal(true)
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|