import {
    closeTestingConnections,
    createTestingConnections,
    reloadTestingDatabases,
} from "../../utils/test-utils"
import { DataSource } from "../../../src/data-source/DataSource"
import { EntitySchema } from "../../../src"
import { Author, AuthorSchema } from "./entity/Author"
import { Post, PostSchema } from "./entity/Post"

describe("github issues > #1123 load relation eagerly by setting isEager property", () => {
    let connections: DataSource[]
    before(
        async () =>
            (connections = await createTestingConnections({
                entities: [
                    new EntitySchema<Author>(AuthorSchema),
                    new EntitySchema<Post>(PostSchema),
                ],
                dropSchema: true,
            })),
    )
    beforeEach(() => reloadTestingDatabases(connections))
    after(() => closeTestingConnections(connections))

    async function prepareData(connection: DataSource) {
        const author = new Author()
        author.id = 1
        author.name = "Jane Doe"
        await connection.manager.save(author)

        const post = new Post()
        post.id = 1
        post.title = "Post 1"
        post.author = author
        await connection.manager.save(post)
    }

    it("should load all eager relations when object is loaded", () =>
        Promise.all(
            connections.map(async (connection) => {
                await prepareData(connection)

                const loadedPost = await connection.manager.findOneBy(Post, {
                    id: 1,
                })
                loadedPost!.should.be.eql({
                    id: 1,
                    title: "Post 1",
                    author: {
                        id: 1,
                        name: "Jane Doe",
                    },
                })
            }),
        ))

    it("should not load eager relations when query builder is used", () =>
        Promise.all(
            connections.map(async (connection) => {
                await prepareData(connection)

                const loadedPost = (await connection.manager
                    .createQueryBuilder("Post", "post")
                    .where("post.id = :id", { id: 1 })
                    .getOne()) as Post

                loadedPost!.should.be.eql({
                    id: 1,
                    title: "Post 1",
                })
            }),
        ))
})