overleaf/services/docstore/test/unit/js/MongoManagerTests.js

const SandboxedModule = require('sandboxed-module')
const sinon = require('sinon')
const modulePath = require('path').join(
  __dirname,
  '../../../app/js/MongoManager'
)
const { ObjectId } = require('mongodb')
const { assert, expect } = require('chai')
const Errors = require('../../../app/js/Errors')

describe('MongoManager', function () {
  beforeEach(function () {
    this.db = {
      docs: {
        updateOne: sinon.stub().resolves({ matchedCount: 1 }),
        insertOne: sinon.stub().resolves(),
      },
      docOps: {},
    }
    this.MongoManager = SandboxedModule.require(modulePath, {
      requires: {
        './mongodb': {
          db: this.db,
          ObjectId,
        },
        '@overleaf/metrics': { timeAsyncMethod: sinon.stub() },
        '@overleaf/settings': {
          max_deleted_docs: 42,
          docstore: { archivingLockDurationMs: 5000 },
        },
        './Errors': Errors,
      },
    })
    this.projectId = new ObjectId().toString()
    this.docId = new ObjectId().toString()
    this.rev = 42
    this.stubbedErr = new Error('hello world')
    this.lines = ['Three French hens', 'Two turtle doves']
  })

  describe('findDoc', function () {
    beforeEach(async function () {
      this.doc = { name: 'mock-doc' }
      this.db.docs.findOne = sinon.stub().resolves(this.doc)
      this.filter = { lines: true }
      this.result = await this.MongoManager.promises.findDoc(
        this.projectId,
        this.docId,
        this.filter
      )
    })

    it('should find the doc', function () {
      this.db.docs.findOne
        .calledWith(
          {
            _id: new ObjectId(this.docId),
            project_id: new ObjectId(this.projectId),
          },
          {
            projection: this.filter,
          }
        )
        .should.equal(true)
    })

    it('should return the doc', function () {
      expect(this.doc).to.deep.equal(this.doc)
    })
  })

  describe('patchDoc', function () {
    beforeEach(async function () {
      this.meta = { name: 'foo.tex' }
      await this.MongoManager.promises.patchDoc(
        this.projectId,
        this.docId,
        this.meta
      )
    })

    it('should pass the parameter along', function () {
      this.db.docs.updateOne.should.have.been.calledWith(
        {
          _id: new ObjectId(this.docId),
          project_id: new ObjectId(this.projectId),
        },
        {
          $set: this.meta,
        }
      )
    })
  })

  describe('getProjectsDocs', function () {
    beforeEach(function () {
      this.filter = { lines: true }
      this.doc1 = { name: 'mock-doc1' }
      this.doc2 = { name: 'mock-doc2' }
      this.doc3 = { name: 'mock-doc3' }
      this.doc4 = { name: 'mock-doc4' }
      this.db.docs.find = sinon.stub().returns({
        toArray: sinon.stub().resolves([this.doc, this.doc3, this.doc4]),
      })
    })

    describe('with included_deleted = false', function () {
      beforeEach(async function () {
        this.result = await this.MongoManager.promises.getProjectsDocs(
          this.projectId,
          { include_deleted: false },
          this.filter
        )
      })

      it('should find the non-deleted docs via the project_id', function () {
        this.db.docs.find
          .calledWith(
            {
              project_id: new ObjectId(this.projectId),
              deleted: { $ne: true },
            },
            {
              projection: this.filter,
            }
          )
          .should.equal(true)
      })

      it('should call return the docs', function () {
        expect(this.result).to.deep.equal([this.doc, this.doc3, this.doc4])
      })
    })

    describe('with included_deleted = true', function () {
      beforeEach(async function () {
        this.result = await this.MongoManager.promises.getProjectsDocs(
          this.projectId,
          { include_deleted: true },
          this.filter
        )
      })

      it('should find all via the project_id', function () {
        this.db.docs.find
          .calledWith(
            {
              project_id: new ObjectId(this.projectId),
            },
            {
              projection: this.filter,
            }
          )
          .should.equal(true)
      })

      it('should return the docs', function () {
        expect(this.result).to.deep.equal([this.doc, this.doc3, this.doc4])
      })
    })
  })

  describe('getProjectsDeletedDocs', function () {
    beforeEach(async function () {
      this.filter = { name: true }
      this.doc1 = { _id: '1', name: 'mock-doc1.tex' }
      this.doc2 = { _id: '2', name: 'mock-doc2.tex' }
      this.doc3 = { _id: '3', name: 'mock-doc3.tex' }
      this.db.docs.find = sinon.stub().returns({
        toArray: sinon.stub().resolves([this.doc1, this.doc2, this.doc3]),
      })
      this.result = await this.MongoManager.promises.getProjectsDeletedDocs(
        this.projectId,
        this.filter
      )
    })

    it('should find the deleted docs via the project_id', function () {
      this.db.docs.find
        .calledWith({
          project_id: new ObjectId(this.projectId),
          deleted: true,
        })
        .should.equal(true)
    })

    it('should filter, sort by deletedAt and limit', function () {
      this.db.docs.find
        .calledWith(sinon.match.any, {
          projection: this.filter,
          sort: { deletedAt: -1 },
          limit: 42,
        })
        .should.equal(true)
    })

    it('should return the docs', function () {
      expect(this.result).to.deep.equal([this.doc1, this.doc2, this.doc3])
    })
  })

  describe('upsertIntoDocCollection', function () {
    beforeEach(function () {
      this.oldRev = 77
    })

    it('should upsert the document', async function () {
      await this.MongoManager.promises.upsertIntoDocCollection(
        this.projectId,
        this.docId,
        this.oldRev,
        { lines: this.lines }
      )

      const args = this.db.docs.updateOne.args[0]
      assert.deepEqual(args[0], {
        _id: new ObjectId(this.docId),
        project_id: new ObjectId(this.projectId),
        rev: this.oldRev,
      })
      assert.equal(args[1].$set.lines, this.lines)
      assert.equal(args[1].$inc.rev, 1)
    })

    it('should handle update error', async function () {
      this.db.docs.updateOne.rejects(this.stubbedErr)
      await expect(
        this.MongoManager.promises.upsertIntoDocCollection(
          this.projectId,
          this.docId,
          this.rev,
          {
            lines: this.lines,
          }
        )
      ).to.be.rejectedWith(this.stubbedErr)
    })

    it('should insert without a previous rev', async function () {
      await this.MongoManager.promises.upsertIntoDocCollection(
        this.projectId,
        this.docId,
        null,
        { lines: this.lines, ranges: this.ranges }
      )

      expect(this.db.docs.insertOne).to.have.been.calledWith({
        _id: new ObjectId(this.docId),
        project_id: new ObjectId(this.projectId),
        rev: 1,
        lines: this.lines,
        ranges: this.ranges,
      })
    })

    it('should handle generic insert error', async function () {
      this.db.docs.insertOne.rejects(this.stubbedErr)
      await expect(
        this.MongoManager.promises.upsertIntoDocCollection(
          this.projectId,
          this.docId,
          null,
          { lines: this.lines, ranges: this.ranges }
        )
      ).to.be.rejectedWith(this.stubbedErr)
    })

    it('should handle duplicate insert error', async function () {
      this.db.docs.insertOne.rejects({ code: 11000 })
      await expect(
        this.MongoManager.promises.upsertIntoDocCollection(
          this.projectId,
          this.docId,
          null,
          { lines: this.lines, ranges: this.ranges }
        )
      ).to.be.rejectedWith(Errors.DocRevValueError)
    })
  })

  describe('destroyProject', function () {
    beforeEach(async function () {
      this.projectId = new ObjectId()
      this.docIds = [new ObjectId(), new ObjectId()]
      this.db.docs.deleteMany = sinon.stub().resolves()
      this.db.docOps.deleteMany = sinon.stub().resolves()
      this.db.docs.find = sinon
        .stub()
        .withArgs({ project_id: this.projectId })
        .returns({
          toArray: sinon.stub().resolves(
            this.docIds.map(id => ({
              _id: id,
            }))
          ),
        })
      await this.MongoManager.promises.destroyProject(this.projectId)
    })

    it('should destroy all docs', function () {
      sinon.assert.calledWith(this.db.docs.deleteMany, {
        project_id: this.projectId,
      })
    })

    it('should destroy the docOps', function () {
      sinon.assert.calledWith(this.db.docOps.deleteMany, {
        doc_id: { $in: this.docIds },
      })
    })
  })

  describe('getDocVersion', function () {
    describe('when the doc exists', function () {
      beforeEach(async function () {
        this.doc = { version: (this.version = 42) }
        this.db.docOps.findOne = sinon.stub().resolves(this.doc)
        this.result = await this.MongoManager.promises.getDocVersion(this.docId)
      })

      it('should look for the doc in the database', function () {
        this.db.docOps.findOne
          .calledWith(
            { doc_id: new ObjectId(this.docId) },
            {
              projection: { version: 1 },
            }
          )
          .should.equal(true)
      })

      it('should return the version', function () {
        expect(this.result).to.equal(this.version)
      })
    })

    describe("when the doc doesn't exist", function () {
      beforeEach(async function () {
        this.db.docOps.findOne = sinon.stub().resolves(null)
        this.result = await this.MongoManager.promises.getDocVersion(this.docId)
      })

      it('should return 0', function () {
        expect(this.result).to.equal(0)
      })
    })
  })

  describe('checkRevUnchanged', function () {
    this.beforeEach(function () {
      this.doc = { _id: new ObjectId(), name: 'mock-doc', rev: 1 }
    })

    it('should not error when the rev has not changed', async function () {
      this.db.docs.findOne = sinon.stub().resolves({ rev: 1 })
      await this.MongoManager.promises.checkRevUnchanged(this.doc)
    })

    it('should return an error when the rev has changed', async function () {
      this.db.docs.findOne = sinon.stub().resolves({ rev: 2 })
      await expect(
        this.MongoManager.promises.checkRevUnchanged(this.doc)
      ).to.be.rejectedWith(Errors.DocModifiedError)
    })

    it('should return a value error if incoming rev is NaN', async function () {
      this.db.docs.findOne = sinon.stub().resolves({ rev: 2 })
      this.doc = { _id: new ObjectId(), name: 'mock-doc', rev: NaN }
      await expect(
        this.MongoManager.promises.checkRevUnchanged(this.doc)
      ).to.be.rejectedWith(Errors.DocRevValueError)
    })

    it('should return a value error if checked doc rev is NaN', async function () {
      this.db.docs.findOne = sinon.stub().resolves({ rev: NaN })
      await expect(
        this.MongoManager.promises.checkRevUnchanged(this.doc)
      ).to.be.rejectedWith(Errors.DocRevValueError)
    })
  })

  describe('restoreArchivedDoc', function () {
    beforeEach(function () {
      this.archivedDoc = {
        lines: ['a', 'b', 'c'],
        ranges: { some: 'ranges' },
        rev: 2,
      }
    })

    describe('complete doc', function () {
      beforeEach(async function () {
        await this.MongoManager.promises.restoreArchivedDoc(
          this.projectId,
          this.docId,
          this.archivedDoc
        )
      })

      it('updates Mongo', function () {
        expect(this.db.docs.updateOne).to.have.been.calledWith(
          {
            _id: new ObjectId(this.docId),
            project_id: new ObjectId(this.projectId),
            rev: this.archivedDoc.rev,
          },
          {
            $set: {
              lines: this.archivedDoc.lines,
              ranges: this.archivedDoc.ranges,
            },
            $unset: {
              inS3: true,
            },
          }
        )
      })
    })

    describe('without ranges', function () {
      beforeEach(async function () {
        delete this.archivedDoc.ranges
        await this.MongoManager.promises.restoreArchivedDoc(
          this.projectId,
          this.docId,
          this.archivedDoc
        )
      })

      it('sets ranges to an empty object', function () {
        expect(this.db.docs.updateOne).to.have.been.calledWith(
          {
            _id: new ObjectId(this.docId),
            project_id: new ObjectId(this.projectId),
            rev: this.archivedDoc.rev,
          },
          {
            $set: {
              lines: this.archivedDoc.lines,
              ranges: {},
            },
            $unset: {
              inS3: true,
            },
          }
        )
      })
    })

    describe("when the update doesn't succeed", function () {
      it('throws a DocRevValueError', async function () {
        this.db.docs.updateOne.resolves({ matchedCount: 0 })
        await expect(
          this.MongoManager.promises.restoreArchivedDoc(
            this.projectId,
            this.docId,
            this.archivedDoc
          )
        ).to.be.rejectedWith(Errors.DocRevValueError)
      })
    })
  })
})