��\T# � @ s> 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 m Z d d l m Z d d l
m Z d d l m
Z
d d l m Z m Z d d l m Z m Z m Z m Z m Z m Z m Z m Z m Z d d l m Z d d l m Z Gd
d � d e � Z Gd d
� d
e � Z! Gd d � d e � Z" d S)� N)�DistutilsError�DistutilsOptionError)�log)�
TestLoader)�six)�map�filter) �resource_listdir�resource_exists�normalize_path� working_set�_namespace_packages�evaluate_marker�add_activation_listener�require�
EntryPoint)�Command)�
unittest_mainc @ s e Z d Z d d d � Z d S)�ScanningLoaderNc C s g } | j t j | | � � t | d � rA | j | j � � t | d � r� x� t | j d � D]� } | j d � r� | d k r� | j d | d d � } n- t | j | d � rc | j d | } n qc | j | j | � � qc Wt
| � d
k r| j | � S| d Sd S)
a Return a suite of all tests cases contained in the given module
If the module is a package, load tests from all the modules in it.
If the module has an ``additional_tests`` function, call it and add
the return value to the tests.
�additional_tests�__path__� z.pyz __init__.py�.N� z /__init__.py� r ���) �appendr �loadTestsFromModule�hasattrr r �__name__�endswithr
ZloadTestsFromName�lenZ
suiteClass)�self�module�patternZtests�fileZ submodule� r& �/test.pyr s
z"ScanningLoader.loadTestsFromModule)r �
__module__� __qualname__r r& r& r& r' r s r c @ s+ e Z d Z d d � Z d d d � Z d S)�NonDataPropertyc C s
| | _ d S)N)�fget)r" r+ r& r& r' �__init__5 s zNonDataProperty.__init__Nc C s | d k r | S| j | � S)N)r+ )r" �objZobjtyper& r& r' �__get__8 s zNonDataProperty.__get__)r r( r) r, r. r& r& r& r' r* 4 s r* c @ s� e Z d Z d Z d Z d% d& d' g Z d d
� Z d d � Z e d d � � Z d d � Z
d d � Z e j
g d d � � Z e e j
d d � � � Z e d d � � Z d d � Z d d � Z e d d! � � Z e d" d# � � Z d$ S)(�testz.Command to run unit tests after in-place buildz#run unit tests after in-place build� test-module=�m�$Run 'test_suite' in specified module� test-suite=�s�9Run single test, case or suite (e.g. 'module.test_suite')� test-runner=�r�Test runner to usec C s( d | _ d | _ d | _ d | _ d S)N)�
test_suite� test_module� test_loader� test_runner)r" r& r& r' �initialize_optionsJ s ztest.initialize_optionsc C s� | j r$ | j r$ d } t | � � | j d k rd | j d k rT | j j | _ n | j d | _ | j d k r� t | j d d � | _ | j d k r� d | _ | j d k r� t | j d d � | _ d S)Nz1You may specify a module or a suite, but not bothz .test_suiter; z&setuptools.command.test:ScanningLoaderr< )r9 r: r � distributionr; �getattrr< )r" �msgr& r& r' �finalize_optionsP s ztest.finalize_optionsc C s t | j � � S)N)�list�
_test_args)r" r& r& r' � test_argsc s ztest.test_argsc c s# | j r d V| j r | j Vd S)Nz --verbose)�verboser9 )r" r& r& r' rC g s ztest._test_argsc C s | j � � | � Wd QRXd S)zI
Backward compatibility for project_on_sys_path context.
N)�project_on_sys_path)r" �funcr& r& r' �with_project_on_sys_pathm s
z test.with_project_on_sys_pathc c s� t j o t | j d d � } | r� | j d d d �| j d � | j d � } t | j � } | j d d | �| j d � | j d d d �| j d � n- | j d � | j d d d �| j d � | j d � } t j
d d � } t j j � } zy t | j
� } t j
j d | � t j � t d
d � � t d | j | j f � | j | g � �
d VWd QRXWd | t j
d d � <t j j � t j j | � t j � Xd S)
N�use_2to3FZbuild_pyZinplacer Zegg_info�egg_baseZ build_extr c S s
| j � S)N)Zactivate)�distr& r& r' �<lambda>� s z*test.project_on_sys_path.<locals>.<lambda>z%s==%s)r �PY3r? r> Zreinitialize_commandZ run_commandZget_finalized_commandr Z build_lib�sys�path�modules�copyrJ �insertr r, r r Zegg_nameZ egg_version�paths_on_pythonpath�clear�update) r" Z
include_distsZ with_2to3Zbpy_cmdZ
build_pathZei_cmdZold_pathZ old_modulesZ project_pathr& r& r' rF t s8
ztest.project_on_sys_pathc c s� t � } t j j d | � } t j j d d � } zU t j j | � } t d | | g � } t j j | � } | r� | t j d <d VWd | | k r� t j j d d � n
| t j d <Xd S)z�
Add the indicated paths to the head of the PYTHONPATH environment
variable so that subprocesses will also see the packages at
these paths.
Do this in a context that restores the value on exit.
�
PYTHONPATHr N)�object�os�environ�get�pathsep�joinr �pop)�pathsZnothingZorig_pythonpathZcurrent_pythonpath�prefixZto_join�new_pathr& r& r' rS � s
ztest.paths_on_pythonpathc C sb | j | j � } | j | j p$ g � } | j d d � | j j � D� � } t j | | | � S)z�
Install the requirements indicated by self.distribution and
return an iterable of the dists that were built.
c s s@ | ]6 \ } } | j d � r t | d d � � r | Vq d S)�:r N)�
startswithr )�.0�k�vr& r& r' � <genexpr>� s z%test.install_dists.<locals>.<genexpr>)Zfetch_build_eggsZinstall_requiresZ
tests_requireZextras_require�items� itertools�chain)rK Zir_dZtr_dZer_dr& r& r' �
install_dists� s
ztest.install_distsc C s� | j | j � } d j | j � } | j rB | j d | � d S| j d | � t t j d � | � } | j | � �# | j
� � | j � Wd QRXWd QRXd S)N� zskipping "%s" (dry run)z running "%s"�location) rj r> r\ �_argv�dry_run�announcer �operator�
attrgetterrS rF � run_tests)r" Zinstalled_dists�cmdr^ r& r& r' �run� s
ztest.runc C sQ t j r� t | j d d � r� | j j d � d } | t k r� g } | t j k rb | j | � | d 7} x- t j D]" } | j
| � rv | j | � qv Wt t t j j
| � � t j d k r� i n d d i } t d d | j d | j | j � d | j | j � | �} | j j � sMd
| j } | j | t j � t | � � d S) NrI Fr r � � �exitZ
testLoaderZ
testRunnerzTest failed: %s)ru rv )r rM r? r> r9 �splitr
rN rP r rb rB r � __delitem__� version_infor rm �_resolve_as_epr; r< �resultZ
wasSuccessfulro r �ERRORr )r" r# Z del_modules�nameZ
exit_kwargr/ r@ r& r& r' rr � s*
!
ztest.run_testsc C s d g | j S)N�unittest)rD )r" r& r& r' rm � s z
test._argvc C s0 | d k r d St j d | � } | j � � S)zu
Load the indicated attribute value, called, as a as if it were
specified as an entry point.
Nzx=)r �parseZresolve)�valZparsedr& r& r' r{ � s ztest._resolve_as_epN)r0 r1 r2 )r3 r4 r5 )r6 r7 r8 )r r( r) �__doc__� descriptionZ user_optionsr= rA r* rD rC rH �
contextlib�contextmanagerrF � staticmethodrS rj rt rr �propertyrm r{ r& r& r& r' r/ > s( - r/ )#rX rp rN r� rh Zdistutils.errorsr r Z distutilsr r r Zsetuptools.externr Zsetuptools.extern.six.movesr r Z
pkg_resourcesr r
r r r
r r r r Z
setuptoolsr Zsetuptools.py31compatr r rW r* r/ r& r& r& r' �<module> s @