From 2c805d6637099b47b519493ae4080f9e0685227d Mon Sep 17 00:00:00 2001 From: Fahad Date: Sat, 14 Jun 2025 09:34:56 +0400 Subject: [PATCH] Fixed mock comparison error --- tests/test_conversation_field_mapping.py | 31 +++++++++++++++++------- 1 file changed, 22 insertions(+), 9 deletions(-) diff --git a/tests/test_conversation_field_mapping.py b/tests/test_conversation_field_mapping.py index 834b18c..589a732 100644 --- a/tests/test_conversation_field_mapping.py +++ b/tests/test_conversation_field_mapping.py @@ -129,17 +129,30 @@ async def test_unknown_tool_defaults_to_prompt(): with patch("utils.conversation_memory.get_thread", return_value=mock_context): with patch("utils.conversation_memory.add_turn", return_value=True): with patch("utils.conversation_memory.build_conversation_history", return_value=("History", 500)): - # The autouse fixture should handle provider mocking - arguments = { - "continuation_id": "test-thread-456", - "prompt": "User input", - } + # Mock ModelContext to avoid calculation errors + with patch("utils.model_context.ModelContext") as mock_model_context_class: + mock_model_context = MagicMock() + mock_model_context.model_name = "gemini-2.5-flash-preview-05-20" + mock_model_context.calculate_token_allocation.return_value = MagicMock( + total_tokens=200000, + content_tokens=120000, + response_tokens=80000, + file_tokens=48000, + history_tokens=48000, + available_for_prompt=24000, + ) + mock_model_context_class.from_arguments.return_value = mock_model_context - enhanced_args = await reconstruct_thread_context(arguments) + arguments = { + "continuation_id": "test-thread-456", + "prompt": "User input", + } - # Should default to 'prompt' field - assert "prompt" in enhanced_args - assert "History" in enhanced_args["prompt"] + enhanced_args = await reconstruct_thread_context(arguments) + + # Should default to 'prompt' field + assert "prompt" in enhanced_args + assert "History" in enhanced_args["prompt"] @pytest.mark.asyncio