Function createFakeClient

packages/db/tx-tripwire.test.ts:24–52

()

}

function createFakeClient() {
  const rootQueries: string[] = []
  const reservedQueries: string[] = []

  const client = instrumentPoolClient(
    {
      unsafe(query: string) {
        rootQueries.push(query)
        return Promise.resolve([])
      },
      // Mirrors postgres-js: begin issues its internal BEGIN through the root
      // client's unsafe (the instrumented one) before running the callback on
      // a reserved connection.
      begin(...args: unknown[]) {
        const callback = args[args.length - 1] as (reserved: FakeReserved) => unknown
        const reserved: FakeReserved = {
          unsafe: (query: string) => {
            reservedQueries.push(query)
            return Promise.resolve([])
          },
        }
        return Promise.resolve(this.unsafe('begin')).then(() => callback(reserved))
      },
    },
    'test-pool'
  )

  return { client, rootQueries, reservedQueries }
}

afterEach(() => {
  vi.unstubAllEnvs()