U
|f, @ 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$ddZ&G dd dZ'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#kre0 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 )Nred)boldfile)r )sysstderrstdout
USE_COLORSr print)msgcolorr r r F/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/runner.pycprintB s
r c @ sX e Zd ZeZdgZdejkr,edddg 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 fddt jD S )Nc s: g | ]2}| d r|dr| jkrtj j|qS )Ztest_.py)
startswithendswith
skip_filesospathjointestdir).0xselfr r
S s
z,TestLoader._get_testmods..)r listdirr# r& r r& r
_get_testmodsR s
zTestLoader._get_testmodsc c sN | D ]@}t|}t|D ]*}t||}t|tr t|tjr |V q qdS )zgIterate over all test files in this directory and return
all TestCase classes in them.
N) r* r dirgetattr
isinstancetype
issubclassunittestTestCase)r' r! modnameobjr r r _iter_testmod_classes[ s
z TestLoader._iter_testmod_classesc C s, t }| D ]}t|}|| q|S N)r0 TestSuiter5 loadTestsFromTestCaseaddTest)r' suiter4 testr r r allh s
zTestLoader.allc C s\ t }tjts|S tt}| }W 5 Q R X |D ]}t j
|}| | q<|S r6 ) r0 r7 r r! isfileFAILED_TESTS_FNAMEopenreadsplitdefaultTestLoaderloadTestsFromNamer9 )r' r: fnamesnr; r r r last_failedo s
zTestLoader.last_failedc C s. | dr"tjtj|d }tj|S )Nr r )r r r! splitextbasenamer0 rB rC )r' r3 r r r from_name{ s
zTestLoader.from_nameN)__name__
__module__ __qualname__HEREr# r r environextendr* r5 r< rG rJ r r r r r K s
r c s< e Zd Zdd Zdd Zdd Zdd Z fd d
Z ZS ) ColouredResultc C s t j| | tdd d S )NOKgreen)r0
TestResult
addSuccessr )r' r; r r r rU s zColouredResult.addSuccessc C s" t j| || tdddd d S )NERRORr Tr )r0 rT addErrorr r' r; errr r r rX s zColouredResult.addErrorc C s t j| || tdd d S )NFAILr )r0 rT
addFailurer rY r r r r\ s zColouredResult.addFailurec C s& t j| || td| d d S )Nz skipped: %sZbrown)r0 rT addSkipr strip)r' r; reasonr r r r] s zColouredResult.addSkipc s$ t |d|dkd}t || d S )Nr rV rW )r r printErrorList)r' flavourerrors __class__r r r` s z ColouredResult.printErrorList) rK rL rM rU rX r\ r] r`
__classcell__r r rc r rQ s
rQ c sf e Zd ZdZerenejZ fddZ fddZ
dd Z dd Z fd
d Z
d d
Zdd Z ZS )ColouredTextRunnerzA 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 r6 )r __init__set
failed_tnames)r' argskwargsrc r r rg s zColouredTextRunner.__init__c s t | _| jS r6 )r _makeResultresultr& rc r r rl s z ColouredTextRunner._makeResultc C s: | j r6ttd }| j D ]}||d qW 5 Q R X d S )Nw
)ri r? r> write)r' rD tnamer r r _write_last_failed s
z%ColouredTextRunner._write_last_failedc C s6 | s2|j|j D ] }|d }| j| qd S )Nr )
wasSuccessfulrb failuresidri add)r' rm trq r r r _save_result s z ColouredTextRunner._save_resultc sV zt |}W n2 ttfk
rB | jj}| tdY nX | | |S d S )Nr
)
r runKeyboardInterrupt
SystemExitrunnerrm printErrorsr exitrx r' r: rm rc r r _run s
zColouredTextRunner._runc C sJ |r&t dddd tt td n t dddd | td d S ) NSUCCESSrS TrW r ZFAILEDr r
)r r r> r r~ rr )r' successr r r _exit s zColouredTextRunner._exitc C s | |}| | d S r6 )r r rs r r r r ry s
zColouredTextRunner.run)rK rL rM __doc__r rQ r0 TextTestResultZ resultclassrg rl rr rx r r ry re r r rc r rf s
rf c @ s, e Zd Zedd Zedd Zdd ZdS )ParallelRunnerc s0 fdd}t j |tj _tt}t| |S )Nc s | |}t |j |S r6 )atexitregisterclose)fdmodekwdsstreamZ orig_fdopenr r fdopen s
z+ParallelRunner._parallelize..fdopen)r r concurrencytestZfork_for_testsNWORKERSZConcurrentTestSuite)r: r Zforkerr r r _parallelize s
zParallelRunner._parallelizec C s t }t }| D ]l}| dkr&qt|t jr@|jd j}n t|t jrR|}n td| t|ddrv| | q| | q||fS )Nr zcan't recognize type %rZ
_serialrunF)
r0 r7 countTestCasesr- Z_testsrd r1 TypeErrorr, r9 )r: serialparallelr; 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\}}}ttd
|j| |
|||
j|||| f
td |j|
j || tf | oH|
}| | d S ) Nz(starting parallel tests using %s workersrS TrW 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 timer psutilProcesschildrenZ
wait_procsr
rs r r} maplenrt rb Zskippedr textwrapdedentZtestsRunr )r' r: Z ser_suiteZ par_suiterw parZ par_elapsedZorphansZ_gonealiveZserZ ser_elapsedZ par_failsZpar_errsZ par_skipsZ ser_failsZser_errsZ ser_skipsokr r r ry sn
zParallelRunner.runN)rK rL rM staticmethodr r ry r r r r r s
r c C sT dd }| rJt jr |d n.td kr.|d n tdkr@|d n
ttdS ttdS )Nc S s t | d d d S )Nz Running serial tests instead.r )r )r r r r warn< s zget_runner..warnz$Can't run parallel tests on Windows.z(concurrencytest module is not installed.r
zOnly 1 CPU available.) verbosity)r WINDOWSr r r VERBOSITYrf )r r r r r
get_runner; s
r c C s* t r
t t | }t }|| d S r6 )r r r rJ r ry )r3 r: r| r r r
run_from_nameL s
r c C s t d d S )NT)r Z
_set_debugr r r r setupT s r c C s t d} tj| dd}|jddddd |jd ddd
d | \}}|jsVtt t }|rt |d kr~|
t d S |
|d }n|jr| }n| }trt 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)actiondefaulthelpz
--parallelzrun tests in parallelr
r )r optparse OptionParser
add_option
parse_argsrG r r> r r print_usager r~ rJ r< r r r r ry )r parseroptsrj loaderr: r| r r r mainX s>
r __main__)FN)F)2r
__future__r r r r r r r r0 ctypes ImportErrorr r Zpsutil._commonr r r Zpsutil._compatr Z psutil.testsr r r r
r r r> cpu_countr r r! abspathdirname__file__rN rB r8 r r r rQ ZTextTestRunnerrf r r r r r rK r r r r sX
68l
(