U
|f2 @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd lm
Z
dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z G dd deZ G d d d eZ G d d d eZ G d d! d!eZ!G d"d# d#eZ"e#d$krdd%l$m%Z% e%e& dS )&zContracts 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)FREEBSD)LINUX)MACOS)NETBSD)OPENBSD)POSIX)SUNOS)WINDOWS)long)GITHUB_ACTIONS) HAS_CPU_FREQ)HAS_NET_IO_COUNTERS)HAS_SENSORS_FANS)HAS_SENSORS_TEMPERATURES)PYPY) QEMU_USER) SKIP_SYSCONS)PsutilTestCase)create_sockets)enum)
is_namedtuple)kernel_versionc @ sD e Zd Zdd Zdd Zdd Zdd Zee o2e
d d
d Z d S )
TestAvailConstantsAPIsc C s | ttdtptpt d S )NZ PROCFS_PATH) assertEqualhasattrpsutilr r r self r N/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_contracts.pytest_PROCFS_PATH3 s z'TestAvailConstantsAPIs.test_PROCFS_PATHc C sj | j }|ttdt |ttdt |ttdt |ttdt |ttdt |ttdt d S )NZABOVE_NORMAL_PRIORITY_CLASSZBELOW_NORMAL_PRIORITY_CLASSZHIGH_PRIORITY_CLASSZIDLE_PRIORITY_CLASSZNORMAL_PRIORITY_CLASSZREALTIME_PRIORITY_CLASSr r r r
r Zaer r r test_win_priority6 s z(TestAvailConstantsAPIs.test_win_priorityc C sJ | j }|ttdt |ttdt |ttdt |ttdt d S )NZIOPRIO_CLASS_NONEZIOPRIO_CLASS_RTZIOPRIO_CLASS_BEZIOPRIO_CLASS_IDLEr r r r r# r r r test_linux_ioprio_linux? s
z.TestAvailConstantsAPIs.test_linux_ioprio_linuxc C sJ | j }|ttdt |ttdt |ttdt |ttdt d S )NZ IOPRIO_HIGHZ
IOPRIO_NORMALZ
IOPRIO_LOWZIOPRIO_VERYLOWr" r# r r r test_linux_ioprio_windowsF s
z0TestAvailConstantsAPIs.test_linux_ioprio_windows%unsupported on GITHUB_ACTIONS + LINUXc C s | j }|ttdtpt |ttdtp*t |ttdtp>t |ttdtpRt |ttdtpft |ttdtpzt |ttdtpt |ttdtpt |ttd tpt |ttd
tpt |ttd tpt |ttd t trt d
kr|ttdt t dkr0|ttdt t dkrL|ttdt t dkrh|ttdt t d
kr|ttdt |ttdt |ttdt |ttdt d S )NZ
RLIM_INFINITYZ RLIMIT_ASZ RLIMIT_COREZ
RLIMIT_CPUZ RLIMIT_DATAZ RLIMIT_FSIZEZRLIMIT_MEMLOCKZ
RLIMIT_NOFILEZ RLIMIT_NPROCZ
RLIMIT_RSSZ RLIMIT_STACKZ RLIMIT_LOCKS) ZRLIMIT_MSGQUEUE)r) r* Z RLIMIT_NICEZ
RLIMIT_RTPRIO)r) r* Z
RLIMIT_RTTIMEZRLIMIT_SIGPENDINGZ RLIMIT_SWAPZ
RLIMIT_SBSIZEZ RLIMIT_NPTS)r r r r r r r r# r r r test_rlimitM s6 z"TestAvailConstantsAPIs.test_rlimitN) __name__
__module__ __qualname__r! r$ r& r' unittestskipIfr r r. r r r r r 2 s r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )TestAvailSystemAPIsc C s | ttdt d S )NZwin_service_iterr" r r r r test_win_service_iterq s z)TestAvailSystemAPIs.test_win_service_iterc C s | ttdt d S )NZwin_service_getr" r r r r test_win_service_gett s z(TestAvailSystemAPIs.test_win_service_getc C s& | ttdtp tp tp tp t d S )Ncpu_freq)r r r r r r
r r r r r r
test_cpu_freqw s z!TestAvailSystemAPIs.test_cpu_freqc C s | ttdtpt d S )Nsensors_temperatures)r r r r r r r r r test_sensors_temperatures} s z-TestAvailSystemAPIs.test_sensors_temperaturesc C s | ttdt d S )N sensors_fansr% r r r r test_sensors_fans s z%TestAvailSystemAPIs.test_sensors_fansc C s" | ttdtptptpt d S )NZsensors_battery)r r r r r
r r r r r r test_battery s z TestAvailSystemAPIs.test_batteryN) r/ r0 r1 r5 r6 r8 r: r< r= r r r r r4 p s r4 c @ s| e Zd Zdd Zdd Zdd Zdd Zd d
Ze e
o:e d d d
Z dd Z
dd Zdd Zdd Zdd Zdd ZdS )TestAvailProcessAPIsc C s4 | ttjdtp,tp,tp,tp,tp,t p,t
p,t d S )Nenviron) r r r Processr r r
r r r r r r r r r test_environ s"
z!TestAvailProcessAPIs.test_environc C s | ttjdt d S NZuidsr r r r@ r r r r r test_uids s z TestAvailProcessAPIs.test_uidsc C s | ttjdt d S rB rC r r r r test_gids s z TestAvailProcessAPIs.test_gidsc C s | ttjdt d S )NterminalrC r r r r
test_terminal s z"TestAvailProcessAPIs.test_terminalc C s | ttjdtpt d S )NZionice)r r r r@ r r
r r r r test_ionice s z TestAvailProcessAPIs.test_ionicer( c C s | ttjdtpt d S )NZrlimit)r r r r@ r r r r r r r. s z TestAvailProcessAPIs.test_rlimitc C s" t tjd}| |tpt d S )NZ io_counters)r r r@ r r r r Zhasitr r r test_io_counters s z%TestAvailProcessAPIs.test_io_countersc C s | ttjdt d S )NZnum_fdsrC r r r r test_num_fds s z!TestAvailProcessAPIs.test_num_fdsc C s | ttjdt d S )NZ num_handles)r r r r@ r
r r r r test_num_handles s z%TestAvailProcessAPIs.test_num_handlesc C s | ttjdtptpt d S )NZ cpu_affinity)r r r r@ r r
r r r r r test_cpu_affinity s
z&TestAvailProcessAPIs.test_cpu_affinityc C s | ttjdtptpt d S )NZcpu_num)r r r r@ r r r r r r r test_cpu_num s
z!TestAvailProcessAPIs.test_cpu_numc C s* t tjd}| |tp tp tp t d S )NZ memory_maps)r r r@ r r r r r rI r r r test_memory_maps s z%TestAvailProcessAPIs.test_memory_mapsN)r/ r0 r1 rA rD rE rG rH r2 r3 r r r. rJ rK rL rM rN rO r r r r r> s
r> c @ s
e Zd ZdZedd ZedfddZdd Zd d
Z d d Z
d
d Z e
eoXe dkde
e ddd Zdd Zdd Ze
eddd Zdd Ze
ed d d Ze
e dd d! Ze
e dd"d# Z e
e dd$d% Z d&d' Z d(d) Z!d*S )+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 C s t | _d S N)r r@ proc)clsr r r
setUpClass s z TestSystemAPITypes.setUpClassTc C s6 t |s t|D ] }| || |r| |d qd S )Nr )r AssertionErrorassertIsInstanceZassertGreaterEqual)r nttype_Zgezeronr r r assert_ntuple_of_nums s
z(TestSystemAPITypes.assert_ntuple_of_numsc C s. | t tjddD ]}| | qd S )NT)Zpercpu)rZ r Z cpu_times)r rW r r r test_cpu_times s z!TestSystemAPITypes.test_cpu_timesc C s, | tjd dt | tjddt d S )Nintervalgh㈵>)rV r Z cpu_percentfloatr r r r test_cpu_percent s z#TestSystemAPITypes.test_cpu_percentc C s( | tjd d | tjdd d S )Nr\ g-C 6?)rZ r Zcpu_times_percentr r r r test_cpu_times_percent s z)TestSystemAPITypes.test_cpu_times_percentc C s | t t d S rQ )rV r cpu_countintr r r r test_cpu_count s z!TestSystemAPITypes.test_cpu_countarm64zskipped due to #1892z
not supportedc C s2 t d krtd| jt tttfd d S )Nzcpu_freq() returns NonerX )r r7 r2 ZSkipTestrZ r^ rb r r r r r r8 s
z TestSystemAPITypes.test_cpu_freqc C s< t jdd D ]&\}}| |t | j|ttfd qd S )NT)Zperdiskre )r Zdisk_io_countersitemsrV strrZ rb r )r kvr r r test_disk_io_counters s z(TestSystemAPITypes.test_disk_io_countersc C sJ t D ]<}| |jt | |jt | |jt | |jt qd S rQ )r Zdisk_partitionsrV Zdevicerg Z
mountpointZfstypeopts)r Zdiskr r r test_disk_partitions s
z'TestSystemAPITypes.test_disk_partitionsz
requires rootc C sN t > td}| t|tt| |D ]}t|s.tq.W 5 Q R X d S )Nall)r r Znet_connectionsr lensetr rU )r retconnr r r test_net_connections s
z'TestSystemAPITypes.test_net_connectionsc C s t D ]\}}| |t |D ]j}td k rFtsF| |jtj n| |jt | |j
t | |j tt d f | |j
tt d f q$q d S rQ )r Z net_if_addrsrf rV rg r r familyIntEnumrb addressnetmasktype broadcast)r ifnameaddrsaddrr r r test_net_if_addrs s z$TestSystemAPITypes.test_net_if_addrszQEMU user not supportedc C sx t D ]f\}}| |t | |jt td k rH| |jtj n| |jt
| |j t
| |j t
q d S rQ )
r Z net_if_statsrf rV rg Zisupboolr Zduplexrt rb speedZmtu)r ry infor r r test_net_if_stats s z$TestSystemAPITypes.test_net_if_statsc C s" t jddD ]}| |t q d S )NT)Zpernic)r Znet_io_countersrV rg )r ry r r r test_net_io_counters& s z'TestSystemAPITypes.test_net_io_countersc C sV t D ]D\}}| |t |D ]*}| |jt | |jttt d f q$q d S rQ )
r r; rf rV rg labelcurrentr^ rb rw r nameZunitsunitr r r r< , s
z$TestSystemAPITypes.test_sensors_fansc C s t D ]t\}}| |t |D ]Z}| |jt | |jttt d f | |j
ttt d f | |j ttt d f q$q d S rQ ) r r9 rf rV rg r r r^ rb rw highcriticalr r r r r: 5 s z,TestSystemAPITypes.test_sensors_temperaturesc C s | t t d S rQ )rV r Z boot_timer^ r r r r test_boot_time@ s z!TestSystemAPITypes.test_boot_timec C sb t D ]T}| |jt | |jttd f | |jttd f | |jt td f qd S rQ )
r ZusersrV r rg rF rw hostpidrb )r userr r r
test_usersD s
z TestSystemAPITypes.test_usersN)"r/ r0 r1 __doc__ classmethodrT r^ rZ r[ r_ r` rc r2 r3 r platformmachiner
r8 rj rl r rr r| r r r r r r< r r: r r r r r r rP s:
rP c @ s" e Zd Zee ddd ZdS )TestProcessWaitTypez not POSIXc C sX t | j}| | }| |tj t d k rH|
|t j n |
|t d S rQ )
r r@ Zspawn_testprocr terminatewaitr signalSIGTERMr rV rt rb )r pcoder r r test_negative_signalN s z(TestProcessWaitType.test_negative_signalN)r/ r0 r1 r2 r3 r r r r r r r M s r __main__)
run_from_name)'r r r r2 r r 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 r4 r> rP r r/ Zpsutil.tests.runnerr __file__r r r r sH > ?