U
Ã|ýfVL ã @ s¾ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
ddlmZ 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" e one Z#dd „ Z$e"G d d „ d e ƒƒZ%G d d „ d e ƒZ&e'd!krºdd"l(m)Z) e)e*ƒ dS )#zeIterate over all process PIDs and for each one of them invoke and
test all psutil.Process() methods.
é N)ÚAIX)ÚBSD)ÚFREEBSD)ÚLINUX)ÚMACOS)ÚNETBSD)ÚOPENBSD)ÚOSX)ÚPOSIX)ÚWINDOWS)ÚPY3)ÚFileNotFoundError)Úlong)Úunicode)Ú
CI_TESTING)Ú QEMU_USER)ÚVALID_PROC_STATUSES)ÚPsutilTestCase)Úcheck_connection_ntuple)Úcreate_sockets)Ú
is_namedtuple)Úis_win_secure_system_proc)Úprocess_namespace)Ú serialrunc s( t ƒ ‰‡‡fdd„‰ ‡ ‡‡‡‡fdd„}zt ˆ¡‰W n$ tjk
rZ ˆ ˆ¡ i Y S X zˆ ddg¡}W n tjk
rŽ ˆ ˆ¡ Y n–X |d |d ‰‰dˆji}tˆƒ}|j |j
dd D ]T\}}z|ƒ ||< W qÄ tj k
r } zˆ |ˆˆˆƒ W Y ¢qÄW 5 d }~X Y qÄX qÄ|ƒ |S d S )
Nc sŽ ˆ | jˆ ¡ | jd k r&ˆ | j|¡ t| tjƒrdˆ |¡ | jd k rzˆ | jd¡ ˆ | j|¡ nt| tj ƒrzˆ
|¡ t | ƒ t | ƒ d S ©Nr )
Ú assertEqualÚpidÚnameÚ
isinstanceÚpsutilZ
ZombieProcessZassertProcessZombieÚppidÚassertGreaterEqualÚ
NoSuchProcessÚassertProcessGoneÚstrÚrepr)ÚexcÚprocr r )r Útcase© úP/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_process_all.pyÚcheck_exception6 s
z"proc_info..check_exceptionc
sN ˆdkrJzˆ d¡ W n2 tjk
rH } zˆ | ˆˆˆƒ W 5 d } ~ X Y nX d S r )Úwaitr ÚError)r& )r+ r r r r' r) r* Údo_waitD s
zproc_info..do_waitr r r F)Ú clear_cache) r r ÚProcessr" Z
assertPidGoneÚas_dictr# r r ÚiterÚgettersr- )r r. ÚdÚinfoÚnsZfunZfun_namer& r) )r+ r r r r' r( r* Ú proc_info3 s.
r7 c @ s8 e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd„ Zd d „ Zd d „ Zd d!„ Zd"d#„ Zd$d%„ Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zd2d3„ Z d4d5„ Z d6d7„ Z d8d9„ Z d:d;„ Z dd?„ Z"d@dA„ Z#dBdC„ Z$dDdE„ Z%dFdG„ Z&dHdI„ Z'dJdK„ Z(dLS )MÚTestFetchAllProcessesz·Test 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 C s t d¡ trt ¡ | _d S ©NF)r Ú
_set_debugÚ
USE_PROC_POOLÚmultiprocessingZPoolÚpool©Úselfr) r) r* ÚsetUpk s
zTestFetchAllProcesses.setUpc C s& t d¡ tr"| j ¡ | j ¡ d S ©NT)r r: r; r= Ú terminateÚjoinr> r) r) r* ÚtearDownr s
z TestFetchAllProcesses.tearDownc C sJ ddl m} tr"| j |t ¡ ¡S g }t ¡ D ]}| ||ƒ¡ q.|S d S )Nr )r7 )Z psutil.tests.test_process_allr7 r; r= Zimap_unorderedr ÚpidsÚappend)r? r7 Zlsr r) r) r* Úiter_proc_infox s z$TestFetchAllProcesses.iter_proc_infoc C sè g }| ¡ D ]Â}| ¡ D ]´\}}t| |ƒ}z|||ƒ W nr tk
rª d}|d||d t|ƒ|f 7 }|d7 }|dt ¡ 7 }d dd„ | ¡ D ƒ¡d }| |¡ Y qX |d d
g d d i fkr|st
|ƒ‚qq |rä| d |¡¡‚d S ) NzH
======================================================================
z+FAIL: name=test_%s, pid=%s, ret=%s
info=%s
r zF----------------------------------------------------------------------z
%sÚ
c s s | ]}d | V qdS )z Nr) )Ú.0Úir) r) r* Ú – s z1TestFetchAllProcesses.test_all..r ç Ú ) rG ÚitemsÚgetattrÚ Exceptionr% Ú tracebackÚ
format_excrC Ú
splitlinesrF ÚAssertionErrorZfail)r? Zfailuresr5 r ÚvalueÚmethÚsr) r) r* Útest_all… s,
ü z TestFetchAllProcesses.test_allc C s& | |t¡ |D ]}| |t¡ qd S ©N)ÚassertIsInstanceÚlistr$ )r? Úretr5 Úpartr) r) r* Úcmdlinež s z TestFetchAllProcesses.cmdlinec C s° | |ttf¡ | | ¡ |¡ |r¬tr6| d¡s6d S tj |¡sJt
|ƒ‚t r¬tj |¡r¬t
tdƒr¬t
tdƒr¬zt |tj¡s‚t
‚W n& t
k
rª tj |¡r¦ts¦‚ Y nX d S )Nz.exeÚaccessÚX_OK)rZ r$ r r Ústripr ÚendswithÚosÚpathÚisabsrT r
ÚisfileÚhasattrr_ r` Úexistsr ©r? r\ r5 r) r) r* Úexe£ s zTestFetchAllProcesses.exec C s | |t¡ | |d¡ d S r ©rZ Úintr! ri r) r) r* r ¶ s zTestFetchAllProcesses.pidc C s( | |ttf¡ | |d¡ t|ƒ d S r )rZ rl r r! r7 ri r) r) r* r º s zTestFetchAllProcesses.ppidc C s@ | |ttf¡ tr(|s(t|d ƒr(d S ts<|s t |ƒs t‚|D ](}| |ttf¡ |dkr| |d¡ qd S )Néÿÿÿÿr ©r rT rZ rl r r! )r? r\ r5 Úfieldr) r) r* Ú io_countersõ s
z!TestFetchAllProcesses.io_countersc C sz t r>| |jt¡ | |jt¡ | |jd¡ | |jd¡ n8tjtjtj tj
g}| |t¡ | |d¡ | ||¡ d S r ) r rZ Zioclassrl rU r! r ZIOPRIO_VERYLOWZ
IOPRIO_LOWZ
IOPRIO_NORMALZ IOPRIO_HIGHrz )r? r\ r5 Úchoicesr) r) r* Úioniceü s ü z TestFetchAllProcesses.ionicec C s8 | |t¡ tr(|dkr(t|d ƒr(d S | |d¡ d S )Nr r é )rZ rl r r r! ri r) r) r* Ú num_threads
s z!TestFetchAllProcesses.num_threadsc C sj | |t¡ |D ]T}t|ƒs t‚| |jd¡ | |jd¡ | |jd¡ |D ]}| |tt f¡ qNqd S r )
rZ r[ r rT r! ÚidZ user_timeZ system_timerl rn )r? r\ r5 Útr} r) r) r* Úthreads s z TestFetchAllProcesses.threadsc C s2 t |ƒs t‚|D ] }| |t¡ | |d¡ qd S r )r rT rZ rn r! )r? r\ r5 Únr) r) r* Ú cpu_times s z TestFetchAllProcesses.cpu_timesc C s. | |t¡ d| kr dks*n t|ƒ‚d S )NrL g Y@©rZ rn rT ri r) r) r* Ú cpu_percent% s z!TestFetchAllProcesses.cpu_percentc C s\ | |t¡ tr |dkr d S | |d¡ t ¡ dkr@| |d¡ | |tt t ¡ ƒƒ¡ d S )Nr{ r r )
rZ rl r r! r Ú cpu_countr rz r[ Úrangeri r) r) r* Úcpu_num) s z TestFetchAllProcesses.cpu_numc C sz t |ƒs t‚|D ] }| |ttf¡ | |d¡ qtrv| |j|j¡ | |j |j
¡ | |j |j ¡ | |j
|j¡ d S r )r rT rZ rl r r! r Z peak_wsetZwsetZpeak_paged_poolZ
paged_poolZpeak_nonpaged_poolZ
nonpaged_poolZ
peak_pagefileZpagefile©r? r\ r5 rU r) r) r* Ú memory_info2 s z!TestFetchAllProcesses.memory_infoc C s t |ƒs t‚t ¡ j}|jD ]Z}t||ƒ}| |tt f¡ | j
|d||fd t s t r`|dkr`q | j
|||||fd q t rŒ|
|j|j¡ d S )Nr )Úmsg)ZvmsÚdata)r rT r Zvirtual_memoryÚtotalÚ_fieldsrO rZ rl r r! r r ZassertLessEqualZpssZuss)r? r\ r5 r‘ r rU r) r) r* Úmemory_full_info= s
z&TestFetchAllProcesses.memory_full_infoc C s | |t¡ |D ]þ}| |jt¡ | |jt¡ | |j ¡ |j¡ trZ| |jd¡ nft r´| |j
t¡ | |j t¡ | |j t¡ |
|j
d¡ | |j d¡ | |j d¡ n trÀ|jsÀqtj |j¡sÖt|ƒ‚zt |j¡}W n tk
rú Y qX t |j¡st|ƒ‚qd S )Nr{ r )ÚrÚwÚazr+za+)rZ r[ Úfdrl rd r$ r ra r r ÚpositionÚmodeÚflagsr! rz Z
assertGreaterr rc re rT Ústatr
ÚS_ISREGÚst_mode)r? r\ r5 ÚfÚstr) r) r* Ú
open_filesM s,
z TestFetchAllProcesses.open_filesc C s | |t¡ | |d¡ d S r rk ri r) r) r* Únum_fdsg s z TestFetchAllProcesses.num_fdsc C sL t ƒ < | t|ƒtt|ƒƒ¡ |D ]}t|ƒs4t‚t|ƒ q$W 5 Q R X d S rY )r r ÚlenÚsetr rT r )r? r\ r5 Úconnr) r) r* Únet_connectionsk s
z%TestFetchAllProcesses.net_connectionsc
C sž | |ttf¡ | | ¡ |¡ |rštj |¡s8t|ƒ‚zt |¡}W nB t
k
rˆ } z$t rjt j
|¡rjn|jtjkrx‚ W 5 d }~X Y nX t |j¡sšt‚d S rY )rZ r$ r r ra rc rd re rT r› ÚOSErrorr r Z _psplatformZis_permission_errÚerrnoÚENOENTÚS_ISDIRr )r? r\ r5 rŸ Úerrr) r) r* Úcwdr s zTestFetchAllProcesses.cwdc C s. | |t¡ d| kr dks*n t|ƒ‚d S )Nr éd rˆ ri r) r) r* Úmemory_percent‚ s z$TestFetchAllProcesses.memory_percentc C s | |t¡ d S rY )rZ Úboolri r) r) r* Ú
is_running† s z TestFetchAllProcesses.is_runningc C sN | |t¡ | |g ¡ ttt ¡ ƒƒ}|D ] }| |t¡ | ||¡ q,d S rY )rZ r[ ry r‹ r rŠ rl rz )r? r\ r5 Zcpusr† r) r) r* Ú cpu_affinity‰ s z"TestFetchAllProcesses.cpu_affinityc C sH | |ttd ƒf¡ |d k rDtj |¡s0t|ƒ‚tj |¡sDt|ƒ‚d S rY )rZ r$ Útyperc rd re rT rh ri r) r) r* Úterminal‘ s z TestFetchAllProcesses.terminalc C sÎ |D ]Ä}| |jt¡ | |jt¡ | |jt¡ |jD ]Ž}t||ƒ}|dkrr| d¡sÆtj |j¡sÆt
|jƒ‚q8|dkrŒ|sÆt
t |ƒƒ‚q8|dkrªt sÆ|sÆt
t |ƒƒ‚q8| |t
tf¡ | |d¡ q8qd S )Nrd )ú[z anon_inode:ÚaddrÚpermsr )rZ r´ r$ rµ rd r’ rO Ú
startswithrc re rT r% r rl r r! )r? r\ r5 ÚntÚfnamerU r) r) r* Ú memory_maps— s
z!TestFetchAllProcesses.memory_mapsc C s | |t¡ | |d¡ d S r rk ri r) r) r* Ú num_handles s z!TestFetchAllProcesses.num_handlesc C sr | |t¡ tr0d| kr$dksnn t|ƒ‚n>dd„ ttƒD ƒ}| ||¡ trb| |tj ¡ n | |t¡ d S )Niìÿÿÿé c S s g | ]}| d ¡rtt|ƒ‘qS )Z_PRIORITY_CLASS)rb rO r )rI Úxr) r) r* Ú
¶ s
þz.TestFetchAllProcesses.nice..)
rZ rl r
rT Údirr rz r ÚenumÚIntEnum)r? r\ r5 Z
prioritiesr) r) r* Únice± s þ zTestFetchAllProcesses.nicec C s6 t |ƒs t‚|D ] }| |ttf¡ | |d¡ qd S r r| r r) r) r* Únum_ctx_switchesÁ s z&TestFetchAllProcesses.num_ctx_switchesc C s@ | |t¡ | t|ƒd¡ | |d d¡ | |d d¡ d S )Né r r{ r )rZ Útupler r¢ r! ri r) r) r* ÚrlimitÇ s z TestFetchAllProcesses.rlimitc C s: | |t¡ | ¡ D ] \}}| |t¡ | |t¡ qd S rY )rZ ÚdictrN r$ )r? r\ r5 ÚkÚvr) r) r* ÚenvironÍ s z TestFetchAllProcesses.environN))Ú__name__Ú
__module__Ú __qualname__Ú__doc__r@ rD rG rX r^ rj r r r rr rt rv rw rm 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* r8 d sL
r8 c @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) Ú
TestPidsRangea@ Given pid_exists() return value for a range of PIDs which may or
may not exist, make sure that psutil.Process() and psutil.pids()
agree with pid_exists(). This guarantees that the 3 APIs are all
consistent with each other. See:
https://github.com/giampaolo/psutil/issues/2359
XXX - Note about Windows: it turns out there are some "hidden" PIDs
which are not returned by psutil.pids() and are also not revealed
by taskmgr.exe and ProcessHacker, still they can be instantiated by
psutil.Process() and queried. One of such PIDs is "conhost.exe".
Running as_dict() for it reveals that some Process() APIs
erroneously raise NoSuchProcess, so we know we have problem there.
Let's ignore this for now, since it's quite a corner case (who even
imagined hidden PIDs existed on Windows?).
c C s t d¡ d S r9 ©r r: r> r) r) r* r@ å s zTestPidsRange.setUpc C s t d¡ d S rA rÏ r> r) r) r* rD è s zTestPidsRange.tearDownc
sV dd„ }‡ fdd„}t ddƒD ]2}tr0||ƒr0q ˆ j|d ||ƒ W 5 Q R X q d S )Nc
S s‚ zt d| dƒ}W n tk
r( Y dS X |J |D ]6}| d¡r4t| ¡ d ƒ}|| k W 5 Q R £ S q4tdƒ‚W 5 Q R X d S )Nz/proc/%s/statusÚrbFs Tgid:r z'Tgid' line not found)Úopenr
r¶ rl ÚsplitÚ
ValueError)r rž ÚlineZtgidr) r) r* Ú is_linux_tidì s
z+TestPidsRange.test_it..is_linux_tidc s¨ d}t | ¡}zb|r4t | ¡ tsnˆ | t ¡ ¡ n:tsZˆ t j¡ t | ¡ W 5 Q R X tsnˆ | t ¡ ¡ W n, t j
t fk
rœ |d8 }|dkr˜‚ Y qX d S qd S )Né r r ) r Z
pid_existsr0 r rz rE r Ú assertRaisesr" Z assertNotInr- rT )r r¼ rh r> r) r* Úcheckû s"
z$TestPidsRange.test_it..checkr i¸ )r )r‹ r ZsubTest)r? rÕ rØ r r) r> r* Útest_itë s zTestPidsRange.test_itN)rÊ rË rÌ rÍ r@ rD rÙ r) r) r) r* rÎ Ô s rÎ Ú__main__)Ú
run_from_name)+rÍ r¿ r§ r< rc r› ro rQ r r r r r r r r r r
r Zpsutil._compatr r
r r Z psutil.testsr r r r r r r r r r r; r7 r8 rÎ rÊ Zpsutil.tests.runnerrÛ Ú__file__r) r) r) r* Ú sT 1 qN