import "reflect-metadata";
import {createTestingConnections, closeTestingConnections, reloadTestingDatabases} from "../../utils/test-utils";
import {Connection} from "../../../src/connection/Connection";
import {Post} from "./entity/Post";
import {Category} from "./entity/Category";

describe("github issues > #1259 Can't sort by fields added with addSelect", () => {

    let connections: Connection[];
    before(async () => connections = await createTestingConnections({
        entities: [__dirname + "/entity/*{.js,.ts}"],
        enabledDrivers: ["postgres"]
    }));
    beforeEach(() => reloadTestingDatabases(connections));
    after(() => closeTestingConnections(connections));

    it("should order by added selects when pagination is used", () => Promise.all(connections.map(async connection => {

        const categories = [new Category(), new Category()];
        await connection.manager.save(categories);

        const posts: Post[] = [];
        for (let i = 0; i < 10; i++) {
            const post = new Post();
            if (i > 5 && i < 8) {
                post.name = `timber`;
            } else {
                post.name = `Tim${i}ber`;
            }
            post.count = 2;
            post.categories = categories;
            posts.push(post);
        }
        await connection.manager.save(posts);

        const loadedPosts = await connection.manager
            .createQueryBuilder(Post, "post")
            .addSelect("ts_rank_cd(to_tsvector(post.name), to_tsquery(:query))", "rank")
            .leftJoinAndSelect("post.categories", "categories")
            .orderBy("rank", "DESC")
            // .addOrderBy("post.id")
            .take(5)
            .setParameter("query", "timber")
            .getMany();

        loadedPosts.length.should.be.equal(5);
        loadedPosts[0].id.should.be.equal(7);
        loadedPosts[0].name.should.be.equal("timber");
        loadedPosts[1].id.should.be.equal(8);
        loadedPosts[1].name.should.be.equal("timber");
    })));

    it("should order by added selects when pagination is used", () => Promise.all(connections.map(async connection => {

        const categories = [new Category(), new Category()];
        await connection.manager.save(categories);

        const posts: Post[] = [];
        for (let i = 0; i < 10; i++) {
            const post = new Post();
            post.name = `timber`;
            post.count = i * -1;
            post.categories = categories;
            posts.push(post);
        }
        await connection.manager.save(posts);

        const loadedPosts = await connection.manager
            .createQueryBuilder(Post, "post")
            .addSelect("post.count * 2", "doublecount")
            .leftJoinAndSelect("post.categories", "categories")
            .orderBy("doublecount")
            .take(5)
            .getMany();

        loadedPosts.length.should.be.equal(5);
        loadedPosts[0].id.should.be.equal(10);
        loadedPosts[1].id.should.be.equal(9);
        loadedPosts[2].id.should.be.equal(8);
        loadedPosts[3].id.should.be.equal(7);
        loadedPosts[4].id.should.be.equal(6);
    })));

});