← Back
Editing: test_osx.cpython-37.pyc
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
Save File
Cancel