const testDatabase = require('../testDatabase');

class TestHelper {
    static async setupDatabase() {
        try {
            await testDatabase.connect();
            await testDatabase.setupTables();
        } catch (error) {
            console.error('Failed to setup test database:', error);
            throw error;
        }
    }

    static async cleanupDatabase() {
        try {
            await testDatabase.cleanupTables();
            await testDatabase.disconnect();
        } catch (error) {
            console.error('Failed to cleanup test database:', error);
        }
    }

    static async clearTables() {
        try {
            await testDatabase.cleanupTables();
        } catch (error) {
            console.error('Failed to clear test tables:', error);
        }
    }

    static mockDbErrorHandler() {
        // Mock the dbErrorHandler to just execute the function
        jest.mock('../../src/middleware/errorHandler', () => ({
            dbErrorHandler: jest.fn((fn) => fn())
        }));
    }
}

module.exports = TestHelper;