sourcegraph/client/browser/scripts/record-integration.js

const { Console } = require('console')

const { readdir, readFile } = require('mz/fs')
const shelljs = require('shelljs')

const { compressRecordings, deleteRecordings } = require('./utils')

const recordSnapshot = grepValue =>
  new Promise((resolve, reject) => {
    shelljs.exec(
      // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
      `POLLYJS_MODE=record SOURCEGRAPH_BASE_URL=https://sourcegraph.com pnpm run-integration --grep='${grepValue}'`,
      (code, stdout, stderr) => {
        console.log(`stdout: ${stdout}`)
        console.log(`stderr: ${stderr}`)

        if (code === 0) {
          resolve()
        }

        const error = new Error()
        error.code = code
        reject(error)
      }
    )
  })

const recordTests = async () => {
  // 1. Record by --grep args
  const args = process.argv.slice(2)
  for (let index = 0; index < args.length; ++index) {
    if (args[index] === '--grep' && !!args[index + 1]) {
      await recordSnapshot(args[index + 1])
      return
    }
  }

  // 2. Record all tests
  const fileNames = await readdir('./src/integration')
  const testFileNames = fileNames.filter(fileName => fileName.endsWith('.test.ts'))
  const testFiles = await Promise.all(
    testFileNames.map(testFileName => readFile(`./src/integration/${testFileName}`, 'utf-8'))
  )

  const testNames = testFiles
    // Ignore template strings for now. If we have lots of tests with parameterized test names, we
    // can use heuristics to still be able to run them.
    .flatMap(testFile => testFile.split('\n').map(line => line.match(/\bit\((["'])(.*)\1/)))
    .filter(Boolean)
    .map(matchArray => matchArray[2])

  for (const testName of testNames) {
    await recordSnapshot(testName)
  }
}

// eslint-disable-next-line no-void
void (async () => {
  try {
    await recordTests()
    await compressRecordings()
    process.exit(0)
  } catch (error) {
    await deleteRecordings()
    process.exit(error.code ?? 1)
  }
})()