Current File : //opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/__pycache__/runner.cpython-37.pyc
B
,.�_C, � @ 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)�msgZcolorr 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- � TestSuiter2 �loadTestsFromTestCase�addTest)r$ �suiter1 �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 )NZrt) r- r3 r r �isfile�FAILED_TESTS_FNAME�open�read�split�defaultTestLoader�loadTestsFromNamer5 )r$ r6 �f�names�nr7 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> r? )r$ r0 r r r � from_names s
zTestLoader.from_nameN)�__name__�
__module__� __qualname__�HEREr! r r �environ�extendr' r2 r8 rC rF 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$ r7 r r r rP { s zColouredResult.addSuccessc C s" t j�| ||� tdddd� d S )N�ERRORr T)r )r- rO �addErrorr )r$ r7 �errr r r rR s zColouredResult.addErrorc C s t j�| ||� tdd� d S )NZFAILr )r- rO �
addFailurer )r$ r7 rS r r r rT � s zColouredResult.addFailurec C s& t j�| ||� td|�� d� d S )Nz skipped: %sZbrown)r- rO �addSkipr �strip)r$ r7 �reasonr r r rU � s zColouredResult.addSkipc s$ t |d|dkd�}t� �||� d S )Nr rQ )r )r r �printErrorList)r$ Zflavour�errors)� __class__r r rX � s z ColouredResult.printErrorList) rG rH rI rP rR rT rU rX �
__classcell__r r )rZ r rM y s
rM 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)rZ r r r] � s zColouredTextRunner.__init__c s t � �� | _| jS )N)r � _makeResult�result)r$ )rZ r r rb � s z ColouredTextRunner._makeResultc C s> | j r:ttd��$}x | j D ]}|�|d � qW W d Q R X d S )NZwt�
)r_ r; r: �write)r$ r@ �tnamer r r �_write_last_failed� s z%ColouredTextRunner._write_last_failedc C s: |� � s6x,|j|j D ] }|d �� }| j�|� qW d S )Nr )�
wasSuccessfulrY �failures�idr_ �add)r$ rc �trf 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�runnerrc � printErrorsr �exitrm )r$ r6 rc )rZ r r �_run� s
zColouredTextRunner._runc C sJ |r&t dddd� tt� t�d� n t dddd� | �� t�d� d S ) N�SUCCESSrN T)r r ZFAILEDr r
)r r r: r rs rg )r$ �successr r r �_exit� s zColouredTextRunner._exitc C s | � |�}| �|�� � d S )N)rt rw rh )r$ r6 rc r r r rn � s
zColouredTextRunner.run)rG rH rI �__doc__r rM r- �TextTestResultZ resultclassr] rb rg rm rt rw rn r[ r r )rZ 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)r6 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- r3 �countTestCasesr* Z_testsrZ r. � TypeErrorr) r5 )r6 �serial�parallelr7 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 workersrN T)r r
)Ztimeoutzalive 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� �timert �psutilZProcessZchildrenZ
wait_procsr
rh r� rr �map�lenri rY Zskippedr �textwrap�dedentZtestsRunrw )r$ r6 Z ser_suiteZ par_suiterl ZparZ par_elapsedZorphansZgoneZaliveZserZ ser_elapsedZ par_failsZpar_errsZ par_skipsZ ser_failsZser_errsZ ser_skips�okr r r rn � s<
zParallelRunner.runN)rG rH rI � staticmethodr� r� rn r r r r rz � s rz 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� ZWINDOWSr� r� rz � VERBOSITYr\ )r� r� r r r �
get_runner s
r� c C s t � �| �}t� }|�|� d S )N)r rF r� rn )r0 r6 rq r r r �
run_from_name, s r� c C s$ dt jkrdt jd< tjj�� d S )NZPSUTIL_TESTING�1)r rK 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� �optparseZ OptionParserZ
add_option�
parse_argsrC r r: r r� Z print_usager rs rF r8 r r r� r� rn )r� �parserZoptsr` �loaderr6 rq r r r �main9 s2
r� �__main__)FN)F)2rx Z
__future__r r{ r� r r r� r� r- Zctypes� ImportErrorr� r� Zpsutil._commonr r r Zpsutil._compatr Z psutil.testsr r r r
r r� r: � cpu_countr� r r �abspath�dirname�__file__rJ r> r4 r r ry rM ZTextTestRunnerr\ rz r� r� r� r� rG r r r r �<module> sV
27U
"
Mini Shell