use custom db reset

Author: devneill

prisma/seed.ts

@@ -3,12 +3,12 @@ import { promiseHash } from 'remix-utils/promise'
 import { prisma } from '#app/utils/db.server.ts'
 import { MOCK_CODE_GITHUB } from '#app/utils/providers/constants'
 import {
+	cleanupDb,
 	createPassword,
 	createUser,
 	getNoteImages,
 	getUserImages,
 	img,
-	resetDb,
 } from '#tests/db-utils.ts'
 import { insertGitHubUser } from '#tests/mocks/github.ts'
 
@@ -17,7 +17,7 @@ async function seed() {
 	console.time(`🌱 Database has been seeded`)
 
 	console.time('🧹 Cleaned up the database...')
-	await resetDb()
+	await cleanupDb(prisma)
 	console.timeEnd('🧹 Cleaned up the database...')
 
 	const totalUsers = 5

tests/db-utils.ts

@@ -1,6 +1,6 @@
-import { spawnSync } from 'node:child_process'
 import fs from 'node:fs'
 import { faker } from '@faker-js/faker'
+import { type PrismaClient } from '@prisma/client'
 import bcrypt from 'bcryptjs'
 import { UniqueEnforcer } from 'enforce-unique'
 
@@ -115,17 +115,50 @@ export async function img({
 	}
 }
 
-export async function resetDb(dbPath?: string) {
-	const databaseUrl = dbPath ? { DATABASE_URL: `file:${dbPath}` } : {}
-
-	spawnSync(
-		'npx',
-		['prisma', 'migrate', 'reset', '--force', '--skip-seed', '--skip-generate'],
-		{
-			env: {
-				...process.env,
-				...databaseUrl,
-			},
-		},
-	)
+export async function cleanupDb(prisma: PrismaClient) {
+	const tables = await prisma.$queryRaw<
+		{ name: string }[]
+	>`SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%' AND name NOT LIKE '_prisma_migrations';`
+
+	const migrationPaths = fs
+		.readdirSync('prisma/migrations')
+		.filter((dir) => dir !== 'migration_lock.toml')
+		.map((dir) => `prisma/migrations/${dir}/migration.sql`)
+
+	const migrations = migrationPaths.map((path) => {
+		// Parse the sql into individual statements
+		const sql = fs
+			.readFileSync(path)
+			.toString()
+			.split(';')
+			.map((statement) => (statement += ';'))
+
+		// Remove empty last line
+		sql.pop()
+
+		return sql
+	})
+
+	try {
+		// Disable FK constraints to avoid relation conflicts during deletion
+		await prisma.$executeRawUnsafe(`PRAGMA foreign_keys = OFF`)
+		await prisma.$transaction([
+			// Delete all tables except the ones that are excluded above
+			...tables.map(({ name }) =>
+				prisma.$executeRawUnsafe(`DROP TABLE "${name}"`),
+			),
+		])
+
+		// Run the migrations sequentially
+		for (const migration of migrations) {
+			await prisma.$transaction([
+				// Run each sql statement in the migration
+				...migration.map((sql) => prisma.$executeRawUnsafe(sql)),
+			])
+		}
+	} catch (error) {
+		console.error('Error cleaning up database:', error)
+	} finally {
+		await prisma.$executeRawUnsafe(`PRAGMA foreign_keys = ON`)
+	}
 }

tests/setup/db-setup.ts

@@ -1,7 +1,7 @@
 import path from 'node:path'
 import fsExtra from 'fs-extra'
 import { afterAll, afterEach, beforeAll } from 'vitest'
-import { resetDb } from '#tests/db-utils.js'
+import { cleanupDb } from '#tests/db-utils.ts'
 import { BASE_DATABASE_PATH } from './global-setup.ts'
 
 const databaseFile = `./tests/prisma/data.${process.env.VITEST_POOL_ID || 0}.db`
@@ -12,8 +12,11 @@ beforeAll(async () => {
 	await fsExtra.copyFile(BASE_DATABASE_PATH, databasePath)
 })
 
+// we *must* use dynamic imports here so the process.env.DATABASE_URL is set
+// before prisma is imported and initialized
 afterEach(async () => {
-	await resetDb()
+	const { prisma } = await import('#app/utils/db.server.ts')
+	await cleanupDb(prisma)
 })
 
 afterAll(async () => {

tests/setup/global-setup.ts

@@ -1,6 +1,6 @@
 import path from 'node:path'
+import { execaCommand } from 'execa'
 import fsExtra from 'fs-extra'
-import { resetDb } from '#tests/db-utils.js'
 
 export const BASE_DATABASE_PATH = path.join(
 	process.cwd(),
@@ -22,5 +22,14 @@ export async function setup() {
 		}
 	}
 
-	await resetDb(BASE_DATABASE_PATH)
+	await execaCommand(
+		'npx prisma migrate reset --force --skip-seed --skip-generate',
+		{
+			stdio: 'inherit',
+			env: {
+				...process.env,
+				DATABASE_URL: `file:${BASE_DATABASE_PATH}`,
+			},
+		},
+	)
 }