mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-08-04 19:08:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fixed #36016 -- Prevented traceback when quitting makemigrations with Ctrl-C.

Browse source
This commit is contained in:
Aman Sharma 2024-12-17 14:00:33 +05:30 • committed by Sarah Boyce
parent 3ee4c6a27a
commit f05edb2b43
2 changed files with 26 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/migrations/test_questioner.py
View file

@ -85,8 +85,9 @@ class QuestionerHelperMethodsTests(SimpleTestCase):
@mock.patch("builtins.input", side_effect=[KeyboardInterrupt()])
def test_questioner_no_default_keyboard_interrupt(self, mock_input):
with self.assertRaises(KeyboardInterrupt):
with self.assertRaises(SystemExit):
self.questioner._ask_default()
self.assertIn("Cancelled.\n", self.prompt.getvalue())
@mock.patch("builtins.input", side_effect=["", "n"])
def test_questioner_no_default_no_user_entry_boolean(self, mock_input):
@ -105,3 +106,18 @@ class QuestionerHelperMethodsTests(SimpleTestCase):
expected_msg = f"{question}\n" f" 1) a\n" f" 2) b\n" f" 3) c\n"
self.assertIn(expected_msg, self.prompt.getvalue())
self.assertEqual(value, 1)
@mock.patch("builtins.input", side_effect=[KeyboardInterrupt()])
def test_questioner_no_choice_keyboard_interrupt(self, mock_input):
question = "Make a choice:"
with self.assertRaises(SystemExit):
self.questioner._choice_input(question, choices="abc")
expected_msg = (
f"{question}\n"
f" 1) a\n"
f" 2) b\n"
f" 3) c\n"
f"Select an option: \n"
f"Cancelled.\n"
)
self.assertIn(expected_msg, self.prompt.getvalue())