U
VF @ s d Z ddlmZ ddlZddlZddlZddlZeeZ G dd dej
Z G dd dej
Z zddl
mZ d Ze ZW n ek
r d
Ze ZY nX dS ) a setuptools/distutils commands to run testr via setup.py
Currently provides 'testr' which runs tests using testr. You can pass
--coverage which will also export PYTHON='coverage run --source <your package>'
and automatically combine the coverage from each testr backend test runner
after the run completes.
To use, just use setuptools/distribute and depend on testr, and it should be
picked up automatically (as the commands are exported in the testrepository
package metadata.
)cmdNc @ s\ e Zd ZdZdddddddgZd d
d gZd d
Zdd Zdd Zdd Z dd Z
dd Z dS ) TestrRealRun unit tests using testr)coverageNzGReplace PYTHON with coverage and merge coverage from each testr worker.)z testr-args=tzRun 'testr' with these args)zomit=oz(Files to omit from coverage calculations)zcoverage-package-name=Nz"Use this name for coverage package)slowestNz-Show slowest test times after tests complete.)z no-parallelNzRun testr serially)z
log-level=lzLog level (default: info)r r no_parallelc G s2 t d| ttjd gt| tjtjtj S )Nz _run_testr called with args = %rr )
loggerdebugcommandsZrun_argvsysargvliststdinstdoutstderr)selfargs r /testr_command.py
_run_testrC s zTestrReal._run_testrc C s. d | _ d | _d| _d | _d | _d | _d| _d S )N info)
testr_argsr omitr coverage_package_namer
log_levelr r r r initialize_optionsH s z TestrReal.initialize_optionsc C sp t t| j tj| _tj| jd td | jd kr@g | _n | j | _| j
r^d| j
| _
td| j d S )N)levelzfinalize_options calledz --omit=%sz$finalize_options: self.__dict__ = %r) getattrloggingr upperINFOZ basicConfigr r r splitr __dict__r r r r finalize_optionsQ s
zTestrReal.finalize_optionsc C s t d tjds | d | jr.| | jsF| jd
| j }n| jd | j }|rjt
j d| | j
rtd | d | jr| d S ) z"Set up testr repo, then run testr.z
run calledz.testrepositoryZinitrun
--parallelztestr failed (%d)z
Slowest Testsr N)r) r* )r) )r r ospathisdirr r _coverage_beforer
r distutilserrorsZDistutilsErrorr print_coverage_after)r Z testr_retr r r r) ` s"
z
TestrReal.runc C sb t d | j }|dr*|dd }| jr6| j}d| }d| tjd< t dtjd d S )Nz_coverage_before calledzpython- z--source %s --parallel-modezcoverage run %sZPYTHONzos.environ['PYTHON'] = %r)r r Z distributionZget_name
startswithr r+ environ)r packageZoptionsr r r r. u s
zTestrReal._coverage_beforec C s( t d td td| j d S )Nz_coverage_after calledzcoverage combinezcoverage html -d ./cover %s)r r r+ systemr r r r r r2 s
zTestrReal._coverage_afterN) __name__
__module__ __qualname__ description user_optionsZboolean_optionsr r r( r) r. r2 r r r r r 2 s
r c @ s, e Zd ZdZg Zdd Zdd Zdd ZdS ) TestrFaker c C s d S Nr r r r r r s z TestrFake.initialize_optionsc C s d S r> r r r r r r( s zTestrFake.finalize_optionsc C s t d d S )Nz7Install testrepository to run 'testr' command properly.)r1 r r r r r) s z
TestrFake.runN)r8 r9 r: r; r< r r( r) r r r r r= s
r= )r
TF)__doc__r/ r Zdistutils.errorsr# r+ r Z getLoggerr8 r ZCommandr r= Ztestrepositoryr
Z
have_testrZTestr ImportErrorr r r r <module> s
V