#!/usr/bin/env python3 # pyre-strict # Copyright (c) 2004-present Facebook All rights reserved. # Use of this source code is governed by a BSD-style # license that can be found in the LICENSE file. import sys from unittest import TestLoader, TextTestRunner import pyinventory_tests from pyinventory_tests.utils.constant import TESTS_PATTERN, XML_OUTPUT_DIRECTORY from xmlrunner import XMLTestRunner if __name__ == "__main__": loader = TestLoader() loader.testNamePatterns = [TESTS_PATTERN] suite = loader.loadTestsFromModule(pyinventory_tests) if XML_OUTPUT_DIRECTORY: runner = XMLTestRunner(output=XML_OUTPUT_DIRECTORY, verbosity=2) else: runner = TextTestRunner(verbosity=2) result = runner.run(suite) if len(result.errors) != 0 or len(result.failures) != 0: sys.exit(1) sys.exit(0)
# "--remote", # help="Run against which tenant in production staging environment", # type=str, # default=None, # ) args: Namespace = parser.parse_args() if args.local is not None: utils.TEST_MODE = TestMode.LOCAL utils.TENANT = args.local # TODO(T64902729): Restore after support for cleaning production between tests # elif args.remote is not None: # utils.TEST_MODE = TestMode.REMOTE # utils.TENANT = args.remote loader = TestLoader() loader.testNamePatterns = [args.pattern] # type: ignore suite_list: List[TestSuite] = [] suite_list.append(loader.loadTestsFromModule(pyinventory_tests)) suite_list.append(loader.loadTestsFromModule(pyworkforce_tests)) suite: TestSuite = TestSuite(suite_list) runner: TextTestRunner = TextTestRunner(verbosity=2) if args.output: runner = XMLTestRunner(output=args.output, verbosity=2) result = runner.run(suite) if len(result.errors) != 0 or len(result.failures) != 0: sys.exit(1) sys.exit(0)
action="store_true", ) parser.add_argument( "-r", "--remote", help="Run against which tenant in production staging environment", type=str, default=None, ) args = parser.parse_args() if args.local: pyinventory_tests.utils.TEST_MODE = TestMode.LOCAL elif args.remote is not None: pyinventory_tests.utils.TEST_MODE = TestMode.REMOTE pyinventory_tests.utils.TENANT = args.remote loader = TestLoader() loader.testNamePatterns = [args.pattern] suite = loader.loadTestsFromModule(pyinventory_tests) if args.output: runner = XMLTestRunner(output=args.output, verbosity=2) else: runner = TextTestRunner(verbosity=2) result = runner.run(suite) if len(result.errors) != 0 or len(result.failures) != 0: sys.exit(1) sys.exit(0)