3
vh# @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dlm Z d dl m
Z
d dlmZmZ d dlmZmZmZmZmZmZmZmZmZ d dlm Z G d d
d
e Z G d d d e Z G d
d de Z dS ) N)DistutilsErrorDistutilsOptionError)log)
TestLoader)six)mapfilter) resource_listdirresource_existsnormalize_path working_set_namespace_packagesevaluate_markeradd_activation_listenerrequire
EntryPoint)Commandc @ s e Zd Zdd ZdddZdS )ScanningLoaderc C s t j| t | _d S )N)r __init__set_visited)self r /usr/lib/python3.6/test.pyr s
zScanningLoader.__init__Nc C s || j krdS | j j| g }|jtj| | t|drH|j|j t|drxpt|jdD ]`}|j dr|dkr|jd |dd }n"t
|j|d r`|jd | }nq`|j| j | q`W t |d
kr| j
|S |d S dS )
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.
Nadditional_tests__path__ z.pyz __init__.py. z /__init__.py r )r addappendr loadTestsFromModulehasattrr r __name__endswithr
ZloadTestsFromNamelenZ
suiteClass)r modulepatternZtestsfileZ submoduler r r r# s$
z"ScanningLoader.loadTestsFromModule)N)r%
__module__ __qualname__r r# r r r r r s r c @ s e Zd Zdd ZdddZdS )NonDataPropertyc C s
|| _ d S )N)fget)r r. r r r r > s zNonDataProperty.__init__Nc C s |d kr | S | j |S )N)r. )r objZobjtyper r r __get__A s zNonDataProperty.__get__)N)r% r+ r, r r0 r r r r r- = s r- c @ s e Zd ZdZdZd%d&d'gZd d
Zdd Zedd Z dd Z
dd Z e j
g fddZee j
dd Zedd Zd d Zd d Zed d! Zed"d# Zd$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_optionsS s ztest.initialize_optionsc C s | j r| jrd}t|| j d krD| jd kr8| jj | _ n | jd | _ | jd kr^t| jdd | _| jd krnd| _| jd krt| jdd | _d S )Nz1You may specify a module or a suite, but not bothz .test_suiter= z&setuptools.command.test:ScanningLoaderr> )r; r< r distributionr= getattrr> )r msgr r r finalize_optionsY s
ztest.finalize_optionsc C s t | j S )N)list
_test_args)r r r r test_argsl s ztest.test_argsc c s6 | j rtjdkrdV | jr$dV | j r2| j V d S )N Zdiscoverz --verbose)rG rH )r; sys version_infoverbose)r r r r rE p s ztest._test_argsc C s | j | W dQ R X dS )zI
Backward compatibility for project_on_sys_path context.
N)project_on_sys_path)r funcr r r with_project_on_sys_pathx s
z test.with_project_on_sys_pathc c sP t jot| jdd}|rv| jddd | jd | jd}t|j}| jd|d | jd | jddd | jd n"| jd | jdd d | jd | jd}t j
d d }t j j }zbt|j
}t j
jd| tj td
d td |j|jf | j|g d V W d Q R X W d |t j
d d < t j j t j j| tj X d S )
Nuse_2to3FZbuild_pyr )ZinplaceZegg_info)egg_baseZ build_extr c S s | j S )N)Zactivate)distr r r s z*test.project_on_sys_path..z%s==%s)r PY3rA r@ Zreinitialize_commandZ run_commandZget_finalized_commandr Z build_librI pathmodulescopyrP 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 rL s8
ztest.project_on_sys_pathc c s t }tjjd|}tjjdd}z>tjj| }td||g}tjj|}|rX|tjd< dV W d||krztjjdd n
|tjd< X dS )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 rX s
ztest.paths_on_pythonpathc C sD | j | j}| j | jpg }| j dd | jj D }tj|||S )z
Install the requirements indicated by self.distribution and
return an iterable of the dists that were built.
c s s0 | ](\}}|j d rt|dd r|V qdS ):r N)
startswithr ).0kvr r r s z%test.install_dists..)Zfetch_build_eggsZinstall_requiresZ
tests_requireZextras_requireitems itertoolschain)rQ Zir_dZtr_dZer_dr r r
install_dists s
ztest.install_distsc C s | j | j}dj| j}| jr0| jd| d S | jd| ttjd|}| j |" | j
| j W d Q R X W d Q R X d S )N zskipping "%s" (dry run)z running "%s"location) ro r@ ra _argvZdry_runannouncer operator
attrgetterrX rL run_tests)r Zinstalled_distscmdrc r r r run s
ztest.runc C s t jrt| jddr| jjdd }|tkrg }|tjkrD|j | |d7 }x"tjD ]}|j
|rT|j | qTW t t tjj
| tjd d | j| j| j| j| jdd}|jj sd|j }| j|tj t|d S )NrO Fr r )Z
testLoaderZ
testRunnerexitzTest failed: %s)r rS rA r@ r; splitr
rI rU r" rg rD r __delitem__unittestmainrr _resolve_as_epr= r> resultZ
wasSuccessfulrs r ZERRORr )r r( Z del_modulesnamer1 rB r r r rv s(
ztest.run_testsc C s dg| j S )Nr| )rF )r r r r rr s z
test._argvc C s$ | dkr dS t jd| }|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)r2 r3 r4 )r5 r6 r7 )r8 r9 r: )r% r+ r, __doc__ descriptionZ user_optionsr? rC r- rF rE rN
contextlibcontextmanagerrL staticmethodrX ro rx rv propertyrr r~ r r r r r1 G s( - r1 )!r] rt rI r rm r| Zdistutils.errorsr r Z distutilsr r Zsetuptools.externr Zsetuptools.extern.six.movesr r Z
pkg_resourcesr r
r r r
r r r r Z
setuptoolsr r r\ r- r1 r r r r s , )