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

describe('github issues > #3946 loadRelationCountAndMap fails cause made a wrong IN calculation, when primary key is string', () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				entities: [__dirname + '/entity/*{.js,.ts}'],
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('should load relation count on owner side', () =>
		Promise.all(
			connections.map(async (connection) => {
				const category1 = new Category();
				category1.name = 'cars';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.name = 'BMW';
				await connection.manager.save(category2);

				const category3 = new Category();
				category3.name = 'Germany';
				await connection.manager.save(category3);

				const category4 = new Category();
				category4.name = 'airplanes';
				await connection.manager.save(category4);

				const category5 = new Category();
				category5.name = 'Boeing';
				await connection.manager.save(category5);

				const post1 = new Post();
				post1.id = 'a26f5a9a-4017-4365-be65-4665081d3f39';
				post1.title = 'about BMW';
				post1.categories = [category1, category2, category3];
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.id = 'b1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Boeing';
				post2.categories = [category4, category5];
				await connection.manager.save(post2);

				let loadedPosts = await connection.manager
					.createQueryBuilder(Post, 'post')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.addOrderBy('post.id')
					.getMany();

				expect(loadedPosts![0].categoryCount).to.be.equal(3);
				expect(loadedPosts![1].categoryCount).to.be.equal(2);

				let loadedPost = await connection.manager
					.createQueryBuilder(Post, 'post')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.where('post.id = :id', { id: post1.id })
					.getOne();

				expect(loadedPost!.categoryCount).to.be.equal(3);
			}),
		));

	it('should load relation count on owner side with limitation', () =>
		Promise.all(
			connections.map(async (connection) => {
				const category1 = new Category();
				category1.id = '126f5a9a-4017-4365-be65-4665081d3f39';
				category1.name = 'cars';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.id = '226f5a9a-4017-4365-be65-4665081d3f39';
				category2.name = 'BMW';
				await connection.manager.save(category2);

				const category3 = new Category();
				category3.id = '326f5a9a-4017-4365-be65-4665081d3f39';
				category3.name = 'Germany';
				await connection.manager.save(category3);

				const category4 = new Category();
				category4.id = '426f5a9a-4017-4365-be65-4665081d3f39';
				category4.name = 'airplanes';
				await connection.manager.save(category4);

				const category5 = new Category();
				category5.id = '526f5a9a-4017-4365-be65-4665081d3f39';
				category5.name = 'Boeing';
				await connection.manager.save(category5);

				const post1 = new Post();
				post1.id = 'a26f5a9a-4017-4365-be65-4665081d3f39';
				post1.title = 'about BMW';
				post1.categories = [category1, category2, category3];
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.id = 'b1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Boeing';
				post2.categories = [category4, category5];
				await connection.manager.save(post2);

				const post3 = new Post();
				post3.id = 'c1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post3.title = 'about Audi';
				await connection.manager.save(post3);

				const post4 = new Post();
				post4.id = 'd1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post4.title = 'about Airbus';
				await connection.manager.save(post4);

				let loadedPosts = await connection.manager
					.createQueryBuilder(Post, 'post')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.orderBy('post.id')
					.offset(0)
					.limit(2)
					.getMany();

				expect(loadedPosts![0].categoryCount).to.be.equal(3);
				expect(loadedPosts![1].categoryCount).to.be.equal(2);
			}),
		));

	it('should load relation count on owner side with additional conditions', () =>
		Promise.all(
			connections.map(async (connection) => {
				const image1 = new Image();
				image1.id = '11d58da3-913d-4696-afb5-a0d2aae41dc9';
				image1.isRemoved = true;
				image1.name = 'image #1';
				await connection.manager.save(image1);

				const image2 = new Image();
				image2.id = '21d58da3-913d-4696-afb5-a0d2aae41dc9';
				image2.name = 'image #2';
				await connection.manager.save(image2);

				const image3 = new Image();
				image3.id = '31d58da3-913d-4696-afb5-a0d2aae41dc9';
				image3.name = 'image #3';
				await connection.manager.save(image3);

				const category1 = new Category();
				category1.id = '41d58da3-913d-4696-afb5-a0d2aae41dc9';
				category1.name = 'cars';
				category1.isRemoved = true;
				category1.images = [image1, image2];
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.id = '51d58da3-913d-4696-afb5-a0d2aae41dc9';
				category2.name = 'BMW';
				await connection.manager.save(category2);

				const category3 = new Category();
				category3.id = '61d58da3-913d-4696-afb5-a0d2aae41dc9';
				category3.name = 'Germany';
				await connection.manager.save(category3);

				const category4 = new Category();
				category4.id = '71d58da3-913d-4696-afb5-a0d2aae41dc9';
				category4.name = 'airplanes';
				category4.images = [image3];
				await connection.manager.save(category4);

				const category5 = new Category();
				category5.id = '81d58da3-913d-4696-afb5-a0d2aae41dc9';
				category5.name = 'Boeing';
				await connection.manager.save(category5);

				const post1 = new Post();
				post1.id = 'a1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post1.title = 'about BMW';
				post1.categories = [category1, category2, category3];
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.id = 'b1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Boeing';
				post2.categories = [category4, category5];
				await connection.manager.save(post2);

				let loadedPosts = await connection.manager
					.createQueryBuilder(Post, 'post')
					.leftJoinAndSelect('post.categories', 'categories')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.loadRelationCountAndMap('post.removedCategoryCount', 'post.categories', 'rc', (qb) =>
						qb.andWhere('rc.isRemoved = :isRemoved', {
							isRemoved: true,
						}),
					)
					.loadRelationCountAndMap('categories.imageCount', 'categories.images', 'ic')
					.loadRelationCountAndMap(
						'categories.removedImageCount',
						'categories.images',
						'removedImages',
						(qb) => qb.andWhere('removedImages.isRemoved = :isRemoved', { isRemoved: true }),
					)
					.addOrderBy('post.id, categories.id')
					.getMany();

				expect(loadedPosts![0].categoryCount).to.be.equal(3);
				expect(loadedPosts![0].removedCategoryCount).to.be.equal(1);
				expect(loadedPosts![0].categories[0].imageCount).to.be.equal(2);
				expect(loadedPosts![0].categories[0].removedImageCount).to.be.equal(1);
				expect(loadedPosts![0].categories[1].imageCount).to.be.equal(0);
				expect(loadedPosts![0].categories[2].imageCount).to.be.equal(0);
				expect(loadedPosts![1].categoryCount).to.be.equal(2);
				expect(loadedPosts![1].categories[0].imageCount).to.be.equal(1);

				let loadedPost = await connection.manager
					.createQueryBuilder(Post, 'post')
					.leftJoinAndSelect('post.categories', 'categories')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.loadRelationCountAndMap('post.removedCategoryCount', 'post.categories', 'rc', (qb) =>
						qb.andWhere('rc.isRemoved = :isRemoved', {
							isRemoved: true,
						}),
					)
					.loadRelationCountAndMap('categories.imageCount', 'categories.images', 'ic')
					.loadRelationCountAndMap(
						'categories.removedImageCount',
						'categories.images',
						'removedImages',
						(qb) => qb.andWhere('removedImages.isRemoved = :isRemoved', { isRemoved: true }),
					)
					.where('post.id = :id', { id: post1.id })
					.addOrderBy('post.id, categories.id')
					.getOne();

				expect(loadedPost!.categoryCount).to.be.equal(3);
				expect(loadedPost!.removedCategoryCount).to.be.equal(1);
				expect(loadedPost!.categories[0].imageCount).to.be.equal(2);
				expect(loadedPost!.categories[0].removedImageCount).to.be.equal(1);
			}),
		));

	it('should load relation count on both sides of relation', () =>
		Promise.all(
			connections.map(async (connection) => {
				const category1 = new Category();
				category1.id = '41d58da3-913d-4696-afb5-a0d2aae41dc9';
				category1.name = 'cars';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.id = '51d58da3-913d-4696-afb5-a0d2aae41dc9';
				category2.name = 'BMW';
				await connection.manager.save(category2);

				const category3 = new Category();
				category3.id = '61d58da3-913d-4696-afb5-a0d2aae41dc9';
				category3.name = 'Germany';
				await connection.manager.save(category3);

				const post1 = new Post();
				post1.id = '71d58da3-913d-4696-afb5-a0d2aae41dc9';
				post1.title = 'about BMW';
				post1.categories = [category1, category2, category3];
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.id = '81d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Audi';
				post2.categories = [category1, category3];
				await connection.manager.save(post2);

				let loadedPosts = await connection.manager
					.createQueryBuilder(Post, 'post')
					.leftJoinAndSelect('post.categories', 'categories')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.loadRelationCountAndMap('categories.postCount', 'categories.posts')
					.addOrderBy('post.id, categories.id')
					.getMany();

				expect(loadedPosts![0].categoryCount).to.be.equal(3);
				expect(loadedPosts![0].categories[0].postCount).to.be.equal(2);
				expect(loadedPosts![0].categories[1].postCount).to.be.equal(1);
				expect(loadedPosts![0].categories[2].postCount).to.be.equal(2);
				expect(loadedPosts![1].categoryCount).to.be.equal(2);
				expect(loadedPosts![1].categories[0].postCount).to.be.equal(2);
				expect(loadedPosts![1].categories[1].postCount).to.be.equal(2);

				let loadedPost = await connection.manager
					.createQueryBuilder(Post, 'post')
					.leftJoinAndSelect('post.categories', 'categories')
					.loadRelationCountAndMap('post.categoryCount', 'post.categories')
					.loadRelationCountAndMap('categories.postCount', 'categories.posts')
					.where('post.id = :id', { id: post1.id })
					.addOrderBy('post.id, categories.id')
					.getOne();

				expect(loadedPost!.categoryCount).to.be.equal(3);
				expect(loadedPost!.categories[0].postCount).to.be.equal(2);
				expect(loadedPost!.categories[1].postCount).to.be.equal(1);
				expect(loadedPost!.categories[2].postCount).to.be.equal(2);
			}),
		));

	it('should load relation count on inverse side', () =>
		Promise.all(
			connections.map(async (connection) => {
				const category1 = new Category();
				category1.id = '41d58da3-913d-4696-afb5-a0d2aae41dc9';
				category1.name = 'cars';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.id = '51d58da3-913d-4696-afb5-a0d2aae41dc9';
				category2.name = 'airplanes';
				await connection.manager.save(category2);

				const post1 = new Post();
				post1.id = '61d58da3-913d-4696-afb5-a0d2aae41dc9';
				post1.title = 'about BMW';
				post1.categories = [category1];
				await connection.manager.save(post1);

				const post2 = new Post();
				post1.id = '71d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Audi';
				post2.categories = [category1];
				await connection.manager.save(post2);

				const post3 = new Post();
				post1.id = '81d58da3-913d-4696-afb5-a0d2aae41dc9';
				post3.title = 'about Mercedes';
				post3.categories = [category1];
				await connection.manager.save(post3);

				const post4 = new Post();
				post1.id = '91d58da3-913d-4696-afb5-a0d2aae41dc9';
				post4.title = 'about Boeing';
				post4.categories = [category2];
				await connection.manager.save(post4);

				const post5 = new Post();
				post1.id = 'a1d58da3-913d-4696-afb5-a0d2aae41dc9';
				post5.title = 'about Airbus';
				post5.categories = [category2];
				await connection.manager.save(post5);

				let loadedCategories = await connection.manager
					.createQueryBuilder(Category, 'category')
					.loadRelationCountAndMap('category.postCount', 'category.posts')
					.addOrderBy('category.id')
					.getMany();

				expect(loadedCategories![0].postCount).to.be.equal(3);
				expect(loadedCategories![1].postCount).to.be.equal(2);

				let loadedCategory = await connection.manager
					.createQueryBuilder(Category, 'category')
					.loadRelationCountAndMap('category.postCount', 'category.posts')
					.where('category.id = :id', { id: category1.id })
					.getOne();

				expect(loadedCategory!.postCount).to.be.equal(3);
			}),
		));

	it('should load relation count on inverse side with limitation', () =>
		Promise.all(
			connections.map(async (connection) => {
				const category1 = new Category();
				category1.id = '11d58da3-913d-4696-afb5-a0d2aae41dc9';
				category1.name = 'cars';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.id = '21d58da3-913d-4696-afb5-a0d2aae41dc9';
				category2.name = 'airplanes';
				await connection.manager.save(category2);

				const category3 = new Category();
				category3.id = '31d58da3-913d-4696-afb5-a0d2aae41dc9';
				category3.name = 'BMW';
				await connection.manager.save(category3);

				const category4 = new Category();
				category4.id = '41d58da3-913d-4696-afb5-a0d2aae41dc9';
				category4.name = 'Boeing';
				await connection.manager.save(category4);

				const post1 = new Post();
				post1.id = '51d58da3-913d-4696-afb5-a0d2aae41dc9';
				post1.title = 'about BMW';
				post1.categories = [category1];
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.id = '61d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Audi';
				post2.categories = [category1];
				await connection.manager.save(post2);

				const post3 = new Post();
				post3.id = '71d58da3-913d-4696-afb5-a0d2aae41dc9';
				post3.title = 'about Mercedes';
				post3.categories = [category1];
				await connection.manager.save(post3);

				const post4 = new Post();
				post4.id = '81d58da3-913d-4696-afb5-a0d2aae41dc9';
				post4.title = 'about Boeing';
				post4.categories = [category2];
				await connection.manager.save(post4);

				const post5 = new Post();
				post5.id = '91d58da3-913d-4696-afb5-a0d2aae41dc9';
				post5.title = 'about Airbus';
				post5.categories = [category2];
				await connection.manager.save(post5);

				let loadedCategories = await connection.manager
					.createQueryBuilder(Category, 'category')
					.loadRelationCountAndMap('category.postCount', 'category.posts')
					.orderBy('category.id')
					.offset(0)
					.limit(2)
					.getMany();

				expect(loadedCategories![0].postCount).to.be.equal(3);
				expect(loadedCategories![1].postCount).to.be.equal(2);
			}),
		));

	it('should load relation count on inverse side with additional conditions', () =>
		Promise.all(
			connections.map(async (connection) => {
				const category1 = new Category();
				category1.id = '11d58da3-913d-4696-afb5-a0d2aae41dc9';
				category1.name = 'cars';
				await connection.manager.save(category1);

				const category2 = new Category();
				category2.id = '21d58da3-913d-4696-afb5-a0d2aae41dc9';
				category2.name = 'airplanes';
				await connection.manager.save(category2);

				const post1 = new Post();
				post1.id = '31d58da3-913d-4696-afb5-a0d2aae41dc9';
				post1.title = 'about BMW';
				post1.isRemoved = true;
				post1.categories = [category1];
				await connection.manager.save(post1);

				const post2 = new Post();
				post2.id = '41d58da3-913d-4696-afb5-a0d2aae41dc9';
				post2.title = 'about Audi';
				post2.isRemoved = true;
				post2.categories = [category1];
				await connection.manager.save(post2);

				const post3 = new Post();
				post3.id = '51d58da3-913d-4696-afb5-a0d2aae41dc9';
				post3.title = 'about Mercedes';
				post3.categories = [category1];
				await connection.manager.save(post3);

				const post4 = new Post();
				post4.id = '61d58da3-913d-4696-afb5-a0d2aae41dc9';
				post4.title = 'about Boeing';
				post4.categories = [category2];
				await connection.manager.save(post4);

				const post5 = new Post();
				post5.id = '71d58da3-913d-4696-afb5-a0d2aae41dc9';
				post5.title = 'about Airbus';
				post5.categories = [category2];
				await connection.manager.save(post5);

				let loadedCategories = await connection.manager
					.createQueryBuilder(Category, 'category')
					.loadRelationCountAndMap('category.postCount', 'category.posts')
					.loadRelationCountAndMap(
						'category.removedPostCount',
						'category.posts',
						'removedPosts',
						(qb) =>
							qb.andWhere('removedPosts.isRemoved = :isRemoved', {
								isRemoved: true,
							}),
					)
					.addOrderBy('category.id')
					.getMany();

				expect(loadedCategories![0].postCount).to.be.equal(3);
				expect(loadedCategories![0].removedPostCount).to.be.equal(2);
				expect(loadedCategories![1].postCount).to.be.equal(2);

				let loadedCategory = await connection.manager
					.createQueryBuilder(Category, 'category')
					.loadRelationCountAndMap('category.postCount', 'category.posts')
					.loadRelationCountAndMap(
						'category.removedPostCount',
						'category.posts',
						'removedPosts',
						(qb) =>
							qb.andWhere('removedPosts.isRemoved = :isRemoved', {
								isRemoved: true,
							}),
					)
					.where('category.id = :id', { id: category1.id })
					.getOne();

				expect(loadedCategory!.postCount).to.be.equal(3);
				expect(loadedCategory!.removedPostCount).to.be.equal(2);
			}),
		));
});