diff options
Diffstat (limited to 'tests/admin_scripts/tests.py')
| -rw-r--r-- | tests/admin_scripts/tests.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/admin_scripts/tests.py b/tests/admin_scripts/tests.py index 5ee3eeb803..29023b74c3 100644 --- a/tests/admin_scripts/tests.py +++ b/tests/admin_scripts/tests.py @@ -25,6 +25,7 @@ from django.core.management import ( color, execute_from_command_line, ) +from django.core.management.base import LabelCommand from django.core.management.commands.loaddata import Command as LoaddataCommand from django.core.management.commands.runserver import Command as RunserverCommand from django.core.management.commands.testserver import Command as TestserverCommand @@ -2280,6 +2281,20 @@ class CommandTypes(AdminScriptTestCase): "('settings', None), ('traceback', False), ('verbosity', 1)]", ) + def test_custom_label_command_custom_missing_args_message(self): + class Command(LabelCommand): + missing_args_message = "Missing argument." + + with self.assertRaisesMessage(CommandError, "Error: Missing argument."): + call_command(Command()) + + def test_custom_label_command_none_missing_args_message(self): + class Command(LabelCommand): + missing_args_message = None + + with self.assertRaisesMessage(CommandError, ""): + call_command(Command()) + def test_suppress_base_options_command_help(self): args = ["suppress_base_options_command", "--help"] out, err = self.run_manage(args) |
