pg-god

import { createDatabase, dropDatabase, DbCredential } from 'pg-god'; async function main() { const dbConfig = { databaseName: 'my_app_test_db' }; const credentials: Partial<DbCredential> = { user: process.env.PG_USER ?? 'postgres', password: process.env.PG_PASSWORD ?? '', host: process.env.PG_HOST ?? 'localhost', port: parseInt(process.env.PG_PORT ?? '5432', 10), database: process.env.PG_INITIAL_DB ?? 'postgres', // Initial database to connect to }; try { console.log(`Attempting to create database: ${dbConfig.databaseName}`); await createDatabase(dbConfig, credentials); console.log(`Database '${dbConfig.databaseName}' created successfully.`); // --- Your application logic or tests would go here --- console.log(`Attempting to drop database: ${dbConfig.databaseName}`); // dropConnections: true is default since v1.0.6, but explicit for clarity await dropDatabase({ ...dbConfig, dropConnections: true }, credentials); console.log(`Database '${dbConfig.databaseName}' dropped successfully.`); } catch (error) { console.error('An error occurred during database operation:', error); // Attempt to clean up even if an error occurred after creation try { await dropDatabase({ ...dbConfig, dropConnections: true, errorIfNonExist: false }, credentials); console.log(`Attempted cleanup for database '${dbConfig.databaseName}'.`); } catch (cleanupError) { console.warn(`Failed to clean up database '${dbConfig.databaseName}':`, cleanupError); } process.exit(1); } } main();