"""Test configs for adjustment_file_upload tests."""
from types import SimpleNamespace

import pytest


@pytest.fixture
def mock_adjustment_file_upload_event():
    """Mock abacus_event that triggers the adjustment_file_upload dag."""
    return {
        'abacus_event_id': 1,
        'event_date': '2022-05-13 18:57:21.590679',
        'event_name': 'adjustment_file_upload',
        'target_id': 1,
        'target_type': 'statement_period_adjustment_file',
        'created_by': 'default_user_id'
    }


@pytest.fixture
def mock_adjustment_file_upload_dag_run(mock_adjustment_file_upload_event):
    """Mock adjustment_file_upload dag run config."""
    return SimpleNamespace(conf=mock_adjustment_file_upload_event)


@pytest.fixture
def mock_adjustment_file_upload_states():
    """Mock abacus_state records for an adjustment_file_upload."""
    return [
        {
            'action_status': 'complete',
            'action_name': 'upload_file',
            'abacus_state_id': 1
        }
    ]


@pytest.fixture
def mock_valid_adjustment_file():
    """Mock xlsx file for an valid adjustment file."""
    adjustment_file_path = \
        'tests/tasks/adjustment_file_upload/test_files/valid_adjustment.xlsx'
    return adjustment_file_path


@pytest.fixture
def mock_invalid_adjustment_file():
    """Mock xlsx file for an invalid adjustment file."""
    adjustment_file_path = \
        'tests/tasks/adjustment_file_upload/test_files/invalid_adjustment.xlsx'
    return adjustment_file_path


@pytest.fixture
def mock_invalid_header_count_file():
    """Mock xlsx file for an invalid header count adjustment file."""
    adjustment_file_path = \
        'tests/tasks/adjustment_file_upload/test_files/invalid_header_count.xlsx'
    return adjustment_file_path


@pytest.fixture
def mock_empty_adjustment_file():
    """Mock xlsx file for an empty adjustment file."""
    adjustment_file_path = \
        'tests/tasks/adjustment_file_upload/test_files/no_adjustments_records.xlsx'
    return adjustment_file_path


@pytest.fixture
def mock_too_many_rows_adjustment_file():
    """Mock xlsx file for an adjustment file with too many rows."""
    adjustment_file_path = \
        'tests/tasks/adjustment_file_upload/test_files/exceed_file_row_limit.xlsx'
    return adjustment_file_path