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

import pytest


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


@pytest.fixture
def mock_adjustment_file_import_dag_run(mock_adjustment_file_import_event):
    """Mock adjustment_file_import dag run config."""
    return SimpleNamespace(conf=mock_adjustment_file_import_event)


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