Your IP : 216.73.216.69
B
�=�_D, � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z y ddl
Z
W n e k
rl dZ
Y nX y ddl Z W n e k
r� dZ Y nX ddl
Z
ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ d
ZdZe
� � �p dZ e �o*e� Z ej � ej �!e"��Z#e j$j%Z%d$dd�Z&G dd� d�Z'G dd� de j(�Z)G dd� de j*�Z+G dd� de+�Z,d%dd �Z-d d � Z.d d � Z/d!d"� Z0e1d#k�r�e0� dS )&aA
Unit test runner, providing new features on top of unittest module:
- colourized output
- parallel run (UNIX only)
- print failures/tracebacks on CTRL+C
- re-run failed tests only (make test-failed)
Invocation examples:
- make test
- make test-failed
Parallel:
- make test-parallel
- make test-process ARGS=--parallel
� )�print_functionN)�hilite)� print_color)�term_supports_colors)�super)�
CI_TESTING)�import_module_by_path)�
print_sysinfo)�
reap_children)� safe_rmpath� z.failed-tests.txt� Fc C sB |d kr |dkrt jnt j}tr2t| |||d� n t| |d� d S )N�red)�bold�file)r )�sys�stderr�stdout�
USE_COLORSr �print)�msg�colorr r � r �F/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/runner.py�cprint> s
r c @ sX e Zd ZeZdgZdejkr,e�dddg� dd� Z dd � Z
d
d � Z d d
� Z dd� Z
dS )�
TestLoaderztest_memleaks.pyZ WHEELHOUSE_UPLOADER_USERNAMEz test_osx.pyz
test_linux.pyz
test_posix.pyc s � fdd�t �� j�D �S )Nc s: g | ]2}|� d �r|�d�r|� jkrtj�� j|��qS )Ztest_z.py)�
startswith�endswith�
skip_files�os�path�join�testdir)�.0�x)�selfr r �
<listcomp>O s z,TestLoader._get_testmods.<locals>.<listcomp>)r �listdirr" )r% r )r% r �
_get_testmodsN s
zTestLoader._get_testmodsc c sV xP| � � D ]D}t|�}x6t|�D ]*}t||�}t|t�r t|tj�r |V q W q
W dS )zgIterate over all test files in this directory and return
all TestCase classes in them.
N) r( r �dir�getattr�
isinstance�type�
issubclass�unittest�TestCase)r% r �mod�name�objr r r �_iter_testmod_classesT s
z TestLoader._iter_testmod_classesc C s0 t �� }x"| �� D ]}t|�}|�|� qW |S )N)r. � TestSuiter3 �loadTestsFromTestCase�addTest)r% �suiter2 �testr r r �all` s
zTestLoader.allc C sb t �� }tj�t�s|S ttd��}|�� �� }W d Q R X x"|D ]}t j �
|�}|� |� q@W |S )N�rt) r. r4 r r �isfile�FAILED_TESTS_FNAME�open�read�split�defaultTestLoader�loadTestsFromNamer6 )r% r7 �f�names�nr8 r r r � last_failedg s
zTestLoader.last_failedc C s. |� d�r"tj�tj�|��d }tj�|�S )Nz.pyr )r r r �splitext�basenamer. r@ rA )r% r1 r r r � from_names s
zTestLoader.from_nameN)�__name__�
__module__� __qualname__�HEREr" r r �environ�extendr( r3 r9 rE rH r r r r r G s
r c s< e Zd Zdd� Zdd� Zdd� Zdd� Z� fd d
�Z� ZS ) �ColouredResultc C s t j�| |� tdd� d S )NZOK�green)r. �
TestResult�
addSuccessr )r% r8 r r r rR { s zColouredResult.addSuccessc C s" t j�| ||� tdddd� d S )N�ERRORr T)r )r. rQ �addErrorr )r% r8 �errr r r rT s zColouredResult.addErrorc C s t j�| ||� tdd� d S )NZFAILr )r. rQ �
addFailurer )r% r8 rU r r r rV � s zColouredResult.addFailurec C s& t j�| ||� td|�� d� d S )Nz skipped: %s�brown)r. rQ �addSkipr �strip)r% r8 �reasonr r r rX � s zColouredResult.addSkipc s$ t |d|dkd�}t� �||� d S )Nr rS )r )r r �printErrorList)r% Zflavour�errors)� __class__r r r[ � s z ColouredResult.printErrorList) rI rJ rK rR rT rV rX r[ �
__classcell__r r )r] r rO y s
rO c sf e Zd ZdZerenejZ� fdd�Z � fdd�Z
dd� Z dd � Z � fd
d �Z
d d
� Zdd� Z� ZS )�ColouredTextRunnerz�
A coloured text runner which also prints failed tests on KeyboardInterrupt
and save failed tests in a file so that they can be re-run.
c s t � j||� t� | _d S )N)r �__init__�set�
failed_tnames)r% �args�kwargs)r] r r r` � s zColouredTextRunner.__init__c s t � �� | _| jS )N)r � _makeResult�result)r% )r] r r re � s z ColouredTextRunner._makeResultc C s> | j r:ttd��$}x | j D ]}|�|d � qW W d Q R X d S )N�wt�
)rb r= r< �write)r% rB �tnamer r r �_write_last_failed� s z%ColouredTextRunner._write_last_failedc C s: |� � s6x,|j|j D ] }|d �� }| j�|� qW d S )Nr )�
wasSuccessfulr\ �failures�idrb �add)r% rf �trj r r r � _save_result� s z ColouredTextRunner._save_resultc sV yt � �|�}W n2 ttfk
rB | jj}|�� t�d��Y nX | � |� |S d S )Nr
)
r �run�KeyboardInterrupt�
SystemExit�runnerrf � printErrorsr �exitrq )r% r7 rf )r] r r �_run� s
zColouredTextRunner._runc C sJ |r&t dddd� tt� t�d� n t dddd� | �� t�d� d S ) N�SUCCESSrP T)r r ZFAILEDr r
)r r r< r rw rk )r% �successr r r �_exit� s zColouredTextRunner._exitc C s | � |�}| �|�� � d S )N)rx r{ rl )r% r7 rf r r r rr � s
zColouredTextRunner.run)rI rJ rK �__doc__r rO r. �TextTestResultZ resultclassr` re rk rq rx r{ rr r^ r r )r] r r_ � s
r_ c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�ParallelRunnerc s0 � fdd�}t j� |tj _t�t�}t�| |�S )Nc s � | |�}t �|j� |S )N)�atexit�register�close)�fd�mode�kwds�stream)� orig_fdopenr r �fdopen� s
z+ParallelRunner._parallelize.<locals>.fdopen)r r� �concurrencytestZfork_for_tests�NWORKERSZConcurrentTestSuite)r7 r� Zforkerr )r� r � _parallelize� s
zParallelRunner._parallelizec C s� t �� }t �� }xv| D ]n}|�� dkr*qn8t|t j�rD|jd j}n t|t j�rV|}n td| ��t|dd�rz|� |� q|� |� qW ||fS )Nr zcan't recognize type %rZ
_serialrunF)
r. r4 �countTestCasesr+ Z_testsr] r/ � TypeErrorr* r6 )r7 �serial�parallelr8 Z
test_classr r r � _split_suite� s
zParallelRunner._split_suitec C sX | � |�\}}| �|�}tdt ddd� t�� }| �|�}t�� | }t�� �� }tj |dd�\}} | r~td| d� t
� t�� }| �|�}
t�� | } |� � s�|� � d kr�|�
� tt|j|j|jf�\} }
}tt|
j|
j|
jf�\}}}tt�d
|j| |
|||
j|||| f
�� td |j|
j || tf � |� � �oH|
� � }| �|� d S ) Nz(starting parallel tests using %s workersrP T)r r
)�timeoutzalive processes %sr r a=
+----------+----------+----------+----------+----------+----------+
| | total | failures | errors | skipped | time |
+----------+----------+----------+----------+----------+----------+
| parallel | %3s | %3s | %3s | %3s | %.2fs |
+----------+----------+----------+----------+----------+----------+
| serial | %3s | %3s | %3s | %3s | %.2fs |
+----------+----------+----------+----------+----------+----------+
z&Ran %s tests in %.3fs using %s workers)r� r� r r� �timerx �psutilZProcess�childrenZ
wait_procsr
rl r� rv �map�lenrm r\ Zskippedr �textwrap�dedentZtestsRunr{ )r% r7 Z ser_suiteZ par_suiterp �parZ par_elapsedZorphansZgone�aliveZserZ ser_elapsedZ par_failsZpar_errsZ par_skipsZ ser_failsZser_errsZ ser_skips�okr r r rr � s<
zParallelRunner.runN)rI rJ rK � staticmethodr� r� rr r r r r r~ � s r~ c C sT dd� }| rJt jr |d� n.td kr.|d� n tdkr@|d� n
ttd�S ttd�S )Nc S s t | d d� d S )Nz Running serial tests instead.r )r )r r r r �warn s zget_runner.<locals>.warnz$Can't run parallel tests on Windows.z(concurrencytest module is not installed.r
zOnly 1 CPU available.)� verbosity)r� �WINDOWSr� r� r~ � VERBOSITYr_ )r� r� r r r �
get_runner s
r� c C s t � �| �}t� }|�|� d S )N)r rH r� rr )r1 r7 ru r r r �
run_from_name, s r� c C s t jj�� d S )N)r� Z _psplatformZcextZ set_testingr r r r �setup2 s r� c C s� t � d} tj| dd�}|jddddd� |jd ddd
d� |�� \}}|jsVtt� t� }|r�t |�d kr~|�
� t � d �S |�
|d �}n|jr�|�� }n|�� }tr�t� t|j�}|�|� d S )
Nz*python3 -m psutil.tests [opts] [test-name]zrun unit tests)�usage� descriptionz
--last-failed�
store_trueFzonly run last failed tests)�action�default�helpz
--parallelzrun tests in parallelr
r )r� �optparse� OptionParser�
add_option�
parse_argsrE r r< r r� Z print_usager rw rH r9 r r r� r� rr )r� �parser�optsrc �loaderr7 ru r r r �main9 s2
r� �__main__)FN)F)2r| �
__future__r r r� r r r� r� r. �ctypes� ImportErrorr� r� Zpsutil._commonr r r Zpsutil._compatr Z psutil.testsr r r r
r r� r< � cpu_countr� r r �abspath�dirname�__file__rL r@ r5 r r r} rO ZTextTestRunnerr_ r~ r� r� r� r� rI r r r r �<module> sV
27U
"