71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
|
# -----------------------------------------------------------------------------
|
||
|
|
# Copyright (c) 2005-2023, PyInstaller Development Team.
|
||
|
|
#
|
||
|
|
# Distributed under the terms of the GNU General Public License (version 2
|
||
|
|
# or later) with exception for distributing the bootloader.
|
||
|
|
#
|
||
|
|
# The full license is in the file COPYING.txt, distributed with this software.
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: (GPL-2.0-or-later WITH Bootloader-exception)
|
||
|
|
# -----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import sys
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from PyInstaller.compat import importlib_metadata
|
||
|
|
|
||
|
|
|
||
|
|
def paths_to_test(include_only=None):
|
||
|
|
"""
|
||
|
|
If ``include_only`` is falsey, this functions returns paths from all entry points. Otherwise, this parameter
|
||
|
|
must be a string or sequence of strings. In this case, this function will return *only* paths from entry points
|
||
|
|
whose ``module_name`` begins with the provided string(s).
|
||
|
|
"""
|
||
|
|
# Convert a string to a list.
|
||
|
|
if isinstance(include_only, str):
|
||
|
|
include_only = [include_only]
|
||
|
|
|
||
|
|
# Walk through all entry points.
|
||
|
|
test_path_list = []
|
||
|
|
for entry_point in importlib_metadata.entry_points(group="pyinstaller40", name="tests"):
|
||
|
|
# Implement ``include_only``.
|
||
|
|
if (
|
||
|
|
not include_only # If falsey, include everything,
|
||
|
|
# Otherwise, include only the specified modules.
|
||
|
|
or any(entry_point.module.startswith(name) for name in include_only)
|
||
|
|
):
|
||
|
|
test_path_list += list(entry_point.load()())
|
||
|
|
return test_path_list
|
||
|
|
|
||
|
|
|
||
|
|
# Run pytest on all tests registered by the PyInstaller setuptools testing entry point. If provided,
|
||
|
|
# the ``include_only`` argument is passed to ``path_to_test``.
|
||
|
|
def run_pytest(*args, **kwargs):
|
||
|
|
paths = paths_to_test(include_only=kwargs.pop("include_only", None))
|
||
|
|
# Return an error code if no tests were discovered.
|
||
|
|
if not paths:
|
||
|
|
print("Error: no tests discovered.", file=sys.stderr)
|
||
|
|
# This indicates no tests were discovered; see
|
||
|
|
# https://docs.pytest.org/en/latest/usage.html#possible-exit-codes.
|
||
|
|
return 5
|
||
|
|
else:
|
||
|
|
# See https://docs.pytest.org/en/latest/usage.html#calling-pytest-from-python-code.
|
||
|
|
# Omit ``args[0]``, which is the name of this script.
|
||
|
|
print("pytest " + " ".join([*paths, *args[1:]]))
|
||
|
|
return pytest.main([*paths, *args[1:]], **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# Look only for the ``--include_only`` argument.
|
||
|
|
parser = argparse.ArgumentParser(description='Run PyInstaller packaging tests.')
|
||
|
|
parser.add_argument(
|
||
|
|
"--include_only",
|
||
|
|
action="append",
|
||
|
|
help="Only run tests from the specified package.",
|
||
|
|
)
|
||
|
|
args, unknown = parser.parse_known_args(sys.argv)
|
||
|
|
# Convert the parsed args into a dict using ``vars(args)``.
|
||
|
|
sys.exit(run_pytest(*unknown, **vars(args)))
|