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

describe("github issues > #5478 Setting enumName doesn't change how migrations get generated", () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				migrations: [],
				enabledDrivers: ['postgres'],
				schemaCreate: true,
				dropSchema: true,
				entities: [UserEntity],
			})),
	);
	after(() => closeTestingConnections(connections));

	it('should correctly rename enum', () =>
		Promise.all(
			connections.map(async (connection) => {
				const queryRunner = connection.createQueryRunner();

				// add `enumName`
				let table = await queryRunner.getTable('user');
				const column = table!.findColumnByName('userType')!;
				const newColumn = column.clone();
				newColumn.enumName = 'UserTypeEnum';

				// change column
				await queryRunner.changeColumn(table!, column, newColumn);

				// check if `enumName` changed
				table = await queryRunner.getTable('user');
				let changedColumn = table!.findColumnByName('userType')!;
				expect(changedColumn.enumName).to.equal('UserTypeEnum');

				// revert changes
				await queryRunner.executeMemoryDownSql();

				// check if `enumName` reverted
				table = await queryRunner.getTable('user');
				changedColumn = table!.findColumnByName('userType')!;
				expect(changedColumn.enumName).to.undefined;

				await queryRunner.release();
			}),
		));
});