karunpoudel commented on code in PR #57184:
URL: https://github.com/apache/airflow/pull/57184#discussion_r2461134645
##########
providers/postgres/tests/unit/postgres/hooks/test_postgres.py:
##########
@@ -474,19 +474,19 @@ def test_get_conn_azure_iam(self, mocker, mock_connect):
def test_get_azure_iam_token_expect_failure_on_get_token(self, mocker):
"""Test get_azure_iam_token method gets token from provided connection
id"""
- class MockAzureBaseHookWithoutGetToken:
- def __init__(self):
+ class MockAzureBaseHookOldVersion:
+ def __init__(self, sdk_client, conn_id="azure_default"):
pass
azure_conn_id = "azure_test_conn"
mock_connection_class =
mocker.patch("airflow.providers.postgres.hooks.postgres.Connection")
- mock_connection_class.get.return_value.get_hook.return_value =
MockAzureBaseHookWithoutGetToken()
+ mock_connection_class.get.return_value.get_hook =
MockAzureBaseHookOldVersion
Review Comment:
updated the docstring for the mock class and test case name accordingly
--
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]
