B �=�_~�@s dZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lmZdd lmZddlm Z ddlmZd d�Zdd�Zdd�Ze�ed�Gdd�de��Ze�ed�Gdd�de��Zedk�rddlmZee�dS)zmacOS specific tests.�N)�MACOS)�HAS_BATTERY)�PsutilTestCase)�retry_on_failure)�sh)�spawn_testproc)� terminate)�TOLERANCE_DISK_USAGE)�TOLERANCE_SYS_MEM)�unittestcCs6t|�}|��d}yt|�Stk r0|SXdS)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�sysctlsrcCsVddlm}td�}x$|�d�D]}||kr Pq Wtd��tt�d|��d��|�S)z)Wrapper around 'vm_stat' cmdline utility.r)�getpagesize�vm_stat� zline not foundz\d+) Zpsutil._psutil_posixrrr rr�re�search�group)�fieldrr�linerrrr%src Cs�ddi}|}d}x@|r&|dd��s6|dd�dkrP||d7}|dd�}qWt|�}|��}xD|��D]\}}||krlPqlW|dkr�|d}|��}ntd|��|ddi}x0t|dd��D]\}}d|dd >||<q�Wt|||�S) NZ customary) �B�K�M�G�T�P�E�Z�Y�rr�.�kzcan't interpret %r� )�isdigit�float�strip�items�upperr� enumerater) �sZSYMBOLS�init�num�letter�nameZsset�prefix�irrr�human2bytes3s&& r7z MACOS onlyc@s,eZdZedd��Zedd��Zdd�ZdS)�TestProcesscCst�j|_dS)N)r�pid)�clsrrr� setUpClassPszTestProcess.setUpClasscCst|j�dS)N)rr9)r:rrr� tearDownClassTszTestProcess.tearDownClassc Cs�td|j�}|�dd��}|�d�d}|�d�d}t�|j��}|�|t � dt �|��|�|t � dt �|��dS) Nzps -o lstart -p %sZSTARTEDr&� ��z%H:%M:%Sz%Y)rr9�replacer,r �psutilZProcessZcreate_time�assertEqual�time�strftime� localtime)�self�outputZstart_psZhhmmss�yearZstart_psutilrrr�test_process_create_timeXsz$TestProcess.test_process_create_timeN)�__name__� __module__�__qualname__�classmethodr;r<rIrrrrr8Msr8c@s�eZdZe�dd��Zdd�Zdd�Zdd�Zd d �Ze�dd��Z e�d d��Z e�dd��Ze�dd��Ze�dd��Z e�dd��Zdd�Ze�ed�dd��ZdS)�TestSystemAPIscCs�dd�}xrtjdd�D]b}t�|j�}||j�\}}}}|�|j|�|�|j|�|j|j|t d�|j|j |t d�qWdS)NcSs�td|��}|�d�}|�d�|�d�}|��dd�\}}}}|dkrRd}t|�d}t|�d}t|�d}||||fS)Nz df -k "%s"rr�Znoner&i)rr,r �popr)�pathr�linesr�dev�total�used�freerrr�dfos z%TestSystemAPIs.test_disks..dfF)�all)�delta)rAZdisk_partitions� disk_usage� mountpointrB�devicerT�assertAlmostEqualrVr rU)rFrW�part�usagerSrTrUrVrrr� test_disksks zTestSystemAPIs.test_diskscCs td�}|�|tjdd��dS)Nzsysctl hw.logicalcpuT)�logical)rrBrA� cpu_count)rFr2rrr�test_cpu_count_logical�sz%TestSystemAPIs.test_cpu_count_logicalcCs td�}|�|tjdd��dS)Nzsysctl hw.physicalcpuF)ra)rrBrArb)rFr2rrr�test_cpu_count_physical�sz&TestSystemAPIs.test_cpu_count_physicalcCsZt��}|�|jddtd��|�|jddtd��|�|jddtd��dS)Ni�zsysctl hw.cpufrequencyzsysctl hw.cpufrequency_minzsysctl hw.cpufrequency_max)rAZcpu_freqrB�currentr�min�max)rFZfreqrrr� test_cpu_freq�szTestSystemAPIs.test_cpu_freqcCstd�}|�|t��j�dS)Nzsysctl hw.memsize)rrBrA�virtual_memoryrT)rFZsysctl_hwphymemrrr�test_vmem_total�szTestSystemAPIs.test_vmem_totalcCs&td�}t��j}|j||td�dS)NrV)rY)rrArirVr]r )rF� vmstat_val� psutil_valrrr�test_vmem_free�s zTestSystemAPIs.test_vmem_freecCs&td�}t��j}|j||td�dS)N�active)rY)rrArirnr]r )rFrkrlrrr�test_vmem_active�s zTestSystemAPIs.test_vmem_activecCs&td�}t��j}|j||td�dS)N�inactive)rY)rrArirpr]r )rFrkrlrrr�test_vmem_inactive�s z!TestSystemAPIs.test_vmem_inactivecCs&td�}t��j}|j||td�dS)N�wired)rY)rrArirrr]r )rFrkrlrrr�test_vmem_wired�s zTestSystemAPIs.test_vmem_wiredcCs"td�}t��j}|�||�dS)NZPageins)rrA�swap_memory�sinrB)rFrkrlrrr�test_swapmem_sin�s zTestSystemAPIs.test_swapmem_sincCs"td�}t��j}|�||�dS)NZPageout)rrArt�soutrB)rFrkrlrrr�test_swapmem_sout�s z TestSystemAPIs.test_swapmem_soutc Csxxrt��D]b\}}ytd|�}Wntk r:YqX|j|jd|k|d�|�|jtt � d|�d��qWdS)Nzifconfig %s�RUNNING)�msgz mtu (\d+)r)rAZnet_if_statsr-r�RuntimeErrorrB�isup�mturr�findall)rFr4Zstatsrrrr�test_net_if_stats�sz TestSystemAPIs.test_net_if_statsz no batterycCs`td�}t�d|��d�}t�d|��d�}|dk}t��}|�|j|�|�|jt |��dS)Nz pmset -g battz(\d+)%rzNow drawing from '([^']+)'zAC Power) rrrrrAZsensors_batteryrB� power_plugged�percentr)rFrr�Zdrawing_fromr�Z psutil_resultrrr�test_sensors_battery�sz#TestSystemAPIs.test_sensors_batteryN)rJrKrLrr`rcrdrhrjrmrorqrsrvrxrr�skipIfrr�rrrrrNfs rN�__main__)� run_from_name)�__doc__rrCrArZpsutil.testsrrrrrrr r rrrr7r�r8rNrJZpsutil.tests.runnerr��__file__rrrr�s0