Return session user

Co-Authored-By: Rafal Skolasinski <rafal.skolasinski@neo4j.com>

Author: FlorentinD

graphdatascience/session/aura_api_responses.py

@@ -29,7 +29,7 @@ class SessionDetails:
     @classmethod
     def fromJson(cls, json: Dict[str, Any]) -> SessionDetails:
         expiry_date = json.get("expiry_date")
-        ttl = json.get("ttl")
+        ttl: Any | None = json.get("ttl")
 
         return cls(
             id=json["id"],

graphdatascience/session/session_info.py

@@ -24,12 +24,13 @@ class SessionInfo:
     @classmethod
     def from_session_details(cls, details: SessionDetails) -> ExtendedSessionInfo:
         return ExtendedSessionInfo(
-            details.name,
-            details.memory,
-            details.instance_id,
-            details.status,
-            details.expiry_date,
-            details.created_at,
+            name=details.name,
+            memory=details.memory,
+            instance_id=details.instance_id,
+            status=details.status,
+            expiry_date=details.expiry_date,
+            created_at=details.created_at,
+            user_id=details.user_id,
         )
 
 
@@ -39,3 +40,4 @@ class ExtendedSessionInfo(SessionInfo):
     status: str
     expiry_date: Optional[datetime]
     created_at: datetime
+    user_id: str

graphdatascience/tests/unit/test_dedicated_sessions.py

@@ -19,7 +19,7 @@
 )
 from graphdatascience.session.dbms_connection_info import DbmsConnectionInfo
 from graphdatascience.session.dedicated_sessions import DedicatedSessions
-from graphdatascience.session.session_info import SessionInfo
+from graphdatascience.session.session_info import ExtendedSessionInfo, SessionInfo
 from graphdatascience.session.session_sizes import SessionMemory, SessionMemoryValue
 
 
@@ -120,12 +120,12 @@ def list_instances(self) -> List[InstanceDetails]:
     def list_session(self, session_id: str, dbid: str) -> Optional[SessionDetails]:
         self._mimic_paused_db_behaviour(dbid)
 
-        matched_instance = self._sessions.get(session_id, None)
+        matched_session = self._sessions.get(session_id, None)
 
-        if matched_instance:
-            old_instance = matched_instance
-            self._sessions[session_id] = dataclasses.replace(old_instance, status=self._status_after_creating)
-            return old_instance
+        if matched_session:
+            old_session = matched_session
+            self._sessions[session_id] = dataclasses.replace(old_session, status=self._status_after_creating)
+            return old_session
         else:
             return None
 
@@ -243,7 +243,13 @@ def test_create_session(mocker: MockerFixture, aura_api: AuraApi) -> None:
         ),
         "session_id": "ffff0-ffff1",
     }
-    assert [i.name for i in sessions.list()] == ["my-session"]
+
+    assert len(sessions.list()) == 1
+    actual_session = sessions.list()[0]
+
+    assert isinstance(actual_session, ExtendedSessionInfo)
+    assert actual_session.name == "my-session"
+    assert actual_session.user_id == "user-1"
 
 
 def test_get_or_create(mocker: MockerFixture, aura_api: AuraApi) -> None: