fix: correct isProductionDatabase() to allow CI database and add schema reset

2026-05-03 16:31:36 -07:00
parent 66e574d5ec
commit fa13ed86c9
2 changed files with 178 additions and 83 deletions
@@ -0,0 +1,126 @@
+/**
+ * Global teardown for Playwright tests
+ * Handles cleanup based on environment:
+ * - CI: Full schema reset (database can be destroyed and recreated)
+ * - Dev: Selective cleanup of test records (preserve real data)
+ * - Prod: Selective cleanup of test records (preserve real data)
+ */
+
+import { type FullConfig } from '@playwright/test';
+import { PrismaClient } from '@prisma/client';
+import { execSync } from 'child_process';
+
+const TEST_PATTERNS = {
+  players: [
+    '%Test%',
+    '%Setup%',
+    '%Home Test%',
+    '%Home Match Player%',
+    '%Admin User%',
+    '%NinePart%',
+    '%Nine Part%',
+    '%Test Player%',
+    '%TestUser%',
+    '%Cucumber%',
+    '%Config Admin%',
+  ],
+  events: [
+    '%Test%',
+    '%Setup%',
+    '%Recent%',
+    '%Test Tournament%',
+    '%Cucumber%',
+  ],
+  users: [
+    '%test%',
+    '%setup%',
+    '%cucumber%',
+    '%TestUser%',
+  ]
+};
+
+function isDatabase(url: string, name: string): boolean {
+  return url.includes(name);
+}
+
+function isProductionDatabase(): boolean {
+  const dbUrl = process.env.DATABASE_URL || '';
+  return isDatabase(dbUrl, 'euchre_camp') && !isDatabase(dbUrl, '_dev') && !isDatabase(dbUrl, '_ci');
+}
+
+function isCIDatabase(): boolean {
+  const dbUrl = process.env.DATABASE_URL || '';
+  return isDatabase(dbUrl, '_ci');
+}
+
+function buildLikeClause(patterns: string[]): string {
+  return patterns.map(p => `name LIKE '${p}'`).join(' OR ');
+}
+
+function buildEmailLikeClause(patterns: string[]): string {
+  return patterns.map(p => `email LIKE '${p}'`).join(' OR ');
+}
+
+async function resetDatabaseSchema(prisma: PrismaClient) {
+  console.log('Resetting database schema...');
+  
+  const tables = await prisma.$queryRaw<{ tablename: string }[]>`
+    SELECT tablename FROM pg_tables 
+    WHERE schemaname = 'public' AND tablename NOT LIKE '_prisma_migrations'
+  `;
+  
+  if (tables.length > 0) {
+    await prisma.$executeRawUnsafe(`
+      DROP SCHEMA public CASCADE;
+      CREATE SCHEMA public;
+      GRANT ALL ON SCHEMA public TO postgres;
+      GRANT ALL ON SCHEMA public TO public;
+    `);
+    console.log(`Dropped ${tables.length} tables`);
+  }
+  
+  console.log('Running migrations...');
+  execSync('bunx prisma migrate deploy', { stdio: 'inherit' });
+}
+
+async function cleanupTestRecords(prisma: PrismaClient) {
+  console.log('Cleaning up test records...');
+  
+  const playerWhere = buildLikeClause(TEST_PATTERNS.players);
+  const eventWhere = buildLikeClause(TEST_PATTERNS.events);
+  const userWhere = buildEmailLikeClause(TEST_PATTERNS.users);
+
+  await prisma.$executeRawUnsafe(`DELETE FROM events WHERE (${eventWhere});`);
+  console.log('Deleted test events');
+
+  await prisma.$executeRawUnsafe(`DELETE FROM players WHERE (${playerWhere});`);
+  console.log('Deleted test players');
+
+  await prisma.$executeRawUnsafe(`DELETE FROM users WHERE (${userWhere}) AND "playerId" IS NULL;`);
+  console.log('Deleted test users (without player associations)');
+
+  await prisma.$disconnect();
+}
+
+export default async function globalTeardown(config: FullConfig) {
+  console.log('\n=== Global Teardown ===');
+  
+  const dbUrl = process.env.DATABASE_URL || '';
+  
+  if (isCIDatabase()) {
+    console.log('CI environment - resetting database schema');
+    const prisma = new PrismaClient();
+    await resetDatabaseSchema(prisma);
+    await prisma.$disconnect();
+  } else if (isProductionDatabase()) {
+    console.log('Production environment - selective cleanup of test records');
+    const prisma = new PrismaClient();
+    await cleanupTestRecords(prisma);
+  } else {
+    console.log('Development environment - selective cleanup of test records');
+    const prisma = new PrismaClient();
+    await cleanupTestRecords(prisma);
+  }
+  
+  console.log('=== Global Teardown Complete ===\n');
+}