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

describe('github issues > #6947 Custom primary column for TreeRepository based entities unable to get tree descendants', () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				entities: [Category],
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('entities with custom primary column names should work', () =>
		Promise.all(
			connections.map(async (connection) => {
				const categoryRepository = connection.getTreeRepository(Category);

				const parent = new Category();
				parent.cat_name = 'parent';
				await categoryRepository.save(parent);

				const child = new Category();
				child.cat_name = 'child';
				child.parent = parent;
				await categoryRepository.save(child);

				const tree = await categoryRepository.findDescendantsTree(
					(await categoryRepository.findOneBy({
						cat_name: 'parent',
					}))!,
				);

				tree.should.deep.include({
					cat_id: 1,
					cat_name: 'parent',
					children: [
						{
							cat_id: 2,
							cat_name: 'child',
							children: [],
						},
					],
				});
			}),
		));
});