From 4dbe08f42ba9f5ba59f1e83ddbca02cd4d14a243 Mon Sep 17 00:00:00 2001 From: kaxil Date: Fri, 10 Jan 2020 01:25:35 +0000 Subject: [PATCH] fixup! [AIRFLOW-6347] BugFix: Can't get task logs when serialization is enabled (#7092) --- tests/serialization/test_dag_serialization.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/serialization/test_dag_serialization.py b/tests/serialization/test_dag_serialization.py index a6ad87c08366cf..558253aabd4150 100644 --- a/tests/serialization/test_dag_serialization.py +++ b/tests/serialization/test_dag_serialization.py @@ -275,7 +275,7 @@ def validate_deserialized_dag(self, serialized_dag, dag): checking fields between Serialized Dags & non-Serialized Dags """ fields_to_check = [ - "task_ids", "params", "fileloc", "max_active_runs", "concurrency", + "params", "fileloc", "max_active_runs", "concurrency", "is_paused_upon_creation", "doc_md", "safe_dag_id", "is_subdag", "catchup", "description", "start_date", "end_date", "parent_dag", "template_searchpath" @@ -285,6 +285,10 @@ def validate_deserialized_dag(self, serialized_dag, dag): for field in fields_to_check: self.assertEqual(getattr(serialized_dag, field), getattr(dag, field)) + self.assertEqual( + sorted(serialized_dag.task_ids), + sorted([str(task) for task in dag.task_ids])) + def validate_deserialized_task(self, task, task_type, ui_color, ui_fgcolor): """Verify non-airflow operators are casted to BaseOperator.""" self.assertTrue(isinstance(task, SerializedBaseOperator))