budibase/packages/server/src/integrations/tests/postgres.spec.ts

const pg = require("pg")
import { default as PostgresIntegration } from "../postgres"
jest.mock("pg")

class TestConfiguration {
  integration: any

  constructor(config: any = {}) {
    this.integration = new PostgresIntegration.integration(config)
  }
}

describe("Postgres Integration", () => {
  let config: any

  beforeEach(() => {
    config = new TestConfiguration()
  })

  it("calls the create method with the correct params", async () => {
    const sql = "insert into users (name, age) values ('Joe', 123);"
    await config.integration.create({
      sql,
    })
    expect(pg.queryMock).toHaveBeenCalledWith(sql, [])
  })

  it("calls the read method with the correct params", async () => {
    const sql = "select * from users;"
    await config.integration.read({
      sql,
    })
    expect(pg.queryMock).toHaveBeenCalledWith(sql, [])
  })

  it("calls the update method with the correct params", async () => {
    const sql = "update table users set name = 'test';"
    const response = await config.integration.update({
      sql,
    })
    expect(pg.queryMock).toHaveBeenCalledWith(sql, [])
  })

  it("calls the delete method with the correct params", async () => {
    const sql = "delete from users where name = 'todelete';"
    await config.integration.delete({
      sql,
    })
    expect(pg.queryMock).toHaveBeenCalledWith(sql, [])
  })

  describe("no rows returned", () => {
    beforeEach(() => {
      pg.queryMock.mockImplementation(() => ({ rows: [] }))
    })

    it("returns the correct response when the create response has no rows", async () => {
      const sql = "insert into users (name, age) values ('Joe', 123);"
      const response = await config.integration.create({
        sql,
      })
      expect(response).toEqual([{ created: true }])
    })

    it("returns the correct response when the update response has no rows", async () => {
      const sql = "update table users set name = 'test';"
      const response = await config.integration.update({
        sql,
      })
      expect(response).toEqual([{ updated: true }])
    })

    it("returns the correct response when the delete response has no rows", async () => {
      const sql = "delete from users where name = 'todelete';"
      const response = await config.integration.delete({
        sql,
      })
      expect(response).toEqual([{ deleted: true }])
    })
  })
})