def test_get_conn_with_key_path(self, mock_connection, mock_get_client_from_auth_file): mock_sdk_client = Mock() auth_sdk_client = AzureBaseHook(mock_sdk_client).get_conn() mock_get_client_from_auth_file.assert_called_once_with( client_class=mock_sdk_client, auth_path=mock_connection.return_value.extra_dejson['key_path'] ) assert auth_sdk_client == mock_get_client_from_auth_file.return_value