Added test for createsuperuser's handling of KeyboardInterrupt.

This commit is contained in:
Hasan Ramezani 2018-05-17 18:54:36 +04:30 committed by Tim Graham
parent 4ab1f559e8
commit e0ff88be4f
1 changed files with 15 additions and 0 deletions

View File

@ -51,6 +51,8 @@ def mock_inputs(inputs):
assert '__proxy__' not in prompt assert '__proxy__' not in prompt
response = None response = None
for key, val in inputs.items(): for key, val in inputs.items():
if val == 'KeyboardInterrupt':
raise KeyboardInterrupt
# get() fallback because sometimes 'key' is the actual # get() fallback because sometimes 'key' is the actual
# prompt rather than a shortcut name. # prompt rather than a shortcut name.
prompt_msgs = MOCK_INPUT_KEY_TO_PROMPTS.get(key, key) prompt_msgs = MOCK_INPUT_KEY_TO_PROMPTS.get(key, key)
@ -626,6 +628,19 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
test(self) test(self)
@mock_inputs({'username': 'KeyboardInterrupt'})
def test_keyboard_interrupt(self):
new_io = StringIO()
with self.assertRaises(SystemExit):
call_command(
'createsuperuser',
interactive=True,
stdin=MockTTY(),
stdout=new_io,
stderr=new_io,
)
self.assertEqual(new_io.getvalue(), '\nOperation cancelled.\n')
def test_existing_username(self): def test_existing_username(self):
"""Creation fails if the username already exists.""" """Creation fails if the username already exists."""
user = User.objects.create(username='janet') user = User.objects.create(username='janet')