def __init__(self, dag: 'DAG'):
from airflow.serialization import SerializedDAG # noqa # pylint: disable=redefined-outer-name
self.dag_id = dag.dag_id
self.fileloc = dag.full_filepath
self.fileloc_hash = self.dag_fileloc_hash(self.fileloc)
self.data = SerializedDAG.to_dict(dag)
self.last_updated = timezone.utcnow()
def test_serialization(self):
"""Serialization and deserialization should work for every DAG and Operator."""
dags = collect_dags()
serialized_dags = {}
for _, v in dags.items():
dag = SerializedDAG.to_dict(v)
SerializedDAG.validate_schema(dag)
serialized_dags[v.dag_id] = dag
# Compares with the ground truth of JSON string.
self.validate_serialized_dag(serialized_dags['simple_dag'],
serialized_simple_dag_ground_truth)