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� 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 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 �split�int�
ValueError)Zcmdline�out�result� r �H/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_osx.py�sysctl s
r c C sT ddl m} td�}|�d�D ]}| |kr q8q 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)Zfieldr r �liner r r r % s r c C s� ddi}| }d}| r$| dd� � � s4| dd� dkrN|| d 7 }| dd � } qt|�}| �� }|�� D ]\}}||krf q�qf|dkr�|d }|�� }n td| ��|d di}t|dd � �D ] \}} d|d d > || < q�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 ) �sZSYMBOLSZinit�numZletter�nameZsset�prefix�ir r r � human2bytes3 s* �$
r4 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 r6 )r r7 r8 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 r7 �replacer+ r
�psutilZProcessZ create_time� assertEqual�time�strftime� localtime)�self�outputZstart_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� Ze�e d�dd� �Zd S ) �TestSystemAPIsc C s| dd� }t jdd�D ]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 �dev�total�used�freer r r �dfo s
z%TestSystemAPIs.test_disks.<locals>.dfF)�all�Zdelta) 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.logicalcpuT�Zlogical�r rA r@ � cpu_count�rE 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 �min�max)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 )N�activerW )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 )N�inactiverW )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 )N�wiredrW )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�}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 r@ Zsensors_batteryrA �
power_plugged�percentr )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