ashb commented on a change in pull request #6715: [AIRFLOW-5945] Make inbuilt
OperatorLinks work when using Serialization
URL: https://github.com/apache/airflow/pull/6715#discussion_r354553508
##########
File path: airflow/utils/tests.py
##########
@@ -71,15 +75,70 @@ class Dummy3TestOperator(BaseOperator):
operator_extra_links = ()
+@attr.s(auto_attribs=True)
+class CustomBaseIndexOpLink(BaseOperatorLink):
+ index: int = attr.ib()
+
+ @property
+ def name(self) -> str:
+ return 'BigQuery Console #{index}'.format(index=self.index + 1)
+
+ def get_link(self, operator, dttm):
+ ti = TaskInstance(task=operator, execution_date=dttm)
+ search_queries = ti.xcom_pull(task_ids=operator.task_id,
key='search_query')
+ if not search_queries:
+ return None
+ if len(search_queries) < self.index:
+ return None
+ search_query = search_queries[self.index]
+ return
'https://console.cloud.google.com/bigquery?j={}'.format(search_query)
+
+
+class CustomBaseOpLink(BaseOperatorLink):
Review comment:
```suggestion
class CustomOpLink(BaseOperatorLink):
```
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
