%Ojkk d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddl m
Z
ddl mZ ddl mZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# dd lm$Z$ dd lm%Z% dd lm&Z& dd lm'Z' dd lm(Z( dd!lm)Z) dd"lm*Z* dd#lm+Z+ ddlZddlZ G d$ d%e Z, G d& d'e Z- G d( d)e Z. G d* d+e Z/ G d, d-e Z0e# G d. d/e' Z1 G d0 d1e Z2 G d2 d3e Z3e4d4k rdd5l5m6Z6 e6e7 dS dS )6z3
Tests for testing utils (psutil.tests namespace).
N)FREEBSD)NETBSD)POSIX) open_binary) open_text)
supports_ipv6) bind_socket)bind_unix_socket)
call_until)chdir)
CI_TESTING)create_sockets)
get_free_port)HAS_CONNECTIONS_UNIX)
is_namedtuple)mock)process_namespace)PsutilTestCase)
PYTHON_EXE)
reap_children)retry)retry_on_failure)
safe_mkdir) safe_rmpath) serialrun)system_namespace)tcp_socketpair) terminate)TestMemoryLeak)unittest)unix_socketpair)
wait_for_file) wait_for_pidc : e Zd Z ej d d Z ej d d Z ej d d Z ej d d Z ej d d Z ej d d Z
dS ) TestRetryDecoratorz
time.sleepc t ddd fd }t t d | | d | |j d d S )N retriesintervallogfunc B r ddz dS Nr( r popqueues }/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/psutil/tests/test_testutils.pyfooz2TestRetryDecorator.test_retry_success..fooC 2
A
1 )r listrange assertEqual
call_countselfsleepr4 r2 s @r3 test_retry_successz%TestRetryDecorator.test_retry_success? s
q1T 2 2 2
3 2 U1XX """ )1-----r6 c t ddd fd }t t d | t | | |j d d S )Nr' r( r) c B r ddz dS r. r/ r1 s r3 r4 z2TestRetryDecorator.test_retry_failure..fooQ r5 r6 )r r8 r9 assertRaisesZeroDivisionErrorr: r; r< s @r3 test_retry_failurez%TestRetryDecorator.test_retry_failureN s
q1T 2 2 2
3 2 U1XX +S111 )1-----r6 c t t d d }| t | | |j d d S )Nr( ) exceptionr+ c t N) TypeError r6 r3 r4 z2TestRetryDecorator.test_exception_arg..foo^ s Or6 r )r
ValueErrorrC rJ r: r; r= r> r4 s r3 test_exception_argz%TestRetryDecorator.test_exception_arg\ s` a 0 0 0
1 0
)S))) )1-----r6 c t dd d d }| t | | |j d d S )Nr' r) c ddz d S r. rK rK r6 r3 r4 z4TestRetryDecorator.test_no_interval_arg..fooi
EEEEr6 r r rC rD r: r; rM s r3 test_no_interval_argz'TestRetryDecorator.test_no_interval_arge se
q4 5 5 5
6 5
+S111 )1-----r6 c t ddd d }| t | | |j d d S )Nr' r( r) c ddz d S r. rK rK r6 r3 r4 z0TestRetryDecorator.test_retries_arg..foos rQ r6 rR rM s r3 test_retries_argz#TestRetryDecorator.test_retries_argp se
q1T 2 2 2
3 2
+S111 )1-----r6 c L | t t dd d S )Nr' r( )r* timeout)rC rL r )r= r> s r3 test_retries_and_timeout_argsz0TestRetryDecorator.test_retries_and_timeout_argsz s% *eQBBBBBr6 N) __name__
__module__ __qualname__r patchr? rE rN rS rV rY rK r6 r3 r% r% = s TZ . . . TZ . . . TZ . . . TZ . . . TZ . . . TZ C C C C Cr6 r% c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestSyncTestUtilsc R t t j t t j dz }t
j dt dg 5 | t j
t | d d d d S # 1 swxY w Y d S )Ni psutil.tests.retry.__iter__r return_value) r# osgetpidmaxpsutilpidsr r] iterrC
NoSuchProcess)r= nopids r3 test_wait_for_pidz#TestSyncTestUtils.test_wait_for_pid s RY[[!!!FKMM""U*
Z5D!II
N
N
N I I f 2L% H H H I I I I I I I I I I I I I I I I I Is ('B B #B c | }t |d 5 }| d d d d n # 1 swxY w Y t | t j | rJ d S )Nwr4
get_testfnopenwriter" rd pathexistsr= testfnfs r3 test_wait_for_filez$TestSyncTestUtils.test_wait_for_file s ""
&#
!
GGENNN f 7>>&))))))) AA A c | }t |d 5 d d d n # 1 swxY w Y t |d t j | rJ d S )Nrn T)empty)rp rq r" rd rs rt r= rv s r3 test_wait_for_file_emptyz*TestSyncTestUtils.test_wait_for_file_empty s ""
&#
fD))))7>>&)))))))s 377c | }t j dt dg 5 | t
t | d d d d S # 1 swxY w Y d S )Nra r rb )rp r r] ri rC IOErrorr" r| s r3 test_wait_for_file_no_filez,TestSyncTestUtils.test_wait_for_file_no_file s ""
Z5D!II
N
N
N > > g}f = = = > > > > > > > > > > > > > > > > > >s "A((A,/A,c | }t |d 5 }| d d d d n # 1 swxY w Y t |d t j | sJ d S )Nrn r4 F)deletero ru s r3 test_wait_for_file_no_deletez.TestSyncTestUtils.test_wait_for_file_no_delete s ""
&#
!
GGENNN fU++++w~~f%%%%%%%ry c T t d d }| |d d S )Nc dS )Nr( rK rK r6 r3 z3TestSyncTestUtils.test_call_until.. s r6 zret == 1r( )r r: )r= rets r3 test_call_untilz!TestSyncTestUtils.test_call_until s/ J// a r6 N) rZ r[ r\ rl rx r} r r r rK r6 r3 r_ r_ sq I I I * * ** * *> > >
& & &! ! ! ! !r6 r_ c , e Zd Zd Zd Zd Zd Zd ZdS )TestFSTestUtilsc t t 5 }| |j d d d d d S # 1 swxY w Y d S )Nrt)r __file__r: moder= rw s r3 test_open_textz TestFSTestUtils.test_open_text s
x
+A QVT * * * + + + + + + + + + + + + + + + + + + >AAc t t 5 }| |j d d d d d S # 1 swxY w Y d S )Nrb)r r r: r r s r3 test_open_binaryz TestFSTestUtils.test_open_binary s
"
" +a QVT * * * + + + + + + + + + + + + + + + + + +r c | }t | t j | sJ t | t j | sJ d S rI )rp r rd rs isdirr| s r3 test_safe_mkdirz TestFSTestUtils.test_safe_mkdir sg ""6w}}V$$$$$6w}}V$$$$$$$r6 c | }t |d t | t j | rJ t | t j | t | t j | rJ t j dt t j d 5 }|
t 5 t | d d d n # 1 swxY w Y |j sJ d d d d S # 1 swxY w Y d S )Nrn zpsutil.tests.os.stat ) side_effect)rp rq closer rd rs rt mkdirr r] OSErrorerrnoEINVALrC called)r= rv ms r3 test_safe_rmpathz TestFSTestUtils.test_safe_rmpath s "" VS !!!F7>>&)))))F
F7>>&)))))
Z.$+EL"$=$=? ? ? BC""7++
$
$F###
$
$
$
$
$
$
$
$
$
$
$
$
$
$
$8OO8O s6 (ED E D# #E&D# ' EEEc | }t j }t j | t | 5 | t j t j || d d d n # 1 swxY w Y | t j | d S rI )rp rd getcwdr r r: rs join)r= rv bases r3
test_chdirzTestFSTestUtils.test_chdir s ""y{{
6]] F F RY[["',,tV*D*D E E E F F F F F F F F F F F F F F F d+++++s AB B!$B!N)rZ r[ r\ r r r r r rK r6 r3 r r s_ + + ++ + +% % % &, , , , ,r6 r c V e Zd Zd Zd Z ej e d d Zd Z dS )TestProcessUtilsc | }t j |j }| sJ t | rJ t j j rJ t j j rJ d S rI ) spawn_testprocrg Processpid
is_runningr tests
_pids_started_subprocesses_started)r= subpps r3 test_reap_childrenz#TestProcessUtils.test_reap_children s| ""$$ N48 $ $||~~ ~<<>>!!!<----<555555r6 c | \ }}| |j |j | sJ | sJ t j }| ||g t j d }| t | d | || | || | |
t j | |
|j t | | rJ | sJ t | | rJ d S )NT) recursive )spawn_children_pairassertNotEqualr r rg r childrenr: lenassertInppidrd re r )r= child
grandchildr s r3 test_spawn_children_pairz)TestProcessUtils.test_spawn_children_pair s 4 4 6 6z EIz~666 !!!!!$$&&&&&>##,,.. E7+++>##,,t,<< X ***
eX&&&
j(+++ ry{{333 **EI666%##%%%%%$$&&&&&* ((*******r6
POSIX onlyc | \ }}| | t j d S rI ) spawn_zombier: statusrg
STATUS_ZOMBIE)r= parentzombies r3 test_spawn_zombiez"TestProcessUtils.test_spawn_zombie s> **,, &*>?????r6 c | }t | | | t | t j | j }t | | | t | t ddg}t j |t j t j }t | | | t | | j }t | | | t | t ra| \ }}t | t | | | | | d S d S )Nz-cz import time; time.sleep(60);)stdoutstderr) r r assertProcessGonerg r r r Popen
subprocessPIPEr r )r= r cmdr r r s r3 test_terminatez TestProcessUtils.test_terminate s ! !! q!!!! N4..004 5 5! q!!!! 4!?@ LZ_Z_ M M M! q!!!! !!##'# q!!!# + ! . . 0 0NFF f f " "6 * * * " "6 * * * * * + +r6 N)
rZ r[ r\ r r r skipIfr r r rK r6 r3 r r sp 6 6 6+ + +* X_Y --@ @ .-@ + + + + +r6 r c e Zd Zd Z ej e d d Zd Z ej e d ej e pe
d d Z d Z dS ) TestNetUtilsc t }t j t d|f 5 }| | d | d d d d S # 1 swxY w Y d S )Nr addrr( )r
contextlibclosingr r: getsockname)r= portss r3 r zTestNetUtils.bind_socket s
"d < < <
=
= 7 Q]]__Q / 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s /A//A36A3r c ( | }t | }t j | 5 | |j t j | |j t j | |
| t j
| sJ t j t j | j sJ d d d n # 1 swxY w Y | }t |t j }t j | 5 | |j t j d d d d S # 1 swxY w Y d S )N)type)rp r
r r r: familysocketAF_UNIXr SOCK_STREAMr rd rs rt statS_ISSOCKst_mode
SOCK_DGRAM)r= namesocks r3 test_bind_unix_socketz"TestNetUtils.test_bind_unix_socket s %%
%
% 8 8 T[&. 9 9 9 TY(: ; ; ; T - - / / 6 6 67>>$'' ' ''= !677 7 77 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 6+<===
%
% ; ; TY(9 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;s$ CDD
D
&FF F c dt f}t t j | \ }}t j | 5 t j | 5 | | | | | | | | | d d d n # 1 swxY w Y d d d d S # 1 swxY w Y d S )Nz 127.0.0.1r )
r r r AF_INETr r r: r getpeernamer )r= r serverclients r3 tcp_tcp_socketpairz TestNetUtils.tcp_tcp_socketpair. s ]__-'TBBB
'
' @ @#F++
@
@ !3!3!5!5t<<< !3!3!5!5t<<<##F$6$6$8$8$???
@
@
@
@
@
@
@
@
@
@
@
@
@
@
@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s7 C5A9C C5 C! !C5$C! %C55C9<C9z*/var/run/log UNIX socket opened by defaultc | t j }| }| d rJ | }t | \ }} t j | sJ t j
t
j | j sJ | | |z
d | t | d d | | | | | | | | d S # | | w xY w)Nunix)kindr )rg r num_fds connectionsrp r! rd rs rt r r r r: r r r r )r= r r r r r s r3 test_unix_socketpairz!TestNetUtils.test_unix_socketpair9 sk
N ))++==f=----- (.. 7>>$'' ' ''= !677 7 77 QYY[[7 2A 6 6 6 S F !;!; < .funb s JJJ!OJJJJJr6 r r
)times warmup_times )executer: )r= r r s @r3
test_timeszTestMemLeakClass.test_timesa s] aj S 444 UR(((((r6 c n | t | j d d | t | j d d | t | j d d | t | j d d | t | j d
d d S ) Nc dS Nr rK rK r6 r3 r z1TestMemLeakClass.test_param_err..i A r6 r r c dS r rK rK r6 r3 r z1TestMemLeakClass.test_param_err..j r r6 c dS r rK rK r6 r3 r z1TestMemLeakClass.test_param_err..k r r6 )r c dS r rK rK r6 r3 r z1TestMemLeakClass.test_param_err..l r r6 ) tolerancec dS r rK rK r6 r3 r z1TestMemLeakClass.test_param_err..m r r6 )r* )rC rL r r= s r3 test_param_errz TestMemLeakClass.test_param_errh s *dlIIQGGG *dlIIRHHH *dlIIBOOO *dlIILLL *dlIIrJJJJJr6 z
skipped on CIc l g }|fd} | t d| j |d ~d S # ~w xY w)Nc 6 | ddz d S Nxxxxxxxxxxxxxxxxxxxxxxxxi appendlss r3 r z+TestMemLeakClass.test_leak_mem..funt s IIho & & & & &r6 z extra-mem2 r )assertRaisesRegexAssertionErrorr )r= r$ r s r3
test_leak_memz TestMemLeakClass.test_leak_memo sf ' ' ' ' " ">;#'<B
#
@
@
@ FFFFs $0 3c z fd}g t rdnd} t d|z j | d S )Nc t t } | j | d S rI )rq r
addCleanupr r" )rw boxr= s r3 r z1TestMemLeakClass.test_unclosed_files..fun s5 XA OOAG $ $ $ JJqMMMMMr6 fdhandlez unclosed )r r& r' r )r= r r r, s ` @r3 test_unclosed_filesz$TestMemLeakClass.test_unclosed_files~ si
*tt( ~{T/A #|S 2 2 2 2 2r6 c fd}g d}| ||dd | t |dz d S )Nc 8 ddz d S r r! r# s r3 r z,TestMemLeakClass.test_tolerance..fun s IIho & & & & &r6 d r i )r r r r( )r r: r )r= r r r$ s @r3 test_tolerancez TestMemLeakClass.test_tolerance sp ' ' ' ' '
SA 0 2 2 2 R%!),,,,,r6 c ~ d }| t | | t 5 | t | d d d n # 1 swxY w Y d }| t 5 | t | d d d d S # 1 swxY w Y d S )Nc ddz d S r. rK rK r6 r3 r z0TestMemLeakClass.test_execute_w_exc..fun s
EEEEr6 c d S rI rK rK r6 r3 r z0TestMemLeakClass.test_execute_w_exc..fun s Dr6 )
execute_w_excrD rC r r' )r= r s r3 test_execute_w_excz#TestMemLeakClass.test_execute_w_exc sF ,c222
0
1
1 - - w , , , - - - - - - - - - - - - - - -
~
.
. 7 7 0# 6 6 6 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7s# A!!A%(A% B22B69B6N)
rZ r[ r\ r r r r r r
r( r/ r3 r8 rK r6 r3 r r ^ s ) ) )K K K X_Z 11 21 2 2 2- - -
7
7
7
7
7r6 r c e Zd Zd Zd ZdS )TestTestingUtilsc 2 t j }t | }| d | |j D d d }| | | d S )Nc * g | ]}|d dk |S )r( r rK .0xs r3
z;TestTestingUtils.test_process_namespace.. s! @@@Q1 q r6 r )rg r r testri gettersr: r )r= r nsr s r3 test_process_namespacez'TestTestingUtils.test_process_namespace s~ N
q
!
!
@@"''"*--@@@CAF )))))r6 c t }d | |j D d d }| | t j d S )Nc * g | ]}|d dk |S )r( net_if_addrsrK r= s r3 r@ z:TestTestingUtils.test_system_namespace.. s& HHHQ1 1G1Gq1G1G1Gr6 r )r ri rB r: rg rG )r= rC r s r3 test_system_namespacez&TestTestingUtils.test_system_namespace sc
HH"''"*--HHHKAN 3 5 566666r6 N)rZ r[ r\ rD rH rK r6 r3 r: r: s2 * * *7 7 7 7 7r6 r: c e Zd Zd ZdS )TestOtherUtilsc t t j dd ddd sJ t t rJ d S )Nr4 za b cr( r r7 )r r
namedtupletupler s r3 test_is_namedtuplez!TestOtherUtils.test_is_namedtuple sP C[ 3E7 C CAq! L LMMMMM )))))))r6 N)rZ r[ r\ rN rK r6 r3 rJ rJ s# * * * * *r6 rJ __main__)
run_from_name)8__doc__r r r rd r r r rg r r r psutil._commonr r r psutil.testsr r
r r r
r r r r r r r r r r r r r r r r r r r r! r" r# r% r_ r r r r r: rJ rZ psutil.tests.runnerrP r rK r6 r3 rU s
& & & & & & $ $ $ $ $ $ ( ( ( ( ( ( $ $ $ $ $ $ ) ) ) ) ) ) # # # # # # # # # # # # ' ' ' ' ' ' & & & & & & - - - - - - & & & & & & * * * * * * ' ' ' ' ' ' # # # # # # & & & & & & ) ) ) ) ) ) # # # # # # $ $ $ $ $ $ " " " " " " ) ) ) ) ) ) ' ' ' ' ' ' " " " " " " ' ' ' ' ' ' ! ! ! ! ! ! ( ( ( ( ( ( & & & & & & % % % % % %
?C ?C ?C ?C ?C ?C ?C ?CD$! $! $! $! $! $! $! $!N*, *, *, *, *,n *, *, *,ZA+ A+ A+ A+ A+~ A+ A+ A+HDA DA DA DA DA> DA DA DAN =7 =7 =7 =7 =7~ =7 =7 =7@ 7 7 7 7 7~ 7 7 7 * * * * *^ * * * z111111M( r6