Skip to content

Commit efeb8aa

Browse files
committed
fix: Fix tests after merge
1 parent 092a72b commit efeb8aa

4 files changed

Lines changed: 6 additions & 8 deletions

File tree

dags/fetch_notices_by_date.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from ted_sws.event_manager.services.log import log_error
1919

2020
DAG_ID = "fetch_notices_by_date"
21-
DAG_NAME = "Fetch notices by date"
21+
FETCHER_DAG_NAME = "Fetch notices by date"
2222
BATCH_SIZE = 2000
2323
WILD_CARD_DAG_KEY = "wild_card"
2424
TRIGGER_COMPLETE_WORKFLOW_DAG_KEY = "trigger_complete_workflow"
@@ -31,7 +31,7 @@
3131

3232
@dag(default_args=DEFAULT_DAG_ARGUMENTS,
3333
dag_id=DAG_ID,
34-
dag_display_name=DAG_NAME,
34+
dag_display_name=FETCHER_DAG_NAME,
3535
catchup=False,
3636
schedule=CronTriggerTimetable(
3737
cron=config.SCHEDULE_DAG_FETCH,

dags/reprocess_notices_from_backlog_by_id.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
11
from airflow.decorators import dag, task
22
from airflow.models import Param
33

4-
from dags import DEFAULT_DAG_ARGUMENTS
4+
from dags import DEFAULT_DAG_ARGUMENTS, NOTICE_NORMALISATION_PIPELINE_TASK_ID
55
from dags.dags_utils import push_dag_downstream, get_dag_param
66
from dags.operators.DagBatchPipelineOperator import NOTICE_IDS_KEY, TriggerNoticeBatchPipelineOperator
7-
from dags.notice_processing_pipeline import NOTICE_NORMALISATION_PIPELINE_TASK_ID
87
from ted_sws.event_manager.adapters.event_log_decorator import event_log
98
from ted_sws.event_manager.model.event_message import TechnicalEventMessage, EventMessageMetadata, EventMessageProcessType
109

dags/reprocess_notices_from_backlog_by_status.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
from airflow.decorators import dag, task
22
from airflow.models import Param
33

4-
from dags import DEFAULT_DAG_ARGUMENTS
4+
from dags import DEFAULT_DAG_ARGUMENTS, NOTICE_NORMALISATION_PIPELINE_TASK_ID
55
from dags.dags_utils import push_dag_downstream, get_dag_param
6-
from dags.notice_processing_pipeline import NOTICE_NORMALISATION_PIPELINE_TASK_ID
76
from dags.operators.DagBatchPipelineOperator import NOTICE_IDS_KEY, TriggerNoticeBatchPipelineOperator
87
from dags.pipelines.notice_selectors_pipelines import notice_ids_selector_by_status
98
from ted_sws.core.model.notice import NoticeStatus

tests/unit/event_manager/services/test_logger_from_context.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def test_get_dag_run_id_from_dag_context(fake_dag_context):
5656

5757

5858
def test_handle_event_message_metadata_dag_context(fake_dag_context):
59-
process_name = "DAG_NAME"
59+
process_name = "FETCHER_DAG_NAME"
6060
process_id = "DAG_RUN_ID"
6161

6262
metadata = handle_event_message_metadata_dag_context(ps_name=process_name, ps_id=process_id, ps_context={})
@@ -70,7 +70,7 @@ def test_handle_event_message_metadata_dag_context(fake_dag_context):
7070

7171

7272
def test_handle_event_message_metadata_context():
73-
process_name = "DAG_NAME"
73+
process_name = "FETCHER_DAG_NAME"
7474
process_id = "DAG_RUN_ID"
7575
process_type = EventMessageProcessType.CLI
7676

0 commit comments

Comments
 (0)