import PostgresDatabase from "../../../src/database/postgresDatabase";

jest.mock("pg", () => {
  const mock = {
    Pool: function (config) {
      expect(config).toEqual({
        host: "localhost",
        user: "user",
        password: "password",
        port: 5432,
      });
      return mock;
    },
    query: (query, callback) => {
      callback({
        name: "name",
        message: "table does not exist",
      });
    },
    end: () => true,
  };
  return mock;
});

test("it calls createConnection", () => {
  const db = new PostgresDatabase("localhost", "user", "password", 5432);
});

test("it calls callback if there is an error", async () => {
  const db = new PostgresDatabase("localhost", "user", "password", 5432);
  const sql = "SELECT some_column FROM some_table WHERE id = 1";

  expect(await db.lintQuery(sql)).toMatchObject({
    sqlMessage: "table does not exist",
    code: "name",
  });
});

test("it calls end on connection", () => {
  const db = new PostgresDatabase("localhost", "user", "password", 5432);
  db.end();
});

test("it does not call callback if there is no error", () => {
  const callback = jest.fn(() => true);

  jest.mock("pg", () => {
    const mock = {
      createConnection: () => mock,
      query: () => true,
    };
    return mock;
  });

  expect(callback.mock.calls.length).toBe(0);
});