Skip to content

Commit

Permalink
fixup! [AIRFLOW-6347] BugFix: Can't get task logs when serialization …
Browse files Browse the repository at this point in the history
…is enabled (apache#7092)
  • Loading branch information
kaxil committed Jan 10, 2020
1 parent 2200f99 commit 4dbe08f
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/serialization/test_dag_serialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand All @@ -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))
Expand Down

0 comments on commit 4dbe08f

Please sign in to comment.