dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix: Improve test_commands.py to capture and verify diff output

Browse source
This commit is contained in:
Paul Gauthier 2024-08-10 08:55:28 -07:00 • committed by Paul Gauthier (aider)
parent 6d64e88478
commit edf98d5138
1 changed files with 22 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

34
tests/basic/test_commands.py
View file

@ -823,21 +823,31 @@ class TestCommands(TestCase):
# Run cmd_commit # Run cmd_commit
commands.cmd_commit() commands.cmd_commit()
# Modify the file again # Capture the output of cmd_diff
file_path.write_text("Further modified content") with mock.patch("builtins.print") as mock_print:
commands.cmd_diff("")
# Run cmd_commit again # Check if the diff output is correct
commands.cmd_commit() mock_print.assert_called_with(mock.ANY)
diff_output = mock_print.call_args[0][0]
self.assertIn("-Initial content", diff_output)
self.assertIn("+Modified content", diff_output)
# Capture the output of cmd_diff # Modify the file again
with mock.patch("builtins.print") as mock_print: file_path.write_text("Further modified content")
commands.cmd_diff("")
# Check if the diff output is correct # Run cmd_commit again
mock_print.assert_called_with(mock.ANY) commands.cmd_commit()
diff_output = mock_print.call_args[0][0]
self.assertIn("-Modified content", diff_output) # Capture the output of cmd_diff
self.assertIn("+Further modified content", diff_output) with mock.patch("builtins.print") as mock_print:
commands.cmd_diff("")
# Check if the diff output is correct
mock_print.assert_called_with(mock.ANY)
diff_output = mock_print.call_args[0][0]
self.assertIn("-Modified content", diff_output)
self.assertIn("+Further modified content", diff_output)
def test_cmd_ask(self): def test_cmd_ask(self):
io = InputOutput(pretty=False, yes=True) io = InputOutput(pretty=False, yes=True)