Resolve deprecations in serialization tests (#40625)

apache · Jul 5, 2024 · 2eda737 · 2eda737
1 parent ec5da92
commit 2eda737
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/tests/deprecations_ignore.yml b/tests/deprecations_ignore.yml
@@ -27,10 +27,6 @@
 - tests/models/test_xcom.py::TestXCom::test_set_serialize_call_old_signature
 
 
-# Serialization
-- tests/serialization/test_dag_serialization.py::TestStringifiedDAGs::test_custom_dep_detector
-
-
 # WWW
 - tests/www/api/experimental/test_dag_runs_endpoint.py::TestDagRunsEndpoint::test_get_dag_runs_success
 - tests/www/api/experimental/test_dag_runs_endpoint.py::TestDagRunsEndpoint::test_get_dag_runs_success_with_capital_state_parameter

diff --git a/tests/serialization/test_dag_serialization.py b/tests/serialization/test_dag_serialization.py
@@ -1712,7 +1712,12 @@ def test_custom_dep_detector(self):
                 mode="reschedule",
             )
             CustomDepOperator(task_id="hello", bash_command="hi")
-            dag = SerializedDAG.to_dict(dag)
+            with pytest.warns(
+                RemovedInAirflow3Warning,
+                match=r"Use of a custom dependency detector is deprecated\. "
+                r"Support will be removed in a future release\.",
+            ):
+                dag = SerializedDAG.to_dict(dag)
             assert sorted(dag["dag"]["dag_dependencies"], key=lambda x: tuple(x.values())) == sorted(
                 [
                     {