U
=_o @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z ddl
m Z ddl
m
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" ddlm#Z# ddlm$Z$ ddlm%Z% ddlm&Z& ddl
Z
ddlZ
e j'dd d kZ(G d d d e Z)e*d d!Z+G d"d# d#e Z,e&-ej./e$ d$G d%d& d&e Z0e1d'kr
dd(l2m3Z3 e3e4 dS ))z
Miscellaneous tests.
N)LINUX)POSIX)WINDOWS)memoize)memoize_when_activated)
supports_ipv6) wrap_numbers)PY3)APPVEYOR)
CI_TESTING) HAS_BATTERY)HAS_MEMORY_MAPS)HAS_NET_IO_COUNTERS)HAS_SENSORS_BATTERY)HAS_SENSORS_FANS)HAS_SENSORS_TEMPERATURES)import_module_by_path)mock)PsutilTestCase)
PYTHON_EXE)
reload_module)ROOT_DIR) SCRIPTS_DIR)sh)unittest ) c @ s e Zd ZefddZdd ZefddZefddZefd d
Zefd d Z d
d Z
dd Z dd Z dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)S )*TestMiscc C s t | j}||}| d| | d|j | | dt| |dd | d| | d| | |
||}| d| | d| t j j
t jd t t d
@ t }||}| d|j | | d | | d | W 5 Q R X t j j
t jd t t d
@ t }||}| d|j | | d
| | d | W 5 Q R X t j j
t jd t t d
4 t }||}| d|j | | d | W 5 Q R X d S )Nzpsutil.Processzpid=%sz name='%s'zname=u'zname='zstatus=z exitcode=zstatus='terminated'nameZ side_effectzstatus='zombie'zname=Z
terminated)psutilProcessZspawn_testprocpidassertInstrr replace assertNotInZ terminatewaitr patchobject
ZombieProcessosgetpid
NoSuchProcess AccessDenied)selffuncpr r4 I/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_misc.pytest_process__repr__= sJ
z TestMisc.test_process__repr__c C s | j td d S )N)r1 )r6 r% r0 r4 r4 r5 test_process__str__a s z TestMisc.test_process__str__c C sN | ttdd | ttjdddd | ttjdddd d S )NA z7psutil.NoSuchProcess process no longer exists (pid=321)foor zCpsutil.NoSuchProcess process no longer exists (pid=321, name='foo')msgzpsutil.NoSuchProcess foo) assertEqualreprr! r. r0 r1 r4 r4 r5 test_no_such_process__repr__d s z%TestMisc.test_no_such_process__repr__c C sj | ttdd | ttjdddd | ttjddddd | ttjddd d
d S ) Nr9 zEpsutil.ZombieProcess process still exists but it's a zombie (pid=321)r: r; zQpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo') )r ZppidzYpsutil.ZombieProcess process still exists but it's a zombie (pid=321, name='foo', ppid=1)r< zpsutil.ZombieProcess foo)r> r? r! r+ r@ r4 r4 r5 test_zombie_process__repr__p s z$TestMisc.test_zombie_process__repr__c C sN | ttdd | ttjdddd | ttjdddd d S )Nr9 z psutil.AccessDenied (pid=321)r: r; z)psutil.AccessDenied (pid=321, name='foo')r< zpsutil.AccessDenied foo)r> r? r! r/ r@ r4 r4 r5 test_access_denied__repr__ s z#TestMisc.test_access_denied__repr__c C sP | ttdd | ttjdddd | ttjddddd d S ) Nr9 z/psutil.TimeoutExpired timeout after 321 secondso )r# z9psutil.TimeoutExpired timeout after 321 seconds (pid=111)r: )r# r zEpsutil.TimeoutExpired timeout after 321 seconds (pid=111, name='foo'))r> r? r! ZTimeoutExpiredr@ r4 r4 r5 test_timeout_expired__repr__ s z%TestMisc.test_timeout_expired__repr__c C s> t }t }| || d|_| || | |d d S )N)r r r: )r! r" r> Z_identZassertNotEqual)r0 Zp1Zp2r4 r4 r5 test_process__eq__ s zTestMisc.test_process__eq__c C s( t t t g}| t|d d S )NrB )setr! r" r> len)r0 sr4 r4 r5 test_process__hash__ s z TestMisc.test_process__hash__c C s t t}|D ]}|dkrq |ds z t| W q tk
r |tjkrtt|}|d kr`Y q |jd k rd|j kr| d| Y q X q tjD ]}|
|| qd S )N)ZlongZtestsZtestPermissionErrorProcessLookupError_Z
deprecatedz%r not in psutil.__all__) dirr!
startswith
__import__ ImportError__all__getattr__doc__lowerfailr$ )r0 Z
dir_psutilr Zfunr4 r4 r5 test__all__ s$
zTestMisc.test__all__c C s$ | ddd tjD tj d S )N.c S s g | ] }t |qS r4 )r% ).0xr4 r4 r5
<listcomp> s z)TestMisc.test_version.<locals>.<listcomp>)r> joinr! version_info __version__r7 r4 r4 r5 test_version s zTestMisc.test_versionc C s" t }d|_| d| d S )N1r: )r! r" r: r' as_dict)r0 r2 r4 r4 r5 !test_process_as_dict_no_new_names s z*TestMisc.test_process_as_dict_no_new_namesc s t fdd}g tdD ].}| }di f}| || | t d q tdD ]0}|d}di f}| || | t d qTtdD ]8}|ddd}dddif}| || | t d q| | }di f}| || | t d
| |jd d S ) Nc s d | |fS )
foo docstringNappend)argskwargsZcallsr4 r5 r: s
z"TestMisc.test_memoize.<locals>.foor r4 rB )rB )barrj r rd )r ranger> rI cache_clearrU )r0 r: r[ retZexpectedr4 ri r5 test_memoize s0 zTestMisc.test_memoizec s G fddd}| }g | | | t d g |j | | | | t d g |j | | | | t d d S )Nc s e Zd Ze fddZdS )z1TestMisc.test_memoize_when_activated.<locals>.Fooc s d d S Nre r7 ri r4 r5 r: s z5TestMisc.test_memoize_when_activated.<locals>.Foo.fooN)__name__
__module__ __qualname__r r: r4 ri r4 r5 Foo s rt r rB )r: r> rI Zcache_activateZcache_deactivate)r0 rt fr4 ri r5 test_memoize_when_activated s z$TestMisc.test_memoize_when_activatedc C s ddl m} dd }| |d|ddi | |d|dd|d d
i | |d |dd|d d i | |d
|dd|d d
i | |d|ddi | |d|ddi | |d|ddi d S )Nr )parse_environ_blockc S s t r | S | S rp )r upper)rJ r4 r4 r5 k s z,TestMisc.test_parse_environ_block.<locals>.kza=1 ara z a=1 b=2 b2za=1 b= z
a=1 b=2 c=3 zxxx a=1 z a=1 =b=2 za=1 b=2)psutil._commonrw r> )r0 rw ry r4 r4 r5 test_parse_environ_block s"
z!TestMisc.test_parse_environ_blockc C s> | tj t rtd }d|_t t r6tW 5 Q R X t tjdtjd}t rdt|j sntW 5 Q R X t tjdtj
d"}t rtt |j stW 5 Q R X t tjdtj
d"}t rtt |j stW 5 Q R X n@| t 0 ttj
tj}z|d W 5 | X W 5 Q R X d S )Nzpsutil._common.socketFz psutil._common.socket.socketr z!psutil._common.socket.socket.bind)z::1r )Z
addCleanupr rm r r) Zhas_ipv6AssertionErrorsocketerrorcalledZgaierror assertRaises ExceptionZAF_INET6Z SOCK_STREAMcloseZbind)r0 rJ Zsockr4 r4 r5 test_supports_ipv6 sD
zTestMisc.test_supports_ipv6c C s ddl m} tjt}||s$t|tj|r8ttj dt
t j dd |
t
|| W 5 Q R X tj dt
t jdd |
t
|| W 5 Q R X tj dt
t jdd ||rtW 5 Q R X tj ddd ||rtW 5 Q R X d S ) Nr )
isfile_strictzpsutil._common.os.statr: r zpsutil._common.stat.S_ISREGFZ return_value)r~ r r, pathabspath__file__r dirnamer r) OSErrorerrnoZEPERMr ZEACCESZENOENT)r0 r Z this_filer4 r4 r5 test_isfile_strict= s$
zTestMisc.test_isfile_strictc s fdd}|t |t |t |t |t jdd |t trnt j
dsnnt s~|t
|t |t t |t d S )Nc s< t d k rt t | t| }t|} | | d S rp )jsonloadsdumpspickler> )rn rz r{ r7 r4 r5 checkO s
z*TestMisc.test_serialization.<locals>.checkr )Zintervalz/proc/diskstats)r! r" rb Zvirtual_memoryZ swap_memoryZ cpu_timesZcpu_times_percentnet_io_countersr r, r existsr
disk_io_countersZdisk_partitionsZ
disk_usagegetcwdusers)r0 r r4 r7 r5 test_serializationN s zTestMisc.test_serializationc C sT t jtd}tr(t j|s(| dS t|}| t |j
| | t
j d S )Nzsetup.pyzcan't find setup.py)r, r r] r r r skipTestr r
SystemExitZsetupr> Z get_versionr! r_ )r0 Zsetup_pymoduler4 r4 r5 test_setup_scripte s
zTestMisc.test_setup_scriptc
C s t jjtjdtjd}t |js*tW 5 Q R X t jjtjdtdd}t |jsbtW 5 Q R X t jjtjdt d.}|
t t W 5 Q R X |jstW 5 Q R X d S )NZ create_timer rB ) r r) r* r! r" r/ r r r+
ValueErrorr )r0 Zmethr4 r4 r5 test_ad_on_process_creationm s& z$TestMisc.test_ad_on_process_creationc
C sR t jddd: | t}tt W 5 Q R X | dt|j W 5 Q R X d S )Nz psutil._psplatform.cext.versionz0.0.0r zversion conflict)
r r) r rR r r! r$ r% Z exceptionrV )r0 cmr4 r4 r5 test_sanity_version_check~ s z"TestMisc.test_sanity_version_checkN)rq rr rs r? r6 r8 rA rC rD rF rG rK rX r` rc ro rv r r r r r r r r4 r4 r4 r5 r ; s( $ ##r r: za b cc @ s e Zd Zdd ZeZdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd Zdd Zdd Zee dd d Zd S ) TestWrapNumbersc C s t d S rp )r rm r7 r4 r4 r5 setUp s zTestWrapNumbers.setUpc C s& dt dddi}| t|d| d S Ndisk1 disk_iontr> r r0 inputr4 r4 r5 test_first_call s z TestWrapNumbers.test_first_callc C s8 dt dddi}| t|d| | t|d| d S r r r r4 r4 r5 test_input_hasnt_changed s z(TestWrapNumbers.test_input_hasnt_changedc C s dt dddi}| t|d| dt dddi}| t|d| dt dddi}| t|d| dt dddi}| t|d| d S ) Nr r r
r r r4 r4 r5 test_increase_but_no_wrap s z)TestWrapNumbers.test_increase_but_no_wrapc C s dt dddi}| t|d| dt dddi}| t|ddt dddi dt dddi}| t|ddt dddi dt dddi}| t|ddt dddi dt dddi}| t|ddt ddd i dt dddi}| t|ddt ddd i dt d
ddi}| t|ddt d dd i dt d ddi}| t|ddt ddd i dt d ddi}| t|ddt ddd i d S )
Nr d r r n Z r 2 ( r r r4 r4 r5 test_wrap sD zTestWrapNumbers.test_wrapc C st dt dddi}| t|d| t dddt dddd}| t|d| dt dddi}| t|d| d S )Nr r r r Zdisk2 r r r4 r4 r5 test_changing_keys s
z"TestWrapNumbers.test_changing_keysc C s. t dddt dddd}| t|d| t dddt dddd}| t|dt dddt dddd dt dddi}| t|d| t dddt dddd}| t|d| t dddt dddd}| t|d| t dddt dddd}| t|dt dddt dddd d S )Nr r r r r r r r r r4 r4 r5 test_changing_keys_w_wrap s<
z)TestWrapNumbers.test_changing_keys_w_wrapc C sb ddddd}| t|d| | t|d| ddddd}t|d}| |d d d
d S ) N) i, # R ȷ ) i r i tU i i r r r r ) 6 r i $ i N rk r i ) iU i i" r i i4 i i i\ )nvme0n1Z nvme0n1p1Z nvme0n1p2Z nvme0n1p3r ) r r r r r r r r r r r i )r> r )r0 doutr4 r4 r5 test_real_data s
z TestWrapNumbers.test_real_datac C sb dt dddi}t|d t }| |d d|i | |d di i | |d di i d S )Nr r r r rB r r r
cache_infor> r0 r cacher4 r4 r5 test_cache_first_call s
z%TestWrapNumbers.test_cache_first_callc C s dt dddi}t|d dt dddi}t|d t }| |d d|i | |d dddddi | |d di i d S ) Nr r r r r rB )r r )r rB r r r r r r4 r4 r5 test_cache_call_twice s
z%TestWrapNumbers.test_cache_call_twicec s dt dddi}t|d dt dddi}t|d t } |d d|i |d dddddi |d ddtd gii fd
d }dt dddi}t|d t } |d d|i | dt ddd i}t|d t } |d d|i | dt ddd
i}t|d t } |d d|i |d dddddi |d ddtd gii d S )Nr r r r r rB r r r c sF t } | d dddddi | d ddtdgii d S ) NrB r r r r r r r )r r r> rH )r r7 r4 r5 assert_, s
z0TestWrapNumbers.test_cache_wrap.<locals>.assert_r r r )r r r r> rH )r0 r r r r4 r7 r5 test_cache_wrap s>
z TestWrapNumbers.test_cache_wrapc C s dt dddi}t|d t dddt dddd}t|d t }| |d d|i | |d dddddi | |d di i d S )
Nr r r r r r rB r r r r r4 r4 r5 test_cache_changing_keysM s
z(TestWrapNumbers.test_cache_changing_keysc C s\ dt dddi}t|d t|d td | t i i i f td td d S )Nr r r z?!?)r r rm r> r r r4 r4 r5 test_cache_clearZ s
z TestWrapNumbers.test_cache_clear
not supportedc C s t rt s| dS t t t }|D ] }| d| | d| q6t j t }|D ] }| d| | d| qjt j t }| |i i i f d S )Nzno disks or NICs availablezpsutil.disk_io_counterszpsutil.net_io_counters)
r! r r r r r r$ rm r' r> )r0 Zcachesr r4 r4 r5 test_cache_clear_public_apisc s
z,TestWrapNumbers.test_cache_clear_public_apisN)rq rr rs r ZtearDownr r r r r r r r r r r r r skipIfr r r4 r4 r4 r5 r s
% 0
r z can't locate scripts directoryc @ s~ e Zd ZdZedd Zed<ddZdd Ze e
d d
d Z d d
Z dd Z
dd Zdd Ze eote ddd Zdd Zdd Zdd Zd d Ze e d d d! Zd"d# Zd$d% Zd&d' Z d(d) Z d*d+ Z d,d- Z e e d.d/d0 Z!d1d2 Z"e e# d d3d4 Z$e e% d d5d6 Z&e e' d e e( d7d8d9 Z)e e' d e e( d7d:d; Z*dS )= TestScriptsz-Tests for scripts in the "scripts" directory.c
O s dt jt| } t| g}|D ]}|| q zt|f| }W n@ tk
r } z"dt |krpt | W Y S W 5 d }~X Y nX |st
||S )Nz%sr/ ) r, r r] r r rf r strip RuntimeErrorr% r )exerg rh cmdargr errr4 r4 r5
assert_stdout s zTestScripts.assert_stdoutNc C sR t jt| } tr"t| ddd}n
t| d}| | }W 5 Q R X t| d S )NZrtutf8)encoding) r, r r] r r openreadastparse)r rg ru srcr4 r4 r5
assert_syntax s
zTestScripts.assert_syntaxc C sV t | }ttD ]>}|drdtj|d |kr| dtjt| qd S )N.pyZtest_r z no test defined for %r script) rO r, listdirr endswithr splitextrW r] )r0 Zmethsr r4 r4 r5
test_coverage s
zTestScripts.test_coveragez
POSIX onlyc C sP t tD ]@}|dr
t jt|}tjt |tj @ s
| d| q
d S )Nr z%r is not executable)
r, r r r r r] statS_IXUSRST_MODErW )r0 r r r4 r4 r5 test_executable s
zTestScripts.test_executablec C s | d d S )Nz
disk_usage.pyr r7 r4 r4 r5 test_disk_usage s zTestScripts.test_disk_usagec C s | d d S )Nzfree.pyr r7 r4 r4 r5 test_free s zTestScripts.test_freec C s | d d S )Nz
meminfo.pyr r7 r4 r4 r5 test_meminfo s zTestScripts.test_meminfoc C s | dtt d S )Nz procinfo.pyr r% r, r- r7 r4 r4 r5
test_procinfo s zTestScripts.test_procinfozno usersc C s | d d S )Nzwho.pyr r7 r4 r4 r5 test_who s zTestScripts.test_whoc C s | d d S )Nzps.pyr r7 r4 r4 r5 test_ps s zTestScripts.test_psc C s | d d S )Nz pstree.pyr r7 r4 r4 r5 test_pstree s zTestScripts.test_pstreec C s | d d S )Nz
netstat.pyr r7 r4 r4 r5 test_netstat s zTestScripts.test_netstatc C s | d d S )Nz ifconfig.pyr r7 r4 r4 r5
test_ifconfig s zTestScripts.test_ifconfigr c C s | dtt d S )Nzpmap.pyr r7 r4 r4 r5 test_pmap s zTestScripts.test_pmapc C s* dt jkr | d| d d S )NZussr z procsmem.py)r! r" Zmemory_full_info_fieldsr r r7 r4 r4 r5
test_procsmem s
zTestScripts.test_procsmemc C s | d d S )Nz
killall.pyr r7 r4 r4 r5 test_killall s zTestScripts.test_killallc C s | d d S )Nz nettop.pyr r7 r4 r4 r5 test_nettop s zTestScripts.test_nettopc C s | d d S )Nztop.pyr r7 r4 r4 r5 test_top s zTestScripts.test_topc C s | d d S )Nziotop.pyr r7 r4 r4 r5
test_iotop s zTestScripts.test_iotopc C s, | dt }| tt | d S )Nzpidof.py)r r! r" r r$ r% r, r- )r0 outputr4 r4 r5
test_pidof s zTestScripts.test_pidofz WINDOWS onlyc C s | d d S )Nzwinservices.pyr r7 r4 r4 r5 test_winservices s z TestScripts.test_winservicesc C s | d d S )Nzcpu_distribution.pyr r7 r4 r4 r5 test_cpu_distribution s z!TestScripts.test_cpu_distributionc C s t s| d | d d S )Nzno temperaturesztemperatures.py)r! Zsensors_temperaturesr r r7 r4 r4 r5 test_temperatures s
z TestScripts.test_temperaturesc C s t s| d | d d S )Nzno fanszfans.py)r! Z sensors_fansr r r7 r4 r4 r5 test_fans s
zTestScripts.test_fansz
no batteryc C s | d d S )Nz
battery.pyr r7 r4 r4 r5 test_battery s zTestScripts.test_batteryc C s | d d S )Nz
sensors.pyr r7 r4 r4 r5 test_sensors s zTestScripts.test_sensors)N)+rq rr rs rU staticmethodr 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 r r r r r4 r4 r4 r5 r ~ sN
r __main__)
run_from_name)5rU r collectionsr r r, r r r sysr! r r r r~ r r r r Zpsutil._compatr Z psutil.testsr
r r r
r r r r r r r r r r r r r r^ Z PYTHON_39r
namedtupler r r r r r rq Zpsutil.tests.runnerr r r4 r4 r4 r5 <module> sf S p