"""Test Setup."""

import os

import pytest
from dotenv import load_dotenv

load_dotenv(override=True)
os.environ['ENVIRONMENT'] = 'TEST'
os.environ['SNOWFLAKE_DATABASE'] = 'DEV_ENGINEERING'
os.environ['SNOWFLAKE_SCHEMA'] = 'TEST'


@pytest.fixture
def mock_snapshot_event():
    """Mock "snapshot_account_contracts" abacus_event."""
    return {
        'abacus_event_id': 1,
        'target_type': 'statement_period',
        'target_id': '318',
        'event_name': 'snapshot_account_contracts',
    }


@pytest.fixture
def mock_closed_statement_period():
    """Mock closed statement period data."""
    return {
        'statement_period_id': 318,
        'statement_period_name': 'June 2025',
        'statement_period_status': 'closed',
        'statement_month': 6,
        'statement_year': 2025,
        'closed_date': '2025-07-31',
        'exchange_rates_delivered': False,
    }


@pytest.fixture
def mock_open_statement_period():
    """Mock open statement period data."""
    return {
        'statement_period_id': 319,
        'statement_period_name': 'July 2025',
        'statement_period_status': 'open',
        'statement_month': 7,
        'statement_year': 2025,
        'closed_date': None,
        'exchange_rates_delivered': False,
    }