Overleaf/services/web/test/acceptance/src/helpers/MongoHelper.js

const { execFile } = require('child_process')
const {
  waitForDb,
  db,
  dropTestDatabase,
} = require('../../../../app/src/infrastructure/mongodb')
const Settings = require('@overleaf/settings')

const DEFAULT_ENV = 'saas'

module.exports = {
  initialize() {
    before(waitForDb)
    if (process.env.CLEANUP_MONGO === 'true') {
      before(dropTestDatabase)
    }

    before(function (done) {
      const args = [
        'run',
        'migrations',
        '--',
        'migrate',
        '-t',
        Settings.env || DEFAULT_ENV,
      ]
      execFile('npm', args, (error, stdout, stderr) => {
        if (error) {
          throw error
        }
        done()
      })
    })

    afterEach(async function () {
      return Promise.all(
        Object.values(db).map(async collection => {
          if (collection === db.migrations) {
            // Do not clear the collection for tracking migrations.
            return
          }
          return collection.deleteMany({})
        })
      )
    })
  },
}