potiuk commented on code in PR #39513:
URL: https://github.com/apache/airflow/pull/39513#discussion_r1597505397
##########
tests/providers/google/cloud/operators/test_dataproc.py:
##########
@@ -1064,11 +1064,10 @@ def test_create_cluster_operator_extra_links(dag_maker,
create_task_instance_of_
serialized_dag = dag_maker.get_serialized_data()
deserialized_dag = SerializedDAG.from_dict(serialized_dag)
deserialized_task = deserialized_dag.task_dict[TASK_ID]
-
# Assert operator links for serialized DAG
- assert serialized_dag["dag"]["tasks"][0]["__var"]["_operator_extra_links"]
== [
- {"airflow.providers.google.cloud.links.dataproc.DataprocClusterLink":
{}}
- ]
+ deserialized_dag = SerializedDAG.deserialize_dag(serialized_dag["dag"])
Review Comment:
@Taragolis -> this is how I corrected the tests that were somewhat relying
on internals of serialization.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
