Files
goodgo-platform/e2e/global-teardown.ts
2026-05-04 17:27:08 +07:00

97 lines
4.6 KiB
TypeScript

import path from 'node:path';
/**
* Playwright globalTeardown — runs once after all E2E tests.
*
* Cleans up test-generated data (users, listings, etc.) while preserving
* seed data for the next run. This ensures isolation between test runs.
*/
export default async function globalTeardown() {
const root = path.resolve(__dirname, '..');
const isCI = !!process.env.CI;
if (!isCI) {
const { config } = await import('dotenv');
config({ path: path.join(root, '.env.test'), override: true });
}
const databaseUrl = process.env.DATABASE_URL;
if (!databaseUrl) {
console.warn('[E2E globalTeardown] DATABASE_URL not set, skipping cleanup.');
return;
}
console.log('\n[E2E globalTeardown] Cleaning up test-generated data...');
// Dynamic import to avoid top-level side effects
const pg = await import('pg');
const pool = new pg.default.Pool({ connectionString: databaseUrl });
try {
// Delete test-generated records (those NOT created by seed).
// Seed data uses known IDs (seed-user-*, prop-1..prop-5, listing-1..listing-5)
// and known phones (0900000001..0900000005).
// Test fixtures generate users with phone starting with '09' + timestamp digits.
//
// Order matters due to foreign key constraints.
// Seed user IDs and phones to preserve between runs
const SEED_USER_IDS = `('seed-admin-001','seed-agent-001','seed-agent-002','seed-agent-003','seed-buyer-001','seed-buyer-002','seed-seller-001','seed-seller-002')`;
const SEED_PHONES = `('+84876677771','+84900000002','+84900000003','+84900000004','+84900000005','+84900000006','+84900000007','+84900000008')`;
const SEED_LISTING_IDS = `('seed-listing-001','seed-listing-002','seed-listing-003','seed-listing-004','seed-listing-005','seed-listing-006','seed-listing-007','seed-listing-008','seed-listing-009','seed-listing-010')`;
const SEED_PROP_IDS = `('seed-prop-001','seed-prop-002','seed-prop-003','seed-prop-004','seed-prop-005','seed-prop-006','seed-prop-007','seed-prop-008','seed-prop-009','seed-prop-010')`;
const NON_SEED_USERS = `SELECT id FROM "User" WHERE id NOT IN ${SEED_USER_IDS} AND phone NOT IN ${SEED_PHONES}`;
await pool.query(`
-- Delete test-generated data in dependency order (FK-safe)
DELETE FROM "NotificationLog" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "NotificationPreference" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "Review" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "Lead" WHERE "agentId" IN (
SELECT a.id FROM "Agent" a
JOIN "User" u ON a."userId" = u.id
WHERE u.id NOT IN ${SEED_USER_IDS} AND u.phone NOT IN ${SEED_PHONES}
);
DELETE FROM "Inquiry" WHERE "listingId" NOT IN ${SEED_LISTING_IDS} OR "userId" IN (${NON_SEED_USERS});
DELETE FROM "Transaction" WHERE "buyerId" IN (${NON_SEED_USERS});
DELETE FROM "Payment" WHERE "userId" IN (${NON_SEED_USERS}) OR "orderId" IN (
SELECT id FROM "Order"
WHERE "buyerId" IN (${NON_SEED_USERS})
OR "sellerId" IN (${NON_SEED_USERS})
OR "listingId" NOT IN ${SEED_LISTING_IDS}
);
DELETE FROM "Escrow" WHERE "orderId" IN (
SELECT id FROM "Order"
WHERE "buyerId" IN (${NON_SEED_USERS})
OR "sellerId" IN (${NON_SEED_USERS})
OR "listingId" NOT IN ${SEED_LISTING_IDS}
);
DELETE FROM "Order"
WHERE "buyerId" IN (${NON_SEED_USERS})
OR "sellerId" IN (${NON_SEED_USERS})
OR "listingId" NOT IN ${SEED_LISTING_IDS};
DELETE FROM "UsageRecord" WHERE "subscriptionId" IN (
SELECT s.id FROM "Subscription" s
JOIN "User" u ON s."userId" = u.id
WHERE u.phone NOT IN ${SEED_PHONES}
);
DELETE FROM "Subscription" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "Valuation" WHERE "propertyId" NOT IN ${SEED_PROP_IDS};
DELETE FROM "Listing" WHERE id NOT IN ${SEED_LISTING_IDS};
DELETE FROM "PropertyMedia" WHERE "propertyId" NOT IN ${SEED_PROP_IDS};
DELETE FROM "Property" WHERE id NOT IN ${SEED_PROP_IDS};
DELETE FROM "Agent" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "RefreshToken" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "OAuthAccount" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "SavedSearch" WHERE "userId" IN (${NON_SEED_USERS});
DELETE FROM "User" WHERE id NOT IN ${SEED_USER_IDS} AND phone NOT IN ${SEED_PHONES};
`);
console.log('[E2E globalTeardown] Test data cleaned up successfully.\n');
} catch (err) {
console.error('[E2E globalTeardown] Cleanup error (non-fatal):', err);
} finally {
await pool.end();
}
}