test(mcp): Fix Mock InvocationContext to include request_state

Author: timof1308

tests/unittests/agents/test_mcp_instruction_provider.py

@@ -68,6 +68,7 @@ async def test_call_success_no_args(self):
 
     mock_invocation_context = MagicMock()
     mock_invocation_context.session.state = {}
+    mock_invocation_context.request_state = {}
     context = ReadonlyContext(mock_invocation_context)
 
     # Call
@@ -98,6 +99,7 @@ async def test_call_success_with_args(self):
 
     mock_invocation_context = MagicMock()
     mock_invocation_context.session.state = {"arg1": "value1", "arg2": "value2"}
+    mock_invocation_context.request_state = {}
     context = ReadonlyContext(mock_invocation_context)
 
     instruction = await self.provider(context)
@@ -119,6 +121,7 @@ async def test_call_prompt_not_found_in_list_prompts(self):
 
     mock_invocation_context = MagicMock()
     mock_invocation_context.session.state = {"arg1": "value1"}
+    mock_invocation_context.request_state = {}
     context = ReadonlyContext(mock_invocation_context)
 
     instruction = await self.provider(context)
@@ -137,6 +140,7 @@ async def test_call_get_prompt_returns_no_messages(self):
 
     mock_invocation_context = MagicMock()
     mock_invocation_context.session.state = {}
+    mock_invocation_context.request_state = {}
     context = ReadonlyContext(mock_invocation_context)
 
     # Call and assert
@@ -179,6 +183,7 @@ async def test_call_ignore_non_text_messages(self):
 
     mock_invocation_context = MagicMock()
     mock_invocation_context.session.state = {}
+    mock_invocation_context.request_state = {}
     context = ReadonlyContext(mock_invocation_context)
 
     # Call

tests/unittests/agents/test_readonly_context.py

@@ -25,6 +25,7 @@ def mock_invocation_context():
   mock_context.invocation_id = "test-invocation-id"
   mock_context.agent.name = "test-agent-name"
   mock_context.session.state = {"key1": "value1", "key2": "value2"}
+  mock_context.request_state = {}
   mock_context.user_id = "test-user-id"
   return mock_context