test(mcp): Fix Mock InvocationContext to include request_state

Author: timof1308

tests/unittests/agents/test_mcp_instruction_provider.py

@@ -67,6 +67,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
@@ -97,6 +98,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)
@@ -118,6 +120,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)
@@ -136,6 +139,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
@@ -178,6 +182,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