overleaf/services/web/test/UnitTests/coffee/FileStore/FileStoreControllerTests.coffee

68 lines
2 KiB
CoffeeScript

assert = require("chai").assert
sinon = require('sinon')
chai = require('chai')
should = chai.should()
expect = chai.expect
modulePath = "../../../../app/js/Features/FileStore/FileStoreController.js"
SandboxedModule = require('sandboxed-module')
describe "FileStoreController", ->
beforeEach ->
@FileStoreHandler =
getFileStream: sinon.stub()
@ProjectLocator =
findElement: sinon.stub()
@controller = SandboxedModule.require modulePath, requires:
"settings-sharelatex": @settings
"logger-sharelatex" : @logger = {log:sinon.stub(), err:sinon.stub()}
"../Project/ProjectLocator": @ProjectLocator
"./FileStoreHandler": @FileStoreHandler
@stream = {}
@project_id = "2k3j1lk3j21lk3j"
@file_id = "12321kklj1lk3jk12"
@req =
params:
Project_id: @project_id
File_id: @file_id
query: "query string here"
@res =
setHeader: sinon.stub()
@file =
name: "myfile.png"
describe "getFile", ->
beforeEach ->
@FileStoreHandler.getFileStream.callsArgWith(3, null, @stream)
@ProjectLocator.findElement.callsArgWith(1, null, @file)
it "should call the file store handler with the project_id file_id and any query string", (done)->
@stream.pipe = (des)=>
@FileStoreHandler.getFileStream.calledWith(@req.params.Project_id, @req.params.File_id, @req.query).should.equal true
done()
@controller.getFile @req, @res
it "should pipe to res", (done)->
@stream.pipe = (des)=>
des.should.equal @res
done()
@controller.getFile @req, @res
it "should get the file from the db", (done)->
@stream.pipe = (des)=>
opts =
project_id: @project_id
element_id: @file_id
type: "file"
@ProjectLocator.findElement.calledWith(opts).should.equal true
done()
@controller.getFile @req, @res
it "should set the Content-Disposition header", (done)->
@stream.pipe = (des)=>
@res.setHeader.calledWith("Content-Disposition", "attachment; filename=#{@file.name}").should.equal true
done()
@controller.getFile @req, @res