import { Entity, MikroORM, PrimaryKey, Property } from '@mikro-orm/core';
import { mockLogger } from '../helpers';
import { PostgreSqlDriver } from '@mikro-orm/postgresql';

@Entity()
export class A {

  @PrimaryKey({ name: 'ID' })
  id!: number;

  @Property({ name: 'NAME' })
  name: string;

  constructor(name: string) {
    this.name = name;
  }

}

describe('GH issue 1595', () => {

  let orm: MikroORM;

  beforeAll(async () => {
    orm = await MikroORM.init({
      entities: [A],
      dbName: 'mikro_orm_test_gh_1595',
      driver: PostgreSqlDriver,
    });
    await orm.schema.refreshDatabase();
  });

  afterAll(() => orm.close(true));

  test('mapping PKs from batch insert with custom field name', async () => {
    const mock = mockLogger(orm, ['query']);

    const items: A[] = [];

    for (let i = 1; i <= 5; i++) {
      items.push(new A(`a${i}`));
    }

    await orm.em.persistAndFlush(items);
    expect(mock.mock.calls[0][0]).toMatch('begin');
    expect(mock.mock.calls[1][0]).toMatch('insert into "a" ("NAME") values ($1), ($2), ($3), ($4), ($5) returning "ID"');
    expect(mock.mock.calls[2][0]).toMatch('commit');

    items.forEach(item => item.name += ' changed');
    await orm.em.flush();
    expect(mock.mock.calls[3][0]).toMatch('begin');
    expect(mock.mock.calls[4][0]).toMatch('update "a" set "NAME" = case when ("ID" = $1) then $2 when ("ID" = $3) then $4 when ("ID" = $5) then $6 when ("ID" = $7) then $8 when ("ID" = $9) then $10 else "NAME" end where "ID" in ($11, $12, $13, $14, $15)');
    expect(mock.mock.calls[5][0]).toMatch('commit');

    items.forEach(item => orm.em.remove(item));
    await orm.em.flush();
    expect(mock.mock.calls[6][0]).toMatch('begin');
    expect(mock.mock.calls[7][0]).toMatch('delete from "a" where "ID" in ($1, $2, $3, $4, $5)');
    expect(mock.mock.calls[8][0]).toMatch('commit');
  });

});