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$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)msgZcolorr r r F/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/runner.pycprint> 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
<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|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_classesT s
z TestLoader._iter_testmod_classesc C s, t }| D ]}t|}|| q|S N)r/ TestSuiter4 loadTestsFromTestCaseaddTest)r& suiter3 testr r r all` s
zTestLoader.allc C s^ t }tjts|S ttd}| }W 5 Q R X |D ]}t j
|}| | q>|S )NZrt) r/ r6 r r isfileFAILED_TESTS_FNAMEopenreadsplitdefaultTestLoaderloadTestsFromNamer8 )r& r9 fnamesnr: r r r last_failedg s zTestLoader.last_failedc C s. | dr"tjtj|d }tj|S )Nr r )r r r splitextbasenamer/ rA rB )r& r2 r r r from_names s
zTestLoader.from_nameN)__name__
__module__ __qualname__HEREr" r r environextendr) 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 )NZOKgreen)r/
TestResult
addSuccessr )r& r: r r r rS { s zColouredResult.addSuccessc C s" t j| || tdddd d S )NERRORr Tr )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& Zflavourerrors __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 fddZ fddZ
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& argskwargsr_ r r rc s zColouredTextRunner.__init__c s t | _| jS r5 )r _makeResultresultr% 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^ failuresidre add)r& ri trl 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
SystemExitrunnerri printErrorsr exitrs r& r9 ri r_ r r _run s
zColouredTextRunner._runc C sJ |r&t dddd tt td n t dddd | td d S ) NSUCCESSrQ 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 _tt}t| |S )Nc s | |}t |j |S r5 )atexitregisterclose)fdmodekwdsstreamZ orig_fdopenr r fdopen s
z+ParallelRunner._parallelize.<locals>.fdopen)r r concurrencytestZfork_for_testsNWORKERSZConcurrentTestSuite)r9 r Zforkerr r r _parallelize s
zParallelRunner._parallelizec C s t }t }| D ]n}| dkr(qn8t|t jrB|jd j}n t|t jrT|}n td| t|ddrx| | q| | q||fS )Nr zcan't recognize type %rZ
_serialrunF)
r/ r6 countTestCasesr, Z_testsr` r0 TypeErrorr+ r8 )r9 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 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 maplenro r^ Zskippedr textwrapdedentZtestsRunr~ )r& r9 Z ser_suiteZ par_suiterr ZparZ par_elapsedZorphansZgoneZaliveZserZ ser_elapsedZ par_failsZpar_errsZ par_skipsZ ser_failsZser_errsZ ser_skipsokr 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
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.<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 }|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 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 abspathdirname__file__rM rA r7 r r r rP ZTextTestRunnerrb r r r r r rJ r r r r <module> sV
27U
"