Overleaf/services/web/test/unit/src/DocumentUpdater/DocumentUpdaterControllerTests.js

const sinon = require('sinon')
const { expect } = require('chai')
const SandboxedModule = require('sandboxed-module')
const MockResponse = require('../helpers/MockResponse')

const MODULE_PATH =
  '../../../../app/src/Features/DocumentUpdater/DocumentUpdaterController.js'

describe('DocumentUpdaterController', function () {
  beforeEach(function () {
    this.DocumentUpdaterHandler = {
      promises: {
        getDocument: sinon.stub(),
      },
    }
    this.ProjectLocator = {
      promises: {
        findElement: sinon.stub(),
      },
    }
    this.controller = SandboxedModule.require(MODULE_PATH, {
      requires: {
        '@overleaf/settings': this.settings,
        '../Project/ProjectLocator': this.ProjectLocator,
        './DocumentUpdaterHandler': this.DocumentUpdaterHandler,
      },
    })
    this.projectId = '2k3j1lk3j21lk3j'
    this.fileId = '12321kklj1lk3jk12'
    this.req = {
      params: {
        Project_id: this.projectId,
        Doc_id: this.docId,
      },
      get(key) {
        return undefined
      },
    }
    this.lines = ['test', '', 'testing']
    this.res = new MockResponse()
    this.doc = { name: 'myfile.tex' }
  })

  describe('getDoc', function () {
    beforeEach(function () {
      this.DocumentUpdaterHandler.promises.getDocument.resolves({
        lines: this.lines,
      })
      this.ProjectLocator.promises.findElement.resolves({
        element: this.doc,
      })
      this.res = new MockResponse()
    })

    it('should call the document updater handler with the project_id and doc_id', async function () {
      await this.controller.promises.getDoc(this.req, this.res)

      expect(
        this.DocumentUpdaterHandler.promises.getDocument
      ).to.have.been.calledOnceWith(
        this.req.params.Project_id,
        this.req.params.Doc_id,
        -1
      )
    })

    it('should return the content', async function () {
      await this.controller.promises.getDoc(this.req, this.res)
      expect(this.res.statusCode).to.equal(200)
      expect(this.res.body).to.equal('test\n\ntesting')
    })

    it('should find the doc in the project', async function () {
      await this.controller.promises.getDoc(this.req, this.res)
      expect(
        this.ProjectLocator.promises.findElement
      ).to.have.been.calledOnceWith({
        project_id: this.projectId,
        element_id: this.docId,
        type: 'doc',
      })
    })

    it('should set the Content-Disposition header', async function () {
      await this.controller.promises.getDoc(this.req, this.res)
      expect(this.res.setContentDisposition).to.have.been.calledWith(
        'attachment',
        { filename: this.doc.name }
      )
    })
  })
})