"""Test configs for accounting_run_calculate_nr task unit tests.""" from types import SimpleNamespace import pytest @pytest.fixture def mock_accounting_run_calculate_nr_event(): """Mock abacus event that would be passed between DAG tasks.""" return { 'abacus_event_id': 1, 'event_date': '2020-07-17 18:57:21.590679', 'event_name': 'accounting_run_calculate_nr', 'target_id': 1021, 'target_type': 'accounting_run', 'user_id': '1', 'user_type': 'orchard-suite' } @pytest.fixture def mock_accounting_run_calculate_nr_dag_run(mock_accounting_run_calculate_nr_event): """Mock accounting run calculate NR DAG run passed between DAG tasks.""" return SimpleNamespace(conf=mock_accounting_run_calculate_nr_event)