3
+��h� � � @ sP d Z ddlmZmZmZmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZ ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZeed�s�ddlZddl Z ddl Z ddl Z ddl Z y ddl Z W n" e!k
�r ddl"m Z Y nX ddl#Z#ddl$Z%ddl&Z&ddl'Z'y.ej(�r>ddl)Z)ddl*Z*nddl+Z)ddl*Z*W n e!k
�rn dZ)dZ*Y nX y ddl,Z-W n e!k
�r� dZ-Y nX y ddl.Z.W n e!k
�r� dZ.Y nX y ddl/Z/W n e!k
�r� dZ/Y nX y ddl0Z0W n e!k
�r dZ0Y nX y ddl1Z1W n e!k
�r6 dZ1Y nX dd 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;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQgJZ2G dRd� de3�Z4G dSd � d e4�Z5G dTd
� d
ej6�Z7ej8d�dVdW��Z9d�dYd �Z:dZd[� Z;d\d]� Z<d^dL� Z=f f dXfd_dH�Z>d`d?� Z?daZ@dZAdaBdaCdXZDdZEdaFdbd� ZGdcd� ZHddd� ZIejjJde��r�d�dfdg�ZKdhdi� ZLdjdk� ZMdldm� ZNnejOZLejPZMejQZNdnd� ZOdodp� ZPdqd� ZQejjJde��rddlRZRddlSZRdrds� ZTndtds� ZTdud� ZUd�dvd�ZVdwdx� ZWdyd� ZXdzd� ZYd{d� ZZd|Z[d}Z\ej]ej^fd~d�Z_e[fdd�Z`d�d�� Zaea� Zbd�Zcd�Zdejee.d��Zfejee0d��Zgejee1d��ZhejjJd��Ziejjd�k�r�d�Zknd�Zkd�jlekejm� �Zkejn� Zoej8d�d�d!��Zpeed���rej8d�d2� �Zqerdfd�d"�Zsd�d#� Ztd�d$� Zud�d�� Zvd�d%� Zwd�d&� ZxG d�d�� d�ey�Zzd�d�d��Z{ej8d�d'� �Z|G d�d(� d(ey�Z}ej(�r�ddl~Zej�j�Z�nddl�Z�e�j�Z�G d�d)� d)e��Z�G d�d�� d�ey�Z�G d�d*� d*ey�Z�e�e�ej�d��Z�e�ej�ej�d��Z�e�e�ej�d��Z�ej8d�f fd�d3��Z�ej8d�d�� �Z�d�d+� Z�d�d-� Z�d�d,� Z�d�d�� Z�ej8d�d�� �Z�d�d�� Z�d�Z�d�Z�eed���rld�e� Z�d�Z�e�d� Z�d�d�� Z�d�d�� Z��d Z��dZ�d�d�� Z�d�d1� Z�d�dM� Z��dZ�d�e� Z�d�e� Z�d�e� Z�ej�Z�d�d4� Z�G d�d d�ey�Z��dd�d5�Z�d�d6� Z�G d�d7� d7ey�Z�d�dDŽ Z�d�dɄ Z�d�d=� Z��dd�d̄Z�d�d΄ Z�d�d>� Z�d�dф Z�d�dӄ Z�d�dՄ Z�d�dׄ Z�d�d8� Z��dd�d9�Z�d�dۄ Z�d�d݄ Z�d�d:� Z�d�d;� Z�d�d� Z�d�d<� Z�ej8d�dA� �Z�ej8d�d@� �Z�d�d� Z�d�d� Z�G d�dC� dCe%j�jÃZ�G d�dD� dDey�Z�da�d�dE� Z�d�dF� Z�da�d�d� Z�d�dG� Z�ejjJde��r6ej8d�dQ� �Z�nej8d�dQ� �Z�d�d� Z�dS ( zwSupporting definitions for the Python regression tests.
Backported for python-future from Python 3.3 test/support.py.
� )�absolute_import�division�print_function�unicode_literals)�utils)�str�range�open�int�map�listN�skip)� sysconfig�Error�
TestFailed�ResourceDenied�
import_module�verbose�
use_resources�
max_memuse�record_original_stdout�get_original_stdout�unload�unlink�rmtreeZforget�is_resource_enabled�requires�requires_freebsd_version�requires_linux_version�requires_mac_ver�find_unused_port� bind_port� IPV6_ENABLED� is_jython�TESTFN�HOST�SAVEDCWD�temp_cwd�findfile�create_empty_file�sortdict�check_syntax_error�open_urlresource�check_warnings� CleanImport�EnvironmentVarGuard�TransientResource�captured_stdout�captured_stdin�captured_stderr�time_out�socket_peer_reset�ioerror_peer_reset�run_with_locale�
temp_umask�transient_internet� set_memlimit�
bigmemtest�bigaddrspacetest�BasicTestRunner� run_unittest� run_doctest�threading_setup�threading_cleanup�
reap_children� cpython_only�check_impl_detail�
get_attribute� swap_item� swap_attrZrequires_IEEE_754� TestHandler�Matcher� can_symlink�skip_unless_symlink�skip_unless_xattr�import_fresh_module�
requires_zlib�
PIPE_MAX_SIZE�failfast�anticipate_failure� run_with_tzZ
requires_gzip� requires_bz2�
requires_lzma�suppress_crash_popupc @ s e Zd ZdZdS )r z*Base class for regression test exceptions.N)�__name__�
__module__� __qualname__�__doc__� rZ rZ �D/root/tmp/pip-build-yeh9mt8d/future/future/backports/test/support.pyr a s c @ s e Zd ZdZdS )r z Test failed.N)rV rW rX rY rZ rZ rZ r[ r d 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)rV rW rX rY rZ rZ rZ r[ r g 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)r\ rZ rZ r[ �_ignore_deprecated_importso s
ra Fc C sT t |��B y
tj| �S tk
rD } ztjt|���W Y dd}~X nX W dQ R X dS )z�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.N)ra � importlibr � ImportError�unittest�SkipTestr )�name�
deprecated�msgrZ rZ r[ 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)�sys�modules�
__import__r �
startswith)rf � orig_modules�modnamerZ rZ r[ �_save_and_remove_module� s
rp 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)rj rk �KeyError)rf rn ZsavedrZ rZ r[ �_save_and_block_module� s
rr 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 )NrZ )�frZ rZ r[ �<lambda>� s z$anticipate_failure.<locals>.<lambda>)rd ZexpectedFailure)� conditionrZ rZ r[ rQ � 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 )aV 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.
If deprecated is True, any module or package deprecation messages
will be suppressed.
N)
ra rp rr �appendrb r rc �itemsrj rk ) rf ZfreshZblockedrg rn Znames_to_removeZ
fresh_nameZ blocked_nameZ fresh_module� orig_name�moduleZname_to_removerZ rZ r[ rM � 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�AttributeErrorrd re )�objrf � attributerZ rZ r[ rE � s
� c C s | a d S )N)�_original_stdout)�stdoutrZ rZ r[ r � s c C s
t ptjS )N)r rj r� rZ rZ rZ r[ r � s c C s& y t j| = W n tk
r Y nX d S )N)rj rk rq )rf rZ rZ r[ r s �winc C s� | |� |r|}nt jj|�\}}|p(d}d}x<|dk rjt j|�}|rJ|n||ksVd S tj|� |d9 }q0W tjd| tdd� d S )Nri g����MbP?g �?� z)tests may fail, delete still pending for � )�
stacklevel) �os�path�split�listdir�time�sleepr] �warn�RuntimeWarning)�func�pathname�waitall�dirnamerf �timeout�LrZ rZ r[ �_waitfor s
r� c C s t tj| � d S )N)r� r� r )�filenamerZ rZ r[ �_unlink* s r� c C s t tj| � d S )N)r� r� �rmdir)r� rZ rZ r[ �_rmdir- s r� c s* � fdd�� t � | dd� t tj| � d S )Nc sV xPt j| �D ]B}t jj| |�}t jj|�rDt� |dd� t j|� q t j|� q W d S )NT)r� )r� r� r� �join�isdirr� r� r )r� rf �fullname)�
_rmtree_innerrZ r[ r� 1 s z _rmtree.<locals>._rmtree_innerT)r� )r� r� r� )r� rZ )r� r[ �_rmtree0 s r� c C sH y t | � W n6 tk
rB } z|jtjtjfkr2� W Y d d }~X nX d S )N)r� �OSError�errno�ENOENT�ENOTDIR)r� �errorrZ rZ r[ r @ s
c C sB y t | � W n0 tk
r< } z|jtjkr,� W Y d d }~X nX d S )N)r� r� r� r� )r� r� rZ rZ r[ r� H s
r� c C sB y t | � W n0 tk
r< } z|jtjkr,� W Y d d }~X nX d S )N)r� r� r� r� )r� r� rZ rZ r[ r P s
c C s� d} d}G dd� dt j�}t jj}|j� }|s6t j� �|� }t jj� }|j|| t j |�t j
|�t j |��}|svt j� �t |j |@ �S )Nr~ c @ s. e Zd ZdejjfdejjfdejjfgZdS )z*_is_gui_available.<locals>.USEROBJECTFLAGSZfInheritZ fReserved�dwFlagsN)rV rW rX �ctypes�wintypesZBOOL�DWORD�_fields_rZ rZ rZ r[ �USEROBJECTFLAGS` s
r� )
r� � Structure�windllZuser32ZGetProcessWindowStationZWinErrorr� r� ZGetUserObjectInformationW�byref�sizeof�boolr� )Z UOI_FLAGSZ WSF_VISIBLEr� �dll�hZuof�needed�resrZ rZ r[ �_is_gui_available] s"
r� c C s dS )NTrZ rZ rZ rZ r[ r� s s c C s t dk o| t kS )zPTest whether a resource is enabled. Known resources are set by
regrtest.py.N)r )�resourcerZ rZ r[ r v s c C sX | dkrt � rtjd��tjd�jjd�dkr4dS t| �sT|dkrLd| }t|��dS )z�Raise ResourceDenied if the specified resource is not available.
If the caller's module is __main__ then automatically return True. The
possibility of False being returned occurs when regrtest.py is
executing.
�guiz Cannot use the 'gui' resourcer~ rV �__main__Nz"Use of the %r resource not enabled) r� rd re rj � _getframe� f_globals�getr r )r� rh rZ rZ r[ 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 ri z(%s version %s or higher required, not %s) �platform�system�releaser� �tupler r
�
ValueErrorr� r rd re )�args�kw� version_txt�version�min_version_txt)r� � min_version�sysnamerZ r[ �wrapper� s z:_requires_unix_version.<locals>.decorator.<locals>.wrapper)� functools�wrapsr� )r� r� )r� r� )r� r[ � decorator� s z)_requires_unix_version.<locals>.decoratorrZ )r� r� r� rZ )r� r� r[ �_requires_unix_version� 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� rZ rZ r[ r � 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.
�Linux)r� )r� rZ rZ r[ r � 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 )N�darwinr ri z&Mac OS X %s or higher required, not %s) rj r� �mac_verr� r r
r� r� r� r rd re )r� r� r� r� r� )r� r� rZ r[ r� � s
z4requires_mac_ver.<locals>.decorator.<locals>.wrapper)r� r� r� )r� r� )r� )r� r[ r� � s z#requires_mac_ver.<locals>.decoratorrZ )r� r� rZ )r� r[ 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
socket.error 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.
)�socketr! �close)�family�socktypeZtempsock�portrZ rZ r[ r � s
8 c C s� | j tjkr�| jtjkr�ttd�r>| jtjtj�dkr>t d��ttd�r�y | jtjtj
�dkrft d��W n tj 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_INET�type� SOCK_STREAM�hasattr�
getsockopt�
SOL_SOCKETr� r r� r� �
setsockoptr� �bind� getsockname)�sock�hostr� rZ rZ r[ r! s
c C s^ t jrZd} z@y t j t jt j�} | jd� dS t jt jfk
rF Y nX W d| rX| j� X dS )z+Check whether IPv6 is enabled on this host.N�::1r TF)r� r )r� �has_ipv6�AF_INET6r� r� r� �gaierrorr� )r� rZ rZ r[ �_is_ipv6_enabled> s
r� r� i � z
requires zlibz requires bz2z
requires lzma�javaz$testz@testz {0}_{1}_tmp�tempcwdc c s� t j� }d}|dkrZ| }yt j| � d}W n. tk
rX |s@� tjd| tdd� Y nX yt j|� W n. tk
r� |s~� tjd| tdd� Y nX zt j� V W dt j|� |r�t| � X dS )a�
Context manager that temporarily changes the CWD.
An existing path may be provided as *path*, in which case this
function makes no changes to the file system.
Otherwise, the new CWD is created in the current directory and it's
named *name*. If *quiet* is False (default) and it's not possible to
create or change the CWD, an error is raised. If it's True, only a
warning is raised and the original CWD is used.
FNTz*tests may fail, unable to create temp CWD � )r� z,tests may fail, unable to change the CWD to ) r� �getcwd�mkdirr� r] r� r� �chdirr )rf �quietr� Z saved_dirZ is_temporaryrZ rZ r[ r' � s0
�umaskc c s& t j| �}z
dV W dt j|� X dS )z8Context manager that temporarily sets the process umask.N)r� r� )r� ZoldmaskrZ rZ r[ r8 s
c C sn t jj| �r| S |dk r&t jj|| �} tj}t jj|�g| }x*|D ]"}t jj|| �}t jj|�rD|S qDW | S )z�Try to find a file on sys.path and the working 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).N)r� r� �isabsr� rj r� �exists)�file�here�subdirr� �dn�fnrZ rZ r[ r( & 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� r �O_WRONLY�O_CREAT�O_TRUNCr� )r� �fdrZ rZ r[ r) 5 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: %rrZ )�.0�pairrZ rZ r[ �
<listcomp>= s z sortdict.<locals>.<listcomp>z, z{%s})�sortedrw r� )�dictrw Z reprpairsZ
withcommasrZ rZ r[ 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� rZ rZ r[ � make_bad_fdA s
r c C s | j tt|dd� d S )Nz
<test string>�exec)Z assertRaises� SyntaxError�compile)ZtestcaseZ statementrZ rZ r[ r+ M s
c s ddl m}m} �jdd ��|j| �d jd�d }tjjtjj t
�d|�}� ��fdd �}tjj |�r�||�}|d k r||S t |� t
d
� td | t� d � |j| d
d�}z>t|d��*} |j� }
x|
r�| j|
� |j� }
q�W W d Q R X W d |j� X ||�}|d k �r |S td| ��d S )Nr )�request�parse�checkr� �/r~ �datac s> t | f� ���}�d kr |S �|�r2|jd� |S |j� d S )Nr )r �seekr� )r� rs )r� r r� rZ r[ �check_valid_file[ s
z*open_urlresource.<locals>.check_valid_fileZurlfetchz fetching %s ...)r� � )r� r� zinvalid resource %r���)Zfuture.backports.urllibr r �pop�urlparser� r� r� r� r� �__file__r� r r �printr �urlopenr �read�writer� r ) �urlr� r� �urllib_request� urllib_parser� r� r
rs �out�srZ )r� r r� r[ r, Q s0
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_listrZ rZ r[ �__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 %rr )�lenr r rz r] �WarningMessage�_WARNING_DETAILSr{ )r �attrrZ rZ r[ � __getattr__� s
z WarningsRecorder.__getattr__c C s | j | jd � S )N)r r )r rZ rZ r[ r] � s zWarningsRecorder.warningsc C s t | j�| _d S )N)r r r )r rZ rZ r[ �reset� s zWarningsRecorder.resetN) rV rW rX rY r r" �propertyr] r# rZ rZ rZ r[ r ~ s
r c
c s2 t jd�}|jjd�}|rHtjr*|j� n x tt|��D ] }|j � q8W t
j dd�� }t j d j
d� t|�V W dQ R X t|�}g }xz| D ]r\}} d}
xH|dd� D ]8}|j} tj|t| �tj�r�t| j| �r�d}
|j|� q�W |
r�| r�|j|| jf� q�W |�rtd |d
��|�r.td |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� �__warningregistry__T)�recordr] �alwaysNFzunhandled warning %sr z)filter (%r, %s) did not catch any warning)rj r� r� r� r �PY3�clearr r r
r] r^ rk � simplefilterr r �message�re�matchr �I�
issubclass� __class__�removerv rV �AssertionError) �filtersr� �frame�registry�i�w�reraise�missingrh �cat�seen�warningrZ rZ r[ �_filterwarnings� s6
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)
r� � NT)r� �Warningr= )r3 �kwargsr� rZ rZ r[ r- � 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)rj rk �copy�original_modulesrV )r Z module_names� module_namery rZ rZ r[ r � s
zCleanImport.__init__c C s | S )NrZ )r rZ rZ r[ � __enter__� s zCleanImport.__enter__c G s t jj| j� d S )N)rj rk �updaterB )r �
ignore_excrZ rZ r[ �__exit__� s zCleanImport.__exit__N)rV rW rX rY r rD rG rZ rZ rZ r[ r. � s
c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )r/ 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)r rZ rZ r[ r s z EnvironmentVarGuard.__init__c C s
| j | S )N)rI )r �envvarrZ rZ r[ � __getitem__
s z EnvironmentVarGuard.__getitem__c C s* || j kr | jj|�| j |< || j|< d S )N)rJ rI r� )r rK �valuerZ rZ r[ � __setitem__
s
z EnvironmentVarGuard.__setitem__c C s2 || j kr | jj|�| j |< || jkr.| j|= d S )N)rJ rI r� )r rK rZ rZ r[ � __delitem__ s
z EnvironmentVarGuard.__delitem__c C s
| j j� S )N)rI �keys)r rZ rZ r[ rP s zEnvironmentVarGuard.keysc C s
t | j�S )N)�iterrI )r rZ rZ r[ �__iter__ s z EnvironmentVarGuard.__iter__c C s
t | j�S )N)r rI )r rZ rZ r[ �__len__ s zEnvironmentVarGuard.__len__c C s || |< d S )NrZ )r rK rM rZ rZ r[ �set# s zEnvironmentVarGuard.setc C s
| |= d S )NrZ )r rK rZ rZ r[ �unset& s zEnvironmentVarGuard.unsetc C s | S )NrZ )r rZ rZ r[ rD ) s z EnvironmentVarGuard.__enter__c G sJ x<| j j� D ].\}}|d kr0|| jkr:| j|= q || j|< q W | jt_d S )N)rJ rw rI r� rH )r rF �k�vrZ rZ r[ rG , s
z EnvironmentVarGuard.__exit__N)rV rW rX rY r rL rN rO rP rR rS rT rU rD rG rZ rZ rZ r[ r/ 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)rj r� �original_value�original_object�extend)r �pathsrZ rZ r[ r B s zDirsOnSysPath.__init__c C s | S )NrZ )r rZ rZ r[ rD G s zDirsOnSysPath.__enter__c G s | j t_| jtjd d �< d S )N)rZ rj r� rY )r rF rZ rZ r[ rG J s zDirsOnSysPath.__exit__N)rV rW rX rY r rD rG rZ rZ rZ r[ rX 6 s
rX c @ s* e Zd ZdZdd� Zdd� Zd dd�ZdS )
r0 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)�exc�attrs)r r] r@ rZ rZ r[ r T s zTransientResource.__init__c C s | S )NrZ )r rZ rZ r[ rD X 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)r/ r] r^ rw r� rz r )r �type_rM � tracebackr! �
attr_valuerZ rZ r[ rG [ s
zTransientResource.__exit__)NNN)rV rW rX rY r rD rG rZ rZ rZ r[ r0 O s )r� g >@c # s$ d d d d!d"g}d$d&d(d*d+g}t d| ��|� g �� sPdd� |D �� dd� |D ��� ��fdd�}tj� }z�y |dk r~tj|� dV W n� tk
�r } zjxX|j}t|�d kr�t|d t�r�|d }q�t|�dk� r�t|d t�� r�|d }q�P q�W ||� � W Y dd}~X nX W dtj|� X dS ),z�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 � EAI_AGAINr� �EAI_FAILr� �
EAI_NONAMEr� �
EAI_NODATA� �
WSANO_DATA�* z Resource %r is not availablec S s g | ]\}}t t||��qS rZ )rz r� )r� rf �numrZ rZ r[ r� � s z&transient_internet.<locals>.<listcomp>c S s g | ]\}}t t||��qS rZ )rz r� )r� rf rs rZ rZ r[ r� � s c s` t | dd �}t| tj�s4t| tj�r,|�ks4|� kr\tsNtjj�j d d � �}| |_
|�d S )Nr� r �
) rz �
isinstancer� r� r� r rj �stderrr r� � __cause__)�err�nr] )�captured_errnos�denied�
gai_errnosrZ r[ � filter_error� s z(transient_internet.<locals>.filter_errorNr~ r )rb rc )rd re )rf rg )rh ri )rj rk ���)rl r~ ���)rm r ���)rn r� ���)ro r� )rq rr )r r� �getdefaulttimeout�setdefaulttimeout�IOErrorr� r ru ) �
resource_namer� ZerrnosZdefault_errnosZdefault_gai_errnosr} Z old_timeoutrx �arZ )rz r{ r| r[ r9 p sF
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)�iorz rj �setattr�StringIO)� stream_namer� � orig_stdoutrZ rZ r[ �captured_output� s
r� c C s t d�S )z�Capture the output of sys.stdout:
with captured_stdout() as s:
print("hello")
self.assertEqual(s.getvalue(), "hello")
r� )r� rZ rZ rZ r[ r1 � s c C s t d�S )Nrv )r� rZ rZ rZ r[ r3 � s c C s t d�S )N�stdin)r� rZ rZ rZ r[ r2 � 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� rZ rZ rZ r[ �
gc_collect� 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_gcrZ rZ r[ �
disable_gc� s
r� c C sJ ddl } | jd�pd}d}x |j� D ]}|jd�r$|}q$W |dkoH|dkS )z,Find if Python was built with optimizations.r N� PY_CFLAGSr> z-Oz-O0)r �get_config_varr� rm )r �cflagsZ final_opt�optrZ rZ r[ �python_is_optimized� s
r� ZnPZ0n�gettotalrefcountZ2PZ0Pry c C s t jt| t �S )N)�struct�calcsize�_header�_align)�fmtrZ rZ r[ � calcobjsize� s r� c C s t jt| t �S )N)r� r� �_vheaderr� )r� rZ rZ r[ � calcvobjsize s r� � � c C sh t j|�}t|�tkr |jt@ s:t|�tkrDt|�jt@ rD|tj7 }dt|�||f }| j|||� d S )Nz&wrong size for %s: got %d, expected %d) rj � getsizeofr� � __flags__�_TPFLAGS_HEAPTYPE�_TPFLAGS_HAVE_GCZ _testcapiZSIZEOF_PYGC_HEADZ assertEqual)�test�o�size�resultrh rZ rZ r[ � check_sizeof 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 )�localerz � setlocaler{ )r� �kwdsr� �categoryZ orig_locale�loc)�catstrr� �localesrZ r[ �inner s$
z1run_with_locale.<locals>.decorator.<locals>.inner)rV rY )r� r� )r� r� )r� r[ r� s z"run_with_locale.<locals>.decoratorrZ )r� r� r� rZ )r� r� r[ r7 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{ rd re r� rH )r� r� r� Zorig_tz)r� �tzrZ r[ r� = s
z-run_with_tz.<locals>.decorator.<locals>.inner)rV rY )r� r� )r� )r� r[ r� < s z run_with_tz.<locals>.decoratorrZ )r� r� rZ )r� r[ rR ; s r� 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 )rV �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�_1Gr, r- �
IGNORECASE�VERBOSEr� r
�float�group�lower�real_max_memuse�MAX_Py_ssize_t�_2Gr )�limit�sizesr� �memlimitrZ rZ r[ r: e 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)�pidF)�formatr� �getpid�procfile�started)r rZ rZ r[ r 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: {0}zmemory_watchdog.py)r� rv T)r r� r� r] r� r� r� rj rv �flushr( �
subprocess�Popen�
executable�DEVNULL� mem_watchdogr� r� )r rs �eZwatchdog_scriptrZ rZ r[ �start� s
z_MemoryWatchdog.startc C s | j r| jj� | jj� d S )N)r� r� � terminate�wait)r rZ rZ r[ �stop� s
z_MemoryWatchdog.stopN)rV rW rX rY r r� r� rZ rZ rZ r[ r� z s r� c s � ��fdd�}|S )aA Decorator for bigmem tests.
'minsize' is the minimum useful size for the test (in arbitrary,
test-interpreted units.) 'memuse' is the number of 'bytes per size' for
the test, or a good estimate of it.
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� rd re r r r� r� r� r� )r r� r� �maxsizeZwatchdog)�dry_runrs r� rZ r[ r� � s*
z.bigmemtest.<locals>.decorator.<locals>.wrapper)r� r� )rs )r� r� r� )rs r� r[ r� � s z bigmemtest.<locals>.decoratorrZ )r� r� r� r� rZ )r� r� r� r[ r; � 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� rd re )r )rs rZ r[ r� � s z!bigaddrspacetest.<locals>.wrapperrZ )rs r� rZ )rs r[ r< � s c @ s e Zd Zdd� ZdS )r= c C s t j� }||� |S )N)rd Z
TestResult)r r� r� rZ rZ r[ �run� s zBasicTestRunner.runN)rV rW rX r� rZ rZ rZ r[ r= � s c C s | S )NrZ )r| rZ rZ r[ �_id� s r� c C s: | dkrt � rtjd�S t| �r&tS tjdj| ��S d S )Nr� z resource 'gui' is not availablez resource {0!r} is not enabled)r� rd r
r r� r� )r� rZ rZ r[ �requires_resource� s
r� c C s t dd�| �S )z9
Decorator for tests only applicable on CPython.
T)�cpython)� impl_detail)r� rZ rZ r[ rC � 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 ) rD r� �
_parse_guardsr� rP r� r� rd r
)rh �guardsZ
guardnames�defaultrZ rZ r[ r� � s
r� c C sH | sddidfS t | j� �d }t | j� �|gt| � ks>t�| | fS )Nr� TFr )r �valuesr r2 )r� Zis_truerZ rZ r[ 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� r� �python_implementationr� )r� r� rZ rZ r[ rD 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)rj r� �settrace)r� r@ Zoriginal_trace)r� rZ r[ r� s
zno_tracing.<locals>.wrapperN)r� rj r� r� )r� r� rZ )r� r[ �
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� rC )r� rZ rZ r[ �
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_testsru rd � TestSuite�
_filter_suiterv )�suite�predZnewtestsr� rZ rZ r[ r� + s
r� c C s� t rtjtjdtd�}nt� }|j| �}|j� s�t |j
�dkrV|j rV|j
d d }n6t |j �dkr||j
r||j d d }nd}t s�|d7 }t |��dS )z2Run tests from a unittest.TestSuite-derived class.r� )� verbosityrP r~ r zmultiple errors occurredz!; run in verbose mode for detailsN)
r rd ZTextTestRunnerrj r� rP r= r� Z
wasSuccessfulr �errorsZfailuresr )r� Zrunnerr� rx rZ rZ r[ �
_run_suite7 s
r� 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 dd� }t ||� t |� dS )z1Run tests from unittest.TestCase-derived classes.z)str arguments must be keys in sys.modulesc S s8 t d kr dS x&| j� jd�D ]}tj|t �r dS q W dS )NTri F)� match_tests�idr� �fnmatch� fnmatchcase)r� rf rZ rZ r[ � case_predY s z run_unittest.<locals>.case_predN)
rd r� ZTestCaseru r rj rk ZaddTestZ
findTestCasesr� Z makeSuiter� r� )�classesZ valid_typesr� �clsr� rZ rZ r[ r> K s
c 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 �testmodr r rV )ry r� r� r� rs r� rZ rZ r[ r? r s c C s t jj� fS )N)rj rk rA rZ rZ rZ r[ �
modules_setup� s r� c C sR dd� t jj� D �}x ttt j��D ]}t jj� q$W t jj|� t jj| � d S )Nc S s" g | ]\}}|j d �r||f�qS )z
encodings.)rm )r� rV rW rZ rZ r[ r� � s z#modules_cleanup.<locals>.<listcomp>)rj rk rw r r r
rE )Z
oldmodules� encodingsr6 rZ rZ r[ �modules_cleanup� s
r� c C s t rt j� fS dS d S )Nr~ )r~ )�_thread�_countrZ rZ rZ r[ r@ � s
c C s> t sd S d}x,t|�D ] }t j� }|| kr,P tjd� qW d S )N�
g�������?)r r r r� r� )Z
nb_threadsZ
_MAX_COUNT�country rZ rZ r[ rA � 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)r@ rA )r� �key)r� rZ r[ r� � s z reap_threads.<locals>.decorator)r r� r� )r� r� rZ )r� r[ � reap_threads� s r c C sH t td�rDd} x4y tj| tj�\}}|dkr.P 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 Nr )r� r� r �WNOHANG)Z any_processr� �statusrZ rZ r[ rB � s
c
c sd t | |�r<t| |�}t| ||� z
dV W dt| ||� X n$t| ||� z
dV W dt| |� X dS )am Temporary swap out an attribute with a new object.
Usage:
with swap_attr(obj, "attr", 5):
...
This will set obj.attr to 5 for the duration of the with: block,
restoring the old value at the end of the block. If `attr` doesn't
exist on `obj`, it will be created and then deleted at the end of the
block.
N)r� rz r� �delattr)r| r! �new_val�real_valrZ rZ r[ rG � s
c c sP || kr0| | }|| |< z
dV W d|| |< X n || |< z
dV W d| |= X dS )ak Temporary swap out an item with a new object.
Usage:
with swap_item(obj, "item", 5):
...
This will set obj["item"] to 5 for the duration of the with: block,
restoring the old value at the end of the block. If `item` doesn't
exist on `obj`, it will be created and then deleted at the end of the
block.
NrZ )r| �itemr
r rZ rZ r[ rF s
c C s t jdd| �j� } | S )z�Strip the stderr of a Python process from potential debug output
emitted by the interpreter.
This will typically be run on the result of the communicate() method
of a subprocess.Popen object.
s \[\d+ refs\]\r?\n?� )r, �sub�strip)rv rZ rZ r[ �strip_python_stderr s r c C s t j� S )znReturn a list of command-line arguments reproducing the current
settings in sys.flags and sys.warnoptions.)r� � _args_from_interpreter_flagsrZ rZ rZ r[ �args_from_interpreter_flags* s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
rH c C s t jjj| d� || _d S )Nr )�logging�handlers�BufferingHandlerr �matcher)r r rZ rZ r[ r 4 s zTestHandler.__init__c C s dS )NFrZ )r rZ rZ r[ � shouldFlush= s zTestHandler.shouldFlushc C s | j |� | jj|j� d S )N)r� �bufferrv �__dict__)r r& rZ rZ r[ �emit@ s
zTestHandler.emitc K s. d}x$| j D ]}| jj|f|�r d}P q W |S )zW
Look for a saved dict whose keys/values match the supplied arguments.
FT)r r �matches)r r@ r� �drZ rZ r[ r D s zTestHandler.matchesN)rV rW rX r r r r rZ rZ rZ r[ rH 3 s c @ s e Zd ZdZdd� Zdd� ZdS ) rI rh r+ c K s< d}x2|D ]*}|| }|j |�}| j|||�s
d}P q
W |S )a.
Try to match a single dict with the supplied arguments.
Keys whose values are strings and which are in self._partial_matches
will be checked for partial (i.e. substring) matches. You can extend
this scheme to (for example) do regular expression matching, etc.
TF)r� � match_value)r r r@ r� rV rW �dvrZ rZ r[ r S s
zMatcher.matchesc C sH t |�t |�krd}n.t |�tk s,|| jkr6||k}n|j|�dk}|S )zT
Try to match a single stored value (dv) with a supplied value (v).
Fr )r� r �_partial_matches�find)r rV r rW r� rZ rZ r[ r d s
zMatcher.match_valueN)rh r+ )rV rW rX r r r rZ rZ rZ r[ rI O s c
C sZ t d k r t S td } ytjt| � d}W n tttfk
rF d}Y n X tj| � |a |S )NrJ TF)� _can_symlinkr$ r� �symlinkr� �NotImplementedErrorr{ r1 )Z symlink_path�canrZ rZ r[ rJ r s
c C s t � }d}|r| S tj|�| �S )z8Skip decorator for tests that require functional symlinkz*Requires functional symlink implementation)rJ rd r
)r� �okrh rZ rZ r[ rK � s c C s� t d k r t S ttd�s d} n�tj� \}}z�ttd��r}yRtj|dd� tj|j� dd� t j
� }t j d|�}|d kp�t
|jd��dk} W n tk
r� d} Y nX W d Q R X W d tt� t|� X | a | S ) N�setxattrFr� s user.testr
z
2.6.(\d{1,2})r~ �' )�
_can_xattrr� r� �tempfile�mkstempr r$ r& r� r� r� r, r- r
r� r� r )r$ Ztmp_fpZtmp_name�fpZkernel_versionr� rZ rZ r[ � can_xattr� s&
r, c C s t � }d}|r| S tj|�| �S )zDSkip decorator for tests that require functional extended attributesz(no non-broken extended attribute support)r, rd r
)r� r% rh rZ rZ r[ rL � s c c sH ddl } | jj}d}|j|�}|j||B � z
dV W d|j|� X dS )z;Disable Windows Error Reporting dialogs using SetErrorMode.r Nr� )r� r� �kernel32Z SetErrorMode)r� Zk32ZSEM_NOGPFAULTERRORBOXZold_error_moderZ rZ r[ rU � s
c c s
d V d S )NrZ rZ rZ rZ r[ rU � s c 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 rZ )�
attr_is_local� attr_name�object_to_patch� old_valuerZ r[ �cleanup� s zpatch.<locals>.cleanup)rz r r{ rq Z
addCleanupr� )Z
test_instancer0 r/ Z new_valuer2 rZ )r. r/ r0 r1 r[ �patch� s
r3 )T)F)F)Ni i @ i @ i @ i i )r� FN)Fi @ i i )T)N)Nr )�rY �
__future__r r r r �futurer Zfuture.builtinsr r r r
r r �
contextlibr� r� r� r� rj r� r� �shutilr] rd r� Z unittest2rb r, r� r� r rc � distutilsr� �logging.handlersr r� r) r( r � threading�threadZmultiprocessing.process�multiprocessing�zlib�gzip�bz2�lzma�__all__� Exceptionr r re r �contextmanagerra r rp rr rQ rM rE r r r r� rP r� r r r r rm r� r� r� r� r r� r r� Zctypes.wintypesr� r r r� r r r r% ZHOSTv6r� r� r r! r� r" rO Z
SOCK_MAX_SIZEZ
skipUnlessrN rS rT r# rf r$ r� r� r� r& r' r8 r r( r) r* r r+ r, �objectr r= r- r. �collections.abc� collections�abc�MutableMappingZmybase�UserDictZ DictMixinr/ rX r0 r� rj r4 r� rd r5 r6 r9 r� r1 r3 r2 r� r� r� r� r� r� r� r� r� r� r� r7 rR r� r� r� Z_4Gr� r� r: r� r; r< r= r� r� rC r� r� rD r� r� r� r� r> r? r� r� r@ rA r rB rG rF r r r r rH rI r! rJ rK r( r, rL rU r3 rZ rZ rZ r[ �<module> s�
4
!
> % } '
-
,
5 D $# -
' #
" |