�VF � @ s� d Z d d l m Z d d l Z d d l Z d d l Z d d l Z e j e � Z Gd d � d e j
� Z Gd d � d e j
� Z y d d l
m Z d Z e Z Wn e k
r� d
Z e Z Yn Xd S) 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 Z d Z d Z d! d" d# d$ d% d& d' g Z d d
d g Z d d � Z d d � Z d d � Z d d � Z d d � Z
d d � Z d S)(� TestrRealzRun unit tests using testr�coverageN�GReplace PYTHON with coverage and merge coverage from each testr worker.� testr-args=�t�Run 'testr' with these args�omit=�o�(Files to omit from coverage calculations�coverage-package-name=�"Use this name for coverage package�slowest�-Show slowest test times after tests complete.� no-parallel�Run testr serially�
log-level=�l�Log level (default: info)� no_parallelc G sC t j d | � t j t j d g t | � t j t j t j � S)Nz _run_testr called with args = %rr )
�logger�debug�commandsZrun_argv�sys�argv�list�stdin�stdout�stderr)�self�args� r! �/testr_command.py�
_run_testrC s zTestrReal._run_testrc C sC d | _ d | _ d | _ d | _ d | _ d | _ d | _ d S)N� �info)�
testr_argsr �omitr �coverage_package_namer � log_level)r r! r! r" �initialize_optionsH s z TestrReal.initialize_optionsc C s� t t | j j � t j � | _ t j d | j � t j d � | j d k r\ g | _ n | j j � | _ | j
r� d | j
| _
t j d | j � d S)N�levelzfinalize_options calledz --omit=%sz$finalize_options: self.__dict__ = %r) �getattr�loggingr) �upper�INFO� basicConfigr r r& �splitr' �__dict__)r r! r! r" �finalize_optionsQ s
zTestrReal.finalize_optionsc C s� t j d � t j j d � s, | j d � | j r? | j � | j sc | j d d | j � } n | j d | j � } | r� t
j j d | � � | j
r� t d � | j d � | j r� | j � d S)
z"Set up testr repo, then run testr.z
run calledz.testrepositoryZinit�runz
--parallelztestr failed (%d)z
Slowest Testsr N)r r �os�path�isdirr# r �_coverage_beforer r& � distutils�errorsZDistutilsErrorr �print�_coverage_after)r Z testr_retr! r! r" r4 ` s
z
TestrReal.runc C s� t j d � | j j � } | j d � r; | d d � } | j rM | j } d | } d | t j d <t j d t j d � d S)Nz_coverage_before calledzpython-� z--source %s --parallel-modezcoverage run %s�PYTHONzos.environ['PYTHON'] = %r)r r Z distribution�get_name�
startswithr( r5 �environ)r �packageZoptionsr! r! r" r8 u s
zTestrReal._coverage_beforec C s2 t j d � t j d � t j d | j � d S)Nz_coverage_after calledzcoverage combinezcoverage html -d ./cover %s)r r r5 �systemr' )r r! r! r" r< � s
zTestrReal._coverage_after)zcoverageNr )r r r )r r
r )r Nr
)zslowestNr )r Nr )r r r ) �__name__�
__module__� __qualname__� description� user_optionsZboolean_optionsr# r* r3 r4 r8 r< r! r! r! r" r 2 s
r c @ s@ e Z d Z d Z g Z d d � Z d d � Z d d � Z d S) � TestrFakezRun unit tests using testrc C s d S)Nr! )r r! r! r" r* � s z TestrFake.initialize_optionsc C s d S)Nr! )r r! r! r" r3 � s zTestrFake.finalize_optionsc C s t d � d S)Nz7Install testrepository to run 'testr' command properly.)r; )r r! r! r" r4 � s z
TestrFake.runN)rD rE rF rG rH r* r3 r4 r! r! r! r" rI � s
rI )r TF)�__doc__r9 r Zdistutils.errorsr- r5 r � getLoggerrD r ZCommandr rI Ztestrepositoryr Z
have_testrZTestr� ImportErrorr! r! r! r" �<module>' s V