U
�=�_D, � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z z ddl
Z
W n e k
rl dZ
Y nX z 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/python38/lib64/python3.8/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_�.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 sN | � � D ]@}t|�}t|�D ]*}t||�}t|t�r t|tj�r |V q qdS )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 s, t �� }| �� D ]}t|�}|�|� q|S �N)r/ � TestSuiter4 �loadTestsFromTestCase�addTest)r& �suiter3 �testr r r �all` s
zTestLoader.allc C s^ t �� }tj�t�s|S ttd��}|�� �� }W 5 Q R X |D ]}t j �
|�}|� |� q>|S )NZrt) r/ r6 r r �isfile�FAILED_TESTS_FNAME�open�read�split�defaultTestLoader�loadTestsFromNamer8 )r& r9 �f�names�nr: r r r � last_failedg s zTestLoader.last_failedc C s. |� d�r"tj�tj�|��d }tj�|�S )Nr r )r r r �splitext�basenamer/ rA rB )r& r2 r r r � from_names s
zTestLoader.from_nameN)�__name__�
__module__� __qualname__�HEREr" r r �environ�extendr) r4 r; rF rI 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& r: r r r rS { s zColouredResult.addSuccessc C s" t j�| ||� tdddd� d S )N�ERRORr T�r )r/ rR �addErrorr �r& r: �errr r r rV s zColouredResult.addErrorc C s t j�| ||� tdd� d S )NZFAILr )r/ rR �
addFailurer rW r r r rY � s zColouredResult.addFailurec C s& t j�| ||� td|�� d� d S )Nz skipped: %sZbrown)r/ rR �addSkipr �strip)r& r: �reasonr r r rZ � s zColouredResult.addSkipc s$ t |d|dkd�}t� �||� d S )Nr rT rU )r r �printErrorList)r& Zflavour�errors�� __class__r r r] � s z ColouredResult.printErrorList) rJ rK rL rS rV rY rZ r] �
__classcell__r r r_ r rP y s
rP 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 r5 )r �__init__�set�
failed_tnames)r& �args�kwargsr_ r r rc � s zColouredTextRunner.__init__c s t � �� | _| jS r5 )r � _makeResult�resultr% r_ r r rh � s z ColouredTextRunner._makeResultc C s: | j r6ttd�� }| j D ]}|�|d � qW 5 Q R X d S )NZwt�
)re r> r= �write)r& rC �tnamer r r �_write_last_failed� s
z%ColouredTextRunner._write_last_failedc C s6 |� � s2|j|j D ] }|d �� }| j�|� qd S )Nr )�
wasSuccessfulr^ �failures�idre �add)r& ri �trl r r r � _save_result� s z ColouredTextRunner._save_resultc sV zt � �|�}W n2 ttfk
rB | jj}|�� t�d��Y nX | � |� |S d S )Nr
)
r �run�KeyboardInterrupt�
SystemExit�runnerri � printErrorsr �exitrs �r& r9 ri r_ r r �_run� s
zColouredTextRunner._runc C sJ |r&t dddd� tt� t�d� n t dddd� | �� t�d� d S ) N�SUCCESSrQ TrU r ZFAILEDr r
)r r r= r ry rm )r& �successr r r �_exit� s zColouredTextRunner._exitc C s | � |�}| �|�� � d S r5 )r{ r~ rn rz r r r rt � s
zColouredTextRunner.run)rJ rK rL �__doc__r rP r/ �TextTestResultZ resultclassrc rh rm rs r{ r~ rt ra r r r_ r rb � s
rb c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�ParallelRunnerc s0 � fdd�}t j� |tj _t�t�}t�| |�S )Nc s � | |�}t �|j� |S r5 )�atexit�register�close)�fd�mode�kwds�stream�Z orig_fdopenr r �fdopen� s
z+ParallelRunner._parallelize.<locals>.fdopen)r r� �concurrencytestZfork_for_tests�NWORKERSZConcurrentTestSuite)r9 r� Zforkerr r� r � _parallelize� s
zParallelRunner._parallelizec C s� t �� }t �� }| D ]n}|�� dkr(qn8t|t j�rB|jd j}n t|t j�rT|}n td| ��t|dd�rx|� |� q|� |� q||fS )Nr zcan't recognize type %rZ
_serialrunF)
r/ r6 �countTestCasesr, Z_testsr` r0 � TypeErrorr+ r8 )r9 �serial�parallelr: 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 workersrQ TrU 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� �timer{ �psutilZProcessZchildrenZ
wait_procsr
rn r� rx �map�lenro r^ Zskippedr �textwrap�dedentZtestsRunr~ )r& r9 Z ser_suiteZ par_suiterr ZparZ par_elapsedZorphansZgoneZaliveZserZ ser_elapsedZ par_failsZpar_errsZ par_skipsZ ser_failsZser_errsZ ser_skips�okr r r rt � sV
�
� � ��
�zParallelRunner.runN)rJ rK rL � staticmethodr� r� rt 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� ZWINDOWSr� r� r� � VERBOSITYrb )r� r� r r r �
get_runner s
r� c C s t � �| �}t� }|�|� d S r5 )r rI r� rt )r2 r9 rw r r r �
run_from_name, s r� c C s t jj�� d S r5 )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_argsrF r r= r r� Z print_usager ry rI r; r r r� r� rt )r� �parserZoptsrf �loaderr9 rw r r r �main9 s: � �
r� �__main__)FN)F)2r 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__rM rA r7 r r r� rP ZTextTestRunnerrb r� r� r� r� r� rJ r r r r �<module> sV
27U
"