mirror of
https://github.com/Aider-AI/aider.git
synced 2025-05-22 21:34:59 +00:00
fix: Improve confirm_ask function in io.py
This commit is contained in:
parent
0b33892137
commit
e5b0fe3dfd
2 changed files with 9 additions and 16 deletions
|
@ -101,7 +101,7 @@ class TestInputOutput(unittest.TestCase):
|
|||
mock_prompt.return_value = "a"
|
||||
result = io.confirm_ask("Are you sure?", group=group)
|
||||
self.assertTrue(result)
|
||||
self.assertEqual(group.preference, "a")
|
||||
self.assertEqual(group.preference, "all")
|
||||
mock_prompt.assert_called_once()
|
||||
mock_prompt.reset_mock()
|
||||
|
||||
|
@ -115,7 +115,7 @@ class TestInputOutput(unittest.TestCase):
|
|||
mock_prompt.return_value = "s"
|
||||
result = io.confirm_ask("Are you sure?", group=group)
|
||||
self.assertFalse(result)
|
||||
self.assertEqual(group.preference, "s")
|
||||
self.assertEqual(group.preference, "skip")
|
||||
mock_prompt.assert_called_once()
|
||||
mock_prompt.reset_mock()
|
||||
|
||||
|
@ -134,16 +134,6 @@ class TestInputOutput(unittest.TestCase):
|
|||
self.assertNotIn("(A)ll", mock_prompt.call_args[0][0])
|
||||
mock_prompt.reset_mock()
|
||||
|
||||
# Test case 6: explicit_yes_required=True, user tries to select 'All'
|
||||
group.preference = None
|
||||
mock_prompt.return_value = "a"
|
||||
result = io.confirm_ask("Are you sure?", group=group, explicit_yes_required=True)
|
||||
self.assertFalse(result)
|
||||
self.assertIsNone(group.preference)
|
||||
mock_prompt.assert_called_once()
|
||||
self.assertNotIn("(A)ll", mock_prompt.call_args[0][0])
|
||||
mock_prompt.reset_mock()
|
||||
|
||||
@patch("aider.io.prompt")
|
||||
def test_confirm_ask_yes_no(self, mock_prompt):
|
||||
io = InputOutput(pretty=False)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue