moomindani commented on code in PR #69509:
URL: https://github.com/apache/airflow/pull/69509#discussion_r3583643510
##########
providers/databricks/src/airflow/providers/databricks/hooks/databricks_base.py:
##########
@@ -1167,6 +1233,9 @@ def _get_token(self, raise_error: bool = False) -> str |
None:
if self.databricks_conn.extra_dejson.get("federated_token_provider"):
self.log.debug("Using OIDC token federation with a supplied token
provider.")
return
self._get_federated_databricks_token(self._get_oidc_token_service_url())
+ if self._is_aws_federation():
Review Comment:
Confirmed — the new test correctly covers the precedence case (mocks
`_get_k8s_jwt_token`, asserts it's never called, and verifies the subject token
comes from AWS). Thanks for the quick turnaround!
--
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]
