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

describe('github issues > #2632 createQueryBuilder relation remove works only if using ID', () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				entities: [__dirname + '/entity/*{.js,.ts}'],
				schemaCreate: true,
				dropSchema: true,
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('should add and remove relations of an entity if given a mix of ids and objects', () =>
		Promise.all(
			connections.map(async (connection) => {
				const post1 = new Post();
				post1.title = 'post #1';
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.title = 'post #2';
				await connection.manager.save(post2);

				const category1 = new Category();
				category1.title = 'category #1';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.title = 'category #2';
				await connection.manager.save(category2);

				await connection.createQueryBuilder().relation(Post, 'categories').of(post1).add(1);

				let loadedPost1 = await connection.manager.findOne(Post, {
					where: { id: 1 },
					relations: { categories: true },
				});
				expect(loadedPost1!.categories).to.deep.include({
					id: 1,
					title: 'category #1',
				});

				await connection.createQueryBuilder().relation(Post, 'categories').of(post1).remove(1);

				loadedPost1 = await connection.manager.findOne(Post, {
					where: { id: 1 },
					relations: { categories: true },
				});
				expect(loadedPost1!.categories).to.be.eql([]);

				await connection.createQueryBuilder().relation(Post, 'categories').of(2).add(category2);

				let loadedPost2 = await connection.manager.findOne(Post, {
					where: { id: 2 },
					relations: { categories: true },
				});
				expect(loadedPost2!.categories).to.deep.include({
					id: 2,
					title: 'category #2',
				});

				await connection.createQueryBuilder().relation(Post, 'categories').of(2).remove(category2);

				loadedPost1 = await connection.manager.findOne(Post, {
					where: { id: 2 },
					relations: { categories: true },
				});
				expect(loadedPost1!.categories).to.be.eql([]);
			}),
		));
});