U
�VF � @ s� d Z ddlmZ ddlZddlZddlZddlZe�e�Z 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 )� TestrReal�Run 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|� t�tjd gt|� tjtjtj �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 s. 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 sp t t| j�� tj�| _tj| jd� t�d� | jd kr@g | _n | j� � | _| j
r^d| j
| _
t�d| j � d S )N)�levelzfinalize_options calledz --omit=%sz$finalize_options: self.__dict__ = %r) �getattr�loggingr �upper�INFOZ basicConfigr r r �splitr �__dict__r r r r �finalize_optionsQ s �
zTestrReal.finalize_optionsc C s� t �d� tj�d�s | �d� | jr.| �� | jsF| jd
| j �� }n| jd | j �� }|rjt
j � d| ��| j
r�td� | �d� | jr�| �� d S ) z"Set up testr repo, then run testr.z
run calledz.testrepositoryZinit�run�
--parallelztestr failed (%d)z
Slowest Testsr N)r) r* )r) )r r �os�path�isdirr r �_coverage_beforer
r � distutils�errorsZDistutilsErrorr �print�_coverage_after)r Z testr_retr r r r) ` s"
�
z
TestrReal.runc C sb t �d� | j�� }|�d�r*|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� t�d� t�d| 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