import 'reflect-metadata';
import { expect } from 'chai';
import { Test, ValueUnion } from './entity/Test';
import { DataSource } from '../../../src/data-source/DataSource';
import {
	closeTestingConnections,
	createTestingConnections,
	reloadTestingDatabases,
} from '../../utils/test-utils';
import { LessThan } from '../../../src';

describe("github issues > #9152 Can't use LessThan for Union field", () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				entities: [Test],
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('should not raise TypeScript error when LessThan with Union is passed to FindOptionsWhere', () =>
		Promise.all(
			connections.map(async (connection) => {
				await connection.getRepository(Test).save({
					value: 1,
				});

				const value = 2 as ValueUnion;

				const count = await connection.getRepository(Test).countBy({
					value: LessThan(value),
				});

				expect(count).to.eq(1);
			}),
		));
});