test(a2a): update executor tests to reflect direct session_service calls

Author: nickchecan

tests/unittests/a2a/executor/test_a2a_agent_executor.py

@@ -224,17 +224,18 @@ async def mock_run_async(**kwargs):
 
   @pytest.mark.asyncio
   async def test_prepare_session_new_session(self):
-    """Test session preparation delegates to runner._get_or_create_session."""
+    """Test session preparation creates a new session when none is found."""
     run_args = AgentRunRequest(
         user_id="test-user",
-        session_id=None,
+        session_id="new-session-id",
         new_message=Mock(spec=Content),
         run_config=Mock(spec=RunConfig),
     )
 
     mock_session = Mock()
     mock_session.id = "new-session-id"
-    self.mock_runner._get_or_create_session = AsyncMock(
+    self.mock_runner.session_service.get_session = AsyncMock(return_value=None)
+    self.mock_runner.session_service.create_session = AsyncMock(
         return_value=mock_session
     )
 
@@ -243,17 +244,23 @@ async def test_prepare_session_new_session(self):
         self.mock_context, run_args, self.mock_runner
     )
 
-    # Verify session was returned and run_request updated
+    # Verify session was created and run_request updated
     assert result == mock_session
     assert run_args.session_id == "new-session-id"
-    self.mock_runner._get_or_create_session.assert_called_once_with(
+    self.mock_runner.session_service.get_session.assert_called_once_with(
+        app_name="test-app",
         user_id="test-user",
-        session_id=None,
+        session_id="new-session-id",
+    )
+    self.mock_runner.session_service.create_session.assert_called_once_with(
+        app_name="test-app",
+        user_id="test-user",
+        session_id="new-session-id",
     )
 
   @pytest.mark.asyncio
   async def test_prepare_session_existing_session(self):
-    """Test session preparation when session exists."""
+    """Test session preparation returns existing session without creating one."""
     run_args = AgentRunRequest(
         user_id="test-user",
         session_id="existing-session",
@@ -263,21 +270,24 @@ async def test_prepare_session_existing_session(self):
 
     mock_session = Mock()
     mock_session.id = "existing-session"
-    self.mock_runner._get_or_create_session = AsyncMock(
+    self.mock_runner.session_service.get_session = AsyncMock(
         return_value=mock_session
     )
+    self.mock_runner.session_service.create_session = AsyncMock()
 
     # Execute
     result = await self.executor._prepare_session(
         self.mock_context, run_args, self.mock_runner
     )
 
-    # Verify existing session was returned
+    # Verify existing session was returned without creating a new one
     assert result == mock_session
-    self.mock_runner._get_or_create_session.assert_called_once_with(
+    self.mock_runner.session_service.get_session.assert_called_once_with(
+        app_name="test-app",
         user_id="test-user",
         session_id="existing-session",
     )
+    self.mock_runner.session_service.create_session.assert_not_called()
 
   def test_constructor_with_callable_runner(self):
     """Test constructor with callable runner."""

tests/unittests/a2a/executor/test_a2a_agent_executor_impl.py

@@ -97,10 +97,10 @@ async def test_execute_success_new_task(self):
         new_message=Mock(spec=Content),
         run_config=Mock(spec=RunConfig),
     )
-    # Mock _get_or_create_session
+    # Mock session lookup returning existing session
     mock_session = Mock()
     mock_session.id = "test-session"
-    self.mock_runner._get_or_create_session = AsyncMock(
+    self.mock_runner.session_service.get_session = AsyncMock(
         return_value=mock_session
     )
 
@@ -200,10 +200,10 @@ async def test_execute_existing_task(self):
         run_config=Mock(spec=RunConfig),
     )
 
-    # Mock _get_or_create_session
+    # Mock session lookup returning existing session
     mock_session = Mock()
     mock_session.id = "test-session"
-    self.mock_runner._get_or_create_session = AsyncMock(
+    self.mock_runner.session_service.get_session = AsyncMock(
         return_value=mock_session
     )
 
@@ -616,7 +616,7 @@ async def test_execute_missing_user_input(self, mock_handle_user_input):
     )
     mock_handle_user_input.return_value = missing_event
 
-    self.mock_runner._get_or_create_session = AsyncMock(
+    self.mock_runner.session_service.get_session = AsyncMock(
         return_value=Mock(id="test-session")
     )
     self.mock_request_converter.return_value = AgentRunRequest(
@@ -638,28 +638,33 @@ async def test_execute_missing_user_input(self, mock_handle_user_input):
 
   @pytest.mark.asyncio
   async def test_resolve_session_creates_new_session(self):
-    """Test that _resolve_session delegates to runner._get_or_create_session."""
+    """Test that _resolve_session creates a session when none is found."""
     new_session = Mock()
     new_session.id = "new-session-id"
-    self.mock_runner._get_or_create_session = AsyncMock(
+    self.mock_runner.session_service.get_session = AsyncMock(return_value=None)
+    self.mock_runner.session_service.create_session = AsyncMock(
         return_value=new_session
     )
 
     run_request = AgentRunRequest(
         user_id="test-user",
-        session_id="old-session-id",
+        session_id="new-session-id",
         new_message=Mock(spec=Content),
         run_config=Mock(spec=RunConfig),
     )
 
     await self.executor._resolve_session(run_request, self.mock_runner)
 
-    self.mock_runner._get_or_create_session.assert_called_once_with(
+    self.mock_runner.session_service.get_session.assert_called_once_with(
+        app_name="test-app",
         user_id="test-user",
-        session_id="old-session-id",
-        get_session_config=GetSessionConfig(
-            num_recent_events=0, after_timestamp=None
-        ),
+        session_id="new-session-id",
+        config=GetSessionConfig(num_recent_events=0, after_timestamp=None),
+    )
+    self.mock_runner.session_service.create_session.assert_called_once_with(
+        app_name="test-app",
+        user_id="test-user",
+        session_id="new-session-id",
     )
     assert run_request.session_id == "new-session-id"
 

tests/unittests/a2a/utils/test_agent_to_a2a.py

@@ -327,7 +327,6 @@ async def test_create_runner_function_creates_runner_correctly(
         session_service=mock_runner_class.call_args[1]["session_service"],
         memory_service=mock_runner_class.call_args[1]["memory_service"],
         credential_service=mock_runner_class.call_args[1]["credential_service"],
-        auto_create_session=True,
     )
 
     # Verify the services are of the correct types
@@ -392,7 +391,6 @@ async def test_create_runner_function_with_agent_without_name(
         session_service=mock_runner_class.call_args[1]["session_service"],
         memory_service=mock_runner_class.call_args[1]["memory_service"],
         credential_service=mock_runner_class.call_args[1]["credential_service"],
-        auto_create_session=True,
     )
 
   @patch("google.adk.a2a.utils.agent_to_a2a.A2aAgentExecutor")