From ac9aaff3cf03265420d68e0dc6c2a3378e5949ea Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 27 Aug 2024 14:05:09 -0700 Subject: [PATCH] test: refactor shell command suggestion test in test_coder.py --- tests/basic/test_coder.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py index 7790596e1..54b362a6d 100644 --- a/tests/basic/test_coder.py +++ b/tests/basic/test_coder.py @@ -750,7 +750,7 @@ two def test_suggest_shell_commands(self): with GitTemporaryDirectory(): io = InputOutput(yes=True) - coder = Coder.create(self.GPT35, None, io=io) + coder = Coder.create(self.GPT35, "diff", io=io) def mock_send(*args, **kwargs): coder.partial_response_content = """Here's a shell command to run: @@ -765,6 +765,9 @@ This command will print 'Hello, World!' to the console.""" coder.send = mock_send + # Mock the handle_shell_commands method to check if it's called + coder.handle_shell_commands = MagicMock() + # Run the coder with a message coder.run(with_message="Suggest a shell command") @@ -772,12 +775,6 @@ This command will print 'Hello, World!' to the console.""" self.assertEqual(len(coder.shell_commands), 1) self.assertEqual(coder.shell_commands[0], 'echo "Hello, World!"') - # Mock the handle_shell_commands method to check if it's called - coder.handle_shell_commands = MagicMock() - - # Run the coder again to trigger the shell command execution - coder.run(with_message="Run the suggested command") - # Check if handle_shell_commands was called with the correct argument coder.handle_shell_commands.assert_called_once_with('echo "Hello, World!"', ANY)