4^# d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
Z
n
# e $ r dZ
Y nw xY w ddl Z n
# e $ r dZ Y nw xY wddl
Z
ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ d
ZdZ e
j pdZ e o e Z ej ej ! e" Z#e j$ j% Z%d dZ& G d d Z' G d de j( Z) G d de j* Z+ G d de+ Z,d dZ-d Z.d Z/d Z0e1d k r e0 dS 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 | |dk r t j n t j }t rt | ||| d S t | | d S )Nred)boldfile)r )sysstderrstdout
USE_COLORSr print)msgcolorr r s u/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/psutil/tests/runner.pycprintr > sZ | "e^^szz CT555555
c c v e Zd ZeZdgZdej v re g d d Z d Z
d Z d Z d Z
d S )
TestLoaderztest_memleaks.py WHEELHOUSE_UPLOADER_USERNAME)z test_osx.pyz
test_linux.pyz
test_posix.pyc N fdt j j D S )Nc g | ]Z}| d rC| d r.|j v5t j j | [S )test_.py)
startswithendswith
skip_filesospathjointestdir).0xselfs r
z,TestLoader._get_testmods..O sn * * *<<((*-.ZZ->->* (( T\1-- )((r )r' listdirr* )r- s `r
_get_testmodszTestLoader._get_testmodsN s: * * * *DL11* * * *r c # K | D ]f}t | }t | D ]E}t || }t |t
r t
|t j r|V FgdS )zgIterate over all test files in this directory and return
all TestCase classes in them.
N) r0 r dirgetattr
isinstancetype
issubclassunittestTestCase)r- r( modnameobjs r _iter_testmod_classesz TestLoader._iter_testmod_classesT s &&(( D'--CC
c4(( c4(( "3(9:: III
r c t j }| D ]&}t | }| | '|S N)r7 TestSuiter< loadTestsFromTestCaseaddTest)r- suiter; tests r allzTestLoader.all` sR "$$--// C(--D MM$ r c t j }t j t
s|S t
t
d 5 }| }d d d n # 1 swxY w Y |D ]6}t j
| }| | 7|S )Nrt) r7 r? r' r( isfileFAILED_TESTS_FNAMEopenreadsplitdefaultTestLoaderloadTestsFromNamerA )r- rB fnamesnrC s r last_failedzTestLoader.last_failedg s "$$w~~ 011 L
$d
+
+ %qFFHHNN$$E % % % % % % % % % % % % % % % A-??BBD MM$ s 'BB Bc | d rBt j t j | d }t
j | S )Nr# r )r% r' r( splitextbasenamer7 rL rM )r- r: s r from_namezTestLoader.from_names sW == ?7##BG$4$4T$:$:;;A>D);;DAAAr N)__name__
__module__ __qualname__HEREr* r& r' environextendr0 r< rD rQ rU r r r r G s G$%J%33KKKLLL* * *
B B B B Br r c 6 e Zd Zd Zd Zd Zd Z fdZ xZS )ColouredResultc f t j | | t dd d S )NOKgreen)r7
TestResult
addSuccessr )r- rC s r rc zColouredResult.addSuccess{ s1 &&tT222tW r c l t j | || t ddd d S )NERRORr Tr )r7 rb addErrorr r- rC errs r rg zColouredResult.addError s8 $$T4555wD))))))r c h t j | || t dd d S )NFAILr )r7 rb
addFailurer rh s r rl zColouredResult.addFailure s3 &&tT3777vu r c t j | || t d| z d d S )Nz skipped: %sbrown)r7 rb addSkipr strip)r- rC reasons r ro zColouredResult.addSkip s@ ##D$777}v||~~-w77777r c x t |d|dk }t || d S )Nr re rf )r r printErrorList)r- flavourerrors __class__s r rs z ColouredResult.printErrorList s< %g.@AAA
w/////r ) rV rW rX rc rg rl ro rs
__classcell__rv s @r r^ r^ y st * * * 8 8 80 0 0 0 0 0 0 0 0r r^ c d e Zd ZdZerenej Z fdZ fdZ
d Z d Z fdZ
d Zd Z xZS ) 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 ` t j |i | t | _ d S r> )r __init__set
failed_tnames)r- argskwargsrv s r r| zColouredTextRunner.__init__ s1 $)&))) UUr c \ t | _ | j S r> )r _makeResultresult)r- rv s r r z ColouredTextRunner._makeResult s$ gg))++ {r c | j rSt t d 5 }| j D ]}| |dz d d d d S # 1 swxY w Y d S d S )Nwt
)r~ rI rH write)r- rN tnames r _write_last_failedz%ColouredTextRunner._write_last_failed s *($//
*1 ! / * *EGGEDL))))*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
* * *s #AAAc | sF|j |j z D ]8}|d }| j | 7d S d S )Nr )
wasSuccessfulru failuresidr~ add)r- r tr s r _save_resultz ColouredTextRunner._save_result sm ##%% .]V_4
.
.! "&&u---- . .
.
.r c t | }| | |S # t t f$ r5 | j j }| t j d w xY w)Nr )
r runr KeyboardInterrupt
SystemExitrunnerr printErrorsr exit)r- rB r rv s r _runzColouredTextRunner._run s WW[[''F
f % % %M
":. ['F (1++ s
!; ABc |r )r r r )r- rB r s r r zColouredTextRunner.run s7 5!!
6''))*****r )rV rW rX __doc__r r^ r7 TextTestResult resultclassr| r r r r r r rw rx s @r rz rz s %/K..H4KK# # # # # * * * . . . + + + + + + +r rz c @ e Zd Zed Zed Zd ZdS )ParallelRunnerc fd}t j |t j _ t j t }t j | | S )Nc R | | }t j |j |S r> )atexitregisterclose)fdmodekwdsstream orig_fdopens r fdopenz+ParallelRunner._parallelize..fdopen s+ [T**F OFL ) ) )Mr )r' r concurrencytestfork_for_testsNWORKERSConcurrentTestSuite)rB r forkerr s @r _parallelizezParallelRunner._parallelize sS i $*! /99 25&AAAr c t j }t j }| D ]}| dk rt |t j r|j d j }n/t |t j r|}nt d|z t |dd r| | | | ||fS )Nr zcan't recognize type %r
_serialrunF)
r7 r? countTestCasesr4 _testsrv r8 TypeErrorr3 rA )rB serialparallelrC
test_classs r _split_suitezParallelRunner._split_suite s #%%%''
'
'D""$$))D("455
B ![^ 5
D("344
B !
9D @AAAz<77
't$$$$ &&&&!!r c | | \ }}| | }t dt z dd t j }| | }t j |z
}t
j }t
j |d \ }} | r!t d| z d t t j }| | }
t j |z
} | s,| d k r|
t t |j |j |j f \ } }
}t t |
j |
j |
j f \ }}}t' t) j d
|j | |
|||
j |||| f
z t' d |j |
j z || z t fz | o|
}| | d S ) Nz(starting parallel tests using %s workersra Trf 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 psutilProcesschildren
wait_procsr r r r maplenr ru skippedr textwrapdedenttestsRunr )r- rB ser_suite par_suiter par par_elapsedorphansgonealiveser ser_elapsed par_failspar_errs par_skips ser_failsser_errs ser_skipsoks r r zParallelRunner.run sd # 0 0 7 7 9%%i00 9HDT # # # # IKKii ""ikkAo .""++-- ';;; e '%/ 7 7 7 OOO
IKKii ""ikkAo "" y'?'?'A'AA'E'E OO ),S3<36:36;3@ *A *A& 8Y *-S3<36:36;3@ *A *A& 8Y ho <Hi<HiN N O O P P P 6 L3< '{)BH:N N O O O
8S%6%6%8%8
2r N)rV rW rX staticmethodr r r r\ r r r r sZ B B \ B " " \"&. . . . .r r c d }| rWt j r |d n?t |d n,t dk r |d nt t
S t
t
S )Nc , t | dz d d S )Nz Running serial tests instead.r )r )r s r warnzget_runner..warn s s55u=====r z$Can't run parallel tests on Windows.z(concurrencytest module is not installed.r zOnly 1 CPU available.) verbosity)r WINDOWSr r r VERBOSITYrz )r r s r
get_runnerr s > > > 7 > 7 D7 8 8 8 8
$ D; < < < <
]] D( ) ) ) )!I666 6 2 2 22r c t | }t }| | d S r> )r rU r r )r: rB r s r
run_from_namer , s: LL " "4 ( (E
\\F
JJur c L t j j d S r> )r _psplatformcext set_testingr\ r r setupr 2 s" '')))))r c t d} t j | d }| dddd | d ddd
| \ }}|j st
t t }|rWt | d k r(|
t j d S |
|d }n0|j r| }n| }t rt! 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_argsrQ r rH r r print_usager r rU rD r r
r r r )r parseroptsr loaderrB r s r mainr 9 se GGG 8E
"r s & % % % % %
MMMM
FFF OOO
! ! ! ! ! ! & & & & & & / / / / / / # # # # # # . . . . . . & & & & & & & & & & & & $ $ $ $ $ $
( 6 "^
6 4 4 6 6
wrwx0011 2H /B /B /B /B /B /B /B /Bd0 0 0 0 0X, 0 0 0.4+ 4+ 4+ 4+ 4+0 4+ 4+ 4+nR R R R R' R R Rj 3 3 3 3 * * * D zDFFFFF s + 55> AA