mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-24 14:25:00 +00:00
refactor: Update test to simulate multi-response content formatting flow
This commit is contained in:
parent
1ab4bf14dc
commit
ad8b5c9d29
1 changed files with 11 additions and 3 deletions
|
@ -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]
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue