"""Test config for the `adjustment_file_generate` tests."""

from types import SimpleNamespace

import pytest


@pytest.fixture
def mock_adjustment_file_generate_event():
    """Mock the Abacus event that triggers the `adjustment_file_generate` DAG."""
    return {
        'abacus_event_id': 1,
        'event_date': '2022-05-13 18:57:21.590679',
        'event_name': 'generate_flowthrough_adjustments',
        'target_id': 1,
        'target_type': 'statement_period_adjustment_file',
        'created_by': 'default_user_id'
    }


@pytest.fixture
def mock_adjustment_file_generate_dag_run(mock_adjustment_file_generate_event):
    """Mock the `adjustment_file_generate` DAG run config."""
    return SimpleNamespace(conf=mock_adjustment_file_generate_event)


@pytest.fixture
def mock_adjustment_file_generate_states():
    """Mock the Abacus state records for the `adjustment_file_generate` DAG."""
    return [
        {
            'action_status': 'complete',
            'action_name': 'upload_file',
            'abacus_state_id': 1
        }
    ]