import 'reflect-metadata';
import { expect } from 'chai';
import { DataSource } from '../../../src';
import { User } from './entity/User';
import { Role } from './entity/Role';
import {
	createTestingConnections,
	reloadTestingDatabases,
	closeTestingConnections,
} from '../../utils/test-utils';

describe('github issues > #5174 `selectQueryBuilder.take` messes up the query when using the `ids` parameter', () => {
	let connections: DataSource[];

	before(async () => {
		connections = await createTestingConnections({
			entities: [User, Role],
			schemaCreate: true,
			dropSchema: true,
		});
	});
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it("should allow the 'ids' parameter without messing up the query when using .take", () =>
		Promise.all(
			connections.map(async (connection) => {
				const roleRepository = connection.getRepository(Role);
				const userRepository = connection.getRepository(User);

				const userWithId = (id: number) => ({ id });

				await roleRepository.save([
					{ id: 'a', users: [1, 2, 3].map(userWithId) },
					{ id: 'b', users: [9, 10, 11, 12, 13].map(userWithId) },
					{ id: 'c', users: [14, 15, 16, 17].map(userWithId) },
				]);

				const results = await userRepository
					.createQueryBuilder('user')
					.leftJoinAndSelect('user.role', 'role')
					.where('role.id IN (:...ids)', { ids: ['a', 'c'] })
					.take(5)
					.orderBy('user.id')
					.getMany();

				expect(results).to.be.deep.equal([
					{ id: 1, role: { id: 'a' } },
					{ id: 2, role: { id: 'a' } },
					{ id: 3, role: { id: 'a' } },
					{ id: 14, role: { id: 'c' } },
					{ id: 15, role: { id: 'c' } },
				]);
			}),
		));
});