overleaf/services/web/test/unit/src/Downloads/ProjectDownloadsControllerTests.js

/* eslint-disable
    max-len,
    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 sinon = require('sinon')
const { expect } = require('chai')
const modulePath =
  '../../../../app/src/Features/Downloads/ProjectDownloadsController.js'
const SandboxedModule = require('sandboxed-module')
const MockRequest = require('../helpers/MockRequest')
const MockResponse = require('../helpers/MockResponse')

describe('ProjectDownloadsController', function () {
  beforeEach(function () {
    this.project_id = 'project-id-123'
    this.req = new MockRequest()
    this.res = new MockResponse()
    this.next = sinon.stub()
    this.DocumentUpdaterHandler = sinon.stub()
    return (this.ProjectDownloadsController = SandboxedModule.require(
      modulePath,
      {
        requires: {
          './ProjectZipStreamManager': (this.ProjectZipStreamManager = {}),
          '../Project/ProjectGetter': (this.ProjectGetter = {}),
          '@overleaf/metrics': (this.metrics = {}),
          '../DocumentUpdater/DocumentUpdaterHandler': this
            .DocumentUpdaterHandler,
        },
      }
    ))
  })

  describe('downloadProject', function () {
    beforeEach(function () {
      this.stream = { pipe: sinon.stub() }
      this.ProjectZipStreamManager.createZipStreamForProject = sinon
        .stub()
        .callsArgWith(1, null, this.stream)
      this.req.params = { Project_id: this.project_id }
      this.res.contentType = sinon.stub()
      this.res.header = sinon.stub()
      this.project_name = 'project name with accênts'
      this.ProjectGetter.getProject = sinon
        .stub()
        .callsArgWith(2, null, { name: this.project_name })
      this.DocumentUpdaterHandler.flushProjectToMongo = sinon
        .stub()
        .callsArgWith(1)
      this.metrics.inc = sinon.stub()
      return this.ProjectDownloadsController.downloadProject(
        this.req,
        this.res,
        this.next
      )
    })

    it('should create a zip from the project', function () {
      return this.ProjectZipStreamManager.createZipStreamForProject
        .calledWith(this.project_id)
        .should.equal(true)
    })

    it('should stream the zip to the request', function () {
      return this.stream.pipe.calledWith(this.res).should.equal(true)
    })

    it('should set the correct content type on the request', function () {
      return this.res.contentType
        .calledWith('application/zip')
        .should.equal(true)
    })

    it('should flush the project to mongo', function () {
      return this.DocumentUpdaterHandler.flushProjectToMongo
        .calledWith(this.project_id)
        .should.equal(true)
    })

    it("should look up the project's name", function () {
      return this.ProjectGetter.getProject
        .calledWith(this.project_id, { name: true })
        .should.equal(true)
    })

    it('should name the downloaded file after the project', function () {
      return this.res.setContentDisposition
        .calledWith('attachment', { filename: `${this.project_name}.zip` })
        .should.equal(true)
    })

    it('should record the action via Metrics', function () {
      return this.metrics.inc.calledWith('zip-downloads').should.equal(true)
    })
  })

  describe('downloadMultipleProjects', function () {
    beforeEach(function () {
      this.stream = { pipe: sinon.stub() }
      this.ProjectZipStreamManager.createZipStreamForMultipleProjects = sinon
        .stub()
        .callsArgWith(1, null, this.stream)
      this.project_ids = ['project-1', 'project-2']
      this.req.query = { project_ids: this.project_ids.join(',') }
      this.res.contentType = sinon.stub()
      this.res.header = sinon.stub()
      this.DocumentUpdaterHandler.flushMultipleProjectsToMongo = sinon
        .stub()
        .callsArgWith(1)
      this.metrics.inc = sinon.stub()
      return this.ProjectDownloadsController.downloadMultipleProjects(
        this.req,
        this.res,
        this.next
      )
    })

    it('should create a zip from the project', function () {
      return this.ProjectZipStreamManager.createZipStreamForMultipleProjects
        .calledWith(this.project_ids)
        .should.equal(true)
    })

    it('should stream the zip to the request', function () {
      return this.stream.pipe.calledWith(this.res).should.equal(true)
    })

    it('should set the correct content type on the request', function () {
      return this.res.contentType
        .calledWith('application/zip')
        .should.equal(true)
    })

    it('should flush the projects to mongo', function () {
      return this.DocumentUpdaterHandler.flushMultipleProjectsToMongo
        .calledWith(this.project_ids)
        .should.equal(true)
    })

    it('should name the downloaded file after the project', function () {
      return this.res.setContentDisposition
        .calledWith('attachment', {
          filename: 'Overleaf Projects (2 items).zip',
        })
        .should.equal(true)
    })

    it('should record the action via Metrics', function () {
      return this.metrics.inc
        .calledWith('zip-downloads-multiple')
        .should.equal(true)
    })
  })
})