Your IP : 216.73.216.213
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)�fieldr 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 ) �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 r9 )r r: 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 r: �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> rK 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� }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��delta) rC Zdisk_partitions�
disk_usage�
mountpointrD �devicerV �assertAlmostEqualrX r rW )rH rY �part�usagerU rV rW rX r r r �
test_disksk s
�
�zTestSystemAPIs.test_disksc C s t d�}| �|tjdd�� d S )Nzsysctl hw.logicalcpuT�Zlogical�r rD rC � cpu_count�rH r2 r r r �test_cpu_count_logical� s z%TestSystemAPIs.test_cpu_count_logicalc C s t d�}| �|tjdd�� d S )Nzsysctl hw.physicalcpuFrd re rg 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)rC Zcpu_freqrD �currentr �min�max)rH Zfreqr r r �
test_cpu_freq� s � � �z TestSystemAPIs.test_cpu_freqc C s t d�}| �|t�� j� d S )Nzsysctl hw.memsize)r rD rC �virtual_memoryrV )rH Zsysctl_hwphymemr r r �test_vmem_total� s z TestSystemAPIs.test_vmem_totalc C s& t d�}t�� j}| j||td� d S )NrX r[ )r rC rn rX r` r
�rH 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�activer[ )r rC rn rr r` r
rp r r r �test_vmem_active� s
z TestSystemAPIs.test_vmem_activec C s& t d�}t�� j}| j||td� d S )N�inactiver[ )r rC rn rt r` r
rp r r r �test_vmem_inactive� s
z!TestSystemAPIs.test_vmem_inactivec C s& t d�}t�� j}| j||td� d S )N�wiredr[ )r rC rn rv r` r
rp r r r �test_vmem_wired� s
z TestSystemAPIs.test_vmem_wiredc C s" t d�}t�� j}| �||� d S )NZPageins)r rC � swap_memory�sinrD rp r r r �test_swapmem_sin� s
z TestSystemAPIs.test_swapmem_sinc C s" t d�}t�� j}| �||� d S )NZPageout)r rC rx �soutrD rp 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 %s�RUNNING)�msgz mtu (\d+)r ) rC Z net_if_statsr- r � RuntimeErrorrD �isup�mtur r �findall)rH 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 rC Zsensors_batteryrD �
power_plugged�percentr )rH r r� Z drawing_fromr� Z
psutil_resultr r r �test_sensors_battery� s z#TestSystemAPIs.test_sensors_batteryN)rL rM rN r rc rh ri rm ro rq rs ru rw rz r| r� r �skipIfr r� r r r r rP f s*
rP �__main__)�
run_from_name)�__doc__r rE rC r Z psutil.testsr r r r r r r r
r r r r7 r� r8 rP rL Zpsutil.tests.runnerr� �__file__r r r r �<module> s0