| d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
l mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z 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, ddl Z G d$ d%e' Z- G d& d'e' Z. G d( d)e' Z/ G d* d+e' Z0 G d, d-e' Z1d. Z2e) G d/ d0e' Z3e4d1k rdd2l5m6Z6 e6e7 dS dS )3zContracts tests. These tests mainly check API sanity in terms of
returned types and APIs availability.
Some of these are duplicates of tests test_system.py and test_process.py
N)AIX)BSD)FREEBSD)LINUX)MACOS)NETBSD)OPENBSD)OSX)POSIX)SUNOS)WINDOWS)FileNotFoundError)long)range)APPVEYOR)check_connection_ntuple)
CI_TESTING)create_sockets)enum)GITHUB_ACTIONS) HAS_CPU_FREQ)HAS_NET_IO_COUNTERS)HAS_SENSORS_FANS)HAS_SENSORS_TEMPERATURES)
is_namedtuple)kernel_version)process_namespace)PsutilTestCase)PYPY) serialrun) SKIP_SYSCONS)unittest)VALID_PROC_STATUSESc ^ e Zd Zd Zd Zd Zd Z ej e oe
d d Z dS )TestAvailConstantsAPIsc ~ | t t d t p
t pt
d S )N PROCFS_PATH) assertEqualhasattrpsutilr r r selfs }/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib64/python3.11/site-packages/psutil/tests/test_contracts.pytest_PROCFS_PATHz'TestAvailConstantsAPIs.test_PROCFS_PATHB s= 77 .%.3 0 0 0 0 0 c | j } |t t d t |t t d t |t t d t |t t d t |t t d t |t t d t d S )NABOVE_NORMAL_PRIORITY_CLASSBELOW_NORMAL_PRIORITY_CLASSHIGH_PRIORITY_CLASSIDLE_PRIORITY_CLASSNORMAL_PRIORITY_CLASSREALTIME_PRIORITY_CLASSr( r) r* r
r, aes r- test_win_priorityz(TestAvailConstantsAPIs.test_win_priorityF s
768 9 97CCC
768 9 97CCC
760 1 17;;;
760 1 17;;;
762 3 3W===
764 5 5w?????r/ c 4 | j } |t t d t |t t d t |t t d t |t t d t d S )NIOPRIO_CLASS_NONEIOPRIO_CLASS_RTIOPRIO_CLASS_BEIOPRIO_CLASS_IDLEr( r) r* r r8 s r- test_linux_ioprio_linuxz.TestAvailConstantsAPIs.test_linux_ioprio_linuxO s
76. / /777
76, - -u555
76, - -u555
76. / /77777r/ c 4 | j } |t t d t |t t d t |t t d t |t t d t d S )N IOPRIO_HIGH
IOPRIO_NORMAL
IOPRIO_LOWIOPRIO_VERYLOWr7 r8 s r- test_linux_ioprio_windowsz0TestAvailConstantsAPIs.test_linux_ioprio_windowsV s
76= ) )7333
76? + +W555
76< ( ('222
76+ , ,g66666r/ %unsupported on GITHUB_ACTIONS + LINUXc | j } |t t d t pt |t t d t pt |t t d t pt |t t d t pt |t t d t pt |t t d t pt |t t d t pt |t t d t pt |t t d t pt |t t d
t pt |t t d t pt |t t d t t
rt
d
k r$ |t t d t t
dk r$ |t t d t t
dk r$ |t t d t t
dk r$ |t t d t t
d
k r$ |t t d t |t t d t |t t d t |t t d t d S )N
RLIM_INFINITY RLIMIT_AS RLIMIT_CORE
RLIMIT_CPU RLIMIT_DATA RLIMIT_FSIZERLIMIT_MEMLOCK
RLIMIT_NOFILE RLIMIT_NPROC
RLIMIT_RSS RLIMIT_STACK RLIMIT_LOCKS) RLIMIT_MSGQUEUE)rV rW RLIMIT_NICE
RLIMIT_RTPRIO)rV rW
RLIMIT_RTTIMERLIMIT_SIGPENDING RLIMIT_SWAP
RLIMIT_SBSIZE RLIMIT_NPTS)r( r) r* r r r r r8 s r- test_rlimitz"TestAvailConstantsAPIs.test_rlimit] s
76? + +U-=g>>>
76; ' ')9':::
76= ) )5+;G<<<
76< ( (%*:7;;;
76= ) )5+;G<<<
76> * *E,w???
76? + +U-=g>>>
76> * *E, * *E, * *E222
@ 9,,76#455u=== :--76=115999 :--76?33U;;; :--76?33U;;; 9,,76#677???
76= ) )7333
76? + +W555
76= ) )733333r/ N) __name__
__module__ __qualname__r. r: rA rG r" skipIfr r rc r/ r- r% r% @ s 0 0 0@ @ @8 8 87 7 7 X_^-<> > 4 4> > 4 4 4r/ r% c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestAvailSystemAPIsc b | t t d t d S )Nwin_service_iterr7 r+ s r- test_win_service_iterz)TestAvailSystemAPIs.test_win_service_iter s) );< F 2 3 3U5Eg G G G G Gr/ c b | t t d t d S )N sensors_fansr@ r+ s r- test_sensors_fansz%TestAvailSystemAPIs.test_sensors_fans s( 88%@@@@@r/ c | t t d t pt p
t
pt d S )Nsensors_battery)r( r) r* r r
r r r+ s r- test_batteryz TestAvailSystemAPIs.test_battery sC ):;; ='=W= ? ? ? ? ?r/ N) rd re rf rm rp rs rv ry r| rh r/ r- rj rj sz G G GF F F? ? ?G G GA A A? ? ? ? ?r/ rj c e Zd Zd Zd Zd Zd Zd Z ej e
oe d d Z d Z
d Zd
Zd Zd Zd
ZdS )TestAvailProcessAPIsc | t t j d t p0t
p)t p"t pt pt p
t pt d S )Nenviron) r( r) r* Processr r r
r r r r r r+ s r- test_environz!TestAvailProcessAPIs.test_environ si ;; 6% 67 6c 6U 6 6$+6/5 7 7 7 7 7r/ c l | t t j d t d S Nuidsr( r) r* r r r+ s r- test_uidsz TestAvailProcessAPIs.test_uids * 88%@@@@@r/ c l | t t j d t d S r r r+ s r- test_gidsz TestAvailProcessAPIs.test_gids r r/ c l | t t j d t d S )Nterminalr r+ s r-
test_terminalz"TestAvailProcessAPIs.test_terminal s* < >N N> >NC C CD D DJ J J6 6 64 4 4I I I I Ir/ r~ c | e Zd ZdZed ZedfdZd Zd Z d Z
d Z e j
e d d
Zd Zd Z e j
ed
d Zd Zd Z e j
e d d Z e j
e d d Z e j
e d d Zd Z d Z dS )TestSystemAPITypeszCheck the return types of system related APIs.
Mainly we want to test we never return unicode on Python 2, see:
https://github.com/giampaolo/psutil/issues/1039
c 6 t j | _ d S N)r* r proc)clss r-
setUpClassz TestSystemAPITypes.setUpClass s >##r/ Tc t | sJ |D ]0}| || |r| |d 1d S Nr )r assertIsInstanceassertGreaterEqual)r, nttype_gezerons r- assert_ntuple_of_numsz(TestSystemAPITypes.assert_ntuple_of_nums sd R . .A ! !!U + + +
.''1--- . .r/ c | t j t j d D ]}| | d S )NT)percpu)r r* cpu_times)r, r s r- test_cpu_timesz!TestSystemAPITypes.test_cpu_times s] ""6#3#5#5666"$/// + +B & &r * * * * + +r/ c | t j d t | t j d t d S )Nintervalgh㈵>)r r* cpu_percentfloatr+ s r- test_cpu_percentz#TestSystemAPITypes.test_cpu_percent sP f 0$ ? ? ?GGG f 0' B B BEJJJJJr/ c | t j d | t j d d S )Nr g-C 6?)r r* cpu_times_percentr+ s r- test_cpu_times_percentz)TestSystemAPITypes.test_cpu_times_percent sL ""6#;T#J#J#JKKK ""6#;V#L#L#LMMMMMr/ c ^ | t j t d S r )r r* cpu_countintr+ s r- test_cpu_countz!TestSystemAPITypes.test_cpu_count s' f . 0 0#66666r/ z
not supportedc t j | d | t j t t
t f d S )Nzcpu_freq() returns Noner )r* rr skipTestr r r r r+ s r- rs z TestSystemAPITypes.test_cpu_freq sQ ? $-- 9:: : ""6?#4#4UC>@@ O OKD% ! !$ , , ,
O
O%%dj#666%%dlUCd4LMMM%%di%d4jj1IJJJ%%dmeS$t**5MNNNN
O O Or/ c ^ | t j t d S r )r r* boot_timer r+ s r- test_boot_timez!TestSystemAPITypes.test_boot_time> s' f . 0 0%88888r/ c t j D ]}| |j t | |j t t
d f | |j t t
d f | |j t t
d f d S r )
r* usersr r r r r hostpidr )r, users r-
test_usersz TestSystemAPITypes.test_usersB s LNN ? ?D ! !$)S 1 1 1 ! !$-#tDzz1B C C C ! !$)c4::-> ? ? ? ! !$(S$t**,= > > > > ? ?r/ N) rd re rf __doc__ classmethodr r r r r r r r" rg r rs r r r! r r r r r r ry r rv r r rh r/ r- r r s
$ $ [$ /4D . . . .+ + +
K K KN N N7 7 7 X_ %77P P 87P
= = = 5 5 5 X_\?33+ + 43+ I I I
1
1
1 X_,,o>>/ / ?>/
X_))?;;N N <;N X_11?CCO O DCO9 9 9? ? ? ? ?r/ r c D e Zd Z ej e d d ZdS )TestProcessWaitTypez not POSIXc v t j | j }| | }| |t j t "|
|t j d S |
|t d S r )
r* r spawn_testprocr terminatewaitr( signalSIGTERMr r r r )r, pcodes r- test_negative_signalz(TestProcessWaitType.test_negative_signalM s N4..004 5 5
vvxx /// ! !$ 5 5 5 5 5 ! !$ , , , , ,r/ N)rd re rf r" rg r r rh r/ r- r r K s> X_Y ,,- - -,- - -r/ r c
t fd
fd} t j ddg }n# t j $ r i cY S w xY w|d |d c
d j i}t
}| |j d D ]=\ }} | ||< # t j $ r} |
Y d }~6d }~ww xY w | |S )Nc | j | j | t | t j r>| j 6 | j d | j | n/t | t j r | t | | j sJ d S r ) r( r r
isinstancer*
ZombieProcessppidr
NoSuchProcessassertProcessGoner msg)excr r r# r tcases r- check_exceptionz"proc_info..check_exceptiona s
#'3'''
#(D))) c6/ 0 0 *x#((1555!!#(D111
V 1
2
2 * # #D ) ) ) Cwwr/ c dk rC d d S # t j $ r} | Y d } ~ d S d } ~ ww xY wd S r )r r* Error)r' r) r r r# r s r- do_waitzproc_info..do_waitm st !88
7 ! <
7
7
7 T4666666666
7 8s AAAr# r r F) clear_cache)
r r* r as_dictr$ r r itergettersr+ )
r r, dr nsfunfun_namer' r) r r# r r( s
` @@@@@r- proc_infor5 ^ sp E
7 7 7 7 7 7 7 7 7~c"" LL&&) * * 6AfIJD$ 48 D 4 B ??
X SUUDNN| OCtT 2 2 2 HHHH GIII Ks) +A A%$A%3
CC(C##C(c e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d! Z#d" Z$d# Z%d$ Z&d% Z'd& Z(d'S )(TestFetchAllProcesseszTest which iterates over all running processes and performs
some sanity checks against Process API's returned values.
Uses a process pool to get info about all processes.
c 6 t j | _ d S r )multiprocessingPoolpoolr+ s r- setUpzTestFetchAllProcesses.setUp s #(** r/ c j | j | j d S r )r; r joinr+ s r- tearDownz TestFetchAllProcesses.tearDown s. r/ c f ddl m} | j |t j S )Nr )r5 )psutil.tests.test_contractsr5 r; imap_unorderedr* pids)r, r5 s r- iter_proc_infoz$TestFetchAllProcesses.iter_proc_info s4 :99999y'' 6;==AAAr/ c ` g }| D ]}| D ]\ }}t | | } ||| |ddg d di fvr |s
J | 9# t $ r d}|d|d|d dt | d z
}|d
z
}|d t j z z
}d d | D }|d z
}| | Y w xY w|r(|
d | d S )
Nr zH
======================================================================
z FAIL: test_z pid=r z, ret=
zF----------------------------------------------------------------------z
%sc 3 K | ] }d |z V
dS )z Nrh ).0is r- z1TestFetchAllProcesses.test_all.. s& !H!HA7a-!H!H!H!H!H!Hr/ ) rD r getattrAssertionErrorrepr traceback
format_excr>
splitlinesappendfail)r, failuresr r valuemethss r- test_allz TestFetchAllProcesses.test_all s '')) , ,D #zz||
,
, e tT**
,D%%% QRr2$>>> $++e++u & ' ' '.AA d5kkk4;;;;8 8AMA)"6"8"888A !H!H !H!H!HHHAIA OOA&&&&&'
, /))BGGH - -.. . / /s A&&BD ?D c | | |t |D ] }| |t d S r )r listr )r, r r parts r- cmdlinez TestFetchAllProcesses.cmdline sJ c4((( - -D ! !$ , , , , - -r/ c | |t t d f |s| |d d S t r| d sd S t j | s
J | t rt j
| rt t d rxt t d re t
j |t j
sJ d S # t $ r. t j | rt s Y d S Y d S w xY wd S d S d S d S )NrG z.exeaccessX_OK)r r r r( r
endswithospathisabsr isfiler) r_ r` rN existsr r, r r s r- exezTestFetchAllProcesses.exe sq cCd#4555 " S" % % % % %
s||F 3 3
7==%% * *s * *%
" ,,
"2x(( "WR-@-@ "" !ybg 6 666 666) " " " 7>>#.. "z " !" " " " " ""
"
"
"
"" " " "s !D 1D87D8c h | |t | |d d S r r r r rg s r- r zTestFetchAllProcesses.pid 4 c3''' Q'''''r/ c v | |t t f | |d d S r )r r r r rg s r- r# zTestFetchAllProcesses.ppid s8 cC;/// Q'''''r/ c | |t t r|s|d dk rd S t s|sJ d S d S )Nstatusstopped)r r r r rg s r- r zTestFetchAllProcesses.name sZ c3''' C DNi$?$? F JJ3 Jr/ c | |t | |d n/# t $ r" t r|d t
j k rn Y nw xY wt j dt j | d S )Nr rn z%Y %m %d %H:%M:%S)
r r r rN r r*
STATUS_ZOMBIEtimestrftime localtimerg s r- create_timez!TestFetchAllProcesses.create_time s c5))) # #C + + + +
4>V-AAA
)4>#+>+>?????s 4 )A A c t | sJ |D ]3}| |t | |d 4d S r )r r r r )r, r r uids r- r zTestFetchAllProcesses.uids s^ S!!!!! , ,C ! !#s + + + # #C + + + + , ,r/ c t | sJ |D ]A}| |t t s t s| |d Bd S r )r r r r r r )r, r r gids r- gidszTestFetchAllProcesses.gids sm S!!!!! 0 0C ! !#s + + +
0
0''Q/// 0 0r/ c D | |t |sJ d S r )r r rg s r- usernamez TestFetchAllProcesses.username s% c3'''
s
r/ c | |t |sJ | |d | |t d S )N?)r r assertNotEqualassertInr# rg s r- rn z TestFetchAllProcesses.status sR c3'''
s C%%%
c./////r/ c t | sJ |D ]@}| |t t f |dk r| |d Ad S )Nr r r r r r )r, r r fields r- r z!TestFetchAllProcesses.io_counters si S!!!!! 2 2E ! !%#t 5 5 5{{''q111 2 2r/ c t rx| |j t | |j t | |j d | |j d d S t j t j t j t j
g}| |t | |d | || d S r ) r r ioclassr rV r r* rF rE rD rC r )r, r r choicess r- r z TestFetchAllProcesses.ionice s
( ! !#+s 3 3 3 ! !#)S 1 1 1 # #CK 3 3 3 # #CIq 1 1 1 1 1 %!$" $G
! !#s + + + # #C + + + MM#w ' ' ' ' 'r/ c | |t t r|s|d dk rd S | |d d S )Nrn ro )r r r r rg s r- num_threadsz!TestFetchAllProcesses.num_threads sV c3''' C DNi$?$? F Q'''''r/ c X | |t |D ]}t | sJ | |j d | |j d | |j d |D ]$}| |t t f %d S r ) r r[ r r id user_time system_timer r )r, r r tr s r- threadsz TestFetchAllProcesses.threads" s c4((( ; ;A ## # ## # #AD! , , , # #AK 3 3 3 # #AM1 5 5 5
;
;%%ec5\::::
; ; ;r/ c t | sJ |D ]3}| |t | |d 4d S r )r r r r )r, r r r s r- r z TestFetchAllProcesses.cpu_times, s^ S!!!!! * *A ! !!U + + + # #Aq ) ) ) ) * *r/ c n | |t d|cxk rdk s n J | d S )NrF g Y@r r rg s r- r z!TestFetchAllProcesses.cpu_percent3 sK c5)))c""""U"""""C"""""r/ c b | |t t r|dk rd S | |d t j dk r| |d | |t t t j d S )Nr r r )
r r r r r* r r( r r[ r rg s r- r z TestFetchAllProcesses.cpu_num7 s c3''' sbyy F Q''' " " S! $ $ $
c4f&6&8&8 9 9::;;;;;r/ c t | sJ |D ]:}| |t t f | |d ;t
r| |j |j | |j |j | |j
|j | |j |j
d S d S r )r r r r r r
peak_wsetwsetpeak_paged_pool
paged_poolpeak_nonpaged_pool
nonpaged_pool
peak_pagefilepagefiler, r r rV s r- memory_infoz!TestFetchAllProcesses.memory_info@ s S!!!!! . .E ! !%#t 5 5 5 # #E1 - - - - E # #CM38 < < < # #C$7 H H H # #C$:C>#&& + + + +& ? + +r/ c |D ]K}| |j t | |j t | |j t |j D ]}t
|| }|dk rI| d s3t j |j sJ |j a|dk r|sJ t | |dk r!t s|sJ t | | |t t f | |d Md S )Nrc [r permsr )r r r r rc r rM
startswithrb rd rO r
r r r )r, r r r fnamerV s r- r z!TestFetchAllProcesses.memory_maps sU 6 6B ! !"'3 / / / ! !"(C 0 0 0 ! !"'3 / / /
6
6 E**F?? ++C00 ? !w}}RW 5 5>>rw>> 5 f__ --$u++--5-g%%" 2 $11d5kk11u))%#t===++E15555
6 6 6r/ c h | |t | |d d S r rj rg s r- r z!TestFetchAllProcesses.num_handles rk r/ c | |t t rd|cxk rdk s n J | d S d t t D }| || t j dk r"| |t j d S | |t d S )Ni c b g | ],}| d t t | -S )_PRIORITY_CLASS)ra rM r* )rJ xs r-
z.TestFetchAllProcesses.nice.. sC < < < ZZ(9 : :<'&!,, < < r s+
, , , , , , ! ! ! ! ! ! 0 0 0 0 0 0 # # # # # # ' ' ' ' ' ' ' ' ' ' ' ' % % % % % % , , , , , , ) ) ) ) ) ) 1 1 1 1 1 1 & & & & & & ' ' ' ' ' ' * * * * * * ' ' ' ' ' ' " " " " " " % % % % % % ! ! ! ! ! ! , , , , , ,
<4 <4 <4 <4 <4^ <4 <4 <4~? ? ? ? ?. ? ? ?0-I -I -I -I -I> -I -I -Ij|? |? |? |? |? |? |? |?~ - - - - -. - - -&( ( (V Q* Q* Q* Q* Q*N Q* Q* Q*h
z111111M( r/