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

feat: Add test for explicit_yes_required in confirm_ask method

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-08-21 21:38:03 -07:00
parent 3c249c6b3c
commit b5c1ae69a9
1 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

32
tests/basic/test_io.py
View file

@ -60,6 +60,38 @@ class TestInputOutput(unittest.TestCase):
result = io.get_input(root, rel_fnames, addable_rel_fnames, commands)
self.assertEqual(result, "test input")
@patch('aider.io.prompt')
def test_confirm_ask_explicit_yes_required(self, mock_prompt):
io = InputOutput(pretty=False)
# Test case 1: explicit_yes_required=True, self.yes=True
io.yes = True
result = io.confirm_ask("Are you sure?", explicit_yes_required=True)
self.assertFalse(result)
mock_prompt.assert_not_called()
# Test case 2: explicit_yes_required=True, self.yes=False
io.yes = False
result = io.confirm_ask("Are you sure?", explicit_yes_required=True)
self.assertFalse(result)
mock_prompt.assert_not_called()
# Test case 3: explicit_yes_required=True, user input required
io.yes = None
mock_prompt.return_value = 'y'
result = io.confirm_ask("Are you sure?", explicit_yes_required=True)
self.assertTrue(result)
mock_prompt.assert_called_once()
# Reset mock_prompt
mock_prompt.reset_mock()
# Test case 4: explicit_yes_required=False, self.yes=True
io.yes = True
result = io.confirm_ask("Are you sure?", explicit_yes_required=False)
self.assertTrue(result)
mock_prompt.assert_not_called()
if __name__ == "__main__":
unittest.main()