3
�[g� � @ s d Z ddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z dd lm Z dd
lm
Z
dd lmZ dd lmZ dd
lmZ er�ddlmZ dd� Zdd� Ze jje dd�G dd� de ��Ze jje dd�G dd� de ��ZdS )zmacOS specific tests.� N)�MACOS)�POSIX)� HAS_BATTERY)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�spawn_testproc)� terminate)� getpagesizec C s6 t | �}|j� 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)�cmdline�out�result� r �A/usr/local/lib64/python3.6/site-packages/psutil/tests/test_osx.py�sysctl s r c C sJ t d�}x$|jd�D ]}| |krP qW td��ttjd|�jd��t� S )z)Wrapper around 'vm_stat' cmdline utility.�vm_stat�
zline not foundz\d+r )r
r r r �re�search�groupr
)�fieldr �liner r r r + s r z
MACOS only)�reasonc @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )� TestProcessc C s t � j| _d S )N)r �pid)�clsr r r �
setUpClass8 s zTestProcess.setUpClassc C s t | j� d S )N)r r! )r" r r r �
tearDownClass< s zTestProcess.tearDownClassc C s� t d| j �}|jdd�j� }|jd�d }|jd�d
}tj| j�j� }|tj dtj
|��ksdt �|tj dtj
|��ks~t �d S ) Nzps -o lstart -p %sZSTARTED� � � r z%H:%M:%Sz%Y������) r
r! �replace�stripr �psutil�ProcessZ create_time�time�strftime� localtime�AssertionError)�self�outputZstart_psZhhmmss�yearZ start_psutilr r r �test_process_create_time@ s z$TestProcess.test_process_create_timeN)�__name__�
__module__� __qualname__� classmethodr# r$ r5 r r r r r 6 s r c @ s� e Zd Ze� dd� �Zdd� Zdd� Zejj e
o:e j � dkdd �d
d � �Z
d d
� Ze� dd� �Ze� dd� �Ze� dd� �Ze� dd� �Ze� dd� �Ze� dd� �Zdd� Zejj e d d �d d � �Zd S ) �TestSystemAPIsc C s� dd� }xzt jdd�D ]j}t j|j�}||j�\}}}}|j|ksFt�|j|ksTt�t|j| �t k sjt�t|j
| �t k st�qW d S )Nc S s� t d| �j� }|jd�}|jd� |jd�}|j� 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 �dfU s
z%TestSystemAPIs.test_disks.<locals>.dfF)�all) r, Zdisk_partitions�
disk_usage�
mountpoint�devicer1 r@ �absrB r rA )r2 rC �part�usager? r@ rA rB r r r �
test_disksQ s
zTestSystemAPIs.test_disksc C s t d�}|tjdd�ks t�d S )Nzsysctl hw.logicalcpuT)�logical)r r, � cpu_countr1 )r2 �numr r r �test_cpu_count_logicall s z%TestSystemAPIs.test_cpu_count_logicalc C s t d�}|tjdd�ks t�d S )Nzsysctl hw.physicalcpuF)rL )r r, rM r1 )r2 rN r r r �test_cpu_count_coresp s z#TestSystemAPIs.test_cpu_count_coresZarm64zskipped due to #1892)r c C sZ t j� }|jd d td�ks"t�|jd d td�ks<t�|jd d td�ksVt�d S )Ni� zsysctl hw.cpufrequencyzsysctl hw.cpufrequency_minzsysctl hw.cpufrequency_max)r, Zcpu_freq�currentr r1 �min�max)r2 Zfreqr r r �
test_cpu_frequ s z TestSystemAPIs.test_cpu_freqc C s t d�}|tj� jkst�d S )Nzsysctl hw.memsize)r r, �virtual_memoryr@ r1 )r2 Zsysctl_hwphymemr r r �test_vmem_total� s z TestSystemAPIs.test_vmem_totalc C s* t d�}tj� j}t|| �tk s&t�d S )NrB )r r, rU rB rH r r1 )r2 �
vmstat_val�
psutil_valr r r �test_vmem_free� s
z TestSystemAPIs.test_vmem_freec C s* t d�}tj� j}t|| �tk s&t�d S )N�active)r r, rU rZ rH r r1 )r2 rW rX r r r �test_vmem_active� s
z TestSystemAPIs.test_vmem_activec C s* t d�}tj� j}t|| �tk s&t�d S )N�inactive)r r, rU r\ rH r r1 )r2 rW rX r r r �test_vmem_inactive� s
z!TestSystemAPIs.test_vmem_inactivec C s* t d�}tj� j}t|| �tk s&t�d S )N�wired)r r, rU r^ rH r r1 )r2 rW rX r r r �test_vmem_wired� s
z TestSystemAPIs.test_vmem_wiredc C s* t d�}tj� j}t|| �tk s&t�d S )NZPageins)r r, � swap_memory�sinrH r r1 )r2 rW rX r r r �test_swapmem_sin� s
z TestSystemAPIs.test_swapmem_sinc C s* t d�}tj� j}t|| �tk s&t�d S )NZPageout)r r, r` �soutrH r r1 )r2 rW rX r r r �test_swapmem_sout� s
z TestSystemAPIs.test_swapmem_soutc C sx xrt j� j� D ]b\}}ytd| �}W n tk
r: Y qX |jd|kksRt|��|jtt j
d|�d �kst�qW d S )Nz ifconfig %sZRUNNINGz mtu (\d+)r ) r, Z net_if_stats�itemsr
� RuntimeError�isupr1 �mtur r �findall)r2 �nameZstatsr r r r �test_net_if_stats� s z TestSystemAPIs.test_net_if_statsz
no batteryc C s` t d�}tjd|�jd�}tjd|�jd�}|dk}tj� }|j|ksJt�|jt |�ks\t�d S )Nz
pmset -g battz(\d+)%r zNow drawing from '([^']+)'zAC Power)
r
r r r r, Zsensors_battery�
power_pluggedr1 �percentr )r2 r rm Z drawing_fromrl Z
psutil_resultr r r �test_sensors_battery� s z#TestSystemAPIs.test_sensors_batteryN)r6 r7 r8 r rK rO rP r �mark�skipifr �platform�machinerT rV rY r[ r] r_ rb rd rk r rn r r r r r: L s
r: )�__doc__rq r r. r, r r Z psutil.testsr r r r r r r
r r Zpsutil._psutil_posixr
r r ro rp r r: r r r r �<module> s. |