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

describe("query runner > drop table", () => {
    let connections: DataSource[]
    before(async () => {
        connections = await createTestingConnections({
            entities: [__dirname + "/entity/*{.js,.ts}"],
            schemaCreate: true,
        })
    })
    beforeEach(() => reloadTestingDatabases(connections))
    after(() => closeTestingConnections(connections))

    it("should correctly drop table without relations and revert drop", () =>
        Promise.all(
            connections.map(async (connection) => {
                const queryRunner = connection.createQueryRunner()

                let table = await queryRunner.getTable("post")
                table!.should.exist

                await queryRunner.dropTable("post")

                table = await queryRunner.getTable("post")
                expect(table).to.be.undefined

                await queryRunner.executeMemoryDownSql()

                table = await queryRunner.getTable("post")
                table!.should.exist

                await queryRunner.release()
            }),
        ))

    it("should correctly drop table with relations and revert drop", () =>
        Promise.all(
            connections.map(async (connection) => {
                const queryRunner = connection.createQueryRunner()

                let studentTable = await queryRunner.getTable("student")
                let teacherTable = await queryRunner.getTable("teacher")
                let facultyTable = await queryRunner.getTable("faculty")
                studentTable!.should.exist
                teacherTable!.should.exist
                facultyTable!.should.exist

                await queryRunner.dropTable(studentTable!)
                await queryRunner.dropTable(teacherTable!)
                await queryRunner.dropTable(facultyTable!)

                studentTable = await queryRunner.getTable("student")
                teacherTable = await queryRunner.getTable("teacher")
                facultyTable = await queryRunner.getTable("faculty")
                expect(studentTable).to.be.undefined
                expect(teacherTable).to.be.undefined
                expect(facultyTable).to.be.undefined

                await queryRunner.executeMemoryDownSql()

                studentTable = await queryRunner.getTable("student")
                teacherTable = await queryRunner.getTable("teacher")
                facultyTable = await queryRunner.getTable("faculty")
                studentTable!.should.exist
                teacherTable!.should.exist
                facultyTable!.should.exist

                await queryRunner.release()
            }),
        ))
})