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

import { CatEntity } from './entity/Cat';
import { AnimalEntity } from './entity/Animal';

describe('github issues > #9033 Cannot manually insert type in discriminator column of parent entity class when\
using single table inheritance when creating instance of parent entity', () => {
	let dataSources: DataSource[];
	before(
		async () =>
			(dataSources = await createTestingConnections({
				entities: [__dirname + '/entity/*{.js,.ts}'],
				schemaCreate: true,
				dropSchema: true,
			})),
	);
	beforeEach(() => reloadTestingDatabases(dataSources));
	after(() => closeTestingConnections(dataSources));

	it('is possible to set the discriminator column manually on the base entity', () =>
		Promise.all(
			dataSources.map(async (dataSource) => {
				const entityManager = dataSource.createEntityManager();
				const animalRepo = entityManager.getRepository(AnimalEntity);

				// Create a base class entity while manually setting discriminator.
				const maybeACat = animalRepo.create({
					name: 'i-am-maybe-a-cat',
					type: 'cat', // This is the discriminator for `CatEntity`.
				});
				await entityManager.save(maybeACat);

				// Load the animal / cat from the database.
				const animals = await animalRepo.find();
				expect(animals.length).to.equal(1);

				expect(animals[0]).to.be.instanceOf(CatEntity);
				expect(animals[0].type).to.equal('cat');
			}),
		));
});