mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-07 20:31:06 -05:00
164 lines
7 KiB
CoffeeScript
164 lines
7 KiB
CoffeeScript
|
|
should = require('chai').should()
|
|
SandboxedModule = require('sandboxed-module')
|
|
assert = require('assert')
|
|
path = require('path')
|
|
sinon = require('sinon')
|
|
modulePath = path.join __dirname, "../../../app/js/ConnectedUsersManager"
|
|
expect = require("chai").expect
|
|
tk = require("timekeeper")
|
|
|
|
|
|
describe "ConnectedUsersManager", ->
|
|
|
|
beforeEach ->
|
|
|
|
@settings =
|
|
redis:
|
|
realtime:
|
|
key_schema:
|
|
clientsInProject: ({project_id}) -> "clients_in_project:#{project_id}"
|
|
connectedUser: ({project_id, client_id})-> "connected_user:#{project_id}:#{client_id}"
|
|
@rClient =
|
|
auth:->
|
|
setex:sinon.stub()
|
|
sadd:sinon.stub()
|
|
get: sinon.stub()
|
|
srem:sinon.stub()
|
|
del:sinon.stub()
|
|
smembers:sinon.stub()
|
|
expire:sinon.stub()
|
|
hset:sinon.stub()
|
|
hgetall:sinon.stub()
|
|
exec:sinon.stub()
|
|
multi: => return @rClient
|
|
tk.freeze(new Date())
|
|
|
|
@ConnectedUsersManager = SandboxedModule.require modulePath, requires:
|
|
"settings-sharelatex":@settings
|
|
"logger-sharelatex": log:->
|
|
"redis-sharelatex": createClient:=>
|
|
return @rClient
|
|
@client_id = "32132132"
|
|
@project_id = "dskjh2u21321"
|
|
@user = {
|
|
_id: "user-id-123"
|
|
first_name: "Joe"
|
|
last_name: "Bloggs"
|
|
email: "joe@example.com"
|
|
}
|
|
@cursorData = { row: 12, column: 9, doc_id: '53c3b8c85fee64000023dc6e' }
|
|
|
|
afterEach ->
|
|
tk.reset()
|
|
|
|
describe "updateUserPosition", ->
|
|
beforeEach ->
|
|
@rClient.exec.callsArgWith(0)
|
|
|
|
it "should set a key with the date and give it a ttl", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.hset.calledWith("connected_user:#{@project_id}:#{@client_id}", "last_updated_at", Date.now()).should.equal true
|
|
done()
|
|
|
|
it "should set a key with the user_id", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.hset.calledWith("connected_user:#{@project_id}:#{@client_id}", "user_id", @user._id).should.equal true
|
|
done()
|
|
|
|
it "should set a key with the first_name", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.hset.calledWith("connected_user:#{@project_id}:#{@client_id}", "first_name", @user.first_name).should.equal true
|
|
done()
|
|
|
|
it "should set a key with the last_name", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.hset.calledWith("connected_user:#{@project_id}:#{@client_id}", "last_name", @user.last_name).should.equal true
|
|
done()
|
|
|
|
it "should set a key with the email", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.hset.calledWith("connected_user:#{@project_id}:#{@client_id}", "email", @user.email).should.equal true
|
|
done()
|
|
|
|
it "should push the client_id on to the project list", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.sadd.calledWith("clients_in_project:#{@project_id}", @client_id).should.equal true
|
|
done()
|
|
|
|
it "should add a ttl to the project set so it stays clean", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.expire.calledWith("clients_in_project:#{@project_id}", 24 * 4 * 60 * 60).should.equal true
|
|
done()
|
|
|
|
it "should add a ttl to the connected user so it stays clean", (done) ->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, null, (err)=>
|
|
@rClient.expire.calledWith("connected_user:#{@project_id}:#{@client_id}", 60 * 15).should.equal true
|
|
done()
|
|
|
|
it "should set the cursor position when provided", (done)->
|
|
@ConnectedUsersManager.updateUserPosition @project_id, @client_id, @user, @cursorData, (err)=>
|
|
@rClient.hset.calledWith("connected_user:#{@project_id}:#{@client_id}", "cursorData", JSON.stringify(@cursorData)).should.equal true
|
|
done()
|
|
|
|
describe "markUserAsDisconnected", ->
|
|
beforeEach ->
|
|
@rClient.exec.callsArgWith(0)
|
|
|
|
it "should remove the user from the set", (done)->
|
|
@ConnectedUsersManager.markUserAsDisconnected @project_id, @client_id, (err)=>
|
|
@rClient.srem.calledWith("clients_in_project:#{@project_id}", @client_id).should.equal true
|
|
done()
|
|
|
|
it "should delete the connected_user string", (done)->
|
|
@ConnectedUsersManager.markUserAsDisconnected @project_id, @client_id, (err)=>
|
|
@rClient.del.calledWith("connected_user:#{@project_id}:#{@client_id}").should.equal true
|
|
done()
|
|
|
|
it "should add a ttl to the connected user set so it stays clean", (done)->
|
|
@ConnectedUsersManager.markUserAsDisconnected @project_id, @client_id, (err)=>
|
|
@rClient.expire.calledWith("clients_in_project:#{@project_id}", 24 * 4 * 60 * 60).should.equal true
|
|
done()
|
|
|
|
describe "_getConnectedUser", ->
|
|
|
|
it "should return a connected user if there is a user object", (done)->
|
|
cursorData = JSON.stringify(cursorData:{row:1})
|
|
@rClient.hgetall.callsArgWith(1, null, {connected_at:new Date(), user_id: @user._id, last_updated_at: "#{Date.now()}", cursorData})
|
|
@ConnectedUsersManager._getConnectedUser @project_id, @client_id, (err, result)=>
|
|
result.connected.should.equal true
|
|
result.client_id.should.equal @client_id
|
|
done()
|
|
|
|
it "should return a not connected user if there is no object", (done)->
|
|
@rClient.hgetall.callsArgWith(1, null, null)
|
|
@ConnectedUsersManager._getConnectedUser @project_id, @client_id, (err, result)=>
|
|
result.connected.should.equal false
|
|
result.client_id.should.equal @client_id
|
|
done()
|
|
|
|
it "should return a not connected user if there is an empty object", (done)->
|
|
@rClient.hgetall.callsArgWith(1, null, {})
|
|
@ConnectedUsersManager._getConnectedUser @project_id, @client_id, (err, result)=>
|
|
result.connected.should.equal false
|
|
result.client_id.should.equal @client_id
|
|
done()
|
|
|
|
describe "getConnectedUsers", ->
|
|
|
|
beforeEach ->
|
|
@users = ["1234", "5678", "9123", "8234"]
|
|
@rClient.smembers.callsArgWith(1, null, @users)
|
|
@ConnectedUsersManager._getConnectedUser = sinon.stub()
|
|
@ConnectedUsersManager._getConnectedUser.withArgs(@project_id, @users[0]).callsArgWith(2, null, {connected:true, client_age: 2, client_id:@users[0]})
|
|
@ConnectedUsersManager._getConnectedUser.withArgs(@project_id, @users[1]).callsArgWith(2, null, {connected:false, client_age: 1, client_id:@users[1]})
|
|
@ConnectedUsersManager._getConnectedUser.withArgs(@project_id, @users[2]).callsArgWith(2, null, {connected:true, client_age: 3, client_id:@users[2]})
|
|
@ConnectedUsersManager._getConnectedUser.withArgs(@project_id, @users[3]).callsArgWith(2, null, {connected:true, client_age: 11, client_id:@users[3]}) # connected but old
|
|
|
|
it "should only return the users in the list which are still in redis and recently updated", (done)->
|
|
@ConnectedUsersManager.getConnectedUsers @project_id, (err, users)=>
|
|
users.length.should.equal 2
|
|
users[0].should.deep.equal {client_id:@users[0], client_age: 2, connected:true}
|
|
users[1].should.deep.equal {client_id:@users[2], client_age: 3, connected:true}
|
|
done()
|
|
|