Your IP : 216.73.216.213
��Yf� � @ s# d 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 m Z d d l m
Z
d d l m Z m
Z
m Z m Z m Z d d l m Z Gd d � d e � Z Gd d
� d
e � Z Gd d � d � Z Gd
d � d e � Z d d � Z d d � Z d d � Z d S)z&Support code for distutils test cases.� N)�deepcopy)�log)�DEBUG�INFO�WARN�ERROR�FATAL)� Distributionc sX e Z d Z � f d d � Z � f d d � Z d d � Z d d � Z d d
� Z � S) �LoggingSilencerc sM t � j � t j t j � | _ t j j | _ | j t j _ g | _ d S)N)
�super�setUpr �
set_thresholdr � threshold�Log�_log�_old_log�logs)�self)� __class__� �</opt/alt/python35/lib64/python3.5/distutils/tests/support.pyr s
zLoggingSilencer.setUpc s0 t j | j � | j t j _ t � j � d S)N)r r
r r r r r �tearDown)r )r r r r s zLoggingSilencer.tearDownc C sv | t t t t t f k r1 t d t | � � � t | t � sY t d t | � j
� � | j j | | | f � d S)Nz%s wrong log levelz msg should be str, not '%.200s')
r r r r r �
ValueError�str�
isinstance� TypeError�type�__name__r �append)r �level�msg�argsr r r r s zLoggingSilencer._logc s& d d � } � f d d � | j D� S)Nc S s | | S)Nr )r r! r r r �_format) s z)LoggingSilencer.get_logs.<locals>._formatc s/ g | ]% \ } } } | � k r | | � q Sr r )�.0r r r! )�levelsr r �
<listcomp>+ s z,LoggingSilencer.get_logs.<locals>.<listcomp>)r )r r$ r" r )r$ r �get_logs( s zLoggingSilencer.get_logsc C s
g | _ d S)N)r )r r r r �
clear_logs. s zLoggingSilencer.clear_logs)r �
__module__� __qualname__r r r r& r' r r )r r r
s
r
c sd e Z d Z d Z � f d d � Z � f d d � Z d d � Z d d d
� Z d d d
� Z � S)�TempdirManagerz}Mix-in class that handles temporary directories for test cases.
This is intended to be used with unittest.TestCase.
c s) t � j � t j � | _ g | _ d S)N)r r �os�getcwd�old_cwd�tempdirs)r )r r r r 8 s
zTempdirManager.setUpc sY t j | j � t � j � x5 | j rT | j j � } t j | t j d k � q Wd S)N�nt�cygwin)r/ r0 )
r+ �chdirr- r r r. �pop�shutilZrmtree�name)r �d)r r r r = s
zTempdirManager.tearDownc C s t j � } | j j | � | S)zjCreate a temporary directory that will be cleaned up.
Returns the path of the directory.
)�tempfile�mkdtempr. r )r r5 r r r r7 F s zTempdirManager.mkdtempZxxxc
C sY t | t t f � r' t j j | � } t | d � } z | j | � Wd | j � Xd S)zWWrites a file in the given path.
path can be a string or a sequence.
�wN) r �list�tupler+ �path�join�open�write�close)r r; Zcontent�fr r r �
write_fileO s zTempdirManager.write_fileZfooc K sG | j � } t j j | | � } t j | � t d | � } | | f S)a Will generate a test environment.
This function creates:
- a Distribution instance using keywords
- a temporary directory with a package structure
It returns the package directory and the distribution
instance.
Zattrs)r7 r+ r; r< �mkdirr )r Zpkg_name�kwZtmp_dirZpkg_dirZdistr r r � create_dist] s
zTempdirManager.create_dist) r r( r) �__doc__r r r7 rA rD r r )r r r* 2 s r* c @ s. e Z d Z d Z d d � Z d d � Z d S)� DummyCommandzAClass to store options for retrieval via set_undefined_options().c K s1 x* | j � D] \ } } t | | | � q
Wd S)N)�items�setattr)r �kwargsrC �valr r r �__init__r s zDummyCommand.__init__c C s d S)Nr )r r r r �ensure_finalizedv s z DummyCommand.ensure_finalizedN)r r( r) rE rK rL r r r r rF o s rF c s4 e Z d Z � f d d � Z � f d d � Z � S)� EnvironGuardc s) t t | � j � t t j � | _ d S)N)r rM r r r+ �environ� old_environ)r )r r r r | s zEnvironGuard.setUpc s� xB | j j � D]1 \ } } t j j | � | k r | t j | <q Wx6 t t j j � � D] } | | j k r[ t j | =q[ Wt t | � j � d S)N)
rO rG r+ rN �getr: �keysr rM r )r �key�value)r r r r � s zEnvironGuard.tearDown)r r( r) r r r r )r r rM z s rM c C s8 t � } | d k r$ t j d � � t j | | � d S)ak Helper for tests that need the xxmodule.c source file.
Example use:
def test_compile(self):
copy_xxmodule_c(self.tmpdir)
self.assertIn('xxmodule.c', os.listdir(self.tmpdir))
If the source file can be found, it will be copied to *directory*. If not,
the test will be skipped. Errors during copy are not caught.
Nz>cannot find xxmodule.c (test must run in the python build dir))�_get_xxmodule_path�unittestZSkipTestr3 �copy)Z directory�filenamer r r �copy_xxmodule_c� s rX c C s� t j d � } t j j t j j t � d � t j j | d d � t j j | d d d d d � g } x$ | D] } t j j | � rm | Sqm Wd S)N�srcdirz
xxmodule.c�Modulesz..)� sysconfig�get_config_varr+ r; r< �dirname�__file__�exists)rY Z
candidatesr; r r r rT � s $
rT c C s� t j d k r' t j j d � | _ n� t j d � r� t j d � } | d k r` d g | _ nU t j d k r{ g | _ n: | j
d � \ } } } d d
� | j t j � D� | _ d S) a� Function needed to make build_ext tests pass.
When Python was built with --enable-shared on Unix, -L. is not enough to
find libpython<blah>.so, because regrtest runs in a tempdir, not in the
source directory where the .so lives.
When Python was built with in debug mode on Windows, build_ext commands
need their debug attribute set, and it is not done automatically for
some reason.
This function handles both of these things. Example use:
cmd = build_ext(dist)
support.fixup_build_ext(cmd)
cmd.ensure_finalized()
Unlike most other Unix platforms, Mac OS X embeds absolute paths
to shared libraries into executables, so the fixup is not needed there.
r/ z_d.exe�Py_ENABLE_SHARED� RUNSHAREDN�.�darwin�=c S s g | ] } | r | � q Sr r )r# r5 r r r r% � s z#fixup_build_ext.<locals>.<listcomp>)
r+ r4 �sys�
executable�endswith�debugr[ r\ Z library_dirs�platform� partition�split�pathsep)�cmdZ runsharedr4 ZequalsrS r r r �fixup_build_ext� s rn )rE r+ re r3 r6 rU r[ rV r Z distutilsr Z
distutils.logr r r r r Zdistutils.corer �objectr
r* rF rM rX rT rn r r r r �<module> s" (#=