mirror of
https://github.com/yu-i-i/overleaf-cep.git
synced 2026-11-14 14:13:43 +01:00
persist cookie in redis for compiles.
This commit is contained in:
@@ -7,6 +7,9 @@ SandboxedModule = require('sandboxed-module')
|
||||
|
||||
describe "ClsiManager", ->
|
||||
beforeEach ->
|
||||
@jar = {cookie:"stuff"}
|
||||
@ClsiRequestManager =
|
||||
getCookieJar: sinon.stub().callsArgWith(1, null, @jar)
|
||||
@ClsiManager = SandboxedModule.require modulePath, requires:
|
||||
"settings-sharelatex": @settings =
|
||||
apis:
|
||||
@@ -19,8 +22,9 @@ describe "ClsiManager", ->
|
||||
url: "https://clsipremium.example.com"
|
||||
"../../models/Project": Project: @Project = {}
|
||||
"../Project/ProjectEntityHandler": @ProjectEntityHandler = {}
|
||||
"./ClsiRequestManager": @ClsiRequestManager
|
||||
"logger-sharelatex": @logger = { log: sinon.stub(), error: sinon.stub(), warn: sinon.stub() }
|
||||
"request": @request = {}
|
||||
"request": @request = sinon.stub()
|
||||
@project_id = "project-id"
|
||||
@callback = sinon.stub()
|
||||
|
||||
@@ -80,15 +84,15 @@ describe "ClsiManager", ->
|
||||
|
||||
describe "deleteAuxFiles", ->
|
||||
beforeEach ->
|
||||
@request.del = sinon.stub().callsArg(1)
|
||||
@ClsiManager._makeRequest = sinon.stub().callsArg(2)
|
||||
|
||||
describe "with the standard compileGroup", ->
|
||||
beforeEach ->
|
||||
@ClsiManager.deleteAuxFiles @project_id, {compileGroup: "standard"}, @callback
|
||||
|
||||
it "should call the delete method in the standard CLSI", ->
|
||||
@request.del
|
||||
.calledWith("#{@settings.apis.clsi.url}/project/#{@project_id}")
|
||||
@ClsiManager._makeRequest
|
||||
.calledWith(@project_id, { method:"DELETE", url:"#{@settings.apis.clsi.url}/project/#{@project_id}"})
|
||||
.should.equal true
|
||||
|
||||
it "should call the callback", ->
|
||||
@@ -99,8 +103,8 @@ describe "ClsiManager", ->
|
||||
@ClsiManager.deleteAuxFiles @project_id, {compileGroup: "priority"}, @callback
|
||||
|
||||
it "should call the delete method in the CLSI", ->
|
||||
@request.del
|
||||
.calledWith("#{@settings.apis.clsi_priority.url}/project/#{@project_id}")
|
||||
@ClsiManager._makeRequest
|
||||
.calledWith(@project_id, { method:"DELETE", url:"#{@settings.apis.clsi_priority.url}/project/#{@project_id}"})
|
||||
.should.equal true
|
||||
|
||||
describe "_buildRequest", ->
|
||||
@@ -235,15 +239,15 @@ describe "ClsiManager", ->
|
||||
|
||||
describe "successfully", ->
|
||||
beforeEach ->
|
||||
@request.post = sinon.stub().callsArgWith(1, null, {statusCode: 204}, @body = { mock: "foo" })
|
||||
@ClsiManager._makeRequest = sinon.stub().callsArgWith(2, null, {statusCode: 204}, @body = { mock: "foo" })
|
||||
@ClsiManager._postToClsi @project_id, @req, "standard", @callback
|
||||
|
||||
it 'should send the request to the CLSI', ->
|
||||
url = "#{@settings.apis.clsi.url}/project/#{@project_id}/compile"
|
||||
@request.post.calledWith({
|
||||
@ClsiManager._makeRequest.calledWith(@project_id, {
|
||||
method: "POST",
|
||||
url: url
|
||||
json: @req
|
||||
jar: false
|
||||
}).should.equal true
|
||||
|
||||
it "should call the callback with the body and no error", ->
|
||||
@@ -251,7 +255,7 @@ describe "ClsiManager", ->
|
||||
|
||||
describe "when the CLSI returns an error", ->
|
||||
beforeEach ->
|
||||
@request.post = sinon.stub().callsArgWith(1, null, {statusCode: 500}, @body = { mock: "foo" })
|
||||
@ClsiManager._makeRequest = sinon.stub().callsArgWith(2, null, {statusCode: 500}, @body = { mock: "foo" })
|
||||
@ClsiManager._postToClsi @project_id, @req, "standard", @callback
|
||||
|
||||
it "should call the callback with the body and the error", ->
|
||||
@@ -259,20 +263,20 @@ describe "ClsiManager", ->
|
||||
|
||||
describe "when the compiler is priority", ->
|
||||
beforeEach ->
|
||||
@request.post = sinon.stub().callsArgWith(1, null, {statusCode: 500}, @body = { mock: "foo" })
|
||||
@ClsiManager._makeRequest = sinon.stub()
|
||||
@ClsiManager._postToClsi @project_id, @req, "priority", @callback
|
||||
|
||||
it "should use the clsi_priority url", ->
|
||||
url = "#{@settings.apis.clsi_priority.url}/project/#{@project_id}/compile"
|
||||
@request.post.calledWith({
|
||||
@ClsiManager._makeRequest.calledWith(@project_id, {
|
||||
method: "POST",
|
||||
url: url
|
||||
json: @req
|
||||
jar: false
|
||||
}).should.equal true
|
||||
|
||||
describe "wordCount", ->
|
||||
beforeEach ->
|
||||
@request.get = sinon.stub().callsArgWith(1, null, {statusCode: 200}, @body = { mock: "foo" })
|
||||
@ClsiManager._makeRequest = sinon.stub().callsArgWith(2, null, {statusCode: 200}, @body = { mock: "foo" })
|
||||
@ClsiManager._buildRequest = sinon.stub().callsArgWith(2, null, @req = { compile: { rootResourcePath: "rootfile.text", options: {} } })
|
||||
@ClsiManager._getCompilerUrl = sinon.stub().returns "compiler.url"
|
||||
|
||||
@@ -281,8 +285,8 @@ describe "ClsiManager", ->
|
||||
@ClsiManager.wordCount @project_id, false, {}, @callback
|
||||
|
||||
it "should call wordCount with root file", ->
|
||||
@request.get
|
||||
.calledWith({ url: "compiler.url/project/#{@project_id}/wordcount?file=rootfile.text" })
|
||||
@ClsiManager._makeRequest
|
||||
.calledWith(@project_id, { method: "GET", url: "compiler.url/project/#{@project_id}/wordcount?file=rootfile.text" })
|
||||
.should.equal true
|
||||
|
||||
it "should call the callback", ->
|
||||
@@ -293,8 +297,8 @@ describe "ClsiManager", ->
|
||||
@ClsiManager.wordCount @project_id, "main.tex", {}, @callback
|
||||
|
||||
it "should call wordCount with param file", ->
|
||||
@request.get
|
||||
.calledWith({ url: "compiler.url/project/#{@project_id}/wordcount?file=main.tex" })
|
||||
@ClsiManager._makeRequest
|
||||
.calledWith(@project_id, { method: "GET", url: "compiler.url/project/#{@project_id}/wordcount?file=main.tex" })
|
||||
.should.equal true
|
||||
|
||||
describe "with image", ->
|
||||
@@ -303,6 +307,6 @@ describe "ClsiManager", ->
|
||||
@ClsiManager.wordCount @project_id, "main.tex", {}, @callback
|
||||
|
||||
it "should call wordCount with file and image", ->
|
||||
@request.get
|
||||
.calledWith({ url: "compiler.url/project/#{@project_id}/wordcount?file=main.tex&image=#{encodeURIComponent(@image)}" })
|
||||
@ClsiManager._makeRequest
|
||||
.calledWith(@project_id, { method: "GET", url: "compiler.url/project/#{@project_id}/wordcount?file=main.tex&image=#{encodeURIComponent(@image)}" })
|
||||
.should.equal true
|
||||
|
||||
@@ -0,0 +1,111 @@
|
||||
sinon = require('sinon')
|
||||
chai = require('chai')
|
||||
should = chai.should()
|
||||
expect = chai.expect
|
||||
modulePath = "../../../../app/js/Features/Compile/ClsiRequestManager.js"
|
||||
SandboxedModule = require('sandboxed-module')
|
||||
realRequst = require("request")
|
||||
describe "ClsiRequestManager", ->
|
||||
beforeEach ->
|
||||
@redisMulti =
|
||||
set:sinon.stub()
|
||||
get:sinon.stub()
|
||||
expire:sinon.stub()
|
||||
exec:sinon.stub()
|
||||
self = @
|
||||
@project_id = "123423431321"
|
||||
@request =
|
||||
get: sinon.stub()
|
||||
cookie:realRequst.cookie
|
||||
jar: realRequst.jar
|
||||
@ClsiRequestManager = SandboxedModule.require modulePath, requires:
|
||||
"redis-sharelatex" :
|
||||
createClient: =>
|
||||
auth:->
|
||||
multi: -> return self.redisMulti
|
||||
"settings-sharelatex": @settings =
|
||||
redis:
|
||||
web:"redis.something"
|
||||
apis:
|
||||
clsi:
|
||||
url: "http://clsi.example.com"
|
||||
"request": @request
|
||||
|
||||
"logger-sharelatex": @logger = { log: sinon.stub(), error: sinon.stub(), warn: sinon.stub() }
|
||||
|
||||
|
||||
|
||||
describe "getServerId", ->
|
||||
|
||||
it "should call get for the key", (done)->
|
||||
@redisMulti.exec.callsArgWith(0, null, ["clsi-7"])
|
||||
@ClsiRequestManager._getServerId @project_id, (err, serverId)=>
|
||||
@redisMulti.get.calledWith("clsiserver:#{@project_id}").should.equal true
|
||||
serverId.should.equal "clsi-7"
|
||||
done()
|
||||
|
||||
it "should expire the key", (done)->
|
||||
@redisMulti.exec.callsArgWith(0, null, ["clsi-7"])
|
||||
@ClsiRequestManager._getServerId @project_id, (err, serverId)=>
|
||||
@redisMulti.expire.calledWith("clsiserver:#{@project_id}", 60 * 60 * 24 * 7).should.equal true
|
||||
done()
|
||||
|
||||
it "should _getServerIdViaRequest if no key is found", (done)->
|
||||
@ClsiRequestManager._getServerIdViaRequest = sinon.stub().callsArgWith(1)
|
||||
@redisMulti.exec.callsArgWith(0, null, [])
|
||||
@ClsiRequestManager._getServerId @project_id, (err, serverId)=>
|
||||
@ClsiRequestManager._getServerIdViaRequest.calledWith(@project_id).should.equal true
|
||||
done()
|
||||
|
||||
|
||||
describe "_getServerIdViaRequest", ->
|
||||
|
||||
it "should make a request to the clsi", (done)->
|
||||
response = "some data"
|
||||
@request.get.callsArgWith(1, null, response)
|
||||
@ClsiRequestManager.setServerId = sinon.stub().callsArgWith(2)
|
||||
@ClsiRequestManager._getServerIdViaRequest @project_id, (err, serverId)=>
|
||||
args = @ClsiRequestManager.setServerId.args[0]
|
||||
args[0].should.equal @project_id
|
||||
args[1].should.deep.equal response
|
||||
done()
|
||||
|
||||
describe "setServerId", ->
|
||||
|
||||
it "should set the server id with a ttl", (done)->
|
||||
@ClsiRequestManager._parseServerIdFromResponse = sinon.stub().returns("clsi-8")
|
||||
response = "dsadsakj"
|
||||
@redisMulti.exec.callsArgWith(0)
|
||||
@ClsiRequestManager.setServerId @project_id, response, (err)=>
|
||||
@redisMulti.set.calledWith("clsiserver:#{@project_id}", "clsi-8").should.equal true
|
||||
@redisMulti.expire.calledWith("clsiserver:#{@project_id}", 60 * 60 * 24 * 7).should.equal true
|
||||
done()
|
||||
|
||||
|
||||
describe "getCookieJar", ->
|
||||
|
||||
it "should return a jar with the cookie set populated from redis", (done)->
|
||||
@ClsiRequestManager._getServerId = sinon.stub().callsArgWith(1, null, "clsi-11")
|
||||
opts = {}
|
||||
@ClsiRequestManager.getCookieJar @project_id, opts, (err, jar)->
|
||||
jar._jar.store.idx["clsi.example.com"]["/"].clsiserver.key.should.equal "clsiserver"
|
||||
jar._jar.store.idx["clsi.example.com"]["/"].clsiserver.value.should.equal "clsi-11"
|
||||
done()
|
||||
|
||||
|
||||
# describe "_parseServerIdFromResponse", ->
|
||||
# it "take the cookie from the response", (done)->
|
||||
|
||||
# a.should.equal
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -22,6 +22,9 @@ describe "CompileController", ->
|
||||
url: "clsi.example.com"
|
||||
clsi_priority:
|
||||
url: "clsi-priority.example.com"
|
||||
@jar = {cookie:"stuff"}
|
||||
@ClsiRequestManager =
|
||||
getCookieJar:sinon.stub().callsArgWith(1, null, @jar)
|
||||
@CompileController = SandboxedModule.require modulePath, requires:
|
||||
"settings-sharelatex": @settings
|
||||
"request": @request = sinon.stub()
|
||||
@@ -33,6 +36,7 @@ describe "CompileController", ->
|
||||
"./ClsiManager": @ClsiManager
|
||||
"../Authentication/AuthenticationController": @AuthenticationController = {}
|
||||
"../../infrastructure/RateLimiter":@RateLimiter
|
||||
"./ClsiRequestManager":@ClsiRequestManager
|
||||
@project_id = "project-id"
|
||||
@user =
|
||||
features:
|
||||
@@ -182,6 +186,7 @@ describe "CompileController", ->
|
||||
it "should open a request to the CLSI", ->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -204,6 +209,7 @@ describe "CompileController", ->
|
||||
it "should proxy to the priority url if the user has the feature", ()->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi_priority.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -218,6 +224,7 @@ describe "CompileController", ->
|
||||
it "should open a request to the CLSI", ->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -240,6 +247,7 @@ describe "CompileController", ->
|
||||
it "should proxy to the priority url if the user has the feature", ()->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi_priority.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -254,6 +262,7 @@ describe "CompileController", ->
|
||||
it "should proxy to the standard url", ()->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -269,6 +278,7 @@ describe "CompileController", ->
|
||||
it "should proxy to the standard url without the build parameter", ()->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -286,6 +296,7 @@ describe "CompileController", ->
|
||||
it "should open a request to the CLSI", ->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -313,6 +324,7 @@ describe "CompileController", ->
|
||||
it "should proxy to the priority url if the user has the feature", ()->
|
||||
@request
|
||||
.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
url: "#{@settings.apis.clsi_priority.url}#{@url}",
|
||||
timeout: 60 * 1000
|
||||
@@ -331,8 +343,8 @@ describe "CompileController", ->
|
||||
@CompileController.proxyToClsi(@project_id, @url = "/test", @req, @res, @next)
|
||||
|
||||
it "should proxy to the standard url with the build parameter", ()->
|
||||
@request
|
||||
.calledWith(
|
||||
@request.calledWith(
|
||||
jar:@jar
|
||||
method: @req.method
|
||||
qs: {build: 1234}
|
||||
url: "#{@settings.apis.clsi.url}#{@url}",
|
||||
|
||||
Reference in New Issue
Block a user