返回
Warning: You may not have write permissions to the file; saving may fail.
�qc Q � � @ s� d Z e d k r e d � � d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l Z d d l! Z! Wn e k r�d Z d Z! Yn Xy d d l" Z# Wn e k r�d Z# Yn Xy d d l$ Z$ Wn e k r�d Z$ Yn Xy d d l% Z% Wn e k rd Z% Yn Xy d d l& Z& Wn e k r>d Z& Yn Xy d d l' Z' Wn e k rid Z' Yn Xy d d l( Z( Wn e k r�d Z( Yn Xd d d d d d d d d d d d d d d d d d d d d d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 d: d; d< d= d> d? d@ dA dB dC dD dE dF dG dH dI dJ dK dL dM dN dO dP dQ dR dS dT dU gQ Z) GdV d � d e* � Z+ GdW d � d e+ � Z, GdX d � d e j- � Z. e j/ dY dZ d[ � � Z0 d\ d] f d^ d �Z1 d_ d` � Z2 da db � Z3 dc d9 � Z4 dd d: � Z5 f f d\ de d � Z6 df d5 � Z7 dg Z8 d Z9 d a: d a; d\ Z< d a= d a> dh d � Z? di d � Z@ dj d � ZA dk dl � ZB e j jC dm � r�d\ dn do � ZD dp dq � ZE dr ds � ZF dt du � ZG n e jH ZE e jI ZF dv du � ZG dw d � ZH dx dy � ZI dz d � ZJ d{ d| � ZK d} d � ZL d~ d � ZM d� d � ZN d d� d! � ZO d� d� � ZP d� d" � ZQ d� d# � ZR d� d$ � ZS d� ZT d� ZU e jV e jW d� dA � ZX eT d� dB � ZY d� d� � ZZ eZ � Z[ d� d� � Z\ d;Z] d>Z^ e j_ e` ja d� � jC d� � d� � Zb e j_ e$ d� � Zc e j_ e% d� � Zd e j_ e& d� � Ze e j_ e' d� � Zf e j jC d� � Zg e jh d� k r�d� Zi n d� Zi d� jj ei e jk � � Zi d Zl xD d?D]< Zm y e jn e jo em � � Wn ep k r�Yq�Xem Zl Pq�Wei d� Zq e j d� k r<d d lr Zr er js d� eq � Zq e jt � Zu d Zv e jh d@k r�e jw � j d� k r ei d� Zv y ev jx eu � Wn ey k r�Yq Xez d� ev eu f � d Zv nK e j d� k r y d� j{ eu � Wn( e| k r ei d� j{ eu d� � Zv Yn Xd Z} xH dAD]@ Zh y eh j{ eu � Wqe| k rVe jo ei � eh Z} PYqXqWel rrei d� el Z~ n d Z~ e j � Z� e j/ d d\ d� d� � � Z� e j/ d\ d� d� � � Z� e j/ d� d\ d� d � � Z� e� e d� � r�e j/ d� dD � � Z� e j� j� e j� j� e� � � Z� e j� j� e� � Z� e j� j� e� d� � Z� d d� d � Z� d� d � Z� d� dT � Z� d� d� � Z� d� d d� d d� d% � Z� d� dC � Z� Gd� d� � d� e� � Z� d\ d� d� � Z� e j/ d� dK � � Z� e j/ d� dL � � Z� Gd� d � d e� � Z� Gd� dM � dM e j� j� � Z� Gd� d� � d� e� � Z� Gd� d& � d& e� � Z� e� e� d� e j� �Z� e� e� d� e j� �Z� e� e� d� e j� �Z� e j/ d� d� d� f d� d* � � Z� e j/ d� d� � � Z� d� d � Z� d� d � Z� d� d � Z� d� d� � Z� e j/ d� d� � � Z� d� d� � Z� d� Z� d� Z� e� e d� � r0 d� e� Z� d� Z� e� d� Z� d� d� � Z� d� d� � Z� dBZ� dCZ� d� d� � Z� d� dN � Z� d� dU � Z� dDZ� d� e� Z� d� e� Z� d� e� Z� e j� Z� d� dR � Z� Gd� d� � d� � Z� dY d� d2 � Z� d� d3 � Z� Gd� d+ � d+ � Z� d� d� � Z� d� d� � Z� d� d4 � Z� d d d� Z� d a� dd< � Z� dd� Z� dd> � Z� dd� Z� dd � Z� d d � Z� d d � Z� dd� Z� dd, � Z� dd� Z� e� � o� e j dk o� e j� d� Z� e� j d k o� e� Z� e j_ e� d� Z� d d dd- � Z� dd� Z� dd� Z� ddG � Z� d dH � Z� d dI � Z� d dE � Z� e j/ d d dJ � � Z� e j/ d dP � � Z� e j/ d!dO � � Z� d"d#� Z� e j� e� e d$� d%� Z� d&d'� Z� Gd(dF � dF e j� j� � Z� Gd)dQ � dQ e� � Z� d a� d*d � Z� d+d. � Z� d a� d,d-� Z� d.d7 � Z� d/d � Z� d0f d1d; � Z� Gd2dS � dS � Z� d3d4� Z� d5d6� Z� f d7d8� Z� d S(E z7Supporting definitions for the Python regression tests.z test.supportz.support must be imported from the test package� N� PIPE_MAX_SIZE�verbose� max_memuse� use_resources�failfast�Error� TestFailed�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�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� requires_multiprocessing_queue� is_jython�check_impl_detail�HOST� IPV6_ENABLED�find_unused_port� bind_port�open_urlresource� 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_tzc @ s e Z d Z d Z d S)r z*Base class for regression test exceptions.N)�__name__� __module__� __qualname__�__doc__� rW rW �:/opt/alt/python35/lib64/python3.5/test/support/__init__.pyr n s c @ s e Z d Z d Z d S)r z Test failed.N)rS rT rU rV rW rW rW rX r q s c @ s e Z d Z d Z d S)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)rS rT rU rV rW rW rW rX r t s Tc c s> | r5 t j � � t j d d t � d VWd QRXn d Vd S)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)rY rW rW rX �_ignore_deprecated_imports| s r^ F� required_onc C s� t | � �n y t j | � SWnU t k ru } z5 t j j t | � � rN � t j t | � � � WYd d } ~ Xn XWd QRXd S)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) r^ � importlibr � ImportError�sys�platform� startswith�tuple�unittest�SkipTest�str)�name� deprecatedr_ �msgrW rW rX r � s c C s{ | t j k r# t | � t j | =xQ t t j � D]@ } | | k sX | j | d � r3 t j | | | <t j | =q3 Wd S)zyHelper function to save and remove a module from sys.modules Raise ImportError if the module can't be imported. �.N)rb �modules� __import__�listrd )ri � orig_modules�modnamerW rW rX �_save_and_remove_module� s rr c C sG d } y t j | | | <Wn t k r5 d } Yn Xd t 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)rb rm �KeyError)ri rp ZsavedrW rW rX �_save_and_block_module� s rt c C s | r t j Sd d � 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)NrW )�frW rW rX �<lambda>� s z$anticipate_failure.<locals>.<lambda>)rf ZexpectedFailure)Z conditionrW rW rX r6 � s c C sk | d k r d } t j j t j j t j j t � � � } | j d | d | 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_testsrw Ztop_dirZ package_testsrW rW rX r7 � s c C s� t | � �� i } g } t | | � z} y^ x | D] } t | | � q3 Wx* | D]" } t | | � sQ | j | � qQ Wt j | � } Wn t k r� d } Yn XWd x' | j � D] \ } } | t j | <q� Wx | D] } t j | =q� WX| SWd QRXd S)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) r^ rr rt �appendr` r ra �itemsrb rm ) ri ZfreshZblockedrj rp Znames_to_removeZ fresh_nameZ blocked_nameZ fresh_moduleZ orig_name�moduleZname_to_removerW rW rX r � s$ c C sI y t | | � } Wn+ t k r@ t j d | | f � � Yn X| Sd S)z?Get an attribute, raising SkipTest if AttributeError is raised.z object %r has no attribute %rN)�getattr�AttributeErrorrf rg )�objri Z attributerW rW rX r2 s � c C s | a d S)N)�_original_stdout)�stdoutrW rW rX r s c C s t p t j S)N)r� rb r� rW rW rW rX r # s c C s' y t j | =Wn t k r" Yn Xd S)N)rb rm rs )ri rW rW rX r & s c G s� y | | � SWn t k r� } z_ t d k r` t d | j j | f � t d | j | f � t j | t j � | | � SWYd d } ~ Xn Xd S)N� z%s: %sz re-run %s%r) �OSErrorr �print� __class__rS rx �chmod�stat�S_IRWXU)ry �func�args�errrW rW rX � _force_run, s r� �winc C s� | | � | r | } n$ t j j | � \ } } | p: d } d } xR | d k r� t j | � } | rm | n | | k s} d St j | � | d 9} qF Wt j d | t d d �d S)Nrl g����MbP?g �?r� z)tests may fail, delete still pending for � stacklevel� ) rx ry �split�listdir�time�sleeprZ �warn�RuntimeWarning)r� �pathname�waitallrz ri �timeout�LrW rW rX �_waitfor7 s r� c C s t t j | � d S)N)r� rx r )�filenamerW rW rX �_unlinkX s r� c C s t t j | � d S)N)r� rx �rmdir)rz rW rW rX �_rmdir[ s r� c s< � f d d � � t � | d d �t d d � | � d S)Nc s� x� t | t j | � D]� } t j j | | � } y t j | � j } WnH t k r� } z( t d | | f d t j �d } WYd d } ~ Xn Xt j | � r� t � | d d �t | t j | � q t | t j | � q Wd S)Nz-support.rmtree(): os.lstat(%r) failed with %s�filer r� T)r� rx r� ry �join�lstat�st_moder� r� rb � __stderr__r� �S_ISDIRr� r� r )ry ri �fullname�mode�exc)� _rmtree_innerrW rX r� _ s z _rmtree.<locals>._rmtree_innerr� Tc S s t | t j | � S)N)r� rx r� )�prW rW rX rv n s z_rmtree.<locals>.<lambda>)r� )ry rW )r� rX �_rmtree^ s r� c sW y t j | � d SWn t k r) Yn X� f d d � � � | � t j | � d S)Nc s� x� t | t j | � D]� } t j j | | � } y t j | � j } Wn t k ra d } Yn Xt j | � r� � | � t | t j | � q t | t j | � q Wd S)Nr ) r� rx r� ry r� r� r� r� r� r� r� r )ry ri r� r� )r� rW rX r� z s z _rmtree.<locals>._rmtree_inner)�shutilr r� rx r� )ry rW )r� rX r� s s c C s- y t | � Wn t t f k r( Yn Xd S)N)r� �FileNotFoundError�NotADirectoryError)r� rW rW rX r � s c C s' y t | � Wn t k r" Yn Xd S)N)r� r� )rz rW rW rX r� � s r� c C s' y t | � Wn t k r" Yn Xd S)N)r� r� )ry rW rW rX r � s c C s] t j j | � } t j j t j j | � � } t j j | | d � } t j | | � | 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) r` �util�cache_from_sourcerx ry rz �abspathr� �rename)�sourceZpyc_fileZup_oneZ legacy_pycrW rW rX �make_legacy_pyc� s r� c C sv t | � xe t j D]Z } t j j | | d � } t | d � x* d D]" } t t j j | d | �� qH Wq Wd S) 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 rb ry rx r� r r` r� r� )rq rz r� �optrW rW rX r � s c s� t t d � r t j Sd } t j j d � rd d l � d d l � d } d } G� f d d � d � j � } � j j } | j � } | s� � j � � | � } � j j � } | j | | � j | � � j | � � j | � � } | s� � j � � t | j | @� s�d } n� t j d k r�d d l m } m � m } m } d d l m } | j | d � � } | j � d k r�d } na G� f d d � d | � } | � } | | � } | j | � d k s�| j | � d k r�d } | s�y; d d l m } | � } | j � | j � | j! � Wnr t" k r�} zR t# | � } t$ | � d k rq| d d � d } d j% t& | � j' | � } WYd d } ~ Xn X| t _( | t _ t j S)N�resultr� r r� c sC e Z d Z d � j j f d � j j f d � j j f g Z d S)z*_is_gui_available.<locals>.USEROBJECTFLAGSZfInheritZ fReserved�dwFlagsN)rS rT rU �wintypesZBOOL�DWORD�_fields_rW )�ctypesrW rX �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 Z d Z d � f d � f g Z d S)z._is_gui_available.<locals>.ProcessSerialNumberZ highLongOfPSNZ lowLongOfPSNN)rS rT rU r� rW )r� rW rX �ProcessSerialNumber� s r� z#cannot run without OS X gui process)�Tk�2 z [...]z Tk unavailable due to {}: {}))�hasattr�_is_gui_availabler� rb rc rd r� Zctypes.wintypesr� �windllZuser32ZGetProcessWindowStationZWinErrorr� r� ZGetUserObjectInformationWZbyrefZsizeof�boolr� r� r� r� Z ctypes.utilr� Z LoadLibraryZCGMainDisplayIDZGetCurrentProcessZSetFrontProcessZtkinterr� Zwithdraw�updateZdestroy� Exceptionrh �len�format�typerS �reason)r� Z UOI_FLAGSZ WSF_VISIBLEr� Zdll�hZuofZneeded�resr� r� r� r� Z app_servicesr� ZpsnZpsn_pr� �root�eZ err_stringrW )r� r� rX r� � sh " r� c C s t d k p | t k S)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 )�resourcerW rW rX r s c C sW t | � s. | d k r" d | } t | � � | d k rS t � rS t t j � � d S)z@Raise ResourceDenied if the specified resource is not available.Nz"Use of the %r resource not enabled�gui)r r r� r� )r� rk rW rW rX r s c s � � f d d � } | 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 s4 t j � � � � � f d d � � } � | _ | S)Nc s� t j � � k r� t j � j d d � d } y" t t t | j d � � � } Wn t k rd YnA X| � k r� d j t t � � � } t j d � | | f � � � | | � S)N�-r� r rl z(%s version %s or higher required, not %s) rc �system�releaser� re �map�int� ValueErrorr� rh rf rg )r� �kw� version_txt�version�min_version_txt)r� � min_version�sysnamerW rX �wrapper s " z:_requires_unix_version.<locals>.decorator.<locals>.wrapper)� functools�wrapsr� )r� r� )r� r� )r� rX � decorator s ' z)_requires_unix_version.<locals>.decoratorrW )r� r� r� rW )r� r� rX �_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� rW rW rX 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. ZLinux)r� )r� rW rW rX r 7 s c s � f d d � } | 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 s1 t j � � � � f d d � � } � | _ | S)Nc s� t j d k r� t j � d } y" t t t | j d � � � } Wn t k rU Yn> X| � k r� d j t t � � � } t j d | | f � � � | | � S)Nr� r rl z&Mac OS X %s or higher required, not %s) rb rc Zmac_verre r� r� r� r� r� rh rf rg )r� r� r� r� r� )r� r� rW rX r� H s " z4requires_mac_ver.<locals>.decorator.<locals>.wrapper)r� r� r� )r� r� )r� )r� rX r� G s $ z#requires_mac_ver.<locals>.decoratorrW )r� r� rW )r� rX 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. )�socketr? �close)�familyZsocktypeZtempsock�portrW rW rX r> b s 8 c C s | j t j k r� | j t j k r� t t d � r] | j t j t j � d k r] t d � � t t d � r� d t j k r� y. | j t j t j � d k r� t d � � Wn t k r� Yn Xt t d � r� | j t j t j d � | j | d f � | 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_REUSEPORTZWITHIN_PYTHON_RPM_BUILDzHtests 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 rx �environr� r� Z setsockoptr� �bindZ getsockname)�sock�hostr� rW rW rX r? � s $ c C ss t j ro d } zL y3 t j t j t j � } | j t d f � d SWn t k rY Yn XWd | rn | j � Xd S)z+Check whether IPv6 is enabled on this host.Nr TF)r� Zhas_ipv6ZAF_INET6r� r� �HOSTv6r� r� )r� rW rW rX �_is_ipv6_enabled� s r� c s% t j � � � f d d � � } | S)z5Skip the test on TLS certificate validation failures.c sa y � | | � WnI t k r\ } z) d t | � k rG t j d � � � WYd d } ~ Xn Xd S)NZCERTIFICATE_VERIFY_FAILEDz.system does not contain necessary certificates)�IOErrorrh rf rg )r� �kwargsr� )ru rW rX �dec� s z&system_must_validate_cert.<locals>.dec)r� r� )ru r� rW )ru rX �system_must_validate_cert� s ! r� r� i � ZdoubleZIEEEz test requires IEEE 754 doublesz requires zlibz requires gzipz requires bz2z requires lzma�javaz$testz@testz {}_{}_tmp� æ� İ� Ł� φ� К� א� ،� ت� ก� � €u -àòɘŁğr� ZNFD�nt�cer� 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 } | d k r9 t j � } d } t j j | � } nO y t j | � d } Wn5 t k r� | si � t j d | t d d �Yn Xz | VWd | r� t | � Xd S)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� � ) �tempfileZmkdtemprx ry �realpath�mkdirr� rZ r� r� r )ry �quietZ dir_createdrW rW rX �temp_dir� s" r c c sy t j � } y t j | � Wn5 t k rT | s6 � t j d | t d d �Yn Xz t j � VWd t j | � Xd S)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)rx �getcwd�chdirr� rZ r� r� )ry r Z saved_dirrW rW rX � change_cwd� s r Ztempcwdc c sD t d | d | � �( } t | d | �� } | VWd QRXWd QRXd S)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. ry r N)r r )ri r Z temp_pathZcwd_dirrW rW rX r � s �umaskc c s- t j | � } z d VWd t j | � Xd S)z8Context manager that temporarily sets the process umask.N)rx r )r ZoldmaskrW rW rX rA � s �datac C s� t j j | � r | S| d k r7 t j j | | � } t g t j } x9 | D]1 } t j j | | � } t j j | � rN | SqN W| 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)rx ry �isabsr� � TEST_HOME_DIRrb �exists)r� Zsubdirry Zdn�fnrW rW rX r � s c C s4 t j | t j t j Bt j B� } t j | � d S)z>Create an empty file. If the file already exists, truncate it.N)rx �open�O_WRONLY�O_CREAT�O_TRUNCr� )r� �fdrW rW rX r � s #c C s< t | j � � } d d � | D� } d j | � } d | S)z%Like repr(dict), but in sorted order.c S s g | ] } d | � q S)z%r: %rrW )�.0ZpairrW rW rX � <listcomp> s z sortdict.<locals>.<listcomp>z, z{%s})�sortedr~ r� )�dictr~ Z reprpairsZ withcommasrW rW rX rQ s c C s9 t t d � } z | j � SWd | j � t t � Xd S)z` Create an invalid file descriptor by opening and closing a file and return its fd. �wbN)r$ r �filenor� r )r� rW rW rX � make_bad_fd s r/ �lineno�offsetc C s� | j t � � } t | d d � Wd QRX| j } | j | j � | d k ra | j | j | � | j | j � | d k r� | j | j | � d S)Nz <test string>�exec)� assertRaises� SyntaxError�compileZ exceptionZassertIsNotNoner0 � assertEqualr1 )�testcaseZ statementr0 r1 �cmr� rW rW rX r" s c s� d d l } d d l } � j d d � � | j j | � d j d � d } t j j t | � } � � � f d d � } t j j | � r� | | � } | d k r� | St | � t d � t r� t d | d t � �| j j � } t r� | j j d � | j | d d �} t rC| j j d � d k rCt j d | � } zO t | d � �9 } | j � } x# | r�| j | � | j � } qgWWd QRXWd | j � X| | � } | d k r�| St d | � � d S)Nr �checkr� �/r� c sM t | � � � } � d k r" | S� | � r? | j d � | S| j � d S)Nr )r$ �seekr� )r# ru )r� r9 r� rW rX �check_valid_file) s z*open_urlresource.<locals>.check_valid_fileZurlfetchz fetching %s ...r� �Accept-Encoding�gzipr� � zContent-EncodingZfileobjr- zinvalid resource %r���)r= r> ) Zurllib.requestZ urllib.parse�pop�parseZurlparser� rx ry r� � TEST_DATA_DIRr" r r r r� r ZrequestZ build_openerr> Z addheadersr} r$ Zheaders�getZGzipFile�read�writer� r ) Zurlr� r� �urllibr� r# r< ru Zopener�out�srW )r� r9 r� rX r@ s< # c @ sL e Z d Z d Z d d � Z d d � Z e d d � � Z d d � Z d 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_listrW rW rX �__init__V s zWarningsRecorder.__init__c C s\ t | j � | j k r, t | j d | � S| t j j k rB d St d | | f � � d S)Nr� z%r has no attribute %rr@ )r� rK rL r� rZ �WarningMessage�_WARNING_DETAILSr� )rM �attrrW rW rX � __getattr__Z s z WarningsRecorder.__getattr__c C s | j | j d � S)N)rK rL )rM rW rW rX rZ a s zWarningsRecorder.warningsc C s t | j � | _ d S)N)r� rK rL )rM rW rW rX �resete s zWarningsRecorder.resetN) rS rT rU rV rN rR �propertyrZ rS rW rW rW rX rJ R s rJ c c s^ t j d � } | j j d � } | r1 | j � t j d d � �& } t j d j d � t | � VWd QRXt | � } g } x� | D]� \ } } d } xd | d d � D]R } | j } t j | t | � t j � r� t | j | � r� d } | j | � q� W| r� | r� | j | | j f � q� W| r@t d | d � � | rZt d | d � � d S) 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__�recordTrZ �alwaysNFzunhandled warning %sr z)filter (%r, %s) did not catch any warning)rb � _getframe� f_globalsrD �clearrZ r[ rm � simplefilterrJ ro �message�re�matchrh �I� issubclassr� �remover} rS �AssertionError) �filtersr �frame�registry�wZreraiseZmissingrk �cat�seenZwarningrW rW rX �_filterwarningsi s0 rh c O sC | j d � } | s6 d t f f } | d k r6 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 r� NT)rD �Warningrh )rb r� r rW rW rX rH � s c c sO t j d d � �& } t j d d t �d Vt � Wd QRX| j | g � d S)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. rU TrV �categoryN)rZ r[ r\ �ResourceWarning� gc_collectr6 )r7 ZwarnsrW rW rX rI � s c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) 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 si t j j � | _ xP | D]H } | t j k r t j | } | j | k rW t j | j =t j | =q Wd S)N)rb rm �copy�original_modulesrS )rM Z module_namesZ module_namer rW rW rX rN � s zCleanImport.__init__c C s | S)NrW )rM rW rW rX � __enter__� s zCleanImport.__enter__c G s t j j | j � d S)N)rb rm r� rn )rM � ignore_excrW rW rX �__exit__� s zCleanImport.__exit__N)rS rT rU rV rN ro rq rW rW rW rX r � s c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d S)rJ 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)rx r� �_environ�_changed)rM rW rW rX rN � s z EnvironmentVarGuard.__init__c C s | j | S)N)rr )rM �envvarrW rW rX � __getitem__� s z EnvironmentVarGuard.__getitem__c C s9 | | j k r( | j j | � | j | <| | j | <d S)N)rs rr rD )rM rt �valuerW rW rX � __setitem__� s z EnvironmentVarGuard.__setitem__c C sE | | j k r( | j j | � | j | <| | j k rA | j | =d S)N)rs rr rD )rM rt rW rW rX � __delitem__� s z EnvironmentVarGuard.__delitem__c C s | j j � S)N)rr �keys)rM rW rW rX ry � s zEnvironmentVarGuard.keysc C s t | j � S)N)�iterrr )rM rW rW rX �__iter__� s z EnvironmentVarGuard.__iter__c C s t | j � S)N)r� rr )rM rW rW rX �__len__� s zEnvironmentVarGuard.__len__c C s | | | <d S)NrW )rM rt rv rW rW rX �set� s zEnvironmentVarGuard.setc C s | | =d S)NrW )rM rt rW rW rX �unset s zEnvironmentVarGuard.unsetc C s | S)NrW )rM rW rW rX ro s z EnvironmentVarGuard.__enter__c G se xR | j j � D]A \ } } | d k rD | | j k rQ | j | =q | | j | <q W| j t _ d S)N)rs r~ rr rx r� )rM rp �k�vrW rW rX rq s z EnvironmentVarGuard.__exit__N)rS rT rU rV rN ru rw rx ry r{ r| r} r~ ro rq rW rW rW rX rJ � s c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) � 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 s6 t j d d � | _ t j | _ t j j | � d S)N)rb ry �original_value�original_object�extend)rM �pathsrW rW rX rN s zDirsOnSysPath.__init__c C s | S)NrW )rM rW rW rX ro # s zDirsOnSysPath.__enter__c G s& | j t _ | j t j d d � <d S)N)r� rb ry r� )rM rp rW rW rX rq & s zDirsOnSysPath.__exit__N)rS rT rU rV rN ro rq rW rW rW rX r� s r� c @ sC e Z d Z d Z d d � Z d d � Z d d d d d � Z d S) 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)rM r� r� rW rW rX rN 0 s zTransientResource.__init__c C s | S)NrW )rM rW rW rX ro 4 s zTransientResource.__enter__Nc C st | d k rp t | j | � rp xO | j j � D]2 \ } } t | | � sJ Pt | | � | k r. Pq. Wt d � � d S)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� r~ r� r� r )rM Ztype_rv � tracebackrQ Z attr_valuerW rW rX rq 7 s zTransientResource.__exit__)rS rT rU rV rN ro rq rW rW rW rX r# + s �errnor� g >@�errnosc # s� d d d! d" d# g } d% d' d) d+ d, g } t d | � � | � g � � sr d d � | D� � d d � | D� � � � � f d d � } t j � } z!y" | d k r� t j | � d VWn� t j k r} z/ t r� t j j � j d d � � | � WYd d } ~ Xn� t k r�} z� xr | j } t | � d k r_t | d t � r_| d } q$t | � d k r�t | d t � r�| d } q$Pq$W| | � � WYd d } ~ Xn XWd t j | � Xd S)-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 | | � � q SrW )r� r� )r) ri �numrW rW rX r* d s z&transient_internet.<locals>.<listcomp>c S s( g | ] \ } } t t | | � � q SrW )r� r� )r) ri r� rW rW rX r* f s c s� t | d d � } t | t j � s� t | t j � rB | � k s� t | t j j � rv d | j k oq d k n s� t | t j j � r� d | j k s� d | j k s� d | j k s� | � k r� t s� t j j � j d d � � | � d S) Nr� i� iW �ConnectionRefusedError� TimeoutError�EOFErrorr � )r� � isinstancer� r� ZgaierrorrG �errorZ HTTPError�codeZURLErrorr� r rb �stderrrF r� )r� �n)�captured_errnos�denied� gai_errnosrW rX � filter_errori s z(transient_internet.<locals>.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� ZgetdefaulttimeoutZsetdefaulttimeout�nntplibZNNTPTemporaryErrorr rb r� rF r� r� r� r� ) Z resource_namer� r� Zdefault_errnosZdefault_gai_errnosr� Z old_timeoutr� �arW )r� r� r� rX r' L sN % % c c s[ d d l } t t | � } t t | | j � � z t t | � VWd t t | | � Xd S)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� rb �setattr�StringIO)Z stream_namer� Z orig_stdoutrW rW rX �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� rW rW rW rX 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� rW rW rW rX 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� rW rW rW rX r � s c C s5 t j � t r t j d � t j � t j � d S)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� rW rW rW rX rl � s rl c c s7 t j � } t j � z d VWd | r2 t j � Xd S)N)r� � isenabled�disable�enable)Zhave_gcrW rW rX � disable_gc� s r� c C sQ t j d � p d } d } x) | j � D] } | j d � r( | } q( W| d k S)z,Find if Python was built with optimizations.� PY_CFLAGSr� z-O�-O0�-Og)r� r� r� )� sysconfig�get_config_varr� rd )ZcflagsZ final_optr� rW rW rX �python_is_optimized� s r� ZnPZ0n�gettotalrefcountZ2PZ0Pr� c C s t j t | t � S)N)�struct�calcsize�_header�_align)�fmtrW rW rX � calcobjsize� s r� c C s t j t | t � S)N)r� r� �_vheaderr� )r� rW rW rX � calcvobjsize� s r� � � c C s� d d l } t j | � } t | � t k r: | j t @s_ t | � t k rl t | � j t @rl | | j 7} d t | � | | f } | j | | | � d S)Nr z&wrong size for %s: got %d, expected %d) � _testcapirb � getsizeofr� � __flags__�_TPFLAGS_HEAPTYPE�_TPFLAGS_HAVE_GCZSIZEOF_PYGC_HEADr6 )�test�o�sizer� r� rk rW rW rX � check_sizeof� s % r� c s � � f d d � } | S)Nc s4 � � � f d d � } � j | _ � j | _ | S)Nc s� y. d d l } t | � � } | j | � } Wn& t k rE � YnC d } } Yn2 Xx. � D]& } y | j | | � PWq^ Yq^ Xq^ Wz � | | � SWd | r� | r� | j | | � Xd S)Nr )�localer� � setlocaler� )r� �kwdsr� rj Z orig_locale�loc)�catstrr� �localesrW rX �inner s$ z1run_with_locale.<locals>.decorator.<locals>.inner)rS rV )r� r� )r� r� )r� rX r� s z"run_with_locale.<locals>.decoratorrW )r� r� r� rW )r� r� rX rK s c s � f d d � } | S)Nc s1 � � f d d � } � j | _ � j | _ | S)Nc s� y t j } Wn! t k r0 t j d � � Yn Xd t j k rP t j d } n d } � t j d <| � z � | | � SWd | d k r� t j d =n | t j d <t j � Xd S)Nztzset requiredZTZ)r� �tzsetr� rf rg rx r� )r� r� r� Zorig_tz)r� �tzrW rX r� - s z-run_with_tz.<locals>.decorator.<locals>.inner)rS rV )r� r� )r� )r� rX r� , s z run_with_tz.<locals>.decoratorrW )r� r� rW )r� rX rR + s c C s� d d d t d t d d t i } t j d | t j t j B� } | d k r` t d | f � � t t | j d � � | | j d � j � � } | a | t k r� t } | t d k r� t d | f � � | a d S) Nr i �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 )�limitZsizesr� ZmemlimitrW rW rX rO U s 2 c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �_MemoryWatchdogz`An object which periodically watches the process' memory consumption and prints it out. c C s( d j d t j � � | _ d | _ d S)Nz/proc/{pid}/statm�pidF)r� rx �getpid�procfile�started)rM rW rW rX rN o s z_MemoryWatchdog.__init__c C s� y t | j d � } WnO t k rg } z/ t j d j | � t � t j j � d SWYd d } ~ Xn Xt d � } t j t j | g d | d t j �| _ | j � d | _ d S)N�rz!/proc not available for stats: {}zmemory_watchdog.pyr� r� T)r$ r� r� rZ r� r� r� rb r� �flushr � subprocess�Popen� executableZDEVNULL� mem_watchdogr� r� )rM ru r� Zwatchdog_scriptrW rW rX �starts s z_MemoryWatchdog.startc C s' | j r# | j j � | j j � d S)N)r� r� Z terminate�wait)rM rW rW rX �stop� s z_MemoryWatchdog.stopN)rS rT rU rV rN r� r� rW rW rW rX r� j s r� c s � � � f d d � } | 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. � � � f d d � � � � _ � � _ � S)Nc s� � j } � j } t s! d } n | } t s4 � r_ t | | k r_ t j d | | d � � t r� t r� t � t d j d | | d � � t � } | j � n d } z � | | � SWd | r� | j � Xd S) Ni z'not enough memory: %.1fG minimum neededi r z* ... expected peak memory use: {peak:.1f}GZpeaki @i @) r� �memuser� rf rg r r� r� r� r� r� )rM r� r� �maxsizeZwatchdog)�dry_runru r� rW rX r� � s* z.bigmemtest.<locals>.decorator.<locals>.wrapper)r� r� )ru )r� r� r� )ru r� rX r� � s z bigmemtest.<locals>.decoratorrW )r� r� r� r� rW )r� r� r� rX r/ � s !c s � f d d � } | S)z0Decorator for tests that fill the address space.c s^ t t k rP t d k r6 t d k r6 t j d � � qZ t j d t d � � n � | � Sd 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� rf rg )rM )ru rW rX r� � s z!bigaddrspacetest.<locals>.wrapperrW )ru r� rW )ru rX r0 � s c @ s e Z d Z d d � Z d S)r( c C s t j � } | | � | S)N)rf Z TestResult)rM r� r� rW rW rX �run� s zBasicTestRunner.runN)rS rT rU r� rW rW rW rX r( � s c C s | S)NrW )r� rW rW rX �_id� s r� c C sP | d k r& t � r& t j t j � St | � r6 t St j d j | � � Sd S)Nr� z resource {0!r} is not enabled)r� rf �skipr� r r� r� )r� rW rW rX �requires_resource� s r c C s t d d � | � S)z9 Decorator for tests only applicable on CPython. �cpythonT)� impl_detail)r� rW rW rX r1 � s c K sz t | � r t S| d k rm t | � \ } } | r= d } n d } t | j � � } | j d j | � � } t j | � S)Nz*implementation detail not available on {0}z%implementation detail specific to {0}z or ) r; r� � _parse_guardsr+ ry r� r� rf r� )rk �guardsZ guardnames�defaultrW rW rX r � s r c C sj t d k rG d d l } y | j � d a Wn t k rF d a Yn Xd } t rW | St j | � | � S)z8Skip decorator for tests that use multiprocessing.Queue.Nr TFz6requires a functioning shared semaphore implementation)�_have_mp_queue�multiprocessingZQueuera rf r� )r� r rk rW rW rX r9 � s c C sb | s d d i d f St | j � � d } t | j � � | g t | � k sW t � | | f S)Nr TFr )ro �valuesr� ra )r Zis_truerW rW rX r � s +r c K s. t | � \ } } | j t j � 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 rD rc Zpython_implementationr� )r r rW rW rX r; s c s< t t d � s � St j � � � f d d � � } | Sd S)zEDecorator to temporarily turn off tracing for the duration of a test.�gettracec s? t j � } z t j d � � | | � SWd t j | � Xd S)N)rb r �settrace)r� r� Zoriginal_trace)r� rW rX r� s zno_tracing.<locals>.wrapperN)r� rb r� r� )r� r� rW )r� rX � 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 r1 )r� rW rW rX � refcount_test# s r c C so g } xY | j D]N } t | t j � rE t | | � | j | � q | | � r | j | � q W| | _ d S)z>Recursively filter test cases in a suite based on a predicate.N)Z_testsr� rf � TestSuite� _filter_suiter} )�suiteZpredZnewtestsr� rW rW rX r . s r c C s� t r' t j t j d d d t �} n t � } | j | � } | j � s� t | j � d k r~ | j r~ | j d d } nI t | j � d k r� | j r� | j d d } n d } t s� | d 7} t | � � d S) z2Run tests from a unittest.TestSuite-derived class.� verbosityr� r r� r zmultiple errors occurredz!; run in verbose mode for detailsN) r rf ZTextTestRunnerrb r� r r( r� Z wasSuccessfulr� �errorsZfailuresr )r Zrunnerr� r� rW rW rX � _run_suite: s r c C sw t d k r d S| j � } xT t D]L } t j | | � r? d Sx- | j d � D] } t j | | � rO d SqO Wq# Wd S)NTrl F)� match_tests�id�fnmatchZ fnmatchcaser� )r� Ztest_idZ match_testri rW rW rX � _match_testN s r c G s� t j t j f } t j � } x� | D]� } t | t � rx | t j k ri | j t j t j | � � q� t d � � q% t | | � r� | j | � q% | j t j | � � q% Wt | t � t | � d S)z1Run tests from unittest.TestCase-derived classes.z)str arguments must be keys in sys.modulesN)rf r ZTestCaser� rh rb rm ZaddTestZ findTestCasesr� Z makeSuiter r r )�classesZ valid_typesr �clsrW rW rX r) _ s c C s d S)z,Just used to check if docstrings are enabledNrW rW rW rW rX �_check_docstringsv s r �win32�WITH_DOC_STRINGSztest requires docstringsc C s� d d l } | d k r! t } n d } | j | d | d | �\ } } | rd t d | | f � � t r� t d | j | f � | | f S)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 Nr � optionflagsz%d of %d doctests failedz,doctest (%s) ... %d tests with zero failures)�doctestr Ztestmodr r� rS )r r r r ru r� rW rW rX r* � s !c C s t j j � f S)N)rb rm rm rW rW rW rX � modules_setup� s r c C sM d d � t j j � D� } t j j � t j j | � t j j | � d S)Nc S s1 g | ]' \ } } | j d � r | | f � q S)z encodings.)rd )r) r r� rW rW rX r* � s z#modules_cleanup.<locals>.<listcomp>)rb rm r~ rY r� )Z oldmodules� encodingsrW rW rX �modules_cleanup� s r c C s- t r t j � t j j � f Sd f f Sd S)Nr� )�_thread�_count� threading� _danglingrm rW rW rW rX rD � s c G sa t s d Sd } xJ t | � D]< } t j � t j f } | | k rE Pt j d � t � q Wd S)N�d g{�G�z�?)r! �ranger" r# r$ r� r� rl )Zoriginal_valuesZ _MAX_COUNT�countr rW rW rX rE � s c s/ t s � St j � � � f d d � � } | 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 � | � SWd t | � Xd S)N)rD rE )r� �key)r� rW rX r� � s z reap_threads.<locals>.decorator)r! r� r� )r� r� rW )r� rX rF � s !c C sX t t d � rT d } x< y, t j | t j � \ } } | d k rC PWq PYq Xq Wd S)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� rx r) �WNOHANG)Z any_processr� ZstatusrW rW rX rB � s c c sv t | � } g } zi y, x% | D] } | j � | j | � q WWn1 t rm t d t | � t | � f � � Yn Xd VWd z� | r� | � t j � } } x� t d d � D]{ } | d 7} x. | D]& } | j t | t j � d � � q� Wd d � | D� } | s Pt r� t d t | � | f � q� WWd d d � | D� } | rpt j t j � t d t | � � � XXd S) Nz/Can't start %d threads, only %d threads startedr� r �< g{�G�z�?c S s" g | ] } | j � r | � q SrW )�isAlive)r) r� rW rW rX r* s z!start_threads.<locals>.<listcomp>z7Unable to join %d threads during a period of %d minutesc S s" g | ] } | j � r | � q SrW )r, )r) r� rW rW rX r* s zUnable to join %d threads)ro r� r} r r� r� r� r& r� �max� faulthandlerZdump_tracebackrb r� ra )ZthreadsZunlockr� r� ZendtimeZ starttimer� rW rW rX rG � s> $ c c s� t | | � rN t | | � } t | | | � z | VWd t | | | � Xn9 t | | | � z d VWd t | | � r� t | | � Xd S)a� 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. The old value (or None if it doesn't exist) will be assigned to the target of the "as" clause, if there is one. N)r� r� r� �delattr)r� rQ �new_val�real_valrW rW rX rM s c c sh | | k r: | | } | | | <z | VWd | | | <Xn* | | | <z d VWd | | k rc | | =Xd S)a� 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. The old value (or None if it doesn't exist) will be assigned to the target of the "as" clause, if there is one. NrW )r� �itemr0 r1 rW rW rX rL : s c C s t j d d | � 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, \d+ blocks\]\r?\n?� )r\ �sub�strip)r� rW rW rX �strip_python_stderrY s r6 Z getcountsz-types are immortal if COUNT_ALLOCS is definedc C s t j � S)znReturn a list of command-line arguments reproducing the current settings in sys.flags and sys.warnoptions.)r� Z _args_from_interpreter_flagsrW rW rW rX �args_from_interpreter_flagsf s r7 c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S) rC c C s# t j j j | d � | | _ d S)Nr )�logging�handlers�BufferingHandlerrN �matcher)rM r; rW rW rX rN p s zTestHandler.__init__c C s d S)NFrW )rM rW rW rX � shouldFlushy s zTestHandler.shouldFlushc C s$ | j | � | j j | j � d S)N)r� �bufferr} �__dict__)rM rU rW rW rX �emit| s zTestHandler.emitc K s: d } x- | j D]" } | j j | | � r d } Pq W| S)zW Look for a saved dict whose keys/values match the supplied arguments. FT)r= r; �matches)rM r� r� �drW rW rX r@ � s zTestHandler.matchesN)rS rT rU rN r< r? r@ rW rW rW rX rC o s c @ s. e Z d Z d Z d d � Z d d � Z d S) rN rk r[ c K sP d } xC | D]; } | | } | j | � } | j | | | � s d } Pq 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)rD � match_value)rM rA r� r� r r� �dvrW rW rX r@ � s zMatcher.matchesc C sj t | � t | � k r! d } nE t | � t k sB | | j k rQ | | k } n | j | � d k } | S)zT Try to match a single stored value (dv) with a supplied value (v). Fr )r� rh �_partial_matches�find)rM r rC r� r� rW rW rX rB � s !zMatcher.match_valueN)rk r[ )rS rT rU rD r@ rB rW rW rW rX rN � s c C so t d k r t St d } y t j t | � d } Wn! t t t f k rW d } Yn Xt j | � | a | S)Nr TF)� _can_symlinkr rx �symlinkr� �NotImplementedErrorr� r` )Z symlink_path�canrW rW rX r � s c C s, t � } d } | r | St j | � | � S)z8Skip decorator for tests that require functional symlinkz*Requires functional symlink implementation)r rf r� )r� �okrk rW rW rX r+ � s c C s t d k r t St t d � s( d } n� t j � \ } } z� t t d � �� } yu t j | d d � t j | j � d d � t j � } t j d | � } | d k p� t | j d � � d k } Wn t k r� d } Yn XWd QRXWd t t � t | � X| a | S) N�setxattrFr- s user.testr3 z 2.6.(\d{1,2})r� �' )� _can_xattrr� rx r Zmkstempr$ r rK r. rc r� r\ r] r� r� r� r )rI Ztmp_fpZtmp_name�fpZkernel_versionr� rW rW rX � can_xattr� s& + rO c C s, t � } d } | r | St j | � | � S)zDSkip decorator for tests that require functional extended attributesz(no non-broken extended attribute support)rO rf r� )r� rJ rk rW rW rX r4 � s c C s} t j d | � �d } | j } | j � } | | k rB | j � } y t j j | | � SWn t k rq d SYn XWd QRXd S)zKDetects if the file system for the specified directory is case-insensitive.�dirFN) r ZNamedTemporaryFileri �upperr� rx ry �samefiler� )Z directory�base� base_pathZ case_pathrW rW rX r � s rY c C sU t t | � � t t | � � } | r8 | t | � 8} t d d � | 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 s4 | ]* } | j d � s( | j d � r | Vq d S)�_�__N)rd �endswith)r) r� rW rW rX � <genexpr>� s z&detect_api_mismatch.<locals>.<genexpr>)r} rP )Zref_apiZ other_apirY Z missing_itemsrW rW rX r8 � s "c @ s: e Z d Z d Z d Z d Z d d � Z d d � Z d S)rP 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 s� t j j d � r� d d l } | j j | _ d } | j j | � | _ | j j | j | B� y d d l } | j Wn t t f k r� Yq�Xi | _ xH| j | j | j g D]C } | j | | j � } | j | | j � } | | f | j | <q� Wn� t d k r]y9 t j t j � | _ t j t j d | j d f � Wn t t f k r\Yn Xt j d k r�d d d d g } t j | d t j d t j �} | � | j � d } Wd QRX| j � d k r�t d d d d d �| 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/defaultsrE zcom.apple.CrashReporterZ DialogTyper� r� s developerz:this test triggers the Crash Reporter, that is intentional�endr� r� T) rb rc rd r� r� Zkernel32�_k32� SetErrorMode� old_value�msvcrt�CrtSetReportModer� ra � old_modesZCRT_WARNZ CRT_ERRORZ CRT_ASSERTZCRTDBG_MODE_FILE�CrtSetReportFileZCRTDBG_FILE_STDERRr� Z getrlimit� RLIMIT_CORE� setrlimitr� r� r� r� �PIPEZ communicater5 r� ) rM r� ZSEM_NOGPFAULTERRORBOXr] � report_type�old_mode�old_file�cmd�procr� rW rW rX ro sN z SuppressCrashReport.__enter__c G s� | j d k r d St j j d � r� | j j | j � | j r� d d l } x� | j j � D]2 \ } \ } } | j | | � | j | | � q] WnA t d k r� y t j t j | j � Wn t t f k r� Yn Xd S)zARestore Windows ErrorMode or core file behavior to initial value.Nr� r )r\ rb rc rd rZ r[ r_ r] r~ r^ r` r� rb ra r� r� )rM rp r] rd re rf rW rW rX rq L s " z SuppressCrashReport.__exit__)rS rT rU rV r\ r_ ro rq rW rW rW rX rP s Ac s� t � � � d � y � j � � Wn* t t f k rP t � � d � � Yn Xd � � � � � f d d � } | j | � t � � | � d S)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* � r t � � � � n t � � � d S)N)r� r/ rW )� attr_is_local� attr_name�object_to_patchr\ rW rX �cleanupw s zpatch.<locals>.cleanup)r� r> r� rs Z addCleanupr� )Z test_instancerk rj Z new_valuerl rW )ri rj rk r\ rX �patcha s rm c C sY y d d l } Wn t k r$ Yn X| j � r@ t j d � � d d l } | 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)� tracemallocra Z is_tracingrf rg r� �run_in_subinterp)r� rn r� rW rW rX ro � s ro c sb G� � f d d � d | � } d � | | | � � � | j t t � � t � | j � � d S)Nc s% e Z d Z � � f d d � Z d S)z%check_free_after_iterating.<locals>.Ac s- d � y t � � Wn t k r( Yn Xd S)NT)�next� StopIteration)rM )�done�itrW rX �__del__� s z-check_free_after_iterating.<locals>.A.__del__N)rS rT rU rt rW )rr rs rW rX �A� s ru F)r3 rq rp rl Z assertTrue)r� rz r r� ru rW )rr rs rX �check_free_after_iterating� s rv i i @ i @ i @ i i ) r r r r r r r r r r r )r r )r r r r r i @ i i )�rV rS ra �collections.abc� collections� contextlibr� r. r r� r� r` �importlib.utilZlogging.handlersr8 r� rx rc r\ r� r� r� r� r� rb r� r r� rf Z urllib.errorrG rZ r! r# Zmultiprocessing.processr �zlibr> �bz2Zlzmar� �__all__r� r r rg r �contextmanagerr^ r rr rt r6 r7 r r2 r r r r� r r r� r r r r� rd 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 Z SOCK_MAX_SIZEZ skipUnlessr� � __getformat__r3 r5 r, r- r. r: ri r r� r� Z FS_NONASCII� character�fsdecode�fsencode� UnicodeErrorZTESTFN_UNICODEZ unicodedata� normalize�getfilesystemencodingZTESTFN_ENCODINGZTESTFN_UNENCODABLEZgetwindowsversion�encode�UnicodeEncodeErrorr� �decode�UnicodeDecodeErrorZTESTFN_UNDECODABLEZTESTFN_NONASCIIr r r r r r� rA ry rz r� r{ ZTEST_SUPPORT_DIRr! r� rC r r rQ r/ r" r@ �objectrJ rh rH rI r �abc�MutableMappingrJ r� r# r� r� r$ r� r% r& r' r� r r r rl r� r� r� r� r� r� r� r� r� r� rK rR r� r� r� Z_4Gr� r� rO r� r/ r0 r( r� r r1 r r r9 r r; r r r r r r) r r� ZMISSING_C_DOCSTRINGSZHAVE_DOCSTRINGSZrequires_docstringsr* r r rD rE rF rB rG rM rL r6 ZskipIfZrequires_type_collectingr7 r9 r: rC rN rF r r+ rM rO r4 r r8 rP rm ro rv rW rW rW rX �<module> sv 2 ! J >& ! 2' 5 J $ # 0 # " _ "
保存