encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: pass context to client inceptors

Browse Source 
PiperOrigin-RevId: 840918430
This commit is contained in:
Google Team Member
2025-12-05 16:26:20 -08:00
committed by Copybara-Service
parent f22bac0b20
commit 143ad44f8c
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/agents/test_remote_a2a_agent.py
+6
View File
@@ -23,6 +23,7 @@ from unittest.mock import patch
from a2a.client.client import ClientConfig
from a2a.client.client import Consumer
from a2a.client.client_factory import ClientFactory
from a2a.client.middleware import ClientCallContext
from a2a.types import AgentCapabilities
from a2a.types import AgentCard
from a2a.types import AgentSkill
@@ -1429,6 +1430,7 @@ class TestRemoteA2aAgentExecution:
self.mock_session = Mock(spec=Session)
self.mock_session.id = "session-123"
self.mock_session.events = []
self.mock_session.state = {}
self.mock_context = Mock(spec=InvocationContext)
self.mock_context.session = self.mock_session
@@ -1682,6 +1684,7 @@ class TestRemoteA2aAgentExecution:
mock_a2a_client.send_message.assert_called_once_with(
request=mock_message,
request_metadata=request_metadata,
context=ClientCallContext(state=self.mock_session.state),
)
@@ -1703,6 +1706,7 @@ class TestRemoteA2aAgentExecutionFromFactory:
self.mock_session = Mock(spec=Session)
self.mock_session.id = "session-123"
self.mock_session.events = []
self.mock_session.state = {}
self.mock_context = Mock(spec=InvocationContext)
self.mock_context.session = self.mock_session
@@ -1989,6 +1993,7 @@ class TestRemoteA2aAgentIntegration:
mock_session = Mock(spec=Session)
mock_session.id = "session-123"
mock_session.events = [mock_event]
mock_session.state = {}
mock_context = Mock(spec=InvocationContext)
mock_context.session = mock_session
@@ -2084,6 +2089,7 @@ class TestRemoteA2aAgentIntegration:
mock_session = Mock(spec=Session)
mock_session.id = "session-123"
mock_session.events = [mock_event]
mock_session.state = {}
mock_context = Mock(spec=InvocationContext)
mock_context.session = mock_session