import { Collection, Entity, LoadStrategy, ManyToOne, MikroORM, OneToMany, PrimaryKey, wrap } from '@mikro-orm/core';
import { SqliteDriver } from '@mikro-orm/sqlite';
import { mockLogger } from '../helpers';

@Entity()
class Order {

  @PrimaryKey()
  id: number;

  @OneToMany('OrderItem', 'order1')
  orderItems1 = new Collection<OrderItem>(this);

  @OneToMany('OrderItem', 'order2')
  orderItems2 = new Collection<OrderItem>(this);

  constructor(id: number) {
    this.id = id;
  }

}

@Entity()
class OrderItem {

  @PrimaryKey()
  id: number;

  @ManyToOne({
    entity: () => Order,
    eager: true,
    nullable: true,
  })
  order1?: Order;

  @ManyToOne({
    entity: () => Order,
    strategy: LoadStrategy.JOINED,
    eager: true,
    nullable: true,
  })
  order2?: Order;

  constructor(id: number) {
    this.id = id;
  }

}

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

  let orm: MikroORM<SqliteDriver>;

  beforeAll(async () => {
    orm = await MikroORM.init({
      entities: [Order, OrderItem],
      dbName: ':memory:',
      driver: SqliteDriver,
    });
    await orm.schema.createSchema();
  });

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

  test('eager loading m:1 with joined strategy', async () => {
    const order1 = new Order(1);
    const order2 = new Order(2);
    const orderItem1 = new OrderItem(3);
    const orderItem2 = new OrderItem(4);
    order1.orderItems1.add(orderItem1);
    order2.orderItems2.add(orderItem2);
    await orm.em.persistAndFlush([order1, order2]);
    orm.em.clear();

    const mock = mockLogger(orm, ['query', 'query-params']);
    const res1 = await orm.em.find(OrderItem, { id: { $lte: 100 } });
    expect(res1).toHaveLength(2);
    expect(wrap(res1[0].order1).isInitialized()).toBe(true);
    expect(wrap(res1[1].order2).isInitialized()).toBe(true);

    // first query loads item and joins the order2 relation (eager + joined strategy)
    expect(mock.mock.calls[0][0]).toMatch('select `o0`.`id`, `o0`.`order1_id`, `o0`.`order2_id`, `o1`.`id` as `o1__id` from `order_item` as `o0` left join `order` as `o1` on `o0`.`order2_id` = `o1`.`id` where `o0`.`id` <= 100');
    // second query loads order1 relation (eager + select-in strategy)
    expect(mock.mock.calls[1][0]).toMatch('select `o0`.* from `order` as `o0` where `o0`.`id` in (1) order by `o0`.`id` asc');

    expect(mock.mock.calls).toHaveLength(2);
  });

});