Fetch tenant after session init

Author: FlorentinD

graphdatascience/session/aura_api.py

@@ -42,12 +42,14 @@ def __init__(self, client_id: str, client_secret: str, tenant_id: Optional[str]
 
         self._auth = AuraApi.Auth(oauth_url=f"{self._base_uri}/oauth/token", credentials=(client_id, client_secret))
         self._logger = logging.getLogger()
-        self._tenant_id = tenant_id if tenant_id else self._get_tenant_id()
-        self._tenant_details: Optional[TenantDetails] = None
+
         self._request_session = requests.Session()
         self._request_session.headers = {"User-agent": f"neo4j-graphdatascience-v{__version__}"}
         self._request_session.auth = self._auth
 
+        self._tenant_id = tenant_id if tenant_id else self._get_tenant_id()
+        self._tenant_details: Optional[TenantDetails] = None
+
     @staticmethod
     def extract_id(uri: str) -> str:
         host = urlparse(uri).hostname

graphdatascience/tests/unit/test_aura_api.py

@@ -3,12 +3,11 @@
 
 import pytest
 from _pytest.logging import LogCaptureFixture
-from requests import HTTPError
 from requests_mock import Mocker
 
 from graphdatascience.session import SessionMemory
 from graphdatascience.session.algorithm_category import AlgorithmCategory
-from graphdatascience.session.aura_api import AuraApi
+from graphdatascience.session.aura_api import AuraApi, AuraApiError
 from graphdatascience.session.aura_api_responses import (
     EstimationDetails,
     InstanceCreateDetails,
@@ -303,7 +302,7 @@ def test_delete_that_fails(requests_mock: Mocker) -> None:
         json={"errors": [{"message": "some failure happened", "reason": "unknown", "field": "string"}]},
     )
 
-    with pytest.raises(HTTPError, match="Internal Server Error"):
+    with pytest.raises(AuraApiError, match="Internal Server Error"):
         api.delete_instance("id0")