overleaf/services/web/test/unit/coffee/Downloads/ProjectDownloadsControllerTests.coffee

124 lines
4.1 KiB
CoffeeScript

sinon = require('sinon')
chai = require('chai')
should = chai.should()
expect = chai.expect
modulePath = "../../../../app/js/Features/Downloads/ProjectDownloadsController.js"
SandboxedModule = require('sandboxed-module')
MockRequest = require "../helpers/MockRequest"
MockResponse = require "../helpers/MockResponse"
describe "ProjectDownloadsController", ->
beforeEach ->
@project_id = "project-id-123"
@req = new MockRequest()
@res = new MockResponse()
@next = sinon.stub()
@DocumentUpdaterHandler = sinon.stub()
@ProjectDownloadsController = SandboxedModule.require modulePath, requires:
"./ProjectZipStreamManager" : @ProjectZipStreamManager = {}
"../Project/ProjectGetter" : @ProjectGetter = {}
"metrics-sharelatex": @metrics = {}
"logger-sharelatex" : @logger = {log: sinon.stub()}
"../DocumentUpdater/DocumentUpdaterHandler": @DocumentUpdaterHandler
describe "downloadProject", ->
beforeEach ->
@stream =
pipe: sinon.stub()
@ProjectZipStreamManager.createZipStreamForProject =
sinon.stub().callsArgWith(1, null, @stream)
@req.params = Project_id: @project_id
@res.contentType = sinon.stub()
@res.header = sinon.stub()
@project_name = "project name with accênts"
@ProjectGetter.getProject = sinon.stub().callsArgWith(2, null, name: @project_name)
@DocumentUpdaterHandler.flushProjectToMongo = sinon.stub().callsArgWith(1)
@metrics.inc = sinon.stub()
@ProjectDownloadsController.downloadProject @req, @res, @next
it "should create a zip from the project", ->
@ProjectZipStreamManager.createZipStreamForProject
.calledWith(@project_id)
.should.equal true
it "should stream the zip to the request", ->
@stream.pipe.calledWith(@res)
.should.equal true
it "should set the correct content type on the request", ->
@res.contentType
.calledWith("application/zip")
.should.equal true
it "should flush the project to mongo", ->
@DocumentUpdaterHandler.flushProjectToMongo
.calledWith(@project_id)
.should.equal true
it "should look up the project's name", ->
@ProjectGetter.getProject
.calledWith(@project_id, name: true)
.should.equal(true)
it "should name the downloaded file after the project", ->
@res.setContentDisposition
.calledWith(
'attachment',
{filename: "#{@project_name}.zip"})
.should.equal true
it "should record the action via Metrics", ->
@metrics.inc.calledWith("zip-downloads").should.equal true
it "should log the action", ->
@logger.log
.calledWith(sinon.match.any, "downloading project")
.should.equal true
describe "downloadMultipleProjects", ->
beforeEach ->
@stream =
pipe: sinon.stub()
@ProjectZipStreamManager.createZipStreamForMultipleProjects =
sinon.stub().callsArgWith(1, null, @stream)
@project_ids = ["project-1", "project-2"]
@req.query = project_ids: @project_ids.join(",")
@res.contentType = sinon.stub()
@res.header = sinon.stub()
@DocumentUpdaterHandler.flushMultipleProjectsToMongo = sinon.stub().callsArgWith(1)
@metrics.inc = sinon.stub()
@ProjectDownloadsController.downloadMultipleProjects @req, @res, @next
it "should create a zip from the project", ->
@ProjectZipStreamManager.createZipStreamForMultipleProjects
.calledWith(@project_ids)
.should.equal true
it "should stream the zip to the request", ->
@stream.pipe.calledWith(@res)
.should.equal true
it "should set the correct content type on the request", ->
@res.contentType
.calledWith("application/zip")
.should.equal true
it "should flush the projects to mongo", ->
@DocumentUpdaterHandler.flushMultipleProjectsToMongo
.calledWith(@project_ids)
.should.equal true
it "should name the downloaded file after the project", ->
@res.setContentDisposition
.calledWith(
'attachment',
{filename: "Overleaf Projects (2 items).zip"})
.should.equal true
it "should record the action via Metrics", ->
@metrics.inc.calledWith("zip-downloads-multiple").should.equal true
it "should log the action", ->
@logger.log
.calledWith(sinon.match.any, "downloading multiple projects")
.should.equal true