def test_system_check(): """ Performs the Django system check. """ base_command = BaseCommand() system_check_errors = base_command.check() assert not system_check_errors
def __init__(self, log_name=None, mapping=None, *args, **kwargs): BaseCommand.__init__(self, *args, **kwargs) if mapping is None: lg = logging mapping = [lg.ERROR, lg.WARNING, lg.INFO, lg.DEBUG] self.mapping = mapping self.logger = logging.getLogger(log_name or self.__class__.__name__)
def test_create_parser_kwargs(self): """BaseCommand.create_parser() passes kwargs to CommandParser.""" epilog = 'some epilog text' parser = BaseCommand().create_parser('prog_name', 'subcommand', epilog=epilog) self.assertEqual(parser.epilog, epilog)
def test_create_parser_kwargs(self): """BaseCommand.create_parser() passes kwargs to CommandParser.""" epilog = "some epilog text" parser = BaseCommand().create_parser("prog_name", "subcommand", epilog=epilog) self.assertEqual(parser.epilog, epilog)
def create_parser(self, *args, **kwargs): """ Enable multiline command help text. https://stackoverflow.com/questions/35470680/django-command-how-to-insert-newline-in-the-help-text """ parser = BaseCommand.create_parser(self, *args, **kwargs) parser.formatter_class = HelpTextFormatter return parser
def test_create_parser_kwargs(self): """BaseCommand.create_parser() passes kwargs to CommandParser.""" epilog = "some epilog text" parser = BaseCommand().create_parser( "prog_name", "subcommand", epilog=epilog, formatter_class=ArgumentDefaultsHelpFormatter, ) self.assertEqual(parser.epilog, epilog) self.assertEqual(parser.formatter_class, ArgumentDefaultsHelpFormatter)
def test_remove_experiment_displays_prompt_and_user_confirm_removing( self, mock_user_input): owner = create_owner('labX') experiment = create_experiment(1, owner=owner) out = StringIO() call_command('remove_experiment', experiment.nes_id, experiment.owner, stdout=out) self.assertEqual(mock_user_input.called, True) (text, ), kwargs = mock_user_input.call_args self.assertEqual( text, BaseCommand().style.WARNING( 'All versions of experiment "%s" will be destroyed and cannot be ' 'recovered. Are you sure? (Yes/n) ' % experiment.title)) self.assertFalse(Experiment.objects.exists())
def __init__(self, *args, **kwargs): self.verbosity = 0 BaseCommand.__init__(self, *args, **kwargs)
def test_checks(self): from django.core.management import BaseCommand BaseCommand().check(display_num_errors=True)
def __init__(self): BaseCommand.__init__(self)