From 0ae332323c6710a68601c1a1f28499408eb1a321 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Thu, 2 Mar 2023 17:17:13 +0100 Subject: Fix test for command call --- ishtar_common/management/commands/ishtar_maintenance.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'ishtar_common/management/commands/ishtar_maintenance.py') diff --git a/ishtar_common/management/commands/ishtar_maintenance.py b/ishtar_common/management/commands/ishtar_maintenance.py index d57fb575e..d6e7ae5de 100644 --- a/ishtar_common/management/commands/ishtar_maintenance.py +++ b/ishtar_common/management/commands/ishtar_maintenance.py @@ -331,8 +331,14 @@ class Command(BaseCommand): '--clean', dest='clean', action='store_true', default=False, help="[operation_missing_parcels] Delete orphan parcel after fix") + parser.add_argument( + '--test', dest='test', action='store_true', + default=False, + help="Used on tests: no sys.exit") def handle(self, *args, **options): + if options["test"]: + options["quiet"] = True if options["task"] not in TASKS.keys(): msg = f"{options['task']} is not a valid task. Available tasks are:\n" msg += "\n".join(TASKS.keys()) @@ -344,7 +350,11 @@ class Command(BaseCommand): if not errors: if not quiet: sys.stdout.write(f"[{get_time()}] Task {options['task']} finished\n") + if options["test"]: + return sys.exit() if not quiet: sys.stdout.write("\n".join(errors)) + if options["test"]: + return sys.exit(1) -- cgit v1.2.3