Overleaf/services/web/test/acceptance/src/BackFillDocNameForDeletedDocsTests.mjs

import { exec } from 'node:child_process'
import { promisify } from 'node:util'
import { expect } from 'chai'
import logger from '@overleaf/logger'
import { db, ObjectId } from '../../../app/src/infrastructure/mongodb.mjs'
import UserHelper from './helpers/User.mjs'
import { renderObjectId } from '@overleaf/mongo-utils/batchedUpdate.js'

const User = UserHelper.promises

async function getDeletedDocs(projectId) {
  return (await db.projects.findOne({ _id: projectId })).deletedDocs
}

async function setDeletedDocs(projectId, deletedDocs) {
  await db.projects.updateOne({ _id: projectId }, { $set: { deletedDocs } })
}

describe('BackFillDocNameForDeletedDocs', function () {
  let user, projectId1, projectId2, docId1, docId2, docId3
  beforeEach('create projects', async function () {
    user = new User()
    await user.login()

    projectId1 = new ObjectId(await user.createProject('project1'))
    projectId2 = new ObjectId(await user.createProject('project2'))
  })
  beforeEach('create docs', async function () {
    docId1 = new ObjectId(
      await user.createDocInProject(projectId1, null, 'doc1.tex')
    )
    docId2 = new ObjectId(
      await user.createDocInProject(projectId1, null, 'doc2.tex')
    )
    docId3 = new ObjectId(
      await user.createDocInProject(projectId2, null, 'doc3.tex')
    )
  })
  beforeEach('deleted docs', async function () {
    await user.deleteItemInProject(projectId1, 'doc', docId1)
    await user.deleteItemInProject(projectId1, 'doc', docId2)
    await user.deleteItemInProject(projectId2, 'doc', docId3)
  })
  beforeEach('insert doc stubs into docs collection', async function () {
    await db.docs.insertMany([
      { _id: docId1, deleted: true },
      { _id: docId2, deleted: true },
      { _id: docId3, deleted: true },
    ])
  })
  let deletedDocs1, deletedDocs2
  let deletedAt1, deletedAt2, deletedAt3
  beforeEach('set deletedDocs details', async function () {
    deletedAt1 = new Date()
    deletedAt2 = new Date()
    deletedAt3 = new Date()
    deletedDocs1 = [
      { _id: docId1, name: 'doc1.tex', deletedAt: deletedAt1 },
      { _id: docId2, name: 'doc2.tex', deletedAt: deletedAt2 },
    ]
    deletedDocs2 = [{ _id: docId3, name: 'doc3.tex', deletedAt: deletedAt3 }]
    await setDeletedDocs(projectId1, deletedDocs1)
    await setDeletedDocs(projectId2, deletedDocs2)
  })

  async function runScript() {
    let result
    try {
      result = await promisify(exec)(
        'cd ../../tools/migrations && yarn run migrations migrate -t saas --force 20210727150530_ce_sp_backfill_deleted_docs'
      )
    } catch (error) {
      // dump details like exit code, stdErr and stdOut
      logger.error({ error }, 'script failed')
      throw error
    }
    const { stderr: stdErr } = result

    expect(stdErr).to.include(
      `Completed batch ending ${renderObjectId(projectId2)}`
    )
  }

  function checkDocsBackFilled() {
    it('should back fill names and deletedAt dates into docs', async function () {
      const docs = await db.docs.find({}).toArray()
      expect(docs).to.deep.equal([
        { _id: docId1, deleted: true, name: 'doc1.tex', deletedAt: deletedAt1 },
        { _id: docId2, deleted: true, name: 'doc2.tex', deletedAt: deletedAt2 },
        { _id: docId3, deleted: true, name: 'doc3.tex', deletedAt: deletedAt3 },
      ])
    })
  }

  describe('back fill and cleanup', function () {
    beforeEach('run script with cleanup flag', async function () {
      await runScript()
    })

    checkDocsBackFilled()

    it('should cleanup the deletedDocs', async function () {
      expect(await getDeletedDocs(projectId1)).to.deep.equal([])
      expect(await getDeletedDocs(projectId2)).to.deep.equal([])
    })
  })
})