From c5c18c54fa83463bc953249dc28edcbf7179da17 Mon Sep 17 00:00:00 2001 From: Alex Ott Date: Wed, 29 Dec 2021 12:55:59 +0100 Subject: [PATCH] Databricks: fix verification of Managed Identity (#20550) --- airflow/providers/databricks/hooks/databricks.py | 2 +- tests/providers/databricks/hooks/test_databricks.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/airflow/providers/databricks/hooks/databricks.py b/airflow/providers/databricks/hooks/databricks.py index e07299c2259b9..7b80e87019349 100644 --- a/airflow/providers/databricks/hooks/databricks.py +++ b/airflow/providers/databricks/hooks/databricks.py @@ -284,7 +284,7 @@ def _check_azure_metadata_service() -> None: """ try: jsn = requests.get( - AZURE_METADATA_SERVICE_TOKEN_URL, + AZURE_METADATA_SERVICE_INSTANCE_URL, params={"api-version": "2021-02-01"}, headers={"Metadata": "true"}, timeout=2, diff --git a/tests/providers/databricks/hooks/test_databricks.py b/tests/providers/databricks/hooks/test_databricks.py index ea688e87dc955..951673a38d700 100644 --- a/tests/providers/databricks/hooks/test_databricks.py +++ b/tests/providers/databricks/hooks/test_databricks.py @@ -32,7 +32,7 @@ from airflow.providers.databricks.hooks.databricks import ( AZURE_DEFAULT_AD_ENDPOINT, AZURE_MANAGEMENT_ENDPOINT, - AZURE_METADATA_SERVICE_TOKEN_URL, + AZURE_METADATA_SERVICE_INSTANCE_URL, AZURE_TOKEN_SERVICE_URL, DEFAULT_DATABRICKS_SCOPE, SUBMIT_RUN_ENDPOINT, @@ -808,7 +808,7 @@ def test_submit_run(self, mock_requests): run_id = self.hook.submit_run(data) ad_call_args = mock_requests.method_calls[0] - assert ad_call_args[1][0] == AZURE_METADATA_SERVICE_TOKEN_URL + assert ad_call_args[1][0] == AZURE_METADATA_SERVICE_INSTANCE_URL assert ad_call_args[2]['params']['api-version'] > '2018-02-01' assert ad_call_args[2]['headers']['Metadata'] == 'true'