mirror of
https://github.com/overleaf/overleaf.git
synced 2024-11-29 10:43:33 -05:00
297 lines
8.9 KiB
JavaScript
297 lines
8.9 KiB
JavaScript
/* eslint-disable
|
|
no-return-assign,
|
|
no-unused-vars,
|
|
*/
|
|
// TODO: This file was created by bulk-decaffeinate.
|
|
// Fix any style issues and re-enable lint.
|
|
/*
|
|
* decaffeinate suggestions:
|
|
* DS102: Remove unnecessary code created because of implicit returns
|
|
* Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
|
|
*/
|
|
const { assert } = require('chai')
|
|
const sinon = require('sinon')
|
|
const chai = require('chai')
|
|
const should = chai.should()
|
|
const { expect } = chai
|
|
const modulePath = '../../../app/js/FileController.js'
|
|
const SandboxedModule = require('sandboxed-module')
|
|
|
|
describe('FileController', function() {
|
|
beforeEach(function() {
|
|
this.PersistorManager = {
|
|
sendStream: sinon.stub(),
|
|
copyFile: sinon.stub(),
|
|
deleteFile: sinon.stub()
|
|
}
|
|
|
|
this.settings = {
|
|
s3: {
|
|
buckets: {
|
|
user_files: 'user_files'
|
|
}
|
|
}
|
|
}
|
|
this.FileHandler = {
|
|
getFile: sinon.stub(),
|
|
getFileSize: sinon.stub(),
|
|
deleteFile: sinon.stub(),
|
|
insertFile: sinon.stub(),
|
|
getDirectorySize: sinon.stub()
|
|
}
|
|
this.LocalFileWriter = {}
|
|
this.controller = SandboxedModule.require(modulePath, {
|
|
requires: {
|
|
'./LocalFileWriter': this.LocalFileWriter,
|
|
'./FileHandler': this.FileHandler,
|
|
'./PersistorManager': this.PersistorManager,
|
|
'./Errors': (this.Errors = { NotFoundError: sinon.stub() }),
|
|
'settings-sharelatex': this.settings,
|
|
'metrics-sharelatex': {
|
|
inc() {}
|
|
},
|
|
'logger-sharelatex': {
|
|
log() {},
|
|
err() {}
|
|
}
|
|
}
|
|
})
|
|
this.project_id = 'project_id'
|
|
this.file_id = 'file_id'
|
|
this.bucket = 'user_files'
|
|
this.key = `${this.project_id}/${this.file_id}`
|
|
this.req = {
|
|
key: this.key,
|
|
bucket: this.bucket,
|
|
query: {},
|
|
params: {
|
|
project_id: this.project_id,
|
|
file_id: this.file_id
|
|
},
|
|
headers: {}
|
|
}
|
|
this.res = {
|
|
set: sinon.stub().returnsThis(),
|
|
status: sinon.stub().returnsThis()
|
|
}
|
|
return (this.fileStream = {})
|
|
})
|
|
|
|
describe('getFile', function() {
|
|
it('should pipe the stream', function(done) {
|
|
this.FileHandler.getFile.callsArgWith(3, null, this.fileStream)
|
|
this.fileStream.pipe = res => {
|
|
res.should.equal(this.res)
|
|
return done()
|
|
}
|
|
return this.controller.getFile(this.req, this.res)
|
|
})
|
|
|
|
it('should send a 200 if the cacheWarm param is true', function(done) {
|
|
this.req.query.cacheWarm = true
|
|
this.FileHandler.getFile.callsArgWith(3, null, this.fileStream)
|
|
this.res.send = statusCode => {
|
|
statusCode.should.equal(200)
|
|
return done()
|
|
}
|
|
return this.controller.getFile(this.req, this.res)
|
|
})
|
|
|
|
it('should send a 500 if there is a problem', function(done) {
|
|
this.FileHandler.getFile.callsArgWith(3, 'error')
|
|
this.res.send = code => {
|
|
code.should.equal(500)
|
|
return done()
|
|
}
|
|
return this.controller.getFile(this.req, this.res)
|
|
})
|
|
|
|
return describe("with a 'Range' header set", function() {
|
|
beforeEach(function() {
|
|
return (this.req.headers.range = 'bytes=0-8')
|
|
})
|
|
|
|
return it("should pass 'start' and 'end' options to FileHandler", function(done) {
|
|
this.FileHandler.getFile.callsArgWith(3, null, this.fileStream)
|
|
this.fileStream.pipe = res => {
|
|
expect(this.FileHandler.getFile.lastCall.args[2].start).to.equal(0)
|
|
expect(this.FileHandler.getFile.lastCall.args[2].end).to.equal(8)
|
|
return done()
|
|
}
|
|
return this.controller.getFile(this.req, this.res)
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('getFileHead', function() {
|
|
it('should return the file size in a Content-Length header', function(done) {
|
|
const expectedFileSize = 84921
|
|
this.FileHandler.getFileSize.yields(
|
|
new Error('FileHandler.getFileSize: unexpected arguments')
|
|
)
|
|
this.FileHandler.getFileSize
|
|
.withArgs(this.bucket, this.key)
|
|
.yields(null, expectedFileSize)
|
|
|
|
this.res.end = () => {
|
|
expect(this.res.status.lastCall.args[0]).to.equal(200)
|
|
expect(
|
|
this.res.set.calledWith('Content-Length', expectedFileSize)
|
|
).to.equal(true)
|
|
return done()
|
|
}
|
|
|
|
return this.controller.getFileHead(this.req, this.res)
|
|
})
|
|
|
|
it('should return a 404 is the file is not found', function(done) {
|
|
this.FileHandler.getFileSize.yields(new this.Errors.NotFoundError())
|
|
|
|
this.res.end = () => {
|
|
expect(this.res.status.lastCall.args[0]).to.equal(404)
|
|
return done()
|
|
}
|
|
|
|
return this.controller.getFileHead(this.req, this.res)
|
|
})
|
|
|
|
return it('should return a 500 on internal errors', function(done) {
|
|
this.FileHandler.getFileSize.yields(new Error())
|
|
|
|
this.res.end = () => {
|
|
expect(this.res.status.lastCall.args[0]).to.equal(500)
|
|
return done()
|
|
}
|
|
|
|
return this.controller.getFileHead(this.req, this.res)
|
|
})
|
|
})
|
|
|
|
describe('insertFile', () =>
|
|
it('should send bucket name key and res to PersistorManager', function(done) {
|
|
this.FileHandler.insertFile.callsArgWith(3)
|
|
this.res.send = () => {
|
|
this.FileHandler.insertFile
|
|
.calledWith(this.bucket, this.key, this.req)
|
|
.should.equal(true)
|
|
return done()
|
|
}
|
|
return this.controller.insertFile(this.req, this.res)
|
|
}))
|
|
|
|
describe('copyFile', function() {
|
|
beforeEach(function() {
|
|
this.oldFile_id = 'old_file_id'
|
|
this.oldProject_id = 'old_project_id'
|
|
return (this.req.body = {
|
|
source: {
|
|
project_id: this.oldProject_id,
|
|
file_id: this.oldFile_id
|
|
}
|
|
})
|
|
})
|
|
|
|
it('should send bucket name and both keys to PersistorManager', function(done) {
|
|
this.PersistorManager.copyFile.callsArgWith(3)
|
|
this.res.send = code => {
|
|
code.should.equal(200)
|
|
this.PersistorManager.copyFile
|
|
.calledWith(
|
|
this.bucket,
|
|
`${this.oldProject_id}/${this.oldFile_id}`,
|
|
this.key
|
|
)
|
|
.should.equal(true)
|
|
return done()
|
|
}
|
|
return this.controller.copyFile(this.req, this.res)
|
|
})
|
|
|
|
it('should send a 404 if the original file was not found', function(done) {
|
|
this.PersistorManager.copyFile.callsArgWith(
|
|
3,
|
|
new this.Errors.NotFoundError()
|
|
)
|
|
this.res.send = code => {
|
|
code.should.equal(404)
|
|
return done()
|
|
}
|
|
return this.controller.copyFile(this.req, this.res)
|
|
})
|
|
|
|
return it('should send a 500 if there was an error', function(done) {
|
|
this.PersistorManager.copyFile.callsArgWith(3, 'error')
|
|
this.res.send = code => {
|
|
code.should.equal(500)
|
|
return done()
|
|
}
|
|
return this.controller.copyFile(this.req, this.res)
|
|
})
|
|
})
|
|
|
|
describe('delete file', function() {
|
|
it('should tell the file handler', function(done) {
|
|
this.FileHandler.deleteFile.callsArgWith(2)
|
|
this.res.send = code => {
|
|
code.should.equal(204)
|
|
this.FileHandler.deleteFile
|
|
.calledWith(this.bucket, this.key)
|
|
.should.equal(true)
|
|
return done()
|
|
}
|
|
return this.controller.deleteFile(this.req, this.res)
|
|
})
|
|
|
|
return it('should send a 500 if there was an error', function(done) {
|
|
this.FileHandler.deleteFile.callsArgWith(2, 'error')
|
|
this.res.send = function(code) {
|
|
code.should.equal(500)
|
|
return done()
|
|
}
|
|
return this.controller.deleteFile(this.req, this.res)
|
|
})
|
|
})
|
|
|
|
describe('_get_range', function() {
|
|
it('should parse a valid Range header', function(done) {
|
|
const result = this.controller._get_range('bytes=0-200')
|
|
expect(result).to.not.equal(null)
|
|
expect(result.start).to.equal(0)
|
|
expect(result.end).to.equal(200)
|
|
return done()
|
|
})
|
|
|
|
it('should return null for an invalid Range header', function(done) {
|
|
const result = this.controller._get_range('wat')
|
|
expect(result).to.equal(null)
|
|
return done()
|
|
})
|
|
|
|
return it("should return null for any type other than 'bytes'", function(done) {
|
|
const result = this.controller._get_range('carrots=0-200')
|
|
expect(result).to.equal(null)
|
|
return done()
|
|
})
|
|
})
|
|
|
|
return describe('directorySize', function() {
|
|
it('should return total directory size bytes', function(done) {
|
|
this.FileHandler.getDirectorySize.callsArgWith(2, null, 1024)
|
|
return this.controller.directorySize(this.req, {
|
|
json: result => {
|
|
expect(result['total bytes']).to.equal(1024)
|
|
return done()
|
|
}
|
|
})
|
|
})
|
|
|
|
return it('should send a 500 if there was an error', function(done) {
|
|
this.FileHandler.getDirectorySize.callsArgWith(2, 'error')
|
|
this.res.send = function(code) {
|
|
code.should.equal(500)
|
|
return done()
|
|
}
|
|
return this.controller.directorySize(this.req, this.res)
|
|
})
|
|
})
|
|
})
|