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

describe("schema builder > update primary keys", () => {
    let connections: DataSource[]
    before(async () => {
        connections = await createTestingConnections({
            entities: [__dirname + "/entity/*{.js,.ts}"],
            schemaCreate: true,
            dropSchema: true,
        })
    })
    after(() => closeTestingConnections(connections))

    it("should correctly update composite primary keys", () =>
        Promise.all(
            connections.map(async (connection) => {
                // CockroachDB and Spanner does not support changing primary key constraint
                if (
                    connection.driver.options.type === "cockroachdb" ||
                    connection.driver.options.type === "spanner"
                )
                    return

                const metadata = connection.getMetadata(Category)
                const nameColumn = metadata.findColumnWithPropertyName("name")
                nameColumn!.isPrimary = true

                await connection.synchronize()

                const queryRunner = connection.createQueryRunner()
                const table = await queryRunner.getTable("category")
                table!.findColumnByName("id")!.isPrimary.should.be.true
                table!.findColumnByName("name")!.isPrimary.should.be.true

                await queryRunner.release()
            }),
        ))

    it("should correctly update composite primary keys when table already have primary generated column", () =>
        Promise.all(
            connections.map(async (connection) => {
                // Sqlite does not support AUTOINCREMENT on composite primary key
                if (DriverUtils.isSQLiteFamily(connection.driver)) return

                // CockroachDB and Spanner does not support changing primary key constraint
                if (
                    connection.driver.options.type === "cockroachdb" ||
                    connection.driver.options.type === "spanner"
                )
                    return

                const metadata = connection.getMetadata(Question)
                const nameColumn = metadata.findColumnWithPropertyName("name")
                nameColumn!.isPrimary = true

                await connection.synchronize()

                const queryRunner = connection.createQueryRunner()
                const table = await queryRunner.getTable("question")
                table!.findColumnByName("id")!.isPrimary.should.be.true
                table!.findColumnByName("name")!.isPrimary.should.be.true

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