speckle-server/packages/server/modules/multiregion/tests/helpers.ts

import { db } from '@/db/knex'
import {
  getRegisteredRegionClients,
  getReplicationDbs
} from '@/modules/multiregion/utils/dbSelector'
import { isMultiRegionTestMode } from '@/test/speckle-helpers/regions'
import type { Knex } from 'knex'

export async function getAllRegisteredTestDbs(): Promise<[Knex, ...Knex[]]> {
  if (!isMultiRegionTestMode()) return [db]

  const regionClients = await getRegisteredRegionClients()
  const regionDbs = Object.values(regionClients)
  return [db, ...regionDbs]
}

export async function getTestRegionClientsForProject({
  regionKey
}: {
  regionKey?: string
}): Promise<[Knex, ...Knex[]]> {
  if (!isMultiRegionTestMode()) return [db]

  if (!regionKey) return [db]
  const regionClients = await getRegisteredRegionClients()

  const regionDb = regionClients[regionKey]
  if (!regionDb) return [db]

  return await getReplicationDbs({ regionKey })
}