fix(BA-5650-G): align remaining test fixtures and adapter call sites

- Tests for SessionEnqueueData / KernelEnqueueData / TerminatingSessionData
  now use ``main_access_key`` and ``owner_id`` (renamed in slice F/G).
- ``SessionAdapter._session_data_to_node`` reads ``data.owner_id`` and
  drops the obsolete ``data.access_key``; the SessionMetadata GQL DTO's
  ``access_key`` field is now an empty string until the read-time
  resolver lands.
- ``cache_invalidation`` uses ``info.access_key`` (the field name on
  SessionTransitionInfo).
- ``ShutdownServiceAction``/``GetContainerLogsAction``/``RenameSessionAction``
  no longer accept ``owner_access_key``; drop the kwarg in the GQL adapter.
- ``ModelServingRepository.get_session_by_id`` drops the now-removed
  positional ``owner_access_key`` arg from ``SessionRow.get_session``.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: jopemachine

src/ai/backend/manager/api/adapters/session.py

@@ -854,7 +854,6 @@ async def shutdown_service(
         """Shut down a service in a session."""
         action = ShutdownServiceAction(
             session_name=str(session_id),
-            owner_access_key=AccessKey(access_key),
             service_name=input.service,
         )
         await self._processors.session.shutdown_service.wait_for_complete(action)
@@ -872,7 +871,6 @@ async def get_logs(
         """Get container logs for a session."""
         action = GetContainerLogsAction(
             session_name=str(session_id),
-            owner_access_key=AccessKey(access_key),
             kernel_id=KernelId(kernel_id) if kernel_id else None,
         )
         result = await self._processors.session.get_container_logs.wait_for_complete(action)
@@ -894,7 +892,6 @@ async def update(
             action = RenameSessionAction(
                 session_name=str(session_id),
                 new_name=input.name,
-                owner_access_key=AccessKey(access_key),
             )
             result = await self._processors.session.rename_session.wait_for_complete(action)
             return UpdateSessionPayload(session=self._session_data_to_node(result.session_data))

src/ai/backend/manager/repositories/model_serving/repository.py

@@ -738,7 +738,7 @@ async def get_session_by_id(
         async with self._db.begin_readonly_session_read_committed() as session:
             try:
                 return await SessionRow.get_session(
-                    session, session_id, None, kernel_loading_strategy=kernel_loading_strategy
+                    session, session_id, kernel_loading_strategy=kernel_loading_strategy
                 )
             except NoResultFound:
                 return None

src/ai/backend/manager/repositories/scheduler/db_source/db_source.py

@@ -1146,9 +1146,7 @@ async def get_terminating_sessions_by_ids(
                     for kernel in session_row.kernels
                 ]
 
-                owner_main_ak = (
-                    session_row.user.main_access_key if session_row.user else None
-                )
+                owner_main_ak = session_row.user.main_access_key if session_row.user else None
                 terminating_sessions.append(
                     TerminatingSessionData(
                         session_id=session_row.id,
@@ -1848,9 +1846,7 @@ async def allocate_sessions(
             session_ids = {alloc.session_id for alloc in allocation_batch.allocations}
             if session_ids:
                 query = (
-                    sa.select(
-                        SessionRow.id, SessionRow.creation_id, UserRow.main_access_key
-                    )
+                    sa.select(SessionRow.id, SessionRow.creation_id, UserRow.main_access_key)
                     .select_from(SessionRow)
                     .join(UserRow, SessionRow.user_uuid == UserRow.uuid)
                     .where(SessionRow.id.in_(session_ids))

src/ai/backend/manager/sokovan/scheduler/post_processors/cache_invalidation.py

@@ -27,11 +27,11 @@ async def execute(self, context: PostProcessorContext) -> None:
         affected_keys: set[AccessKey] = set()
 
         for info in context.result.successes:
-            if info.main_access_key:
-                affected_keys.add(info.main_access_key)
+            if info.access_key:
+                affected_keys.add(info.access_key)
         for info in context.result.failures:
-            if info.main_access_key:
-                affected_keys.add(info.main_access_key)
+            if info.access_key:
+                affected_keys.add(info.access_key)
 
         if affected_keys:
             await self._repository.invalidate_kernel_related_cache(list(affected_keys))

tests/unit/manager/repositories/scheduler/test_scheduling_history_recording.py

@@ -316,8 +316,8 @@ async def test_enqueue_session_creates_scheduling_history(
             id=session_id,
             creation_id=creation_id,
             name=f"test-session-{uuid.uuid4().hex[:8]}",
-            access_key=test_access_key,
-            user_uuid=test_user_uuid,
+            main_access_key=test_access_key,
+            owner_id=test_user_uuid,
             group_id=test_group_id,
             domain_name=test_domain_name,
             scaling_group_name=test_scaling_group_name,
@@ -354,8 +354,8 @@ async def test_enqueue_session_creates_scheduling_history(
                     scaling_group=test_scaling_group_name,
                     domain_name=test_domain_name,
                     group_id=test_group_id,
-                    user_uuid=test_user_uuid,
-                    access_key=test_access_key,
+                    owner_id=test_user_uuid,
+                    main_access_key=test_access_key,
                     image="python:3.8",
                     architecture="x86_64",
                     registry="docker.io",

tests/unit/manager/sokovan/scheduler/terminator/conftest.py

@@ -137,7 +137,7 @@ def _create_terminating_session_data(
 
     return TerminatingSessionData(
         session_id=session_id or SessionId(uuid4()),
-        access_key=AccessKey("test-key"),
+        main_access_key=AccessKey("test-key"),
         creation_id=str(uuid4()),
         status=SessionStatus.TERMINATING,
         status_info=status_info,

tests/unit/manager/sokovan/scheduler/test_terminate_sessions.py

@@ -126,7 +126,7 @@ async def test_terminate_sessions_single_success(
 
         terminating_session = TerminatingSessionData(
             session_id=session_id,
-            access_key=AccessKey("test-key"),
+            main_access_key=AccessKey("test-key"),
             creation_id="test-creation",
             status=SessionStatus.TERMINATING,
             status_info="USER_REQUESTED",
@@ -181,7 +181,7 @@ async def test_terminate_sessions_multiple_kernels(
 
         terminating_session = TerminatingSessionData(
             session_id=session_id,
-            access_key=AccessKey("test-key"),
+            main_access_key=AccessKey("test-key"),
             creation_id="test-creation",
             status=SessionStatus.TERMINATING,
             status_info="FORCED_TERMINATION",
@@ -233,7 +233,7 @@ async def test_terminate_sessions_partial_failure(
 
         terminating_session = TerminatingSessionData(
             session_id=session_id,
-            access_key=AccessKey("test-key"),
+            main_access_key=AccessKey("test-key"),
             creation_id="test-creation",
             status=SessionStatus.TERMINATING,
             status_info="TEST_PARTIAL",
@@ -310,7 +310,7 @@ async def test_terminate_sessions_concurrent_execution(
             sessions.append(
                 TerminatingSessionData(
                     session_id=session_id,
-                    access_key=AccessKey(f"key-{i}"),
+                    main_access_key=AccessKey(f"key-{i}"),
                     creation_id=f"creation-{i}",
                     status=SessionStatus.TERMINATING,
                     status_info="BATCH_TERMINATION",
@@ -364,7 +364,7 @@ async def test_terminate_sessions_empty_kernel_list(
 
         terminating_session = TerminatingSessionData(
             session_id=session_id,
-            access_key=AccessKey("test-key"),
+            main_access_key=AccessKey("test-key"),
             creation_id="test-creation",
             status=SessionStatus.TERMINATING,
             status_info="NO_KERNELS",