Cs[fI � @ s9 d Z d d l 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 d d
l
m Z d d l
m Z d d l
m Z e rQe j d � Z e j � d k rHe d � Z n d Z n 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 j e d � Gd d! � d! e j � � Z e d" k r5e e! � d S)#z$Tests specific to all BSD platforms.� N)�BSD)�FREEBSD)�NETBSD)�OPENBSD)�get_test_subprocess)� HAS_BATTERY)�MEMORY_TOLERANCE)�
reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�unittest)�which� SC_PAGE_SIZE�museFc C s� t d | � } t r6 | | j d � d d � } n) t sB t r_ | | j d � d d � } y t | � SWn t k
r� | SYn Xd S)zmExpects a sysctl command with an argument and parse the result
returning only the value of interest.
zsysctl z: � N�=� )r r �findr r �int�
ValueError)�cmdline�result� r � /test_bsd.py�sysctl, s
r c C sV t d � } x3 | j d � D] } | j | � r Pq Wt d � � t | j � d � S)z+Thin wrapper around 'muse' cmdline utility.r �
zline not foundr )r �split�
startswithr r )Zfield�out�liner r r r ; s zBSD onlyc @ s� e Z d Z d Z e d d � � Z e d d � � Z e j e d � d d � � Z
d d
� Z e j e d � d � d
d � � Z
e j e d � d � d d � � Z d d � Z d S)�BSDSpecificTestCasez)Generic tests common to all BSD variants.c C s t � j | _ d S)N)r �pid)�clsr r r �
setUpClassO s z BSDSpecificTestCase.setUpClassc C s t � d S)N)r )r# r r r �
tearDownClassS s z!BSDSpecificTestCase.tearDownClassz -o lstart doesn't work on NETBSDc C sr t d | j � } | j d d � j � } t j | j � j � } t j d t j | � � } | j
| | � d S)Nzps -o lstart -p %sZSTARTED� z%a %b %e %H:%M:%S %Y) r r" �replace�strip�psutil�ProcessZ create_time�time�strftime� localtime� assertEqual)�self�outputZstart_psZ start_psutilr r r �test_process_create_timeW s z,BSDSpecificTestCase.test_process_create_timec 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 | f � t | j
| � d k r | j d | j
| f � 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 �dfc s
" z*BSDSpecificTestCase.test_disks.<locals>.df�allF�
i zpsutil=%s, df=%si ( i � i ( i � ) r) Zdisk_partitions�
disk_usageZ
mountpointr. Zdevicer7 �absr9 Zfailr8 )r/ r: �part�usager6 r7 r8 r9 r r r �
test_disks` s
z BSDSpecificTestCase.test_disksr zsysctl cmd not availablec C s, t d � } | j t j d d � | � d S)Nzhw.ncpuZlogicalT)r r. r) � cpu_count)r/ �systr r r �test_cpu_count_logical{ s z*BSDSpecificTestCase.test_cpu_count_logicalc C s) t d � } | j | t j � j � d S)Nz
hw.physmem)r r. r) �virtual_memoryr7 )r/ �numr r r �test_virtual_memory_total� s z-BSDSpecificTestCase.test_virtual_memory_totalc C s� x� t j � j � D]� \ } } y t d | � } Wn t k
rG Yq X| j | j d | k d | �d | k r | j | j t t j
d | � d � � q Wd S)Nz ifconfig %s�RUNNING�msg�mtuz mtu (\d+)r ) r) Z net_if_stats�itemsr � RuntimeErrorr. ZisuprJ r �re�findall)r/ �nameZstatsr r r r �test_net_if_stats� s
z%BSDSpecificTestCase.test_net_if_statsN)�__name__�
__module__� __qualname__�__doc__� classmethodr$ r% r
�skipIfr r1 rA r rD rG rP r r r r r! K s %%r! z FREEBSD onlyc @ s� e Z d Z e d d � � Z e d d � � Z e d d � � Z e � 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 e � d d � � Z e j e d � d d! � � Z e j e d � e � d" d# � � � Z e j e d � e � d$ d% � � � Z e j e d � e � d& d' � � � Z e j e d � e � d( d) � � � Z e j e d � e � d* d+ � � � Z e j e d � e � d, d- � � � Z d. d/ � Z d0 d1 � Z d2 d3 � Z! d4 d5 � Z" d6 d7 � Z# d8 d9 � Z$ d: d; � Z% d< d= � Z& e j e' d> � d? d@ � � Z( e j e' d> � dA dB � � Z) e j e' dC � dD dE � � Z* dF S)G�FreeBSDSpecificTestCasec C s t � j | _ d S)N)r r" )r# r r r r$ � s z"FreeBSDSpecificTestCase.setUpClassc C s t � d S)N)r )r# r r r r% � s z%FreeBSDSpecificTestCase.tearDownClassc C sq t d � j � d } t j d | � } | s> t d | � � d d � | d d � D� \ } } } | | | f S) Nz swapinfo -kr z\s+zCan't parse swapinfo: %sc s s | ] } t | � d Vq d S)i N)r )�.0�pr r r � <genexpr>� s z9FreeBSDSpecificTestCase.parse_swapinfo.<locals>.<genexpr>r2 ���)r �
splitlinesrM r r )r0 �partsr7 r8 r9 r r r �parse_swapinfo� s &z&FreeBSDSpecificTestCase.parse_swapinfoc C s� t d | j � } t j | j � j d d � } | j d � d d � } x� | r� | j � } | j � } | d d � \ } } } } }
| j � } | j d | | f | j � | j t |
� | j
� | j j d � sM | j | d | j � qM Wd S)
Nzprocstat -v %sZgroupedFr r � z%s-%s�[r< )
r r" r) r* Z memory_mapsr r3 r. �addrr Zrssr4 r ) r/ r �mapsr5 r �fields�_�start�stopZperms�res�mapr r r �test_proc_memory_maps� s z-FreeBSDSpecificTestCase.test_proc_memory_mapsc C sP t d | j � } | j t j | j � j � | j d � d j � d � d S)Nzprocstat -b %sr r r[ )r r" r. r) r* Zexer )r/ r r r r �
test_proc_exe� s z%FreeBSDSpecificTestCase.test_proc_exec C sh t d | j � } | j d j t j | j � j � � d j | j d � d j � d d � � � d S)Nzprocstat -c %s� r r r )r r" r. �joinr) r* r r )r/ r r r r �test_proc_cmdline� s $z)FreeBSDSpecificTestCase.test_proc_cmdlinec C s t d | j � } | j d � d j � d d � \ } } } } } } t j | j � } | j � } | j � }
| j | j t | � � | j | j
t | � � | j | j t | � � | j |
j t | � � | j |
j
t | � � | j |
j t | � � d S)Nzprocstat -s %sr r r � ) r r" r r) r* �uids�gidsr. �realr Z effectiveZsaved) r/ r ZeuidZruidZsuidZegidZrgidZsgidrY ro rp r r r �test_proc_uids_gids� s 5 z+FreeBSDSpecificTestCase.test_proc_uids_gidsc C s g } t d | j � } t j | j � } x� | j d � D]� } | j � j � } d | k r� t | j � d � } | j � j } | j
| | � | j d � q; d | k r; t | j � d � } | j � j } | j
| | � | j d � q; Wt
| � d k rt d � � d S)
Nzprocstat -r %sr z voluntary contextr z involuntary contextr z)couldn't find lines match in procstat outr[ r[ )r r" r) r* r �lowerr( r Znum_ctx_switchesZ voluntaryr. �appendZ involuntary�lenrL )r/ �testedr rY r � pstat_value� psutil_valuer r r �test_proc_ctx_switches� s" z.FreeBSDSpecificTestCase.test_proc_ctx_switchesc C s: g } t d | j � } t j | j � } x� | j d � D]� } | j � j � } d | k r� t d | j � d
j d � d � } | j � j } | j
| | � | j d � q; d | k r; t d | j � d j d � d
� } | j � j } | j
| | � | j d � q; Wt
| � d k r6t d � � d S)Nzprocstat -r %sr z user timez0.r �.z system timer z)couldn't find lines match in procstat outr[ r[ r[ r[ )r r" r) r* r rs r( �floatZ cpu_times�userr. rt �systemru rL )r/ rv r rY r rw rx r r r �test_proc_cpu_times� s" ' 'z+FreeBSDSpecificTestCase.test_proc_cpu_timesc C s3 t d � t } | j t j � j | d t �d S)Nzvm.stats.vm.v_active_count�delta)r �PAGESIZE�assertAlmostEqualr) rE �activer )r/ rC r r r �test_vmem_active� s z(FreeBSDSpecificTestCase.test_vmem_activec C s3 t d � t } | j t j � j | d t �d S)Nz vm.stats.vm.v_inactive_countr )r r� r� r) rE �inactiver )r/ rC r r r �test_vmem_inactive s z*FreeBSDSpecificTestCase.test_vmem_inactivec C s3 t d � t } | j t j � j | d t �d S)Nzvm.stats.vm.v_wire_countr )r r� r� r) rE �wiredr )r/ rC r r r �test_vmem_wired s z'FreeBSDSpecificTestCase.test_vmem_wiredc C s3 t d � t } | j t j � j | d t �d S)Nzvm.stats.vm.v_cache_countr )r r� r� r) rE �cachedr )r/ rC r r r �test_vmem_cached s z(FreeBSDSpecificTestCase.test_vmem_cachedc C s3 t d � t } | j t j � j | d t �d S)Nzvm.stats.vm.v_free_countr )r r� r� r) rE r9 r )r/ rC r r r �test_vmem_free s z&FreeBSDSpecificTestCase.test_vmem_freec C s/ t d � } | j t j � j | d t �d S)Nz vfs.bufspacer )r r� r) rE �buffersr )r/ rC r r r �test_vmem_buffers s z)FreeBSDSpecificTestCase.test_vmem_bufferszmuse not installedc C s) t d � } | j t j � j | � d S)NZTotal)r r. r) rE r7 )r/ rF r r r �test_muse_vmem_total% s z,FreeBSDSpecificTestCase.test_muse_vmem_totalc C s/ t d � } | j t j � j | d t �d S)NZActiver )r r� r) rE r� r )r/ rF r r r �test_muse_vmem_active* s z-FreeBSDSpecificTestCase.test_muse_vmem_activec C s/ t d � } | j t j � j | d t �d S)NZInactiver )r r� r) rE r� r )r/ rF r r r �test_muse_vmem_inactive1 s z/FreeBSDSpecificTestCase.test_muse_vmem_inactivec C s/ t d � } | j t j � j | d t �d S)NZWiredr )r r� r) rE r� r )r/ rF r r r �test_muse_vmem_wired8 s z,FreeBSDSpecificTestCase.test_muse_vmem_wiredc C s/ t d � } | j t j � j | d t �d S)NZCacher )r r� r) rE r� r )r/ rF r r r �test_muse_vmem_cached? s z-FreeBSDSpecificTestCase.test_muse_vmem_cachedc C s/ t d � } | j t j � j | d t �d S)NZFreer )r r� r) rE r9 r )r/ rF r r r �test_muse_vmem_freeF s z+FreeBSDSpecificTestCase.test_muse_vmem_freec C s/ t d � } | j t j � j | d t �d S)NZBufferr )r r� r) rE r� r )r/ rF r r r �test_muse_vmem_buffersM s z.FreeBSDSpecificTestCase.test_muse_vmem_buffersc C s) | j t j � j t d � d d �d S)Nzvm.stats.sys.v_swtchr i� )r� r) � cpu_stats� ctx_switchesr )r/ r r r �test_cpu_stats_ctx_switchesT s z3FreeBSDSpecificTestCase.test_cpu_stats_ctx_switchesc C s) | j t j � j t d � d d �d S)Nzvm.stats.sys.v_intrr i� )r� r) r� �
interruptsr )r/ r r r �test_cpu_stats_interruptsX s z1FreeBSDSpecificTestCase.test_cpu_stats_interruptsc C s) | j t j � j t d � d d �d S)Nzvm.stats.sys.v_softr i� )r� r) r� Zsoft_interruptsr )r/ r r r � test_cpu_stats_soft_interrupts\ s z6FreeBSDSpecificTestCase.test_cpu_stats_soft_interruptsc C s) | j t j � j t d � d d �d S)Nzvm.stats.sys.v_syscallr i� )r� r) r� Zsyscallsr )r/ r r r �test_cpu_stats_syscalls` s z/FreeBSDSpecificTestCase.test_cpu_stats_syscallsc C s8 | j � \ } } } | j t j � j | d t �d S)Nr )r^ r� r) � swap_memoryr9 r )r/ r7 r8 r9 r r r �test_swapmem_freej s z)FreeBSDSpecificTestCase.test_swapmem_freec C s8 | j � \ } } } | j t j � j | d t �d S)Nr )r^ r� r) r� r8 r )r/ r7 r8 r9 r r r �test_swapmem_usedo s z)FreeBSDSpecificTestCase.test_swapmem_usedc C s8 | j � \ } } } | j t j � j | d t �d S)Nr )r^ r� r) r� r7 r )r/ r7 r8 r9 r r r �test_swapmem_totalt s z*FreeBSDSpecificTestCase.test_swapmem_totalc C sh t d � } | | j d � d d � } | d | j d � � } t | � } | j | t j � � d S)Nzsysctl kern.boottimez sec = � �,)r r r r. r) � boot_time)r/ �sZbtimer r r �test_boot_time{ s
z&FreeBSDSpecificTestCase.test_boot_timez
no batteryc C s� d d � } t d � } t d d � | j d � D� � } t j � } t | d j d d � � } | d
} | j | j | � | d k r� | j | j t j
� n | j | | j � | � d S) Nc S s8 t | d � \ } } t | d � \ } } d | | f S)N�< z%d:%02d)�divmod)Zsecs�mr� �hr r r �
secs2hours� s z@FreeBSDSpecificTestCase.test_sensors_battery.<locals>.secs2hoursz
acpiconf -i 0c S s6 g | ], } | j d � d | j d � d f � q S)� r r r[ )r )rX �xr r r �
<listcomp>� s z@FreeBSDSpecificTestCase.test_sensors_battery.<locals>.<listcomp>r zRemaining capacity:�%r& zRemaining time:�unknown) r �dictr r) �sensors_batteryr r' r. �percent�secsleftZPOWER_TIME_UNLIMITED)r/ r� r rc Zmetricsr� Zremaining_timer r r �test_sensors_battery� s
z,FreeBSDSpecificTestCase.test_sensors_batteryc C s� | j t j � j t d � � | j t j � j t d � d k � t j � j } | d k rx | j t d � d � n | j | t d � d � d S)Nzhw.acpi.battery.lifezhw.acpi.acliner r zhw.acpi.battery.timer� r[ )r. r) r� r� r Z
power_pluggedr� )r/ r� r r r �#test_sensors_battery_against_sysctl� s
z;FreeBSDSpecificTestCase.test_sensors_battery_against_sysctlz has batteryc C sL | j t � �# t d � t d � t d � Wd QRX| j t j � � d S)Nzhw.acpi.battery.lifezhw.acpi.battery.timezhw.acpi.acline)Z assertRaisesrL r Z assertIsNoner) r� )r/ r r r � test_sensors_battery_no_battery� s
z7FreeBSDSpecificTestCase.test_sensors_battery_no_batteryN)+rQ rR rS rU r$ r% � staticmethodr^ r
ri rj rm rr ry r~ r� r� r� r� r� r� r
rV �MUSE_AVAILABLEr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r r r r rW � sN
rW z OPENBSD onlyc @ s e Z d Z d d � Z d S)�OpenBSDSpecificTestCasec C sM t d � } t j j | d � } t j j t j � � } | j | | � d S)Nz
kern.boottimez%a %b %d %H:%M:%S %Y)r �datetime�strptimeZ
fromtimestampr) r� r. )r/ r� Zsys_btZ psutil_btr r r r� � s z&OpenBSDSpecificTestCase.test_boot_timeN)rQ rR rS r� r r r r r� � s r� z NETBSD onlyc @ s� e Z 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 S)�NetBSDSpecificTestCasec C se t d d � �? } x5 | D]- } | j | � r t | j � d � d Sq WWd QRXt d | � � d S)Nz
/proc/meminfo�rbr i z
can't find %s)�openr r r r )Zlook_for�fr r r r �
parse_meminfo� s
#z$NetBSDSpecificTestCase.parse_meminfoc C s& | j t j � j | j d � � d S)Nz MemTotal:)r. r) rE r7 r� )r/ r r r �test_vmem_total� s z&NetBSDSpecificTestCase.test_vmem_totalc C s, | j t j � j | j d � d t �d S)NzMemFree:r )r� r) rE r9 r� r )r/ r r r r� � s z%NetBSDSpecificTestCase.test_vmem_freec C s, | j t j � j | j d � d t �d S)NzBuffers:r )r� r) rE r� r� r )r/ r r r r� � s z(NetBSDSpecificTestCase.test_vmem_buffersc C s, | j t j � j | j d � d t �d S)Nz
MemShared:r )r� r) rE Zsharedr� r )r/ r r r �test_vmem_shared� s z'NetBSDSpecificTestCase.test_vmem_sharedc C s, | j t j � j | j d � d t �d S)Nz
SwapTotal:r )r� r) r� r7 r� r )r/ r r r r� � s z)NetBSDSpecificTestCase.test_swapmem_totalc C s, | j t j � j | j d � d t �d S)Nz SwapFree:r )r� r) r� r9 r� r )r/ r r r r� � s z(NetBSDSpecificTestCase.test_swapmem_freec C s- t j � } | j | j | j | j � d S)N)r) r� r. r8 r7 r9 )r/ Zsmemr r r r� � s z(NetBSDSpecificTestCase.test_swapmem_usedc C s t d d � �J } x@ | D], } | j d � r t | j � d � } Pq Wt d � � Wd QRX| j t j � j | d d �d S)Nz
/proc/statr� s intrr zcouldn't find liner i� ) r� r r r r r� r) r� r� )r/ r� r r� r r r r� � s
z0NetBSDSpecificTestCase.test_cpu_stats_interruptsc C s t d d � �J } x@ | D], } | j d � r t | j � d � } Pq Wt d � � Wd QRX| j t j � j | d d �d S)Nz
/proc/statr� s ctxtr zcouldn't find liner i� ) r� r r r r r� r) r� r� )r/ r� r r� r r r r� � s
z2NetBSDSpecificTestCase.test_cpu_stats_ctx_switchesN)rQ rR rS r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� �__main__)"rT r� �osrM r+ r) r r r r Z psutil.testsr r r r r
r r r
r �sysconfr� �getuidr� r r rV ZTestCaser! rW r� r� rQ �__file__r r r r �<module>
sJ K � B