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

const MySQLIntegration = require("../mysql")
jest.mock("mysql2")

class TestConfiguration {
  constructor(config = { ssl: {} }) {
    this.integration = new MySQLIntegration.integration(config)
  }
}

describe("MySQL 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);"
    await config.integration.create({
      sql
    })
    expect(config.integration.client.query).toHaveBeenCalledWith(sql, {}, expect.any(Function))
  })

  it("calls the read method with the correct params", async () => {
    const sql = "select * from users;"
    await config.integration.read({
      sql
    })
    expect(config.integration.client.query).toHaveBeenCalledWith(sql, {}, expect.any(Function))
  })

  it("calls the update method with the correct params", async () => {
    const sql = "update table users set name = 'test';"
    await config.integration.update({
      sql
    })
    expect(config.integration.client.query).toHaveBeenCalledWith(sql, {}, expect.any(Function))
  })

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

  describe("no rows returned", () => {
    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 }])
    })
  })
})