mirror of
https://github.com/overleaf/overleaf.git
synced 2024-09-23 02:55:13 -04:00
81 lines
1.9 KiB
JavaScript
81 lines
1.9 KiB
JavaScript
RealTimeClient = require "./helpers/RealTimeClient"
|
|
FixturesManager = require "./helpers/FixturesManager"
|
|
|
|
expect = require("chai").expect
|
|
|
|
async = require "async"
|
|
request = require "request"
|
|
|
|
Settings = require "settings-sharelatex"
|
|
|
|
drain = (rate, callback) ->
|
|
request.post {
|
|
url: "http://localhost:3026/drain?rate=#{rate}"
|
|
auth: {
|
|
user: Settings.internal.realTime.user,
|
|
pass: Settings.internal.realTime.pass
|
|
}
|
|
}, (error, response, data) ->
|
|
callback error, data
|
|
return null
|
|
|
|
describe "DrainManagerTests", ->
|
|
before (done) ->
|
|
FixturesManager.setUpProject {
|
|
privilegeLevel: "owner"
|
|
project: {
|
|
name: "Test Project"
|
|
}
|
|
}, (e, {@project_id, @user_id}) => done()
|
|
return null
|
|
|
|
before (done) ->
|
|
# cleanup to speedup reconnecting
|
|
@timeout(10000)
|
|
RealTimeClient.disconnectAllClients done
|
|
|
|
# trigger and check cleanup
|
|
it "should have disconnected all previous clients", (done) ->
|
|
RealTimeClient.getConnectedClients (error, data) ->
|
|
return done(error) if error
|
|
expect(data.length).to.equal(0)
|
|
done()
|
|
|
|
describe "with two clients in the project", ->
|
|
beforeEach (done) ->
|
|
async.series [
|
|
(cb) =>
|
|
@clientA = RealTimeClient.connect()
|
|
@clientA.on "connectionAccepted", cb
|
|
|
|
(cb) =>
|
|
@clientB = RealTimeClient.connect()
|
|
@clientB.on "connectionAccepted", cb
|
|
|
|
(cb) =>
|
|
@clientA.emit "joinProject", project_id: @project_id, cb
|
|
|
|
(cb) =>
|
|
@clientB.emit "joinProject", project_id: @project_id, cb
|
|
], done
|
|
|
|
describe "starting to drain", () ->
|
|
beforeEach (done) ->
|
|
async.parallel [
|
|
(cb) =>
|
|
@clientA.on "reconnectGracefully", cb
|
|
(cb) =>
|
|
@clientB.on "reconnectGracefully", cb
|
|
|
|
(cb) -> drain(2, cb)
|
|
], done
|
|
|
|
afterEach (done) ->
|
|
drain(0, done) # reset drain
|
|
|
|
it "should not timeout", ->
|
|
expect(true).to.equal(true)
|
|
|
|
it "should not have disconnected", ->
|
|
expect(@clientA.socket.connected).to.equal true
|
|
expect(@clientB.socket.connected).to.equal true
|