Updating tests, registering mark

astronomer · Oct 30, 2024 · 2bbe1ee · 2bbe1ee
1 parent 3ffc6c5
commit 2bbe1ee
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 13 deletions.
diff --git a/dev/dags/dags b/dev/dags/dags
diff --git a/dev/dags/example_callbacks.yml b/dev/dags/example_callbacks.yml
@@ -2,20 +2,19 @@ example_callbacks:
   default_args:
     start_date: "2024-01-01"
     on_failure_callback:
-      callback: customized.callbacks.custom_callbacks.output_message
-      param1: param1
-      param2: param2
+      callback: airflow.providers.slack.notifications.slack.send_slack_notification
+      slack_conn_id: slack_conn_id
+      text: |
+        :red_circle: Task Failed.
+        This task has failed and needs to be addressed.
+        Please remediate this issue ASAP.
+      channel: "#channel"
   schedule_interval: "@daily"
   catchup: False
   on_failure_callback:
-    callback: airflow.providers.slack.notifications.slack.send_slack_notification
-    slack_conn_id: slack_conn_id
-    text: |
-      :red_circle: Task Failed.
-      This task has failed and needs to be addressed.
-      Please remediate this issue ASAP.
-    channel: "#channel"
-    username: username
+    callback: customized.callbacks.custom_callbacks.output_message
+    param1: param1
+    param2: param2
   tasks:
     start:
       operator: airflow.operators.python.PythonOperator

diff --git a/pyproject.toml b/pyproject.toml
@@ -96,7 +96,7 @@ universal = true
 [tool.pytest.ini_options]
 filterwarnings = ["ignore::DeprecationWarning"]
 minversion = "6.0"
-markers = ["integration"]
+markers = ["integration", "callbacks"]
 
 ######################################
 # THIRD PARTY TOOLS

diff --git a/tests/test_example_dags.py b/tests/test_example_dags.py
@@ -54,9 +54,11 @@ def get_dag_bag() -> DagBag:
             print(f"Adding {dagfile} to .airflowignore")
             file.writelines([f"{dagfile}\n"])
 
+    # Print the contents of the .airflowignore file, and build the DagBag
     print(".airflowignore contents: ")
     print(AIRFLOW_IGNORE_FILE.read_text())
     db = DagBag(EXAMPLE_DAGS_DIR, include_examples=False)
+
     assert db.dags
     assert not db.import_errors
     return db