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

test: update shell command handling checks in test_apply_shell_commands_with_no_suggest

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-27 13:56:59 -07:00
parent 6a55708357
commit d01e9ef7a4
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/basic/test_main.py
View file

@ -635,25 +635,25 @@ class TestMain(TestCase):
shell_md = Path("shell.md")
shell_md.write_text("```bash\ntouch no_suggest_file.txt\n```")
with patch("aider.io.InputOutput.confirm_ask") as mock_confirm_ask:
with patch("aider.coders.base_coder.Coder.handle_shell_commands") as mock_handle_shell_commands:
main(
["--apply", "shell.md", "--no-git"],
# input=DummyInput(),
# output=DummyOutput(),
input=DummyInput(),
output=DummyOutput(),
)
# Make sure confirm_ask IS called when --no-suggest-shell-commands is not used
mock_confirm_ask.assert_called_once()
# Make sure handle_shell_commands IS called when --no-suggest-shell-commands is not used
mock_handle_shell_commands.assert_called_once()
with patch("aider.io.InputOutput.confirm_ask") as mock_confirm_ask:
with patch("aider.coders.base_coder.Coder.handle_shell_commands") as mock_handle_shell_commands:
main(
["--apply", "shell.md", "--no-suggest-shell-commands", "--no-git"],
# input=DummyInput(),
# output=DummyOutput(),
input=DummyInput(),
output=DummyOutput(),
)
# Make sure confirm_ask is not called when --no-suggest-shell-commands is used
mock_confirm_ask.assert_not_called()
# Make sure handle_shell_commands is not called when --no-suggest-shell-commands is used
mock_handle_shell_commands.assert_not_called()
# Check that the file was not created in either case
self.assertFalse(Path("no_suggest_file.txt").exists())