From 91f7fd6c856ff19f8a7600dceeeca378cf3127fa Mon Sep 17 00:00:00 2001 From: Eric Joanis Date: Thu, 7 Nov 2024 14:59:13 -0500 Subject: [PATCH] feat: add --quiet option to test/run.py --- test/run.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/test/run.py b/test/run.py index 51dcf71c..b6eea875 100755 --- a/test/run.py +++ b/test/run.py @@ -97,7 +97,7 @@ def describe_suite(suite: TestSuite): SUITES = ["all", "dev", "e2e", "prod", "api", "other"] -def run_tests(suite: str, describe: bool = False) -> bool: +def run_tests(suite: str, describe: bool = False, verbosity=3) -> bool: """Run the specified test suite. Args: @@ -131,7 +131,7 @@ def run_tests(suite: str, describe: bool = False) -> bool: describe_suite(test_suite) return True else: - runner = TextTestRunner(verbosity=3) + runner = TextTestRunner(verbosity=verbosity) success = runner.run(test_suite).wasSuccessful() if not success: LOGGER.error("Some tests failed. Please see log above.") @@ -139,10 +139,18 @@ def run_tests(suite: str, describe: bool = False) -> bool: if __name__ == "__main__": - describe = "--describe" in sys.argv - if describe: + if "--describe" in sys.argv: + describe = True sys.argv.remove("--describe") + else: + describe = False + + if "--quiet" in sys.argv: + verbosity = 1 + sys.argv.remove("--quiet") + else: + verbosity = 3 - result = run_tests("" if len(sys.argv) <= 1 else sys.argv[1], describe) + result = run_tests("" if len(sys.argv) <= 1 else sys.argv[1], describe, verbosity) if not result: sys.exit(1)