budibase/packages/server/__mocks__/pg.ts

module PgMock {
  const pg: any = {}

  const query = jest.fn(() => ({
    rows: [
      {
        a: "string",
        b: 1,
      },
    ],
  }))

// constructor
  function Client() {
  }

  Client.prototype.query = query
  Client.prototype.connect = jest.fn()
  Client.prototype.release = jest.fn()

  function Pool() {
  }

  Pool.prototype.query = query
  Pool.prototype.connect = jest.fn(() => {
    // @ts-ignore
    return new Client()
  })

  pg.Client = Client
  pg.Pool = Pool
  pg.queryMock = query

  module.exports = pg
}