mirror of
https://github.com/python/cpython.git
synced 2026-04-04 19:13:16 +00:00
a84cb74d42
libregrtest now calls random.seed() before running each test file when -r/--randomize command line option is used. Moreover, it's also called in worker processes. It should help to make tests more deterministic. Previously, it was only called once in the main process before running all test files and it was not called in worker processes. * Convert some f-strings to regular strings in test_regrtest when f-string is not needed. * Remove unused all_methods variable from test_regrtest. * Add RunTests members are now mandatory.
133 lines
4.8 KiB
Python
133 lines
4.8 KiB
Python
import faulthandler
|
|
import os
|
|
import random
|
|
import signal
|
|
import sys
|
|
import unittest
|
|
from test import support
|
|
from test.support.os_helper import TESTFN_UNDECODABLE, FS_NONASCII
|
|
try:
|
|
import gc
|
|
except ImportError:
|
|
gc = None
|
|
|
|
from .runtests import RunTests
|
|
from .utils import (
|
|
setup_unraisable_hook, setup_threading_excepthook, fix_umask,
|
|
replace_stdout, adjust_rlimit_nofile)
|
|
|
|
|
|
UNICODE_GUARD_ENV = "PYTHONREGRTEST_UNICODE_GUARD"
|
|
|
|
|
|
def setup_test_dir(testdir: str | None) -> None:
|
|
if testdir:
|
|
# Prepend test directory to sys.path, so runtest() will be able
|
|
# to locate tests
|
|
sys.path.insert(0, os.path.abspath(testdir))
|
|
|
|
|
|
def setup_process():
|
|
fix_umask()
|
|
|
|
try:
|
|
stderr_fd = sys.__stderr__.fileno()
|
|
except (ValueError, AttributeError):
|
|
# Catch ValueError to catch io.UnsupportedOperation on TextIOBase
|
|
# and ValueError on a closed stream.
|
|
#
|
|
# Catch AttributeError for stderr being None.
|
|
stderr_fd = None
|
|
else:
|
|
# Display the Python traceback on fatal errors (e.g. segfault)
|
|
faulthandler.enable(all_threads=True, file=stderr_fd)
|
|
|
|
# Display the Python traceback on SIGALRM or SIGUSR1 signal
|
|
signals = []
|
|
if hasattr(signal, 'SIGALRM'):
|
|
signals.append(signal.SIGALRM)
|
|
if hasattr(signal, 'SIGUSR1'):
|
|
signals.append(signal.SIGUSR1)
|
|
for signum in signals:
|
|
faulthandler.register(signum, chain=True, file=stderr_fd)
|
|
|
|
adjust_rlimit_nofile()
|
|
replace_stdout()
|
|
support.record_original_stdout(sys.stdout)
|
|
|
|
# Some times __path__ and __file__ are not absolute (e.g. while running from
|
|
# Lib/) and, if we change the CWD to run the tests in a temporary dir, some
|
|
# imports might fail. This affects only the modules imported before os.chdir().
|
|
# These modules are searched first in sys.path[0] (so '' -- the CWD) and if
|
|
# they are found in the CWD their __file__ and __path__ will be relative (this
|
|
# happens before the chdir). All the modules imported after the chdir, are
|
|
# not found in the CWD, and since the other paths in sys.path[1:] are absolute
|
|
# (site.py absolutize them), the __file__ and __path__ will be absolute too.
|
|
# Therefore it is necessary to absolutize manually the __file__ and __path__ of
|
|
# the packages to prevent later imports to fail when the CWD is different.
|
|
for module in sys.modules.values():
|
|
if hasattr(module, '__path__'):
|
|
for index, path in enumerate(module.__path__):
|
|
module.__path__[index] = os.path.abspath(path)
|
|
if getattr(module, '__file__', None):
|
|
module.__file__ = os.path.abspath(module.__file__)
|
|
|
|
if hasattr(sys, 'addaudithook'):
|
|
# Add an auditing hook for all tests to ensure PySys_Audit is tested
|
|
def _test_audit_hook(name, args):
|
|
pass
|
|
sys.addaudithook(_test_audit_hook)
|
|
|
|
setup_unraisable_hook()
|
|
setup_threading_excepthook()
|
|
|
|
# Ensure there's a non-ASCII character in env vars at all times to force
|
|
# tests consider this case. See BPO-44647 for details.
|
|
if TESTFN_UNDECODABLE and os.supports_bytes_environ:
|
|
os.environb.setdefault(UNICODE_GUARD_ENV.encode(), TESTFN_UNDECODABLE)
|
|
elif FS_NONASCII:
|
|
os.environ.setdefault(UNICODE_GUARD_ENV, FS_NONASCII)
|
|
|
|
|
|
def setup_tests(runtests: RunTests):
|
|
support.verbose = runtests.verbose
|
|
support.failfast = runtests.fail_fast
|
|
support.PGO = runtests.pgo
|
|
support.PGO_EXTENDED = runtests.pgo_extended
|
|
|
|
support.set_match_tests(runtests.match_tests, runtests.ignore_tests)
|
|
|
|
if runtests.use_junit:
|
|
support.junit_xml_list = []
|
|
from test.support.testresult import RegressionTestResult
|
|
RegressionTestResult.USE_XML = True
|
|
else:
|
|
support.junit_xml_list = None
|
|
|
|
if runtests.memory_limit is not None:
|
|
support.set_memlimit(runtests.memory_limit)
|
|
|
|
support.suppress_msvcrt_asserts(runtests.verbose >= 2)
|
|
|
|
support.use_resources = runtests.use_resources
|
|
|
|
timeout = runtests.timeout
|
|
if timeout is not None:
|
|
# For a slow buildbot worker, increase SHORT_TIMEOUT and LONG_TIMEOUT
|
|
support.SHORT_TIMEOUT = max(support.SHORT_TIMEOUT, timeout / 40)
|
|
support.LONG_TIMEOUT = max(support.LONG_TIMEOUT, timeout / 4)
|
|
|
|
# If --timeout is short: reduce timeouts
|
|
support.LOOPBACK_TIMEOUT = min(support.LOOPBACK_TIMEOUT, timeout)
|
|
support.INTERNET_TIMEOUT = min(support.INTERNET_TIMEOUT, timeout)
|
|
support.SHORT_TIMEOUT = min(support.SHORT_TIMEOUT, timeout)
|
|
support.LONG_TIMEOUT = min(support.LONG_TIMEOUT, timeout)
|
|
|
|
if runtests.hunt_refleak:
|
|
unittest.BaseTestSuite._cleanup = False
|
|
|
|
if runtests.gc_threshold is not None:
|
|
gc.set_threshold(runtests.gc_threshold)
|
|
|
|
if runtests.randomize:
|
|
random.seed(runtests.random_seed)
|