Your IP : 216.73.216.69
B
�=�_~ � @ 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� Ze�e d�G dd� de��Ze�e d�G dd� de��Zedk�rddlmZ 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 s6 t | �}|�� d }yt|�S tk
r0 |S X dS )zmExpects a sysctl command with an argument and parse the result
returning only the value of interest.
� N)r �split�int�
ValueError)Zcmdline�out�result� r �H/opt/alt/python37/lib64/python3.7/site-packages/psutil/tests/test_osx.py�sysctl s r c C sV ddl m} td�}x$|�d�D ]}| |kr P q W td��tt�d|��d��|� S )z)Wrapper around 'vm_stat' cmdline utility.r )� getpagesize�vm_stat�
zline not foundz\d+) Zpsutil._psutil_posixr r r
r r �re�search�group)�fieldr r �liner r r r % s r c C s� ddi}| }d}x@| r&| dd� � � s6| dd� dkrP|| d 7 }| dd � } qW t|�}| �� }xD|�� D ]\}}||krlP qlW |dkr�|d }|�� }n td| ��|d di}x0t|dd � �D ] \}} d|d d > || < q�W t||| �S )
NZ customary) �B�K�M�G�T�P�E�Z�Y� r r �.�kzcan't interpret %r�
)�isdigit�float�strip�items�upperr � enumerater ) �sZSYMBOLS�init�num�letter�nameZsset�prefix�ir r r � human2bytes3 s& &
r7 z
MACOS onlyc @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )� TestProcessc C s t � j| _d S )N)r �pid)�clsr r r �
setUpClassP s zTestProcess.setUpClassc C s t | j� d S )N)r r9 )r: r r r �
tearDownClassT s zTestProcess.tearDownClassc C s� t d| j �}|�dd��� }|�d�d }|�d�d }t�| j��� }| �|t �
dt � |��� | �|t �
dt � |��� d S ) Nzps -o lstart -p %sZSTARTEDr& � ������z%H:%M:%Sz%Y) r r9 �replacer, r
�psutilZProcessZ create_time� assertEqual�time�strftime� localtime)�self�outputZstart_psZhhmmss�yearZ start_psutilr r r �test_process_create_timeX s z$TestProcess.test_process_create_timeN)�__name__�
__module__� __qualname__� classmethodr; r< rI r r r r r8 M s r8 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� Ze�e d�dd� �Zd S ) �TestSystemAPIsc C s� dd� }xrt jdd�D ]b}t �|j�}||j�\}}}}| �|j|� | �|j|� | j|j|t d� | j|j
|t d� qW d 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 �dev�total�used�freer r r �dfo s
z%TestSystemAPIs.test_disks.<locals>.dfF)�all)�delta) rA Zdisk_partitions�
disk_usage�
mountpointrB �devicerT �assertAlmostEqualrV r rU )rF rW �part�usagerS rT rU rV r r r �
test_disksk s
zTestSystemAPIs.test_disksc C s t d�}| �|tjdd�� d S )Nzsysctl hw.logicalcpuT)�logical)r rB rA � cpu_count)rF r2 r r r �test_cpu_count_logical� s z%TestSystemAPIs.test_cpu_count_logicalc C s t d�}| �|tjdd�� d S )Nzsysctl hw.physicalcpuF)ra )r rB rA rb )rF r2 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)rA Zcpu_freqrB �currentr �min�max)rF Zfreqr r r �
test_cpu_freq� s z TestSystemAPIs.test_cpu_freqc C s t d�}| �|t�� j� d S )Nzsysctl hw.memsize)r rB rA �virtual_memoryrT )rF Zsysctl_hwphymemr r r �test_vmem_total� s z TestSystemAPIs.test_vmem_totalc C s& t d�}t�� j}| j||td� d S )NrV )rY )r rA ri rV r] r
)rF �
vmstat_val�
psutil_valr r r �test_vmem_free� s
z TestSystemAPIs.test_vmem_freec C s& t d�}t�� j}| j||td� d S )N�active)rY )r rA ri rn r] r
)rF rk rl r r r �test_vmem_active� s
z TestSystemAPIs.test_vmem_activec C s& t d�}t�� j}| j||td� d S )N�inactive)rY )r rA ri rp r] r
)rF rk rl r r r �test_vmem_inactive� s
z!TestSystemAPIs.test_vmem_inactivec C s& t d�}t�� j}| j||td� d S )N�wired)rY )r rA ri rr r] r
)rF rk rl r r r �test_vmem_wired� s
z TestSystemAPIs.test_vmem_wiredc C s" t d�}t�� j}| �||� d S )NZPageins)r rA � swap_memory�sinrB )rF rk rl r r r �test_swapmem_sin� s
z TestSystemAPIs.test_swapmem_sinc C s" t d�}t�� j}| �||� d S )NZPageout)r rA rt �soutrB )rF rk rl r r r �test_swapmem_sout� s
z TestSystemAPIs.test_swapmem_soutc C sx xrt �� �� D ]b\}}ytd| �}W n tk
r: Y qX | j|jd|k|d� | �|jtt �
d|�d �� qW d S )Nz ifconfig %s�RUNNING)�msgz mtu (\d+)r ) rA Z net_if_statsr- r � RuntimeErrorrB �isup�mtur r �findall)rF r4 Zstatsr r r r �test_net_if_stats� s z TestSystemAPIs.test_net_if_statsz
no batteryc C s` t d�}t�d|��d�}t�d|��d�}|dk}t�� }| �|j|� | �|jt |�� d S )Nz
pmset -g battz(\d+)%r zNow drawing from '([^']+)'zAC Power)
r r r r rA Zsensors_batteryrB �
power_plugged�percentr )rF r r� Z drawing_fromr� Z
psutil_resultr r r �test_sensors_battery� s z#TestSystemAPIs.test_sensors_batteryN)rJ rK rL r r` rc rd rh rj rm ro rq rs rv rx r r �skipIfr r� r r r r rN f s
rN �__main__)�
run_from_name)�__doc__r rC rA r Z psutil.testsr r r r r r r r
r r r r7 r� r8 rN rJ Zpsutil.tests.runnerr� �__file__r r r r �<module> s0