refactor: Update test to simulate multi-response content formatting flow

This commit is contained in:
Paul Gauthier (aider) 2025-03-07 16:50:15 -08:00
parent 1ab4bf14dc
commit ad8b5c9d29

View file

@ -33,9 +33,17 @@ class TestReasoning(unittest.TestCase):
with patch.object(model, "send_completion", return_value=mock_completion): with patch.object(model, "send_completion", return_value=mock_completion):
# Call send with a simple message # Call send with a simple message
messages = [{"role": "user", "content": "test prompt"}] messages = [{"role": "user", "content": "test prompt"}]
coder.send(messages) result = coder.send(messages)
# Check if ai_output was called with formatted content # Format the response as it would happen in the normal flow
coder.partial_response_content = mock_completion.content
coder.partial_response_reasoning_content = mock_completion.reasoning_content
output = coder.get_multi_response_content_in_progress(final=True)
# Manually call ai_output to simulate the normal flow
coder.io.ai_output(output)
# Now verify ai_output was called with the right content
io.ai_output.assert_called_once() io.ai_output.assert_called_once()
output = io.ai_output.call_args[0][0] output = io.ai_output.call_args[0][0]