U
=_~ @ s
d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
dd lm Z dd
lm Z dd lm
Z
dd lmZ d
d Zdd Zdd Zee dG dd deZee dG dd deZedkrddlmZ ee dS )zmacOS specific tests. N)MACOS) HAS_BATTERY)PsutilTestCase)retry_on_failure)sh)spawn_testproc) terminate)TOLERANCE_DISK_USAGE)TOLERANCE_SYS_MEM)unittestc C s< t | }| d }z
t|W S tk
r6 | Y S X dS )zmExpects a sysctl command with an argument and parse the result
returning only the value of interest.
N)r splitint
ValueError)Zcmdlineoutresult r H/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_osx.pysysctl s
r c C sT ddl m} td}|dD ]}| |kr q8q tdttd|d| S )z)Wrapper around 'vm_stat' cmdline utility.r ) getpagesizevm_stat
zline not foundz\d+) Zpsutil._psutil_posixr r r
r r researchgroup)Zfieldr r liner r r r % s r c C s ddi}| }d}| r$| dd s4| dd dkrN|| d 7 }| dd } qt|}| }| D ]\}}||krf qqf|dkr|d }| }n td| |d di}t|dd D ] \}} d|d d > || < qt||| S )
NZ customary) BKMGTPEZY r r .kzcan't interpret %r
)isdigitfloatstripitemsupperr enumerater ) sZSYMBOLSZinitnumZletternameZssetprefixir r r human2bytes3 s* $
r4 z
MACOS onlyc @ s, e Zd Zedd Zedd Zdd ZdS ) TestProcessc C s t j| _d S N)r pidclsr r r
setUpClassP s zTestProcess.setUpClassc C s t | j d S r6 )r r7 r8 r r r
tearDownClassT s zTestProcess.tearDownClassc C s t d| j }|dd }|dd }|dd }t| j }| |t
dt | | |t
dt | d S ) Nzps -o lstart -p %sZSTARTEDr% z%H:%M:%Sz%Y) r r7 replacer+ r
psutilZProcessZ create_time assertEqualtimestrftime localtime)selfoutputZstart_psZhhmmssZyearZ start_psutilr r r test_process_create_timeX s z$TestProcess.test_process_create_timeN)__name__
__module__ __qualname__ classmethodr: r; rG r r r r r5 M s
r5 c @ s e Zd Ze dd Zdd Zdd Zdd Zd d
Ze d d Z e d
d Z
e dd Z e dd Z e dd Z
e dd Zdd Zee ddd Zd S ) TestSystemAPIsc C s| dd }t jddD ]b}t |j}||j\}}}}| |j| | |j| | j|j|t d | j|j
|t d qd S )Nc S s t d| }|d}|d |d}| d d \}}}}|dkrRd}t|d }t|d }t|d }||||fS )Nz
df -k "%s"r r Znoner% i )r r+ r
popr )pathr linesr devtotalusedfreer r r dfo s
z%TestSystemAPIs.test_disks.<locals>.dfF)allZdelta) r@ Zdisk_partitionsZ
disk_usageZ
mountpointrA ZdevicerR assertAlmostEqualrT r rS )rE rU partZusagerQ rR rS rT r r r
test_disksk s
zTestSystemAPIs.test_disksc C s t d}| |tjdd d S )Nzsysctl hw.logicalcpuTZlogicalr rA r@ cpu_countrE r0 r r r test_cpu_count_logical s z%TestSystemAPIs.test_cpu_count_logicalc C s t d}| |tjdd d S )Nzsysctl hw.physicalcpuFr[ r\ r^ r r r test_cpu_count_physical s z&TestSystemAPIs.test_cpu_count_physicalc C sZ t }| |jd d td | |jd d td | |jd d td d S )Ni zsysctl hw.cpufrequencyzsysctl hw.cpufrequency_minzsysctl hw.cpufrequency_max)r@ Zcpu_freqrA Zcurrentr minmax)rE Zfreqr r r
test_cpu_freq s z TestSystemAPIs.test_cpu_freqc C s t d}| |t j d S )Nzsysctl hw.memsize)r rA r@ virtual_memoryrR )rE Zsysctl_hwphymemr r r test_vmem_total s z TestSystemAPIs.test_vmem_totalc C s& t d}t j}| j||td d S )NrT rW )r r@ rd rT rX r
rE Z
vmstat_valZ
psutil_valr r r test_vmem_free s
z TestSystemAPIs.test_vmem_freec C s& t d}t j}| j||td d S )NactiverW )r r@ rd rh rX r
rf r r r test_vmem_active s
z TestSystemAPIs.test_vmem_activec C s& t d}t j}| j||td d S )NinactiverW )r r@ rd rj rX r
rf r r r test_vmem_inactive s
z!TestSystemAPIs.test_vmem_inactivec C s& t d}t j}| j||td d S )NwiredrW )r r@ rd rl rX r
rf r r r test_vmem_wired s
z TestSystemAPIs.test_vmem_wiredc C s" t d}t j}| || d S )NZPageins)r r@ swap_memoryZsinrA rf r r r test_swapmem_sin s
z TestSystemAPIs.test_swapmem_sinc C s" t d}t j}| || d S )NZPageout)r r@ rn ZsoutrA rf r r r test_swapmem_sout s
z TestSystemAPIs.test_swapmem_soutc C st t D ]b\}}ztd| }W n tk
r8 Y q X | j|jd|k|d | |jtt
d|d q d S )Nz ifconfig %sZRUNNING)msgz mtu (\d+)r ) r@ Z net_if_statsr, r RuntimeErrorrA ZisupZmtur r findall)rE r1 Zstatsr r r r test_net_if_stats s z TestSystemAPIs.test_net_if_statsz
no batteryc C s` t d}td|d}td|d}|dk}t }| |j| | |jt | d S )Nz
pmset -g battz(\d+)%r zNow drawing from '([^']+)'zAC Power)
r r r r r@ Zsensors_batteryrA
power_pluggedpercentr )rE r rv Z drawing_fromru Z
psutil_resultr r r test_sensors_battery s z#TestSystemAPIs.test_sensors_batteryN)rH rI rJ r rZ r_ r` rc re rg ri rk rm ro rp rt r skipIfr rw r r r r rL f s*
rL __main__)
run_from_name)__doc__r rB r@ r Z psutil.testsr r r r r r r r
r r r r4 rx r5 rL rH Zpsutil.tests.runnerrz __file__r r r r <module> s0