customRunner.ts

import { resolve } from 'path'
import { Client } from 'pg'
import runner, { RunnerOption } from '../../dist'
import { RunnerOptionDir, RunnerOptionMigrations } from '../../dist/types'

type TestOptions = {
  count?: number
  expectedUpLength?: number
  expectedDownLength?: number
}

type Options = ({ databaseUrl: string } & TestOptions) | ({ dbClient: Client } & TestOptions)

/* eslint-disable no-console */
// eslint-disable-next-line import/prefer-default-export
export const run = async (options: Options): Promise<boolean> => {
  const opts: Omit<RunnerOption, 'direction'> & Options & (RunnerOptionDir | RunnerOptionMigrations) = {
    migrationsTable: 'migrations',
    dir: resolve(__dirname, 'migrations'),
    expectedUpLength: 2,
    expectedDownLength: 2,
    ...options,
  }
  try {
    const upResult = await runner({
      ...opts,
      direction: 'up',
    })
    if (upResult.length !== opts.expectedUpLength) {
      console.error(`There should be exactly ${opts.expectedUpLength} migrations processed`)
      return false
    }
    console.log('Up success')
    console.log(upResult)
    const downResult = await runner({
      ...opts,
      direction: 'down',
    })
    if (downResult.length !== opts.expectedDownLength) {
      console.error(`There should be exactly ${opts.expectedDownLength} migrations processed`)
      return false
    }
    console.log('Down success')
    console.log(downResult)
    return true
  } catch (err) {
    console.error(err)
    return false
  }
}