"""Configuration for handler tests."""

from abacus_common_logic.connectors.database import db
import pytest


@pytest.fixture
def fresh_db_handlers():
    """Refresh the test database."""
    top_level_tables = (
        'run_controller',
        'run_controller_contract',
        'legacy_contract',
        'contract',
        'abacus_event',
        'abacus_state',
        'account_contract',
        'account_payee',
        'account_payment_term',
        'account_tax_info',
        'account',
        'contract_advance',
        'contract_flowthrough',
        'contract_lifecycle_schedule_detail',
        'contract_lifecycle_schedule',
        'contract_lifecycle',
        'contract_mechanical_deduction',
        'contract_party',
        'contract_term',
        'contract_term_condition',
        'contract_term_schedule',
        'contract_term_schedule_history',
        'contract_template',
        'contract_exclusion',
        'contract_reserve',
        'ledger_contract_advance_applied',
        'reference_flowthrough_calculation',
        'reference_sap_profit_center',
        'reference_signing_entity',
        'reference_payment_entity',
        'schedule',
        'statement_period',
        'reference_flowthrough_calculation',
        'reference_mechanical_rate',
        'reference_payment_type',
        'reference_transaction_type_group_transaction_type',
        'reference_transaction_type_group',
        'reference_transaction_type'
    )
    con = db.engine.connect()
    con.execute('SET FOREIGN_KEY_CHECKS = 0;')
    trans = con.begin()
    for table_name in top_level_tables:
        con.execute(f'TRUNCATE TABLE `{table_name}`')
    trans.commit()
    con.execute('SET FOREIGN_KEY_CHECKS = 1;')


@pytest.fixture(scope='function')
def create_mock_account_handlers(fresh_db_handlers):
    """Create mock account rows."""
    ADD_ACCOUNT = """
        INSERT INTO account(
            `account_id`,
            `account_name`,
            `created_by`,
            `created_at`,
            `last_modified_by`,
            `last_modified`
        ) VALUES
            (3, 'Test Account 1', '11', NOW(), '11', NOW()),
            (4, 'Test Account 2', '22', NOW(), '22', NOW())
    """
    db.engine.execute(ADD_ACCOUNT)