PrivacyScanData/Target/Azure/test_base_azure_1.py

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