@budibase/server/src/integrations/tests/utils/postgres.ts

Budibase Web Server

import { Datasource, SourceName } from "@budibase/types"
import { GenericContainer, Wait } from "testcontainers"
import { generator, testContainerUtils } from "@budibase/backend-core/tests"
import { startContainer } from "."
import knex, { Knex } from "knex"
import { POSTGRES_IMAGE, POSTGRES_LEGACY_IMAGE } from "./images"

let ports: Promise<testContainerUtils.Port[]>

async function datasourceWithImage(image: string): Promise<Datasource> {
  if (!ports) {
    ports = startContainer(
      new GenericContainer(image)
        .withExposedPorts(5432)
        .withEnvironment({ POSTGRES_PASSWORD: "password" })
        .withWaitStrategy(
          // Legacy Postgres images can be slow to initialize on CI/arm64
          // Increase startup timeout to reduce flakiness
          Wait.forSuccessfulCommand(
            "pg_isready -h localhost -p 5432"
          ).withStartupTimeout(20000)
        )
    )
  }

  const port = (await ports).find(x => x.container === 5432)?.host
  if (!port) {
    throw new Error("Postgres port not found")
  }

  const datasource: Datasource = {
    type: "datasource_plus",
    source: SourceName.POSTGRES,
    plus: true,
    config: {
      host: "127.0.0.1",
      port,
      database: "postgres",
      user: "postgres",
      password: "password",
      schema: "public",
      ssl: false,
      rejectUnauthorized: false,
      ca: false,
    },
  }

  const database = generator.guid().replaceAll("-", "")
  const client = await knexClient(datasource)
  await client.raw(`CREATE DATABASE "${database}"`)
  datasource.config!.database = database

  return datasource
}

export async function getDatasource(): Promise<Datasource> {
  return datasourceWithImage(POSTGRES_IMAGE)
}

export async function getLegacyDatasource(): Promise<Datasource> {
  return datasourceWithImage(POSTGRES_LEGACY_IMAGE)
}

export async function knexClient(
  ds: Datasource,
  opts?: Knex.Config
): Promise<Knex> {
  if (!ds.config) {
    throw new Error("Datasource config is missing")
  }
  if (ds.source !== SourceName.POSTGRES) {
    throw new Error("Datasource source is not Postgres")
  }

  return knex({
    client: "pg",
    connection: ds.config,
    ...opts,
  })
}