Cs[�% � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m
Z
d d l m Z d d l m Z d d
l m
Z
d d l m Z d d l m Z e r� e j d
� n d Z d d � Z d d � Z d d � Z e j e d � Gd d � d e j � � Z e j e d � Gd d � d e j � � Z e j e d � Gd d � d e j � � Z e d k r�e
e � d S) zMACOS specific tests.� N)�MACOS)�create_zombie_proc)�get_test_subprocess)� HAS_BATTERY)�MEMORY_TOLERANCE)�
reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�unittest� SC_PAGE_SIZEc C sG t | � } | j � d } y t | � SWn t k
rB | SYn Xd S)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 � /test_osx.py�sysctl s
r c C sb t d � } x0 | j d � D] } | | k r Pq Wt d � � t t j d | � j d � � t S)z)Wrapper around 'vm_stat' cmdline utility.�vm_stat�
zline not foundz\d+r )r
r r r �re�search�group�PAGESIZE)Zfieldr �liner r r r ) s r c
C sE d d i } | } d } xT | r7 | d d
� j � sM | d d
� d k rn | | d 7} | d
d � } q Wt | � } | j � } x\ | j � D] \ } } | | k r� Pq� W| d k r� | d } | j � } n t d | � � | d d
i } x: t | d
d � � D]" \ } } d
| d
d >| | <q
Wt | | | � S)NZ customary�B�K�M�G�T�P�E�Z�Y� r r
�.�kzcan't interpret %r�
) r r r r! r" r# r$ r% r& )�isdigit�float�strip�items�upperr � enumerater ) �sZSYMBOLSZinit�numZletter�nameZsset�prefix�ir r r � human2bytes5 s& 5
#r6 z
MACOS onlyc @ s@ e Z d Z e d d � � Z e d d � � Z d d � Z d S)� TestProcessc C s t � j | _ d S)N)r �pid)�clsr r r �
setUpClassR s zTestProcess.setUpClassc C s t � d S)N)r )r9 r r r �
tearDownClassV s zTestProcess.tearDownClassc C s� t d | j � } | j d d � j � } | j d � d } | j d � d
} t j | j � j � } | j | t j
d t j | � � � | j | t j
d t j | � � � d S) Nzps -o lstart -p %sZSTARTEDr' � � r
z%H:%M:%Sz%Y������) r
r8 �replacer- r �psutil�Process� create_time� assertEqual�time�strftime� localtime)�self�outputZstart_psZhhmmssZyearZ start_psutilr r r �test_process_create_timeZ s z$TestProcess.test_process_create_timeN)�__name__�
__module__� __qualname__� classmethodr: r; rJ r r r r r7 O s r7 c @ s� e Z d Z e d d � � Z e d d � � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! S)"�TestZombieProcessAPIsc C s t � } t j | � | _ d S)N)r rA rB �p)r9 Zzpidr r r r: k s z TestZombieProcessAPIs.setUpClassc C s t d d � d S)NZ recursiveT)r )r9 r r r r; p s z#TestZombieProcessAPIs.tearDownClassc C sa | j | j j � t j � | j j � | j j � | j j � | j j � | j j � d S)N)
rD rP �statusrA Z
STATUS_ZOMBIEZppidZuidsZgidsZterminalrC )rH r r r �test_pidtask_infot s
z'TestZombieProcessAPIs.test_pidtask_infoc C s | j t j | j j � d S)N)� assertRaisesrA �
ZombieProcessrP Zexe)rH r r r �test_exe| s z TestZombieProcessAPIs.test_exec C s | j t j | j j � d S)N)rS rA rT rP r )rH r r r � test_cmdline s z"TestZombieProcessAPIs.test_cmdlinec C s | j t j | j j � d S)N)rS rA rT rP �environ)rH r r r � test_environ� s z"TestZombieProcessAPIs.test_environc C s | j t j | j j � d S)N)rS rA rT rP �cwd)rH r r r �test_cwd� s z TestZombieProcessAPIs.test_cwdc C s | j t j | j j � d S)N)rS rA rT rP Zmemory_full_info)rH r r r �test_memory_full_info� s z+TestZombieProcessAPIs.test_memory_full_infoc C s | j t j | j j � d S)N)rS rA rT rP Z cpu_times)rH r r r �test_cpu_times� s z$TestZombieProcessAPIs.test_cpu_timesc C s | j t j | j j � d S)N)rS rA rT rP Znum_ctx_switches)rH r r r �test_num_ctx_switches� s z+TestZombieProcessAPIs.test_num_ctx_switchesc C s | j t j | j j � d S)N)rS rA rT rP Z num_threads)rH r r r �test_num_threads� s z&TestZombieProcessAPIs.test_num_threadsc C s | j t j | j j � d S)N)rS rA rT rP Z
open_files)rH r r r �test_open_files� s z%TestZombieProcessAPIs.test_open_filesc C s | j t j | j j � d S)N)rS rA rT rP Z connections)rH r r r �test_connections� s z&TestZombieProcessAPIs.test_connectionsc C s | j t j | j j � d S)N)rS rA rT rP Znum_fds)rH r r r � test_num_fds� s z"TestZombieProcessAPIs.test_num_fdsc C s& | j t j t j f | j j � d S)N)rS rA rT Z AccessDeniedrP �threads)rH r r r � test_threads� s z"TestZombieProcessAPIs.test_threadsc C s | j t j | j j � d S)N)rS rA rT rP Z memory_maps)rH r r r �test_memory_maps� s z&TestZombieProcessAPIs.test_memory_mapsN)rK rL rM rN r: r; rR rU rV rX rZ r[ r\ r] r^ r_ r` ra rc rd r r r r rO h s rO c @ s
e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e � d d � � Z e � d
d � � Z
e � d d � � Z e � d d � � Z e � d d � � Z
e � d d � � Z e � d d � � Z d d � Z e j e d � d d � � Z d S) �TestSystemAPIsc C s� d d � } x� t j d d � D]� } t j | j � } | | j � \ } } } } | j | j | � | j | j | � t | j | � d k r� | j d | j | � t | j
| � d k r | j d | j
| � q Wd S) Nc S s� t d | � j � } | j d � } | j d � | j d � } | j � d d � \ } } } } | d k ru d } t | � d } t | � d } t | � d } | | | | f S)Nz
df -k "%s"r r � Znoner' i )r
r- r �popr )�pathr �linesr �dev�total�used�freer r r �df� s
" z%TestSystemAPIs.test_disks.<locals>.df�allFr* i zpsutil=%s, df=%si ( i � i ( i � ) rA Zdisk_partitions�
disk_usageZ
mountpointrD Zdevicerk �absrm Zfailrl )rH rn �part�usagerj rk rl rm r r r �
test_disks� s
zTestSystemAPIs.test_disksc C s, t d � } | j | t j d d � � d S)Nzsysctl hw.logicalcpu�logicalT)r rD rA � cpu_count)rH r2 r r r �test_cpu_count_logical� s z%TestSystemAPIs.test_cpu_count_logicalc C s, t d � } | j | t j d d � � d S)Nzsysctl hw.physicalcpuru F)r rD rA rv )rH r2 r r r �test_cpu_count_physical� s z&TestSystemAPIs.test_cpu_count_physicalc C ss t j � } | j | j d d t d � � | j | j d d t d � � | j | j d d t d � � d S)Ni� zsysctl hw.cpufrequencyzsysctl hw.cpufrequency_minzsysctl hw.cpufrequency_max)rA Zcpu_freqrD �currentr �min�max)rH Zfreqr r r �
test_cpu_freq� s z TestSystemAPIs.test_cpu_freqc C s) t d � } | j | t j � j � d S)Nzsysctl hw.memsize)r rD rA �virtual_memoryrk )rH Zsysctl_hwphymemr r r �test_vmem_total� s z TestSystemAPIs.test_vmem_totalc C s5 t d � } t j � j } | j | | d t �d S)Nrm �delta)r rA r} rm �assertAlmostEqualr )rH �
vmstat_val�
psutil_valr r r �test_vmem_free� s z TestSystemAPIs.test_vmem_freec C s? t d � t d � } t j � j } | j | | d t �d S)N�inactiverm r )r rA r} Z availabler� r )rH r� r� r r r �test_vmem_available� s z"TestSystemAPIs.test_vmem_availablec C s5 t d � } t j � j } | j | | d t �d S)N�activer )r rA r} r� r� r )rH r� r� r r r �test_vmem_active� s z TestSystemAPIs.test_vmem_activec C s5 t d � } t j � j } | j | | d t �d S)Nr� r )r rA r} r� r� r )rH r� r� r r r �test_vmem_inactive� s z!TestSystemAPIs.test_vmem_inactivec C s5 t d � } t j � j } | j | | d t �d S)N�wiredr )r rA r} r� r� r )rH r� r� r r r �test_vmem_wired� s z TestSystemAPIs.test_vmem_wiredc C s/ t d � } t j � j } | j | | � d S)NZPageins)r rA � swap_memory�sinrD )rH r� r� r r r �test_swapmem_sin� s z TestSystemAPIs.test_swapmem_sinc C s/ t d � } t j � j } | j | | � d S)NZPageout)r rA r� ZsoutrD )rH r� r� r r r �test_swapmem_sout s z TestSystemAPIs.test_swapmem_soutc C s� x� t j � j � D]} \ } } y t d | � } Wn t k
rG Yq X| j | j d | k d | �| j | j t t j
d | � d � � q Wd S)Nz ifconfig %s�RUNNING�msgz mtu (\d+)r ) rA Z net_if_statsr. r
� RuntimeErrorrD ZisupZmtur r �findall)rH r3 Zstatsr r r r �test_net_if_stats s
z TestSystemAPIs.test_net_if_statsz
no batteryc C s� t d � } t j d | � j d � } t j d | � j d � } | d k } t j � } | j | j | � | j | j t | � � d S)Nz
pmset -g battz(\d+)%r
zNow drawing from '([^']+)'zAC Power)
r
r r r rA 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)rK rL rM rt rw rx r| r~ r r� r� r� r� r� r� r� r� r �skipIfr r� r r r r re � s
re �__main__) �__doc__�osr rE rA r Z psutil.testsr r r r r r r r
r �sysconfr r r r6 r� ZTestCaser7 rO re rK �__file__r r r r �<module> s4 < �