import 'reflect-metadata';
import {
	createTestingConnections,
	closeTestingConnections,
	reloadTestingDatabases,
} from '../../utils/test-utils';
import { DataSource } from '../../../src';
import { Simple } from './entity/Simple';
import { Complex } from './entity/Complex';

describe('github issues > #2103 query builder regression', () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				entities: [__dirname + '/entity/*{.js,.ts}'],
				schemaCreate: true,
				dropSchema: true,
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('whereInIds should respect logical operator precedence > single simple primary key (in is used)', () =>
		Promise.all(
			connections.map(async (connection) => {
				const repository = connection.getRepository(Simple);

				const savedEntities = await repository.save([
					repository.create({ x: 1 }),
					repository.create({ x: 2 }),
					repository.create({ x: 1 }),
					repository.create({ x: 3 }),
				]);

				savedEntities.length.should.be.equal(4); // check if they all are saved

				const ids = savedEntities.map((entity) => entity.id);

				const entities = await repository
					.createQueryBuilder('s')
					.whereInIds(ids)
					.andWhere('s.x = 1')
					.addOrderBy('s.id')
					.getMany();

				entities
					.map((entity) => entity.id)
					.should.be.eql(
						savedEntities.filter((entity) => entity.x === 1).map((entity) => entity.id),
					);
			}),
		));

	it('whereInIds should respect logical operator precedence > multiple primary keys', () =>
		Promise.all(
			connections.map(async (connection) => {
				const repository = connection.getRepository(Complex);

				// sqlite does not support autoincrement for composite primary key, so we pass ids by ourselves here
				const savedEntities = await repository.save([
					repository.create({ id: 1, code: 1, x: 1 }),
					repository.create({ id: 2, code: 1, x: 2 }),
					repository.create({ id: 3, code: 1, x: 1 }),
					repository.create({ id: 4, code: 1, x: 3 }),
				]);

				savedEntities.length.should.be.equal(4); // check if they all are saved

				const ids = savedEntities.map((entity) => entity.id);

				const entities = await repository
					.createQueryBuilder('s')
					.whereInIds(
						ids.map((id) => {
							return { id, code: 1 };
						}),
					)
					.andWhere('s.x = 1')
					.addOrderBy('s.id')
					.getMany();

				entities
					.map((entity) => entity.id)
					.should.be.eql(
						savedEntities.filter((entity) => entity.x === 1).map((entity) => entity.id),
					);
			}),
		));
});