3
�L�j�� � � @ sl
d Z edkred��ddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZ
ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddl!Z"ddl#Z#ddl$m%Z% yddl&Z&ddl'Z'W n ek
�rB dZ&dZ'Y nX y ddl(Z)W n ek
�rj dZ)Y nX y ddl*Z*W n ek
�r� dZ*Y nX y ddl+Z+W n ek
�r� dZ+Y nX y ddl,Z,W n ek
�r� dZ,Y nX y ddl-Z-W n ek
�r
dZ-Y nX y ddl.Z.W n ek
�r2 dZ.Y nX y ddl/Z/W n ek
�rZ dZ/Y nX ddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbg\Z0G dcd � d e1�Z2G ddd
� d
e2�Z3G ded� de2�Z4G dfd� de j5�Z6ej7�dhdhdi��Z8�dif dk�dld�Z9dmdn� Z:dodp� Z;dqd=� Z� Z=f f djfdsd�Z>dtd9� Z?dZ@dZAdaBdaCdZDdjZEdaFdud� ZGdvd� ZHdwd� ZIdxdy� ZJejjKdz��r.�djd{d|�ZLd}d~� ZMdd�� ZNd�d�� ZOd�d�� ZPn ejQZMejRZNd�d�� ZOd�d�� ZPd�d � ZQd�d�� ZRd�d � ZSd�d�� ZTd�d� ZUd�d�� ZVd�d#� ZW�dkd�d$�ZXd�d�� ZYd�d%� ZZd�d&� Z[d�d'� Z\�dld�d(�Z]d�Z^d�Z_ej`ejafd�dJ�Zbe^fd�dK�Zcd�dM� Zdd�d�� Zeee� Zfd�d�� Zg�doZh�drZie jjekjld��jKd��d��Zme jje*d��Zne jje+d��Zoe jje,d��Zpe jje-d��ZqejjKd��Zre jsd��Ztetdk �oxetdkZuejd�k�r�eu�r�d�nd�ZvndZvejwd�k�r�d�Zxnd�Zxd�jyexejz� �ZxdZ{xL�dsD ]BZ|y ej}ej~e|��e|k�r�e�W n ek
�r Y nX e|Z{P �q�W exd� Z�ejd�k�r:ddl�Z�e�j�d�e��Z�ej�� Z�dZ�ejwd�k�r�ej�� jd�k�r�exd� Z�ye�j�e�� W n e�k
�r� Y nX e�d�e�e�f � dZ�nBejd�k�r�yd�j�e�� W n& e�k
�r� exd�j�e�dǃ Z�Y nX dZ�xF�dtD ]dY� �Z��d?�d@� Z�e j�e�e�dA��dB�Z��dC�dD� Z��dE�dF� Z�G �dGdP� dPej�j��Z�G �dHd[� d[e���Z d�a�dId!� �Z�dJd2� �Zd�a�dK�dL� �Z�dMd;� �Z�dN�dO� �Z�dPd"� �Zf �dQ��dRd?��Z df f f�dSd@��Z
G �dTd]� d]��Z �dU�dV� �Z �dW�dX� �Z
f f�dY�dZ��Zg f�d[da��Zd�a�d\dG� �Zej7�d]�d^� ��Z�d_db� �ZG �d`�da� �da��ZG �db�dc� �dc��Zej7�dd�de� ��Z�df�dg� �ZdS (� z7Supporting definitions for the Python regression tests.z test.supportz.support must be imported from the test package� N� )�get_test_runner�
PIPE_MAX_SIZE�verbose�
max_memuse�
use_resources�failfast�Error�
TestFailed�
TestDidNotRun�ResourceDenied�
import_module�import_fresh_module� CleanImport�unload�forget�record_original_stdout�get_original_stdout�captured_stdout�captured_stdin�captured_stderr�TESTFN�SAVEDCWD�unlink�rmtree�temp_cwd�findfile�create_empty_file� can_symlink�fs_is_case_insensitive�is_resource_enabled�requires�requires_freebsd_version�requires_linux_version�requires_mac_ver�requires_hashdigest�check_syntax_error�TransientResource�time_out�socket_peer_reset�ioerror_peer_reset�transient_internet�BasicTestRunner� run_unittest� run_doctest�skip_unless_symlink�
requires_gzip� requires_bz2�
requires_lzma�
bigmemtest�bigaddrspacetest� cpython_only�
get_attribute�requires_IEEE_754�skip_unless_xattr�
requires_zlib�anticipate_failure�load_package_tests�detect_api_mismatch� check__all__�requires_android_level� requires_multiprocessing_queue� is_jython�
is_android�check_impl_detail�
unix_shell�setswitchinterval�HOST� IPV6_ENABLED�find_unused_port� bind_port�open_urlresource�bind_unix_socket�
temp_umask�
reap_children� TestHandler�threading_setup�threading_cleanup� reap_threads�
start_threads�check_warnings�check_no_resource_warning�EnvironmentVarGuard�run_with_locale� swap_item� swap_attr�Matcher� set_memlimit�SuppressCrashReport�sortdict� run_with_tz�PGO�missing_compiler_executable�fd_countc @ s e Zd ZdZdS )r z*Base class for regression test exceptions.N)�__name__�
__module__� __qualname__�__doc__� rd rd �-/usr/lib64/python3.6/test/support/__init__.pyr | s c @ s e Zd ZdZdS )r
z Test failed.N)r` ra rb rc rd rd rd re r
s c @ s e Zd ZdZdS )r z Test did not run any subtests.N)r` ra rb rc rd rd rd re r � s c @ s e Zd ZdZdS )r z�Test skipped because it requested a disallowed resource.
This is raised when a test calls requires() for a resource that
has not be enabled. It is used to distinguish between expected
and unexpected skips.
N)r` ra rb rc rd rd rd re r � s Tc c s8 | r.t j� � t jddt� dV W dQ R X ndV dS )z�Context manager to suppress package and module deprecation
warnings when importing them.
If ignore is False, this context manager has no effect.
�ignorez.+ (module|package)N)�warnings�catch_warnings�filterwarnings�DeprecationWarning)rf rd rd re �_ignore_deprecated_imports� s
rk F)� required_onc C sf t |��T y
tj| �S tk
rV } z&tjjt|��r8� tj t
|���W Y dd}~X nX W dQ R X dS )ac Import and return the module to be tested, raising SkipTest if
it is not available.
If deprecated is True, any module or package deprecation messages
will be suppressed. If a module is required on a platform but optional for
others, set required_on to an iterable of platform prefixes which will be
compared against sys.platform.
N) rk � importlibr
� ImportError�sys�platform�
startswith�tuple�unittest�SkipTest�str)�name�
deprecatedrl �msgrd rd re r
� s
c C s^ | t jkrt| � t j| = x>tt j�D ]0}|| ks@|j| d �r&t j| ||< t j|= q&W dS )zyHelper function to save and remove a module from sys.modules
Raise ImportError if the module can't be imported.
�.N)ro �modules�
__import__�listrq )rv � orig_modules�modnamerd rd re �_save_and_remove_module� s
r c C s> d}yt j| || < W n tk
r. d}Y nX dt j| < |S )z�Helper function to save and block a module in sys.modules
Return True if the module was in sys.modules, False otherwise.
TFN)ro rz �KeyError)rv r} Zsavedrd rd re �_save_and_block_module� s
r� c C s | r
t jS dd� S )z�Decorator to mark a test that is known to be broken in some cases
Any use of this decorator should have a comment identifying the
associated tracker issue.
c S s | S )Nrd )�frd rd re �� s z$anticipate_failure..)rs ZexpectedFailure)Z conditionrd rd re r: � s c C sF |dkr d}t jjt jjt jjt���}|j| ||d�}|j|� |S )z�Generic load_tests implementation for simple test packages.
Most packages can implement load_tests using this function as follows:
def load_tests(*args):
return load_package_tests(os.path.dirname(__file__), *args)
Nztest*)Z start_dirZ
top_level_dir�pattern)�os�path�dirname�__file__ZdiscoverZaddTests)Zpkg_dir�loaderZstandard_testsr� Ztop_dirZ
package_testsrd rd re r; � s
c C s� t |��� i }g }t| |� zfyHx|D ]}t||� q&W x |D ]}t||�s>|j|� q>W tj| �}W n tk
r~ d}Y nX W dx |j� D ]\} }
|
tj | < q�W x|D ] } tj | = q�W X |S Q R X dS )a� Import and return a module, deliberately bypassing sys.modules.
This function imports and returns a fresh copy of the named Python module
by removing the named module from sys.modules before doing the import.
Note that unlike reload, the original module is not affected by
this operation.
*fresh* is an iterable of additional module names that are also removed
from the sys.modules cache before doing the import.
*blocked* is an iterable of module names that are replaced with None
in the module cache during the import to ensure that attempts to import
them raise ImportError.
The named module and any modules named in the *fresh* and *blocked*
parameters are saved before starting the import and then reinserted into
sys.modules when the fresh import is complete.
Module and package deprecation messages are suppressed during this import
if *deprecated* is True.
This function will raise ImportError if the named module cannot be
imported.
N)
rk r r� �appendrm r
rn �itemsro rz ) rv ZfreshZblockedrw r} Znames_to_removeZ
fresh_nameZ blocked_nameZ fresh_moduleZ orig_name�moduleZname_to_removerd rd re r � s$
c C s> yt | |�}W n& tk
r4 tjd| |f ��Y nX |S dS )z?Get an attribute, raising SkipTest if AttributeError is raised.z object %r has no attribute %rN)�getattr�AttributeErrorrs rt )�objrv Z attributerd rd re r6 s
c C s | a d S )N)�_original_stdout)�stdoutrd rd re r 0 s c C s
t ptjS )N)r� ro r� rd rd rd re r 4 s c C s& y t j| = W n tk
r Y nX d S )N)ro rz r� )rv rd rd re r 7 s c G sn y||� S t k
rh } zDtdkrHtd|jj|f � td|j|f � tj| tj� ||� S d }~X nX d S )N� z%s: %sz re-run %s%r) �OSErrorr �print� __class__r` r� �chmod�stat�S_IRWXU)r� �func�args�errrd rd re �
_force_run= s r� �winc C s� | |� |r|}nt jj|�\}}|p(d}d}x<|dk rjt j|�}|rJ|n||ksVd S tj|� |d9 }q0W tjd| tdd� d S )Nry g����MbP?g �?r� z)tests may fail, delete still pending for � )�
stacklevel) r� r� �split�listdir�time�sleeprg �warn�RuntimeWarning)r� �pathname�waitallr� rv �timeout�Lrd rd re �_waitforH s
r� c C s t tj| � d S )N)r� r� r )�filenamerd rd re �_unlinki s r� c C s t tj| � d S )N)r� r� �rmdir)r� rd rd re �_rmdirl s r� c s, � fdd�� t � | dd� t dd� | � d S )Nc s� x�t | tj| �D ]�}tjj| |�}ytj|�j}W n< tk
rn } z td||f t j
d� d}W Y d d }~X nX t j |�r�t
� |dd� t |tj|� qt |tj|� qW d S )Nz-support.rmtree(): os.lstat(%r) failed with %s)�filer T)r� )r� r� r� r� �join�lstat�st_moder� r� ro �
__stderr__r� �S_ISDIRr� r� r )r� rv �fullname�mode�exc)�
_rmtree_innerrd re r� p s
z _rmtree.._rmtree_innerT)r� c S s t | tj| �S )N)r� r� r� )�prd rd re r� s z_rmtree..)r� )r� rd )r� re �_rmtreeo s r� c
C s^ y dd l }W n tk
r Y n:X |jt| �d �}|jjj| |t|��}|rZ|d |� S | S )Nr r� )�ctypesrn Zcreate_unicode_buffer�len�windll�kernel32ZGetLongPathNameW)r� r� �bufferZlengthrd rd re � _longpath� s
r� c sF yt j| � d S tk
r" Y nX � fdd�� � | � tj| � d S )Nc
s� x~t | tj| �D ]l}tjj| |�}ytj|�j}W n tk
rJ d}Y nX tj |�rn� |� t | tj
|� qt | tj |� qW d S )Nr ) r� r� r� r� r� r� r� r� r� r� r� r )r� rv r� r� )r� rd re r� � s
z _rmtree.._rmtree_inner)�shutilr r� r� r� )r� rd )r� re r� � s
c C s | S )Nrd )r� rd rd re r� � s c C s* y t | � W n ttfk
r$ Y nX d S )N)r� �FileNotFoundError�NotADirectoryError)r� rd rd re r � s c C s& y t | � W n tk
r Y nX d S )N)r� r� )r� rd rd re r� � s r� c C s& y t | � W n tk
r Y nX d S )N)r� r� )r� rd rd re r � s c C sB t jj| �}tjjtjj| ��}tjj|| d �}tj||� |S )a Move a PEP 3147/488 pyc file to its legacy pyc location.
:param source: The file system path to the source file. The source file
does not need to exist, however the PEP 3147/488 pyc file must exist.
:return: The file system path to the legacy pyc file.
�c) rm �util�cache_from_sourcer� r� r� �abspathr� �rename)�sourceZpyc_fileZup_oneZ
legacy_pycrd rd re �make_legacy_pyc� s
r� c C s\ t | � xNtjD ]D}tjj|| d �}t|d � x dD ]}ttjj||d�� q8W qW dS ) z�'Forget' a module was ever imported.
This removes the module from sys.modules and deletes any PEP 3147/488 or
legacy .pyc files.
z.pyr� � r r� )� optimizationN)r� r r� ) r ro r� r� r� r rm r� r� )r~ r� r� �optrd rd re r � s
c s� t td�rtjS d } tjjd�r�dd l�dd l�d}d}G �fdd�d�j�}�j j
}|j � }|sj�j � �|� }�j
j� }|j||�j|��j|��j|��}|s��j � �t|j|@ �s�d} n�tjdk�rVdd lm} m� m}
m} dd
lm} | j| d ��}
|
j� dk�rd } nFG � fd
d�d| �}|� }|
|�}|
j|�dk �sR|
j |�dk �rVd} | �s�y.ddl m } |� }|j � |j � |j!� W n\ t"k
�r� } z>t#|�}t$|�dk�r�|d d� d }dj%t&|�j'|�} W Y d d }~X nX | t_(| t_tjS )N�resultr� r r c s. e Zd Zd� jjfd� jjfd� jjfgZdS )z*_is_gui_available..USEROBJECTFLAGSZfInheritZ fReserved�dwFlagsN)r` ra rb �wintypesZBOOL�DWORD�_fields_rd )r� rd re �USEROBJECTFLAGS� s
r� z,gui not available (WSF_VISIBLE flag not set)�darwin)�cdll�c_int�pointer� Structure)� find_libraryZApplicationServicesz0gui tests cannot run without OS X window managerc s e Zd Zd� fd� fgZdS )z._is_gui_available..ProcessSerialNumberZ
highLongOfPSNZ lowLongOfPSNN)r` ra rb r� rd )r� rd re �ProcessSerialNumber s r� z#cannot run without OS X gui process)�Tk�2 z [...]z Tk unavailable due to {}: {}))�hasattr�_is_gui_availabler� ro rp rq r� Zctypes.wintypesr� r� Zuser32ZGetProcessWindowStationZWinErrorr� r� ZGetUserObjectInformationWZbyrefZsizeof�boolr� r� r� r� Z ctypes.utilr� Z LoadLibraryZCGMainDisplayIDZGetCurrentProcessZSetFrontProcessZtkinterr� Zwithdraw�updateZdestroy� Exceptionru r� �format�typer` �reason)r� Z UOI_FLAGSZ WSF_VISIBLEr� Zdll�hZuofZneeded�resr� r� r� r� Z app_servicesr� ZpsnZpsn_pr� �root�eZ
err_stringrd )r� r� re r� � sh
r� c C s t dkp| t kS )z�Test whether a resource is enabled.
Known resources are set by regrtest.py. If not running under regrtest.py,
all resources are assumed enabled unless use_resources has been set.
N)r )�resourcerd rd re r $ s c C s> t | �s |dkrd| }t|��| dkr:t� r:ttj��dS )z@Raise ResourceDenied if the specified resource is not available.Nz"Use of the %r resource not enabled�gui)r r r� r� )r� rx rd rd re r! , s c s � �fdd�}|S )z�Decorator raising SkipTest if the OS is `sysname` and the version is less
than `min_version`.
For example, @_requires_unix_version('FreeBSD', (7, 2)) raises SkipTest if
the FreeBSD version is less than 7.2.
c s$ t j� �� ��fdd��}�|_|S )Nc
s� t j� �krzt j� jdd�d }yttt|jd���}W n tk
rL Y n.X |�k rzdjtt ���}t
j d�||f ��� | |�S )N�-r r ry z(%s version %s or higher required, not %s) rp �system�releaser� rr �map�int�
ValueErrorr� ru rs rt )r� �kw� version_txt�version�min_version_txt)r� � min_version�sysnamerd re �wrapper= s z:_requires_unix_version..decorator..wrapper)� functools�wrapsr� )r� r� )r� r� )r� re � decorator< s z)_requires_unix_version..decoratorrd )r� r� r� rd )r� r� re �_requires_unix_version5 s r� c G s
t d| �S )z�Decorator raising SkipTest if the OS is FreeBSD and the FreeBSD version is
less than `min_version`.
For example, @requires_freebsd_version(7, 2) raises SkipTest if the FreeBSD
version is less than 7.2.
ZFreeBSD)r� )r� rd rd re r" P s c G s
t d| �S )z�Decorator raising SkipTest if the OS is Linux and the Linux version is
less than `min_version`.
For example, @requires_linux_version(2, 6, 32) raises SkipTest if the Linux
version is less than 2.6.32.
ZLinux)r� )r� rd rd re r# Y s c s � fdd�}|S )z�Decorator raising SkipTest if the OS is Mac OS X and the OS X
version if less than min_version.
For example, @requires_mac_ver(10, 5) raises SkipTest if the OS X version
is lesser than 10.5.
c s" t j� �� �fdd��}�|_|S )Nc sx t jdkrntj� d }yttt|jd���}W n tk
rB Y n,X |�k rndjtt ���}t
j d||f ��� | |�S )Nr� r ry z&Mac OS X %s or higher required, not %s) ro rp Zmac_verrr r� r� r� r� r� ru rs rt )r� r� r� r� r� )r� r� rd re r� j s
z4requires_mac_ver..decorator..wrapper)r� r� r� )r� r� )r� )r� re r� i s z#requires_mac_ver..decoratorrd )r� r� rd )r� re r$ b s c s � �fdd�}|S )a� Decorator raising SkipTest if a hashing algorithm is not available
The hashing algorithm could be missing or blocked by a strict crypto
policy.
If 'openssl' is True, then the decorator checks that OpenSSL provides
the algorithm. Otherwise the check falls back to built-in
implementations.
ValueError: [digital envelope routines: EVP_DigestInit_ex] disabled for FIPS
ValueError: unsupported hash type md4
c s t j� ��� �fdd��}|S )Nc sX y&�rt d k rt j� � n
tj� � W n& tk
rL tjd� � d���Y nX �| |�S )Nz
hash digest 'z' is not available.)�_hashlib�new�hashlibr� rs rt )r� �kwargs)�
digestnamer� �opensslrd re r� � s z7requires_hashdigest..decorator..wrapper)r� r� )r� r� )r r )r� re r� � s z&requires_hashdigest..decoratorrd )r r r� rd )r r re r% } s
z 127.0.0.1z::1c C s" t j | |�}t|�}|j� ~|S )a�
Returns an unused port that should be suitable for binding. This is
achieved by creating a temporary socket with the same family and type as
the 'sock' parameter (default is AF_INET, SOCK_STREAM), and binding it to
the specified host address (defaults to 0.0.0.0) with the port set to 0,
eliciting an unused ephemeral port from the OS. The temporary socket is
then closed and deleted, and the ephemeral port is returned.
Either this method or bind_port() should be used for any tests where a
server socket needs to be bound to a particular port for the duration of
the test. Which one to use depends on whether the calling code is creating
a python socket, or if an unused port needs to be provided in a constructor
or passed to an external program (i.e. the -accept argument to openssl's
s_server mode). Always prefer bind_port() over find_unused_port() where
possible. Hard coded ports should *NEVER* be used. As soon as a server
socket is bound to a hard coded port, the ability to run multiple instances
of the test simultaneously on the same host is compromised, which makes the
test a ticking time bomb in a buildbot environment. On Unix buildbots, this
may simply manifest as a failed test, which can be recovered from without
intervention in most cases, but on Windows, the entire python process can
completely and utterly wedge, requiring someone to log in to the buildbot
and manually kill the affected process.
(This is easy to reproduce on Windows, unfortunately, and can be traced to
the SO_REUSEADDR socket option having different semantics on Windows versus
Unix/Linux. On Unix, you can't have two AF_INET SOCK_STREAM sockets bind,
listen and then accept connections on identical host/ports. An EADDRINUSE
OSError will be raised at some point (depending on the platform and
the order bind and listen were called on each socket).
However, on Windows, if SO_REUSEADDR is set on the sockets, no EADDRINUSE
will ever be raised when attempting to bind two identical host/ports. When
accept() is called on each socket, the second caller's process will steal
the port from the first caller, leaving them both in an awkwardly wedged
state where they'll no longer respond to any signals or graceful kills, and
must be forcibly killed via OpenProcess()/TerminateProcess().
The solution on Windows is to use the SO_EXCLUSIVEADDRUSE socket option
instead of SO_REUSEADDR, which effectively affords the same semantics as
SO_REUSEADDR on Unix. Given the propensity of Unix developers in the Open
Source world compared to Windows ones, this is a common mistake. A quick
look over OpenSSL's 0.9.8g source shows that they use SO_REUSEADDR when
openssl.exe is called with the 's_server' option, for example. See
http://bugs.python.org/issue2550 for more info. The following site also
has a very thorough description about the implications of both REUSEADDR
and EXCLUSIVEADDRUSE on Windows:
http://msdn2.microsoft.com/en-us/library/ms740621(VS.85).aspx)
XXX: although this approach is a vast improvement on previous attempts to
elicit unused ports, it rests heavily on the assumption that the ephemeral
port returned to us by the OS won't immediately be dished back out to some
other process when we close and delete our temporary socket but before our
calling code has a chance to bind the returned port. We can deal with this
issue if/when we come across it.
)�socketrH �close)�familyZsocktypeZtempsock�portrd rd re rG � s
8 c C s� | j tjkr�| jtjkr�ttd�r>| jtjtj�dkr>t d��ttd�r~y | jtjtj
�dkrft d��W n t k
r| Y nX ttd�r�| j tjtj
d� | j|df� | j� d }|S )a% Bind the socket to a free port and return the port number. Relies on
ephemeral ports in order to ensure we are using an unbound port. This is
important as many tests may be running simultaneously, especially in a
buildbot environment. This method raises an exception if the sock.family
is AF_INET and sock.type is SOCK_STREAM, *and* the socket has SO_REUSEADDR
or SO_REUSEPORT set on it. Tests should *never* set these socket options
for TCP/IP sockets. The only case for setting these options is testing
multicasting via multiple UDP sockets.
Additionally, if the SO_EXCLUSIVEADDRUSE socket option is available (i.e.
on Windows), it will be set on the socket. This will prevent anyone else
from bind()'ing to our host/port for the duration of the test.
� SO_REUSEADDRr zHtests should never set the SO_REUSEADDR socket option on TCP/IP sockets!� SO_REUSEPORTzHtests should never set the SO_REUSEPORT socket option on TCP/IP sockets!�SO_EXCLUSIVEADDRUSEr )r r �AF_INETr� � SOCK_STREAMr� Z
getsockoptZ
SOL_SOCKETr r
r r� Z
setsockoptr �bindZ getsockname)�sock�hostr rd rd re rH � s
c C s: y| j |� W n& tk
r4 | j� tjd��Y nX dS )zBBind a unix socket, raising SkipTest if PermissionError is raised.zcannot bind AF_UNIX socketsN)r �PermissionErrorr rs rt )r Zaddrrd rd re rJ s
c C sZ t jrVd} z.dec)r� r� )r� r rd )r� re �system_must_validate_cert s r r� i � ZdoubleZIEEEz test requires IEEE 754 doublesz
requires zlibz
requires gzipz requires bz2z
requires lzma�java�ANDROID_API_LEVEL�win32z/system/bin/shz/bin/shz$testz@testz {}_{}_tmp� æ� İ� Ł� φ� К� א� ،� ت� ก� � €u -àòɘŁğr� ZNFD�ntr� u -共Ł♡ͣ���ztWARNING: The filename %r CAN be encoded by the filesystem encoding (%s). Unicode filename tests may not be effective� �s -��surrogateescape� �w�� ������ ���� ��r� c c s� d}| dkr&t j� } d}tjj| �} nBytj| � d}W n. tk
rf |sN� tjd| t dd� Y nX |rttj
� }z
| V W d|r�|tj
� kr�t | � X dS )a� Return a context manager that creates a temporary directory.
Arguments:
path: the directory to create temporarily. If omitted or None,
defaults to creating a temporary directory using tempfile.mkdtemp.
quiet: if False (the default), the context manager raises an exception
on error. Otherwise, if the path is specified and cannot be
created, only a warning is issued.
FNTz+tests may fail, unable to create temp dir: � )r� ) �tempfile�mkdtempr� r� �realpath�mkdirr� rg r� r� �getpidr )r� �quietZ dir_created�pidrd rd re �temp_dir� s&
r2 c c sf t j� }yt j| � W n. tk
rD |s,� tjd| tdd� Y nX zt j� V W dt j|� X dS )ag Return a context manager that changes the current working directory.
Arguments:
path: the directory to use as the temporary current working directory.
quiet: if False (the default), the context manager raises an exception
on error. Otherwise, it issues only a warning and keeps the current
working directory the same.
z)tests may fail, unable to change CWD to: r* )r� N)r� �getcwd�chdirr� rg r� r� )r� r0 Z saved_dirrd rd re �
change_cwd s
r5 �tempcwdc c s: t | |d��$}t||d�� }|V W dQ R X W dQ R X dS )a�
Context manager that temporarily creates and changes the CWD.
The function temporarily changes the current working directory
after creating a temporary directory in the current directory with
name *name*. If *name* is None, the temporary directory is
created using tempfile.mkdtemp.
If *quiet* is False (default) and it is not possible to
create or change the CWD, an error is raised. If *quiet* is True,
only a warning is raised and the original CWD is used.
)r� r0 )r0 N)r2 r5 )rv r0 Z temp_pathZcwd_dirrd rd re r $ s �umaskc c s& t j| �}z
dV W dt j|� X dS )z8Context manager that temporarily sets the process umask.N)r� r7 )r7 Zoldmaskrd rd re rK 8 s
�datac C sb t jj| �r| S |dk r&t jj|| �} tgtj }x*|D ]"}t jj|| �}t jj|�r8|S q8W | S )a[ Try to find a file on sys.path or in the test directory. If it is not
found the argument passed to the function is returned (this does not
necessarily signal failure; could still be the legitimate path).
Setting *subdir* indicates a relative path to use to find the file
rather than looking directly in the path directories.
N)r� r� �isabsr� �
TEST_HOME_DIRro �exists)r� Zsubdirr� Zdn�fnrd rd re r I s
c C s( t j| t jt jB t jB �}t j|� dS )z>Create an empty file. If the file already exists, truncate it.N)r� �open�O_WRONLY�O_CREAT�O_TRUNCr )r� �fdrd rd re r [ s c C s, t | j� �}dd� |D �}dj|�}d| S )z%Like repr(dict), but in sorted order.c S s g | ] }d | �qS )z%r: %rrd )�.0Zpairrd rd re �
c s z sortdict..z, z{%s})�sortedr� r� )�dictr� Z reprpairsZ
withcommasrd rd re r[ ` s
c C s* t td�} z| j� S | j� tt� X dS )z`
Create an invalid file descriptor by opening and closing a file and return
its fd.
�wbN)r= r �filenor r )r� rd rd re � make_bad_fdg s
rH )�lineno�offsetc C sp | j t��}t|dd� W d Q R X |j}| j|j� |d k rJ| j|j|� | j|j� |d k rl| j|j|� d S )Nz
�exec)� assertRaises� SyntaxError�compileZ exceptionZassertIsNotNonerI � assertEqualrJ )�testcaseZ statementrI rJ �cmr� rd rd re r& s s c sV dd l }dd l}�jdd ��|jj| �d jd�d }tjjt |�}� ��fdd�}tjj
|�r|||�}|d k rt|S t |� t d� t
r�td | t� d
� |jj� }tr�|jjd� |j| d
d�}tr�|jjd�d kr�tj|d�}zBt|d��.} |j� }
x |
�r| j|
� |j� }
� q�W W d Q R X W d |j� X ||�}|d k �rF|S t d| ��d S )Nr �checkr� �/r c s> t | f� ���}�d kr |S �|�r2|jd� |S |j� d S )Nr )r= �seekr )r< r� )r� rR r� rd re �check_valid_file� s
z*open_urlresource..check_valid_fileZurlfetchz fetching %s ...)r� �Accept-Encoding�gzip� )r� zContent-Encoding)ZfileobjrF zinvalid resource %r�����)rV rW ) Zurllib.requestZ urllib.parse�pop�parseZurlparser� r� r� r� �
TEST_DATA_DIRr; r r! r r� r ZrequestZ build_openerrW Z
addheadersr� r= Zheaders�getZGzipFile�read�writer r
) Zurlr� r� �urllibr� r< rU r� �opener�out�srd )r� rR r� re rI ~ s<
c @ s4 e Zd ZdZdd� Zdd� Zedd� �Zdd � Zd
S ) �WarningsRecorderzyConvenience wrapper for the warnings list returned on
entry to the warnings.catch_warnings() context manager.
c C s || _ d| _d S )Nr )� _warnings�_last)�selfZ
warnings_listrd rd re �__init__� s zWarningsRecorder.__init__c C sD t | j�| jkr t| jd |�S |tjjkr0d S td| |f ��d S )Nr z%r has no attribute %rrY )r� re rf r� rg �WarningMessage�_WARNING_DETAILSr� )rg �attrrd rd re � __getattr__� s
z WarningsRecorder.__getattr__c C s | j | jd � S )N)re rf )rg rd rd re rg � s zWarningsRecorder.warningsc C s t | j�| _d S )N)r� re rf )rg rd rd re �reset� s zWarningsRecorder.resetN) r` ra rb rc rh rl �propertyrg rm rd rd rd re rd � s
rd c
c s
t jd�}|jjd�}|r"|j� tjdd�� }t jd jd� t |�V W dQ R X t
|�}g }xz| D ]r\}}d} xH|dd� D ]8}|j }
t j
|t|
�t j�r�t|
j|�r�d} |j|� q�W | rf| rf|j||jf� qfW |r�td |d
��|�rtd |d
��dS ) z�Catch the warnings, then check if all the expected
warnings have been raised and re-raise unexpected warnings.
If 'quiet' is True, only re-raise the unexpected warnings.
r� Z__warningregistry__T)�recordrg �alwaysNFzunhandled warning %sr z)filter (%r, %s) did not catch any warning)ro � _getframe� f_globalsr] �clearrg rh rz � simplefilterrd r| �message�re�matchru �I�
issubclassr� �remover� r` �AssertionError) �filtersr0 �frame�registry�wZreraiseZmissingrx �cat�seenZwarningrd rd re �_filterwarnings� s0
r� c O s. |j d�}| s$dtff} |dkr$d}t| |�S )a� Context manager to silence warnings.
Accept 2-tuples as positional arguments:
("message regexp", WarningCategory)
Optional argument:
- if 'quiet' is True, it does not fail if a filter catches nothing
(default True without argument,
default False if some filters are defined)
Without argument, it defaults to:
check_warnings(("", Warning), quiet=True)
r0 r� NT)r] �Warningr� )r| r� r0 rd rd re rR � s
r� c c sH t jdd��&}t jd||d� dV |r.t� W dQ R X | j|g � dS )a� Context manager to check that no warnings are emitted.
This context manager enables a given warning within its scope
and checks that no warnings are emitted even with that warning
enabled.
If force_gc is True, a garbage collection is attempted before checking
for warnings. This may help to catch warnings emitted when objects
are deleted, such as ResourceWarning.
Other keyword arguments are passed to warnings.filterwarnings().
T)ro rp )ru �categoryN)rg rh ri �
gc_collectrO )rP ru r� Zforce_gc�warnsrd rd re �check_no_warnings s r� c c sB t jdd�� }t jdtd� dV t� W dQ R X | j|g � dS )a" Context manager to check that no ResourceWarning is emitted.
Usage:
with check_no_resource_warning(self):
f = open(...)
...
del f
You must remove the object which may emit ResourceWarning before
the end of the context manager.
T)ro rp )r� N)rg rh ri �ResourceWarningr� rO )rP r� rd rd re rS s
c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r a, Context manager to force import to return a new module reference.
This is useful for testing module-level behaviours, such as
the emission of a DeprecationWarning on import.
Use like this:
with CleanImport("foo"):
importlib.import_module("foo") # new reference
c G sN t jj� | _x<|D ]4}|t jkrt j| }|j|kr>t j|j= t j|= qW d S )N)ro rz �copy�original_modulesr` )rg Z module_namesZ module_namer� rd rd re rh ? s
zCleanImport.__init__c C s | S )Nrd )rg rd rd re � __enter__L s zCleanImport.__enter__c G s t jj| j� d S )N)ro rz r� r� )rg �
ignore_excrd rd re �__exit__O s zCleanImport.__exit__N)r` ra rb rc rh r� r� rd rd rd re r 3 s
c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )rT z_Class to help protect the environment variable properly. Can be used as
a context manager.c C s t j| _i | _d S )N)r� �environ�_environ�_changed)rg rd rd re rh X s z EnvironmentVarGuard.__init__c C s
| j | S )N)r� )rg �envvarrd rd re � __getitem__\ s z EnvironmentVarGuard.__getitem__c C s* || j kr | jj|�| j |< || j|< d S )N)r� r� r] )rg r� �valuerd rd re � __setitem___ s
z EnvironmentVarGuard.__setitem__c C s2 || j kr | jj|�| j |< || jkr.| j|= d S )N)r� r� r] )rg r� rd rd re � __delitem__e s
z EnvironmentVarGuard.__delitem__c C s
| j j� S )N)r� �keys)rg rd rd re r� l s zEnvironmentVarGuard.keysc C s
t | j�S )N)�iterr� )rg rd rd re �__iter__o s z EnvironmentVarGuard.__iter__c C s
t | j�S )N)r� r� )rg rd rd re �__len__r s zEnvironmentVarGuard.__len__c C s || |< d S )Nrd )rg r� r� rd rd re �setu s zEnvironmentVarGuard.setc C s
| |= d S )Nrd )rg r� rd rd re �unsetx s zEnvironmentVarGuard.unsetc C s | S )Nrd )rg rd rd re r� { s z EnvironmentVarGuard.__enter__c G sJ x<| j j� D ].\}}|d kr0|| jkr:| j|= q || j|< q W | jt_d S )N)r� r� r� r� r� )rg r� �k�vrd rd re r� ~ s
z EnvironmentVarGuard.__exit__N)r` ra rb rc rh r� r� r� r� r� r� r� r� r� r� rd rd rd re rT S s c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
DirsOnSysPatha� Context manager to temporarily add directories to sys.path.
This makes a copy of sys.path, appends any directories given
as positional arguments, then reverts sys.path to the copied
settings when the context ends.
Note that *all* sys.path modifications in the body of the
context manager, including replacement of the object,
will be reverted at the end of the block.
c G s( t jd d � | _t j| _t jj|� d S )N)ro r� �original_value�original_object�extend)rg �pathsrd rd re rh � s zDirsOnSysPath.__init__c C s | S )Nrd )rg rd rd re r� � s zDirsOnSysPath.__enter__c G s | j t_| jtjd d �< d S )N)r� ro r� r� )rg r� rd rd re r� � s zDirsOnSysPath.__exit__N)r` ra rb rc rh r� r� rd rd rd re r� � s
r� c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS )
r' z�Raise ResourceDenied if an exception is raised while the context manager
is in effect that matches the specified exception and attributes.c K s || _ || _d S )N)r� �attrs)rg r� r� rd rd re rh � s zTransientResource.__init__c C s | S )Nrd )rg rd rd re r� � s zTransientResource.__enter__Nc C sT |dk rPt | j|�rPx:| jj� D ]$\}}t||�s4P t||�|kr P q W td��dS )z�If type_ is a subclass of self.exc and value has attributes matching
self.attrs, raise ResourceDenied. Otherwise let the exception
propagate (if any).Nz%an optional resource is not available)ry r� r� r� r� r� r )rg Ztype_r� � tracebackrk Z
attr_valuerd rd re r� � s
zTransientResource.__exit__)NNN)r` ra rb rc rh r� r� rd rd rd re r' � s )�errnog >@)r� �errnosc # sp d!d"d#d$d%d&g}d(d*d,d.d/g}t d| ��|� g �� sRdd� |D �� dd� |D ��� ��fdd �}tj� }z�y |d k r�tj|� d V W n� tjk
� r� } z&tr�tjj �j
d d � �|�W Y d d }~X n� t k
�rZ } zpx^|j
}t |�d k�rt
|d t ��r|d }n*t |�dk�r8t
|d t ��r8|d }nP � q�W ||� � W Y d d }~X nX W d tj|� X d S )0z�Return a context manager that raises ResourceDenied when various issues
with the Internet connection manifest themselves as exceptions.� ECONNREFUSED�o �
ECONNRESET�h � EHOSTUNREACH�q � ENETUNREACH�e � ETIMEDOUT�n �
EADDRNOTAVAIL�c � EAI_AGAINr* �EAI_FAILr� �
EAI_NONAMEr� �
EAI_NODATA� �
WSANO_DATA��* z Resource %r is not availablec S s g | ]\}}t t||��qS rd )r� r� )rB rv �numrd rd re rC � s z&transient_internet..c S s g | ]\}}t t||��qS rd )r� r )rB rv r� rd rd re rC � s c s� t | dd �}t| tj�s�t| tj�r,|�ks�t| tjj�rTd| j koNdkn s�t| tjj �r�d| j
ks�d| j
ks�d| j
ks�|� kr�t s�t j
j�jd d � �| �d S ) Nr� i� iW �ConnectionRefusedError� TimeoutError�EOFErrorr �
)r� �
isinstancer r� Zgaierrorr` �errorZ HTTPError�codeZURLErrorr� r ro �stderrr_ r� )r� �n)�captured_errnos�denied�
gai_errnosrd re � filter_error� s
z(transient_internet..filter_errorNr r� r )r� r� )r� r� )r� r� )r� r� )r� r� )r� r� �����)r� r� �����)r� r� �����)r� r� �����)r� r� )r� r� )r r ZgetdefaulttimeoutZsetdefaulttimeout�nntplibZNNTPTemporaryErrorr ro r� r_ r� r� r� r� ) Z
resource_namer� r� Zdefault_errnosZdefault_gai_errnosr� Z old_timeoutr� �ard )r� r� r� re r+ � sP
c
c sF ddl }tt| �}tt| |j� � ztt| �V W dtt| |� X dS )z�Return a context manager used by captured_stdout/stdin/stderr
that temporarily replaces the sys stream *stream_name* with a StringIO.r N)�ior� ro �setattr�StringIO)Z stream_namer� Z orig_stdoutrd rd re �captured_output s
r� c C s t d�S )z�Capture the output of sys.stdout:
with captured_stdout() as stdout:
print("hello")
self.assertEqual(stdout.getvalue(), "hello\n")
r� )r� rd rd rd re r s c C s t d�S )z�Capture the output of sys.stderr:
with captured_stderr() as stderr:
print("hello", file=sys.stderr)
self.assertEqual(stderr.getvalue(), "hello\n")
r� )r� rd rd rd re r % s c C s t d�S )a Capture the input to sys.stdin:
with captured_stdin() as stdin:
stdin.write('hello\n')
stdin.seek(0)
# call test code that consumes from sys.stdin
captured = input()
self.assertEqual(captured, "hello")
�stdin)r� rd rd rd re r . s
c C s* t j� trtjd� t j� t j� dS )a� Force as many objects as possible to be collected.
In non-CPython implementations of Python, this is needed because timely
deallocation is not guaranteed by the garbage collector. (Even in CPython
this can be the case in case of reference cycles.) This means that __del__
methods may be called later than expected and weakrefs may remain alive for
longer than expected. This function tries its best to force all garbage
objects to disappear.
g�������?N)�gcZcollectr@ r� r� rd rd rd re r� ; s
r� c
c s. t j� } t j� z
d V W d | r(t j� X d S )N)r� � isenabled�disable�enable)Zhave_gcrd rd re �
disable_gcK s
r� c C s: t jd�p d} d}x | j� D ]}|jd�r |}q W |dkS )z,Find if Python was built with optimizations.� PY_CFLAGSr� z-O�-O0�-Og)r� r� r� )� sysconfig�get_config_varr� rq )ZcflagsZ final_optr� rd rd re �python_is_optimizedV s
r� ZnPZ0n�gettotalrefcountZ2PZ0Pr� c C s t jt| t �S )N)�struct�calcsize�_header�_align)�fmtrd rd re � calcobjsizeg s r� c C s t jt| t �S )N)r� r� �_vheaderr� )r� rd rd re � calcvobjsizej s r� � � c C sp dd l }tj|�}t|�tkr(|jt@ sBt|�tkrLt|�jt@ rL||j7 }dt|�||f }| j|||� d S )Nr z&wrong size for %s: got %d, expected %d) � _testcapiro � getsizeofr� � __flags__�_TPFLAGS_HEAPTYPE�_TPFLAGS_HAVE_GCZSIZEOF_PYGC_HEADrO )�test�o�sizer� r� rx rd rd re � check_sizeofq s
r� c s � �fdd�}|S )Nc s$ �� �fdd�}� j |_ � j|_|S )Nc s� y dd l }t|� �}|j|�}W n( tk
r6 � Y nB d }}Y n0X x,�D ]$}y|j||� P W qP Y qPX qPW z
�| |�S |r�|r�|j||� X d S )Nr )�localer� � setlocaler� )r� �kwdsr� r� Z orig_locale�loc)�catstrr� �localesrd re �inner� s$
z1run_with_locale..decorator..inner)r` rc )r� r� )r� r� )r� re r� � s z"run_with_locale..decoratorrd )r� r� r� rd )r� r� re rU � s c s � fdd�}|S )Nc s" � �fdd�}� j |_ � j|_|S )Nc s� y
t j}W n tk
r( tjd��Y nX dtjkr@tjd }nd }�tjd< |� z
� | |�S |d krrtjd= n
|tjd< t j� X d S )Nztzset requiredZTZ)r� �tzsetr� rs rt r� r� )r� r� r� Zorig_tz)r� �tzrd re r� � s
z-run_with_tz..decorator..inner)r` rc )r� r� )r� )r� re r� � s z run_with_tz..decoratorrd )r� r� rd )r� re r\ � s c C s� dt tdt d�}tjd| tjtjB �}|d kr>td| f ��tt|j d��||j d�j
� �}|a |t krrt }|t
d k r�td| f ��|ad S )Ni )r� �m�g�tz(\d+(\.\d+)?) (K|M|G|T)b?$zInvalid memory limit %rr r* z$Memory limit %r too low to be useful)�_1M�_1Grv rw �
IGNORECASE�VERBOSEr� r� �float�group�lower�real_max_memuse�MAX_Py_ssize_t�_2Gr )�limitZsizesr� Zmemlimitrd rd re rY � s $ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �_MemoryWatchdogz`An object which periodically watches the process' memory consumption
and prints it out.
c C s dj tj� d�| _d| _d S )Nz/proc/{pid}/statm)r1 F)r� r� r/ �procfile�started)rg rd rd re rh � s z_MemoryWatchdog.__init__c C s� yt | jd�}W n< tk
rL } z tjdj|�t� tjj � d S d }~X nX t
d�}t j tj
|g|t jd�| _|j� d| _d S )N�rz!/proc not available for stats: {}zmemory_watchdog.py)r� r� T)r= r
r� rg r� r� r� ro r� �flushr �
subprocess�Popen�
executableZDEVNULL� mem_watchdogr r )rg r� r� Zwatchdog_scriptrd rd re �start� s
z_MemoryWatchdog.startc C s | j r| jj� | jj� d S )N)r r Z terminate�wait)rg rd rd re �stop� s
z_MemoryWatchdog.stopN)r` ra rb rc rh r r rd rd rd re r � s r c s � ��fdd�}|S )at Decorator for bigmem tests.
'size' is a requested size for the test (in arbitrary, test-interpreted
units.) 'memuse' is the number of bytes per unit for the test, or a good
estimate of it. For example, a test that needs two byte buffers, of 4 GiB
each, could be decorated with @bigmemtest(size=_4G, memuse=2).
The 'size' argument is normally passed to the decorated test method as an
extra argument. If 'dry_run' is true, the value passed to the test method
may be less than the requested value. If 'dry_run' is false, it means the
test doesn't support dummy runs when -M is not specified.
c s �� �fdd����_ ��_�S )Nc
s� �j }�j}tsd}n|}ts$� rFt|| k rFtjd|| d ��tr|tr|t� tdj|| d d�� t� }|j � nd }z
�| |�S |r�|j
� X d S ) Ni z'not enough memory: %.1fG minimum neededi r* z* ... expected peak memory use: {peak:.1f}G)Zpeaki @i @) r� �memuser rs rt r r� r� r r r )rg r� r �maxsizeZwatchdog)�dry_runr� r� rd re r� s*
z.bigmemtest..decorator..wrapper)r� r )r� )r r r� )r� r� re r� s z bigmemtest..decoratorrd )r� r r r� rd )r r r� re r3 s
!c s � fdd�}|S )z0Decorator for tests that fill the address space.c sD t tk r8td
kr$t d kr$tjd��q@tjdtd ��n� | �S d S )
Nr� �? r � z-not enough memory: try a 32-bit build insteadz'not enough memory: %.1fG minimum neededi r* l l ���� l i @)r r rs rt )rg )r� rd re r� 3 s z!bigaddrspacetest..wrapperrd )r� r� rd )r� re r4 1 s c @ s e Zd Zdd� ZdS )r, c C s t j� }||� |S )N)rs Z
TestResult)rg r� r� rd rd re �runD s zBasicTestRunner.runN)r` ra rb r rd rd rd re r, C s c C s | S )Nrd )r� rd rd re �_idI s r c C s< | dkr t � r tjt j�S t| �r(tS tjdj| ��S d S )Nr� z resource {0!r} is not enabled)r� rs �skipr� r r r� )r� rd rd re �requires_resourceL s
r c C s& t r t| k r tjd|tf �S tS d S )Nz%s at Android API level %d)rA �_ANDROID_API_LEVELrs r r )�levelr� rd rd re r> T s c C s t dd�| �S )z9
Decorator for tests only applicable on CPython.
T)�cpython)� impl_detail)r� rd rd re r5 [ s c K sV t f |�rtS | d krLt|�\}}|r,d} nd} t|j� �}| jdj|��} tj| �S )Nz*implementation detail not available on {0}z%implementation detail specific to {0}z or ) rB r �
_parse_guardsrD r� r� r� rs r )rx �guardsZ
guardnames�defaultrd rd re r! a s
r! c C sT t dkr:ddl}y|j� da W n tk
r8 da Y nX d}t rF| S tj|�| �S )z8Skip decorator for tests that use multiprocessing.Queue.Nr TFz6requires a functioning shared semaphore implementation)�_have_mp_queue�multiprocessingZQueuern rs r )r� r& rx rd rd re r? o s
c C s* | sddidfS t | j� �d }| | fS )Nr TFr )r| �values)r# Zis_truerd rd re r" ~ s r" c K s t | �\} }| jtj� j� |�S )a5 This function returns True or False depending on the host platform.
Examples:
if check_impl_detail(): # only on CPython (default)
if check_impl_detail(jython=True): # only on Jython
if check_impl_detail(cpython=False): # everywhere except on CPython
)r" r] rp Zpython_implementationr )r# r$ rd rd re rB � s c s, t td�s� S tj� �� fdd��}|S dS )zEDecorator to temporarily turn off tracing for the duration of a test.�gettracec s. t j� }zt jd � � | |�S t j|� X d S )N)ro r( �settrace)r� r� Zoriginal_trace)r� rd re r� � s
zno_tracing..wrapperN)r� ro r� r� )r� r� rd )r� re �
no_tracing� s
r* c C s t t| ��S )a Decorator for tests which involve reference counting.
To start, the decorator does not run the test if is not run by CPython.
After that, any trace function is unset during the test to prevent
unexpected refcounts caused by the trace function.
)r* r5 )r� rd rd re �
refcount_test� s r+ c C sR g }xB| j D ]8}t|tj�r2t||� |j|� q ||�r |j|� q W || _ dS )z>Recursively filter test cases in a suite based on a predicate.N)Z_testsr� rs � TestSuite�
_filter_suiter� )�suiteZpredZnewtestsr� rd rd re r- � s
r- c C s� t tjttdk d�}|j| �}tdk r4tj|j� � |js>t �|j
� s�t |j �dkrl|j
rl|j d d }n6t |j
�dkr�|j r�|j
d d }nd}ts�|d7 }t|��dS )z2Run tests from a unittest.TestSuite-derived class.N)� verbosityZcapture_outputr r zmultiple errors occurredz!; run in verbose mode for details)r ro r� r �junit_xml_listr r� Zget_xml_elementZtestsRunr Z
wasSuccessfulr� �errorsZfailuresr
)r. Zrunnerr� r� rd rd re �
_run_suite� s"
r2 c C s t d kr dS t | j� �S d S )NT)�_match_test_func�id)r� rd rd re �
match_test� s r5 c C s d| kot jd| � S )Nry z[?*\[\]])rv �search)r� rd rd re �_is_full_match_test� s r7 c sr | t kr d S | sd }f } nHttt| ��r4t| �j}n.djttj| ��}t j
|�j � � fdd�}|}t | �a |a
d S )N�|c s$ � | �r dS t t� | jd���S d S )NTry )�anyr� r� )Ztest_id)� regex_matchrd re �match_test_regex� s z)set_match_tests..match_test_regex)�_match_test_patterns�allr� r7 r� � __contains__r� �fnmatch� translaterv rN rw rr r3 )Zpatternsr� Zregexr; rd )r: re �set_match_tests� s rA c G s� t jt jf}t j� }xh| D ]`}t|t�rT|tjkrJ|jt jtj| �� qzt d��qt||�rj|j|� q|jt j
|�� qW t |t � t
|� dS )z1Run tests from unittest.TestCase-derived classes.z)str arguments must be keys in sys.modulesN)rs r, ZTestCaser� ru ro rz ZaddTestZ
findTestCasesr� Z makeSuiter- r5 r2 )�classesZ valid_typesr. �clsrd rd re r- s
c C s dS )z,Just used to check if docstrings are enabledNrd rd rd rd re �_check_docstrings( s rD �WITH_DOC_STRINGSztest requires docstringsc C s` ddl }|dkrt}nd}|j| ||d�\}}|rBtd||f ��trXtd| j|f � ||fS )a Run doctest on the given module. Return (#failures, #tests).
If optional argument verbosity is not specified (or is None), pass
support's belief about verbosity on to doctest. Else doctest's
usual behavior is used (it searches sys.argv for -v).
r N)r � optionflagsz%d of %d doctests failedz,doctest (%s) ... %d tests with zero failures)�doctestr Ztestmodr
r� r` )r� r/ rF rG r� r� rd rd re r. 9 s c C s t jj� fS )N)ro rz r� rd rd rd re �
modules_setupT s rH c C s: dd� t jj� D �}t jj� t jj|� t jj| � d S )Nc S s" g | ]\}}|j d �r||f�qS )z
encodings.)rq )rB r� r� rd rd re rC [ s z#modules_cleanup..)ro rz r� rs r� )Z
oldmodulesZ encodingsrd rd re �modules_cleanupW s
rI c C s" t rt j� tjj� fS df fS d S )Nr )�_thread�_count� threading� _danglingr� rd rd rd re rN z s c G sJ t sd S d}x8t|�D ],}t j� tjf}|| kr2P tjd� t� qW d S )N�d g{�G�z�?)rJ �rangerK rL rM r� r� r� )Zoriginal_valuesZ
_MAX_COUNT�countr' rd rd re rO � s
c s" t s� S tj� �� fdd��}|S )z�Use this function when threads are being used. This will
ensure that the threads are cleaned up even when the test fails.
If threading is unavailable this function does nothing.
c s t � }z� | � S t|� X d S )N)rN rO )r� �key)r� rd re r� � s z reap_threads..decorator)rJ r� r� )r� r� rd )r� re rP � s � N@c c s� t j� }z
dV W dtj� }|| }xjt j� }||kr8P tj� |kr|tj� | }d|| � d|d�d|� d|� d� }t|��tjd� t� q&W X dS ) aH
bpo-31234: Context manager to wait until all threads created in the with
statement exit.
Use _thread.count() to check if threads exited. Indirectly, wait until
threads exit the internal t_bootstrap() C function of the _thread module.
threading_setup() and threading_cleanup() are designed to emit a warning
if a test leaves running threads in the background. This context manager
is designed to cleanup threads started by the _thread.start_new_thread()
which doesn't allow to wait for thread exit, whereas thread.Thread has a
join() method.
Nz!wait_threads() failed to cleanup z threads after z.1fz seconds (count: z
, old count: �)g{�G�z�?)rJ rK r� Z monotonicr{ r� r� )r� Z old_countZ
start_timeZdeadlinerP Zdtrx rd rd re �wait_threads_exit� s
$
rT c
C sZ t td�rVd} xFy2tj| tj�\}}|dkr.P td| tjd� W q P Y qX qW dS )z�Use this function at the end of test_main() whenever sub-processes
are started. This will help ensure that no extra children (zombies)
stick around to hog resources and create problems when looking
for refleaks.
�waitpidr r z2Warning -- reap_children() reaped child process %s)r� NrY )r� r� rU �WNOHANGr� ro r� )Z any_processr1 Zstatusrd rd re rL � s
c c s* t | �} g }zZy$x | D ]}|j� |j|� qW W n* trVtdt| �t|�f � � Y nX d V W d z�|rt|� tj� }}xltdd�D ]^}|d7 }x$|D ] }|jt |tj� d�� q�W dd� |D �}|s�P tr�tdt|�|f � q�W W d d d� |D �}|�r"t
j t j
� td
t|� ��X X d S ) Nz/Can't start %d threads, only %d threads startedr r �< g{�G�z�?c S s g | ]}|j � r|�qS rd )�isAlive)rB r� rd rd re rC � s z!start_threads..z7Unable to join %d threads during a period of %d minutesc S s g | ]}|j � r|�qS rd )rX )rB r� rd rd re rC � s zUnable to join %d threads)r| r r� r r� r� r� rO r� �max� faulthandlerZdump_tracebackro r� r{ )ZthreadsZunlockr r� ZendtimeZ starttimer� rd rd re rQ � s>
c
c sn t | |�r t t| ��t t|�� }|r(|t |�8 }t dd� |D ��}|S )a Returns the set of items in ref_api not in other_api, except for a
defined list of items to be ignored in this check.
By default this skips private attributes beginning with '_' but
includes all magic methods, i.e. those starting and ending in '__'.
c s s( | ] }|j d � s |jd�r|V qdS )�_�__N)rq �endswith)rB r� rd rd re � � s z&detect_api_mismatch..)r� rx )Zref_apiZ other_apirf Z
missing_itemsrd rd re r< � s
c C s� |dkr|j f}nt|t�r"|f}t|�}xbt|�D ]V}|jd�s4||krLq4t||�}t|dd�|ks�t|d� r4t|tj � r4|j
|� q4W | j |j |� dS )a Assert that the __all__ variable of 'module' contains all public names.
The module's public names (its API) are detected automatically based on
whether they match the public name convention and were defined in
'module'.
The 'name_of_module' argument can specify (as a string or tuple thereof)
what module(s) an API could be defined in in order to be detected as a
public API. One case for this is when 'module' imports part of its public
API from other modules, possibly a C backend (like 'csv' and its '_csv').
The 'extra' argument can be a set of names that wouldn't otherwise be
automatically detected as "public", like objects without a proper
'__module__' attribute. If provided, it will be added to the
automatically detected ones.
The 'blacklist' argument can be a set of names that must not be treated
as part of the public API even though their names indicate otherwise.
Usage:
import bar
import foo
import unittest
from test import support
class MiscTestCase(unittest.TestCase):
def test__all__(self):
support.check__all__(self, foo)
class OtherTestCase(unittest.TestCase):
def test__all__(self):
extra = {'BAR_CONST', 'FOO_CONST'}
blacklist = {'baz'} # Undocumented name.
# bar imports part of its API from _bar.
support.check__all__(self, bar, ('bar', '_bar'),
extra=extra, blacklist=blacklist)
Nr� ra )
r` r� ru r� rx rq r� r� �types�
ModuleType�addZassertCountEqual�__all__)Z test_caser� Zname_of_moduleZextraZ blacklistZexpectedrv r� rd rd re r= � s )
c @ s( e Zd ZdZdZdZdd� Zdd� ZdS )rZ z�Try to prevent a crash report from popping up.
On Windows, don't display the Windows Error Reporting dialog. On UNIX,
disable the creation of coredump file.
Nc
C sr t jjd�r�ddl}|jj| _d}| jj|�| _| jj| j|B � yddl }|j
W n t t fk
rl Y nLX i | _
x�|j|j|jgD ].}|j
||j�}|j||j�}||f| j
|< q�W n�tdk �r
y*tjtj�| _tjtjd| jd f� W n ttfk
�r Y nX t jdk�rnddd d
g}tj|tj tj d �}|� |j � d } W dQ R X | j � d k�rnt d
ddd� | S )z�On Windows, disable Windows Error Reporting dialogs using
SetErrorMode.
On UNIX, try to save the previous core file size limit, then set
soft limit to 0.
r� r Nr� r r� z/usr/bin/defaultsr^ zcom.apple.CrashReporterZ
DialogType)r� r� s developerz:this test triggers the Crash Reporter, that is intentionalr� T)�endr
) ro rp rq r� r� r� �_k32� SetErrorMode� old_value�msvcrt�CrtSetReportModer� rn � old_modes�CRT_WARN� CRT_ERROR�
CRT_ASSERTZCRTDBG_MODE_FILE�CrtSetReportFileZCRTDBG_FILE_STDERRr� Z getrlimit� RLIMIT_CORE� setrlimitr� r� r r �PIPEZ communicatera r� )
rg r� ZSEM_NOGPFAULTERRORBOXr� � report_type�old_mode�old_file�cmd�procr� rd rd re r� 2
sN
z SuppressCrashReport.__enter__c G s� | j dkrdS tjjd�rl| jj| j � | jr�ddl}xj| jj� D ]$\}\}}|j ||� |j
||� qBW n6t dk r�yt j t j
| j � W n ttfk
r� Y nX dS )zARestore Windows ErrorMode or core file behavior to initial value.Nr� r )r� ro rp rq r� r� r� r� r� r� r� r� r� r� r� r� )rg r� r� r� r� r� rd rd re r� s
s
z SuppressCrashReport.__exit__)r` ra rb rc r� r� r� r� rd rd rd re rZ )
s
Ac sr t ��� d� y�j� �W n$ ttfk
r@ t ��d��Y nX d� � ���fdd�}| j|� t��|� dS )z�Override 'object_to_patch'.'attr_name' with 'new_value'.
Also, add a cleanup procedure to 'test_instance' to restore
'object_to_patch' value for 'attr_name'.
The 'attr_name' should be a valid attribute for 'object_to_patch'.
FNTc s � rt ���� n
t��� d S )N)r� r[ rd )�
attr_is_local� attr_name�object_to_patchr� rd re �cleanup�
s zpatch..cleanup)r� rj r� r� Z
addCleanupr� )Z
test_instancer� r� Z new_valuer� rd )r� r� r� r� re �patch�
s
r� c C sF y ddl }W n tk
r Y nX |j� r4tjd��ddl}|j| �S )zi
Run code in a subinterpreter. Raise unittest.SkipTest if the tracemalloc
module is enabled.
r NzUrun_in_subinterp() cannot be used if tracemalloc module is tracing memory allocations)� tracemallocrn Z
is_tracingrs rt r� �run_in_subinterp)r� r� r� rd rd re r� �
s
r� c sH G � �fdd�d|�}d� |||� ��| j tt�� t� | j� � d S )Nc s e Zd Z� �fdd�ZdS )z%check_free_after_iterating..Ac s* d� y t �� W n tk
r$ Y nX d S )NT)�next�
StopIteration)rg )�done�itrd re �__del__�
s
z-check_free_after_iterating..A.__del__N)r` ra rb r� rd )r� r� rd re �A�
s r� F)rL r� r� r� Z
assertTrue)r� r� rC r� r� rd )r� r� re �check_free_after_iterating�
s r� c C s| ddl m}m}m} |j� }|j|� xP|jD ]F}| r@|| k[email protected]||�}| rPn
|dkrZq.|j|d �dkr.|d S q.W dS )a< Check if the compiler components used to build the interpreter exist.
Check for the existence of the compiler executables whose names are listed
in 'cmd_names' or all the compiler executables when 'cmd_names' is empty
and return the first missing executable or None when none is found
missing.
r )� ccompilerr� �spawnN) Z distutilsr� r� r� Z new_compilerZcustomize_compilerZ executablesr� Zfind_executable)Z cmd_namesr� r� r� Zcompilerrv r� rd rd re r^ �
s
c C s@ d}t r6| |k r6td kr.tjddg�j� dkatr6|} tj| �S )Ng�h㈵��>Zgetpropzro.kernel.qemu� 1)rA �_is_android_emulatorr Z check_outputra ro rD )ZintervalZminimum_intervalrd rd re rD �
s c
c s> t jj� } tj� }ztj� d V W d |r8tj| dd� X d S )NT)r� Z all_threads)ro r� rG rZ �
is_enabledr� r� )rA r� rd rd re �disable_faulthandler�
s
r� c / C s� t jjd �r8ytjd�} t| �d S tk
r6 Y nX d}ttd�rjytjd�}W n t k
rh Y nX d}t jd kr�yd
dl
}|j W n t t
fk
r� Y n0X i }x(|j|j|jfD ]}|j |d
�||< q�W zpd
}xft|�D ]Z}ytj|�}W n4 t k
�r( } z|jtjk�r� W Y dd}~X q�X tj|� |d7 }q�W W d|dk �rzx*|j|j|jfD ]}|j ||| � �q`W X |S ) z/Count the number of open file descriptors.
�linux�freebsdz
/proc/self/fdr � �sysconf� SC_OPEN_MAXNr r )r� r� )ro rp rq r� r� r� r� r� r� r� r� r� r� rn r� r� r� rO �dupr� ZEBADFr ) �namesZMAXFDr� r� r� rP rA Zfd2r� rd rd re r_ sP
c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � SaveSignalsz�
Save an restore signal handlers.
This class is only able to save/restore signal handlers registered
by the Python signal module: see bpo-13285 for "external" signal
handlers.
c C sj dd l }|| _ ttd|j��| _x>dD ]6}yt||�}W n tk
rN w&Y nX | jj|� q&W i | _d S )Nr r �SIGKILL�SIGSTOP)r� r� ) �signalr| rO �NSIG�signalsr� r� rz rf )rg r� Zsigname�signumrd rd re rh M s
zSaveSignals.__init__c C s4 x.| j D ]$}| jj|�}|d kr"q|| j|< qW d S )N)r� r� � getsignalrf )rg r� �handlerrd rd re �saveZ s
zSaveSignals.savec C s* x$| j j� D ]\}}| jj||� q W d S )N)rf r� r� )rg r� r� rd rd re �restoref s zSaveSignals.restoreN)r` ra rb rc rh r� r� rd rd rd re r� D s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FakePathz.Simple implementing of the path protocol.
c C s
|| _ d S )N)r� )rg r� rd rd re rh n s zFakePath.__init__c C s d| j �d�S )Nz
)r� )rg rd rd re �__repr__q s zFakePath.__repr__c C s6 t | jt�s$t | jt�r,t| jt�r,| j�n| jS d S )N)r� r� �
BaseExceptionr� ry )rg rd rd re �
__fspath__t s
zFakePath.__fspath__N)r` ra rb rc rh r� r� rd rd rd re r� k s r� c c s. t j� }zt j| � dV W dt j|� X dS )z>Temporarily change the integer string conversion length limit.N)ro �get_int_max_str_digits�set_int_max_str_digits)Z
max_digitsZcurrentrd rd re �adjust_int_max_str_digits| s
r� c C s dd� t dd�D �dg S )z�Returns a list of C0 control characters as strings.
C0 control characters defined as the byte range 0x00-0x1F, and 0x7F.
c S s g | ] }t |��qS rd )�chr)rB r� rd rd re rC � s z)control_characters_c0..r � �)rO rd rd rd re �control_characters_c0� s r� )T)F)F)N)Ni i @ i @ i @ i i ) r r r r r r r r r r! r" )r& r$ r' r( r) )NF)F)r6 F)N)Fi @ i i )T)N)Nr )rR )N( rc r` rn �collections.abc� collections�
contextlibZdatetimer� rZ r? r� r� r� rm �importlib.utilr� Zlogging.handlersre r� r� rp rv r� r r� r� r ro r� r+ r� r� rs Z urllib.errorr` rg Z
testresultr rJ rL Zmultiprocessing.processr& �zlibrW �bz2Zlzmar� r� r� r� r r
r rt r �contextmanagerrk r
r r� r: r; r r6 r r r r r0 r r� r r r r� rq r� r� r� r� r� r r� r r� r r� r r! r� r" r# r$ r% rE r r r
rG rH rJ r rF r r Z
SOCK_MAX_SIZEZ
skipUnlessr �
__getformat__r7 r9 r0 r1 r2 r@ r� r rA rC rv r r� r/ Z FS_NONASCII� character�fsdecode�fsencode� UnicodeErrorZTESTFN_UNICODEZ unicodedata� normalize�getfilesystemencodingZTESTFN_ENCODINGZTESTFN_UNENCODABLEZgetwindowsversion�encode�UnicodeEncodeErrorr� �decode�UnicodeDecodeErrorZTESTFN_UNDECODABLEZTESTFN_NONASCIIr3 r r] r} r2 r5 r r� rK r� r� r� r� ZTEST_SUPPORT_DIRr: r� r\ r r r[ rH r& rI �objectrd r� rR r� r� rS r �abc�MutableMappingrT r� r' r� r� r( r� r) r* r+ r� r r r r� r� r� r� r� r� r� r� r� r� r� rU r\ r� r� r Z_4Gr r rY r r3 r4 r, r r r> r5 r! r% r? r" rB r* r+ r- r2 r3 r< r5 r7 rA r- rD ZMISSING_C_DOCSTRINGSZHAVE_DOCSTRINGSZrequires_docstringsr. rH rI Zenvironment_alteredrN rO rP rT rL rQ rW rV rb ZskipIfZrequires_type_collectingrc rd rf rg rM rX rr r r/ rz r| r8 r~ r r< r= rZ r� r� r� r^ r� rD r� r_ r� r� r� r� rd rd rd re � s�
2
!
J > %
% 2 ' 5 M
$
#
0
(
"
#
"
:_" ;'