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

const pg = require("pg")
const PostgresIntegration = require("../postgres")
jest.mock("pg")

class TestConfiguration {
  constructor(config = {}) {
    this.integration = new PostgresIntegration.integration(config) 
  }
}

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

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

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

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

  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, undefined)
  })

  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, undefined)
  })

  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 }])
    })
  })
})