dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-21 21:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

fix: Catch SwitchCoder exception in test_cmd_ask

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-10 14:45:25 -07:00
parent 156f11248f
commit e7c035def4
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/basic/test_commands.py
View file

@ -876,16 +876,17 @@ class TestCommands(TestCase):
with mock.patch("aider.coders.Coder.run") as mock_run: with mock.patch("aider.coders.Coder.run") as mock_run:
mock_run.return_value = canned_reply mock_run.return_value = canned_reply
commands.cmd_ask(question) with self.assertRaises(SwitchCoder):
commands.cmd_ask(question)
mock_run.assert_called_once() mock_run.assert_called_once()
mock_run.assert_called_once_with(question) mock_run.assert_called_once_with(question)
self.assertEqual(len(coder.cur_messages), 2) self.assertEqual(len(coder.cur_messages), 2)
self.assertEqual(coder.cur_messages[0]["role"], "user") self.assertEqual(coder.cur_messages[0]["role"], "user")
self.assertEqual(coder.cur_messages[0]["content"], question) self.assertEqual(coder.cur_messages[0]["content"], question)
self.assertEqual(coder.cur_messages[1]["role"], "assistant") self.assertEqual(coder.cur_messages[1]["role"], "assistant")
self.assertEqual(coder.cur_messages[1]["content"], canned_reply) self.assertEqual(coder.cur_messages[1]["content"], canned_reply)
def test_cmd_lint_with_dirty_file(self): def test_cmd_lint_with_dirty_file(self):
with GitTemporaryDirectory() as repo_dir: with GitTemporaryDirectory() as repo_dir: