From d8b99515b6c5e1e89c232834402c0bbaf5f034d8 Mon Sep 17 00:00:00 2001 From: Charles Cabergs Date: Fri, 5 Feb 2021 18:13:53 +0100 Subject: Refactoring Suite with custom exception instead of print then exit --- minishell_test/__main__.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'minishell_test/__main__.py') diff --git a/minishell_test/__main__.py b/minishell_test/__main__.py index 424867c..0b01ba8 100755 --- a/minishell_test/__main__.py +++ b/minishell_test/__main__.py @@ -21,7 +21,7 @@ import subprocess import minishell_test.config as config import minishell_test.sandbox as sandbox from minishell_test.args import parse_args -from minishell_test.suite.suite import Suite +from minishell_test.suite.suite import Suite, SuiteException from minishell_test.suites import * # noqa: F403,F401 @@ -72,7 +72,11 @@ def main(): if config.RANGE is not None or config.CHECK_LEAKS: config.SHOW_RANGE = True - Suite.setup(args.suites) + try: + Suite.setup(args.suites) + except SuiteException as e: + print(e) + sys.exit(1) try: Suite.run_all() except KeyboardInterrupt: -- cgit