From 9bc1788003985e6384c4c0e255c9e8084e244d3f Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Fri, 23 Aug 2024 16:41:52 -0700 Subject: [PATCH] feat: add test case for explicit_yes_required true and 'a' input --- tests/basic/test_io.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/basic/test_io.py b/tests/basic/test_io.py index baf7f7915..7ea7f0e83 100644 --- a/tests/basic/test_io.py +++ b/tests/basic/test_io.py @@ -134,6 +134,16 @@ 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)