import "reflect-metadata"
import {
    createTestingConnections,
    closeTestingConnections,
    reloadTestingDatabases,
} from "../../utils/test-utils"
import { DataSource } from "../../../src/data-source/DataSource"
import { Category } from "./entity/Category"

describe("github issues > #3783 Tree functionality broken", () => {
    let connections: DataSource[]
    before(
        async () =>
            (connections = await createTestingConnections({
                entities: [__dirname + "/entity/*{.js,.ts}"],
            })),
    )
    beforeEach(() => reloadTestingDatabases(connections))
    after(() => closeTestingConnections(connections))

    it("should work correctly", () =>
        Promise.all(
            connections.map(async (connection) => {
                const categoryRepository =
                    connection.getTreeRepository(Category)

                const a1 = new Category()
                a1.name = "a1"

                const b1 = new Category()
                b1.name = "b1"

                const c1 = new Category()
                c1.name = "c1"

                const c11 = new Category()
                c11.name = "c11"

                const c12 = new Category()
                c12.name = "c12"

                c11.parentCategory = c1
                c12.parentCategory = c1

                await categoryRepository.save(a1)
                await categoryRepository.save(b1)
                await categoryRepository.save(c1)
                await categoryRepository.save(c11)
                await categoryRepository.save(c12)

                const roots = await categoryRepository.findRoots()
                roots.length.should.be.eql(3)

                const c1Tree = await categoryRepository.findDescendantsTree(c1)
                c1Tree.should.be.equal(c1)
                c1Tree!.childCategories.length.should.be.eql(2)
            }),
        ))
})