budibase/packages/datastores/tests/index.js

import setup from "./setup"
import records from "./records"
import { register } from "./diagnosticPlugin"
import pLimit from "p-limit"
import papa from "papaparse"
import { writeFileSync } from "fs"

const limit = pLimit(1)

const iterateActions = async (apis, getIterator) => {
  const iterator = getIterator(apis)
  let result = iterator()
  while (!result.done) {
    try {
      const runPromises = result.action.iterator(i =>
        limit(() => result.action.run(i))
      )

      let n = 1
      await Promise.all(runPromises)
      result = iterator()
    } catch (e) {
      e.message = `FAILED: ${result.action.name}: ${e.message}`
      throw e
    }
  }
}

export default async (datastore, config) => {
  const apis = await setup(datastore)

  const diagnostics = []
  let currentRecordCount = 0

  register(
    apis,
    (ev, elapsed, info) => {
      if (ev === "recordApi:save:onComplete") {
        currentRecordCount++
      } else if (ev === "recordApi:delete:onComplete") {
        currentRecordCount--
      }
      diagnostics.push({ method: ev, elapsed, info, count: currentRecordCount })
      console.log(`${ev}    ${info}    ${elapsed / 1000} s`)
    },
    [
      "recordApi:save",
      "recordApi:load",
      "viewApi:listItems",
      "recordApi:delete",
    ]
  )

  await iterateActions(apis, records)

  const diagnosticscsv = papa.unparse(diagnostics)

  writeFileSync(config.root + "\\results.csv", diagnosticscsv, {
    encoding: "utf8",
  })
}