Fix broken import in tests

WordPress · Nov 22, 2023 · f318d62 · f318d62
1 parent 28f43aa
commit f318d62
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/catalog/dags/database/staging_database_restore/staging_database_restore_dag.py b/catalog/dags/database/staging_database_restore/staging_database_restore_dag.py
@@ -31,6 +31,9 @@
 from airflow.sensors.external_task import ExternalTaskSensor
 from airflow.utils.state import State
 from airflow.utils.trigger_rule import TriggerRule
+from elasticsearch.recreate_staging_index.recreate_full_staging_index import (
+    DAG_ID as RECREATE_STAGING_INDEX_DAG_ID,
+)
 
 from common.constants import (
     AWS_RDS_CONN_ID,
@@ -77,7 +80,6 @@ def restore_staging_database():
     # If the `recreate_full_staging_index` DAG was manually triggered prior
     # to the database restoration starting, we should wait for it to
     # finish.
-    RECREATE_STAGING_INDEX_DAG_ID = "recreate_full_staging_index"
     wait_for_recreate_full_staging_index = ExternalTaskSensor(
         task_id="wait_for_recreate_full_staging_index",
         external_dag_id=RECREATE_STAGING_INDEX_DAG_ID,

diff --git a/catalog/tests/dags/test_dag_parsing.py b/catalog/tests/dags/test_dag_parsing.py
@@ -24,7 +24,7 @@
     "popularity/popularity_refresh_dag_factory.py",
     "data_refresh/dag_factory.py",
     "data_refresh/create_filtered_index_dag.py",
-    "elasticsearch/recreate_staging_index/recreate_full_staging_index_dag.py",
+    "elasticsearch/recreate_staging_index/recreate_full_staging_index.py",
     "oauth2/authorize_dag.py",
     "oauth2/token_refresh_dag.py",
     "database/delete_records/delete_records_dag.py",