mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-25 06:44:59 +00:00
fix: Add mdstream mock to resolve WholeFileCoder test failure
This commit is contained in:
parent
c53833072f
commit
f8c069132e
1 changed files with 3 additions and 0 deletions
|
@ -121,6 +121,9 @@ class TestReasoning(unittest.TestCase):
|
||||||
|
|
||||||
# Mock the model's send_completion to return the hash and completion
|
# Mock the model's send_completion to return the hash and completion
|
||||||
with patch.object(model, "send_completion", return_value=(mock_hash, chunks)):
|
with patch.object(model, "send_completion", return_value=(mock_hash, chunks)):
|
||||||
|
# Set mdstream directly on the coder object
|
||||||
|
coder.mdstream = mock_mdstream
|
||||||
|
|
||||||
# Call send with a simple message
|
# Call send with a simple message
|
||||||
messages = [{"role": "user", "content": "test prompt"}]
|
messages = [{"role": "user", "content": "test prompt"}]
|
||||||
list(coder.send(messages))
|
list(coder.send(messages))
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue