summaryrefslogtreecommitdiff
path: root/tests/user_commands/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/user_commands/tests.py')
-rw-r--r--tests/user_commands/tests.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/user_commands/tests.py b/tests/user_commands/tests.py
index b770e8e459..b1d00f278d 100644
--- a/tests/user_commands/tests.py
+++ b/tests/user_commands/tests.py
@@ -57,12 +57,14 @@ class CommandTests(SimpleTestCase):
""" Exception raised in a command should raise CommandError with
call_command, but SystemExit when run from command line
"""
- with self.assertRaises(CommandError):
+ with self.assertRaises(CommandError) as cm:
management.call_command('dance', example="raise")
+ self.assertEqual(cm.exception.returncode, 3)
dance.Command.requires_system_checks = False
try:
- with captured_stderr() as stderr, self.assertRaises(SystemExit):
+ with captured_stderr() as stderr, self.assertRaises(SystemExit) as cm:
management.ManagementUtility(['manage.py', 'dance', '--example=raise']).execute()
+ self.assertEqual(cm.exception.code, 3)
finally:
dance.Command.requires_system_checks = True
self.assertIn("CommandError", stderr.getvalue())