Initial commit

api/tests/support/clean-database.ts

@@ -0,0 +1,57 @@
+import { QueryTypes } from "@sequelize/core"
+
+import { isNil } from "lodash"
+
+import db from "@/db/db-client"
+
+async function getTableNames() {
+  const query = /* sql */ `
+    SELECT
+      table_name as "tableName"
+    FROM
+      information_schema.tables
+    WHERE
+      table_schema = 'public'
+      AND table_type = 'BASE TABLE'
+      AND table_name != 'SequelizeMeta'
+      AND table_name != 'knex_migrations'
+      AND table_name != 'knex_migrations_lock';
+  `
+
+  try {
+    const result = await db.query<{ tableName: string }>(query, { type: QueryTypes.SELECT })
+    const tableNames = result.map((row) => row.tableName)
+    return tableNames
+  } catch (error) {
+    console.error("Error fetching table names:", error)
+    throw error
+  }
+}
+
+async function buildCleanDatabaseQuery() {
+  const tableNames = await getTableNames()
+  const quotedTableNames = tableNames.map((name) => `"${name}"`)
+  return /* sql */ `
+    TRUNCATE TABLE ${quotedTableNames.join(",\n      ")} RESTART IDENTITY CASCADE;
+  `
+}
+
+let cleanDatabaseQuery: string | null = null
+
+export async function cleanDatabase() {
+  if (isNil(cleanDatabaseQuery)) {
+    cleanDatabaseQuery = await buildCleanDatabaseQuery()
+  }
+
+  try {
+    // TODO: once all tables are in Sequelize models, use this instead:
+    // await db.truncate({ cascade: true, restartIdentity: true })
+    await db.query(cleanDatabaseQuery, { raw: true })
+    return true
+  } catch (error) {
+    console.error(error)
+    return false
+  }
+}
+
+export default cleanDatabase