Fix test for command call

1 files changed, 10 insertions, 0 deletions

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)