\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)DistutilsErrorDistutilsOptionError)log)
TestLoader)six)mapfilter) resource_listdirresource_existsnormalize_path working_set_namespace_packagesevaluate_markeradd_activation_listenerrequire
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 loadTestsFromModulehasattrr r __name__endswithr
ZloadTestsFromNamelenZ
suiteClass)selfmodulepatternZtestsfileZ 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=s9Run single test, case or suite (e.g. 'module.test_suite') test-runner=rTest 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)
Nuse_2to3FZbuild_pyZinplacer Zegg_infoegg_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_libsyspathmodulescopyrJ insertr r, r r Zegg_nameZ egg_versionpaths_on_pythonpathclearupdate) 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)objectosenvirongetpathsepjoinr pop)pathsZnothingZorig_pythonpathZcurrent_pythonpathprefixZto_joinnew_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 ).0kvr& r& r' <genexpr> s z%test.install_dists.<locals>.<genexpr>)Zfetch_build_eggsZinstall_requiresZ
tests_requireZextras_requireitems itertoolschain)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\ _argvdry_runannouncer operator
attrgetterrS rF run_tests)r" Zinstalled_distscmdr^ 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_modulesnameZ
exit_kwargr/ r@ r& r& r' rr s*
!
ztest.run_testsc C s d g | j S)Nunittest)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
contextlibcontextmanagerrF 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 @