From e7c035def4f079b49f5b6a4686a84b605cac83c8 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Sat, 10 Aug 2024 14:45:25 -0700 Subject: [PATCH] fix: Catch SwitchCoder exception in test_cmd_ask --- tests/basic/test_commands.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/basic/test_commands.py b/tests/basic/test_commands.py index c981418ea..59e875b8a 100644 --- a/tests/basic/test_commands.py +++ b/tests/basic/test_commands.py @@ -876,16 +876,17 @@ class TestCommands(TestCase): with mock.patch("aider.coders.Coder.run") as mock_run: 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_with(question) - self.assertEqual(len(coder.cur_messages), 2) - self.assertEqual(coder.cur_messages[0]["role"], "user") - self.assertEqual(coder.cur_messages[0]["content"], question) - self.assertEqual(coder.cur_messages[1]["role"], "assistant") - self.assertEqual(coder.cur_messages[1]["content"], canned_reply) + self.assertEqual(len(coder.cur_messages), 2) + self.assertEqual(coder.cur_messages[0]["role"], "user") + self.assertEqual(coder.cur_messages[0]["content"], question) + self.assertEqual(coder.cur_messages[1]["role"], "assistant") + self.assertEqual(coder.cur_messages[1]["content"], canned_reply) def test_cmd_lint_with_dirty_file(self): with GitTemporaryDirectory() as repo_dir: