"""Test Setup."""

import pytest
from dotenv import load_dotenv

load_dotenv()


@pytest.fixture
def mock_file_upload():
    """Mock file_upload record."""
    return {
        'file_upload_id': 1,
        'file_upload_config_id': 1,
        'file_key': 'c5a23a6e-7dc0-4a14-b4df-817b1651ca68',
        'original_file_name': 'Test File',
        'file_size_bytes': 123,
        'file_type': 'xlsx',
        'mime_type': None,
        's3_bucket': 'test-abacus-adjustments',
        's3_key': 'uploads/2025/11/c5a23a6e-7dc0-4a14-b4df-817b1651ca68.csv',
        'md5sum': None,
        'upload_status': 'complete',
        'multipart_upload_id': None,
        'total_parts': 1,
        'upload_metadata': None,
        'error_message': None,
        'completed_at': '2025-12-02T14:44:33.000000',
        'expires_at': None,
    }


@pytest.fixture
def mock_file_upload_config():
    """Mock file_upload_config record."""
    return {
        'file_upload_config_id': 1,
        'upload_type': 'adjustments',
        's3_key_template': 'uploads/{year}/{month}/{file_key}.{ext}',
        'allowed_file_types': ['csv', 'xlsx'],
        'max_file_size_bytes': 104857600,
        'multipart_threshold_bytes': 52428800,
        'min_multipart_chunk_size_bytes': 10485760,
        'description': 'Adjustments upload configuration.',
        'event_name': None,
    }


@pytest.fixture
def mock_abacus_event():
    """Mock abacus_event record."""
    return {
        'abacus_event_id': 1,
        'statement_period_id': 322,
        'target_id': 1,
        'target_type': 'file_upload',
        'event_date': '2025-12-04T21:11:38.963711',
        'event_name': 'test_event',
        'previous_abacus_event_id': None,
        'rolled_back_at': None,
        'created_by': 'test_user',
    }