관리-도구
편집 파일: runner.cpython-37.pyc
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 "