U=Z d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m Z ddl m
Z
ddl mZ ddl mZ ddl mZ ddl mZ dd l mZ dd
l mZ dd l mZ dd l mZ dd
l mZ ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl m Z ddl m Z ddl m!Z! ddl m"Z" ddl m#Z# ddl m$Z$ ddl m%Z% ddl m&Z& dd l m'Z' dd l m(Z( dd l m)Z) dd l m*Z* dd l m+Z+ dd!l m,Z, dd"l m-Z- dd#l m.Z. G d$ d%e' Z/ G d& d'e' Z0 G d( d)e' Z1 G d* d+e' Z2 G d, d-e' Z3d. Z4e. G d/ d0e' Z5e6d1k rdd2l7m8Z8 e8e9 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)unicode)APPVEYOR)
CI_TESTING)GITHUB_ACTIONS) HAS_CPU_FREQ)HAS_NET_IO_COUNTERS)HAS_SENSORS_FANS)HAS_SENSORS_TEMPERATURES)PYPY) SKIP_SYSCONS)VALID_PROC_STATUSES)PsutilTestCase)check_connection_ntuple)create_sockets)enum)
is_namedtuple)kernel_version)process_namespace) serialrunc ^ 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 o/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/psutil/tests/test_contracts.pytest_PROCFS_PATHz'TestAvailConstantsAPIs.test_PROCFS_PATHD 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_priorityH 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_linuxQ 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_windowsX 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 unittestskipIfr r rc r/ r- r% r% B 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 rn rq rt rw rz r} ri r/ r- rk rk s} G G GF F FJ J JG G GA A A? ? ? ? ?r/ rk 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> >N6 6 6D 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
eo ej d k d
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 Z d 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/ arm64zskipped due to #1892z
not supportedc t j | d | t j t t
t f d S )Nzcpu_freq() returns Noner )r* rs skipTestr r r r r+ s r- rt z TestSystemAPITypes.test_cpu_freq sS ? $-- 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_timeC 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_usersG 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 rg rh r platformmachiner rt r r r r r r r r r rz r rw r
r ri r/ r- r r s
$ $ [$ /4D . . . .+ + +
K K KN N N7 7 7 X_U< /x / 1 1W <+- -X_ %77P P 87- -P
= = = C C C 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_signalR s N4..004 5 5
vvxx /// ! !$ 5 5 5 5 5 ! !$ , , , , ,r/ N)rd re rf rg rh r r" ri r/ r- r r P 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 | d S r ) r( r r
isinstancer*
ZombieProcessppidr
NoSuchProcessassertProcessGoner )excr r r' r tcases r- check_exceptionz"proc_info..check_exceptionf s
#'3'''
#(D))) c6/ 0 0 *x#((1555!!#(D111
V 1
2
2 * # #D ) ) ) Cr/ 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_waitq 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_infor8 c 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 H t st j | _ d S d S r )r multiprocessingPoolpoolr+ s r- setUpzTestFetchAllProcesses.setUp s* /',..DIII / /r/ c | t s4| j | j d S d S r )r r> r joinr+ s r- tearDownz TestFetchAllProcesses.tearDown s@ I ! ! ! INN r/ c ddl m} t s,| j |t j S g }t j D ] }| || !|S )Nr )r8 )psutil.tests.test_contractsr8 r r> imap_unorderedr* pidsappend)r, r8 lsr s r- iter_proc_infoz$TestFetchAllProcesses.iter_proc_info sx :99999 9++Iv{}}EE EB {}}
*
* ))C..))))Ir/ 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 Nri ).0is r- z1TestFetchAllProcesses.test_all.. s& !H!HA7a-!H!H!H!H!H!Hr/ ) rI r getattrAssertionErrorrepr traceback
format_excrA
splitlinesrG fail)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!HHH4OA OOA&&&&&'
, /))BGGH - -.. . / /s A&&BC>=C>c | | |t |D ] }| |t d S r )r listr )r, r r parts r- cmdlinez TestFetchAllProcesses.cmdline sJ c4((( - -D ! !$ , , , , - -r/ c | |t t f | | | |rt
r| d sd S t j | s
J | t rt j | rt t d rzt t d rg 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 d S )Nz.exeaccessX_OK)r r r r( stripr
endswithospathisabsr isfiler) rc rd rS existsr r, r r s r- exezTestFetchAllProcesses.exe sv cC>222 c*** "
s||F 3 3
7==%% * *s * *%
" ,,
"2x(( "WR-@-@ "" !ybg 6 666 666) " " " 7>>#.. "z " !" " " " " "" " "
"
"
"
"" " " "s %!D 1E ?E c h | |t | |d d S r r r r rl s r- r zTestFetchAllProcesses.pid 4 c3''' Q'''''r/ c v | |t t f | |d d S r )r r r r rl s r- r' zTestFetchAllProcesses.ppid s8 cC;/// Q'''''r/ c | |t t f t r|s|d dk rd S t s|sJ d S d S )Nstatusstopped)r r r r r rl s r- r zTestFetchAllProcesses.name s^ cC>222 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 rs z%Y %m %d %H:%M:%S)
r r r rS r r*
STATUS_ZOMBIEtimestrftime localtimerl 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 | |t | | | | sJ d S r )r r r( re rl s r- usernamez TestFetchAllProcesses.username sO c3''' c***yy{{{r/ c | |t |sJ | |d | |t d S )N?)r r assertNotEqualassertInr rl s r- rs 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 rZ 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 )Nrs rt )r r r r rl 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.threads0 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 )NrK g Y@r r rl s r- r z!TestFetchAllProcesses.cpu_percentA 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 rl s r- r z TestFetchAllProcesses.cpu_numE 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 rZ s r- memory_infoz!TestFetchAllProcesses.memory_infoN s S!!!!! . .E ! !%#t 5 5 5 # #E1 - - - - E # #CM38 < < < # #C$7 H H H # #C$:C222 c*** 07==%% * *s * *%
0WS\\ |BJ///////
sy *< = =Y%,.. /..... DDDDD
0 0s 7B( (
C926C44C9c n | |t d|cxk rdk s n J | d S )Nr d r rl s r- memory_percentz$TestFetchAllProcesses.memory_percent s= c5)))C3r/ c < | |t d S r )r r rl s r-
is_runningz TestFetchAllProcesses.is_running s c4(((((r/ c " | |t |g k s
J | t t t j }|D ]3}| |t
| || 4d S r )r r_ r r* r r r )r, r r cpusr s r- r z"TestFetchAllProcesses.cpu_affinity s c4(((byyy#yyyE&*,,--.. # #A ! !!S ) ) ) MM!T " " " " # #r/ c | |t t d f |Rt j | s
J | t j | s J | d S d S r )r r r rg rh ri rk rl s r- r z TestFetchAllProcesses.terminal sy cCd#4555 ?7==%% * *s * *%7>>#&& + + + +& ? + +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 )Nrh )[z anon_inode:r permsr )r r r r rh r rR
startswithrg ri rT r
r r r )r, r r r fnamerZ s r- r z!TestFetchAllProcesses.memory_maps sV 6 6B ! !"'3 / / / ! !"(C 0 0 0 ! !"'3 / / /
6
6 E**F?? ++,@AA ? !w}}RW 5 5>>rw>> 5 f__ --$u++--5-g%%" 2 $11d5kk11u))%#t===++E15555
6 6 6r/ c h | |t | |d d S r ro rl s r- r z!TestFetchAllProcesses.num_handles rp 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)rf rR r* )rO xs r-
z.TestFetchAllProcesses.nice.. sC < < < ZZ(9 : :<'&!,, < < r s=
, , , , , , " " " " " " ! ! ! ! ! ! # # # # # # ' ' ' ' ' ' % % % % % % , , , , , , ) ) ) ) ) ) 1 1 1 1 1 1 % % % % % % , , , , , , ' ' ' ' ' ' 0 0 0 0 0 0 ' ' ' ' ' ' & & & & & & ' ' ' ' ' ' * * * * * * " " " " " "<4 <4 <4 <4 <4^ <4 <4 <4~? ? ? ? ?. ? ? ?0-I -I -I -I -I> -I -I -Ij? ? ? ? ? ? ? ?D - - - - -. - - -&' ' 'T ]* ]* ]* ]* ]*N ]* ]* ]*@ z111111M( r/