Function resetDB

packages/core/fumadb/test/shared.ts:440–569

(provider: SQLProvider)

}

export async function resetDB(provider: SQLProvider) {
  const kysely = kyselyTests.find((kysely) => kysely.provider === provider)!;
  const db = kysely.db as Kysely<any>;

  if (provider === "mysql") {
    await db.transaction().execute(async () => {
      await sql`SET FOREIGN_KEY_CHECKS = 0`.execute(db);
      const tables = await db
        .selectFrom("information_schema.tables")
        .select("TABLE_NAME")
        .where((b) =>
          b.or([
            b("TABLE_SCHEMA", "not in", [
              "mysql",
              "performance_schema",
              "information_schema",
              "sys",
            ]),
            b("TABLE_NAME", "=", "__drizzle_migrations"),
          ]),
        )
        .where("TABLE_TYPE", "=", "BASE TABLE")
        .execute();
      for (const table of tables) {
        await db.schema.dropTable(table.TABLE_NAME).ifExists().execute();
      }

      await sql`SET FOREIGN_KEY_CHECKS = 1`.execute(db);
    });

    return;
  }

  if (provider === "sqlite") {
    const tables = await db
      .selectFrom("sqlite_master")
      .select("name")
      .where("type", "=", "table")
      .where("name", "not like", "sqlite_%")
      .execute();

    await sql`PRAGMA foreign_keys = OFF`.execute(db);
    await Promise.all(
      tables.map((table) =>
        db.schema.dropTable(table.name).ifExists().execute(),
      ),
    );
    await sql`PRAGMA foreign_keys = ON`.execute(db);
    return;
  }

  if (provider === "postgresql" || provider === "cockroachdb") {
    const tables = await db
      .selectFrom("information_schema.tables")
      .select(["table_schema", "table_name"])
      .where("table_type", "=", "BASE TABLE")
      .where("table_schema", "not in", [
        "pg_catalog",