#!/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)
