3
�[g�N � @ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl
m Z dd l
m
Z
dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ er�ddlmZ e� Zej� dko�ed�ZndZdZdd� Zdd� Zejj e dd�G dd� de
��Z ejj e dd�G dd � d e
��Z ejj e dd�G d d � d e
��Z ejj e d d�G d d!� d!e
��Z ejj e d"d�G d#d$� d$e
��Z!dS )%z$Tests specific to all BSD platforms.� N)�BSD)�FREEBSD)�NETBSD)�OPENBSD)� HAS_BATTERY)�TOLERANCE_SYS_MEM)�PsutilTestCase)�pytest)�retry_on_failure)�sh)�spawn_testproc)� terminate)�which)� getpagesize�museFc C sh t d| �}tr(||jd�d d� }n ts0trF||jd�d d� }yt|�S tk
rb |S X dS )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 �A/usr/local/lib64/python3.6/site-packages/psutil/tests/test_bsd.py�sysctl- s r c C s@ t d�}x&|jd�D ]}|j| �rP qW td��t|j� d �S )z+Thin wrapper around 'muse' cmdline utility.r �
zline not foundr )r �split�
startswithr r )�field�out�liner r r r < s
zBSD only)�reasonc @ s� e Zd ZdZedd� �Zedd� �Zejj e
dd�dd � �Z d
d � Z ejj e
d � d
d�dd� �Zejj e
d � d
d�ejj e
dd�dd� ��Zejj e
d� dd�dd� �ZdS )� BSDTestCasez)Generic tests common to all BSD variants.c C s t � j| _d S )N)r �pid)�clsr r r �
setUpClassP s zBSDTestCase.setUpClassc C s t | j� d S )N)r
r$ )r% r r r �
tearDownClassT s zBSDTestCase.tearDownClassz -o lstart doesn't work on NETBSD)r" c C sP t d| j �}|jdd�j� }tj| j�j� }tjdtj |��}||ksLt
�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�AssertionError)�self�outputZstart_psZ start_psutilr r r �test_process_create_timeX s z$BSDTestCase.test_process_create_timec C s� dd� }x�t jdd�D ]�}t j|j�}||j�\}}}}|j|ksFt�|j|ksTt�t|j| �d krz| j d|j|f ��t|j
| �d kr| j d|j
|f ��qW d S ) Nc S s� t d| �j� }|jd�}|jd� |jd�}|j� 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 �dfe s
z"BSDTestCase.test_disks.<locals>.dfF)�all�
i zpsutil=%s, df=%si ( i � i ( i � ) r+ Zdisk_partitions�
disk_usage�
mountpoint�devicer0 r9 �absr; Zfailr: )r1 r< �part�usager8 r9 r: r; r r r �
test_disksb s
zBSDTestCase.test_disksr zsysctl cmd not availablec C s t d�}tjdd�|ks t�d S )Nzhw.ncpuT)Zlogical)r r+ � cpu_countr0 )r1 �systr r r �test_cpu_count_logical} s z"BSDTestCase.test_cpu_count_logicalzskipped on NETBSDc C s t d�}|tj� jkst�d S )Nz
hw.physmem)r r+ �virtual_memoryr9 r0 )r1 �numr r r �test_virtual_memory_total� s z%BSDTestCase.test_virtual_memory_total�ifconfigzifconfig cmd not availablec C s| xvt j� j� D ]f\}}ytd| �}W n tk
r: Y qX |jd|kksNt�d|kr|jtt j
d|�d �kst�qW d S )Nz ifconfig %sZRUNNING�mtuz mtu (\d+)r ) r+ Z net_if_stats�itemsr � RuntimeError�isupr0 rM r �re�findall)r1 �nameZstatsr r r r �test_net_if_stats� s z BSDTestCase.test_net_if_statsN)�__name__�
__module__� __qualname__�__doc__� classmethodr& r' r �mark�skipifr r3 rE r rH rK rT r r r r r# L s
r# z FREEBSD onlyc @ sf e Zd Zedd� �Zedd� �Ze� dd� �Zdd� Zd d
� Z d d � Z
e� d
d� �Z e� dd� �Z dS )�FreeBSDPsutilTestCasec C s t � j| _d S )N)r r$ )r% r r r r& � s z FreeBSDPsutilTestCase.setUpClassc C s t | j� d S )N)r
r$ )r% r r r r' � s z#FreeBSDPsutilTestCase.tearDownClassc C s� t d| j �}tj| j�jdd�}|jd�dd � }x||r�|j� }|j� }|d d� \}}}} }
|j� } d||f | jks~t�t |
�| j
ks�t�| j j d�s6|d | j ks6t�q6W d S )
Nzprocstat -v %sF)Zgroupedr r � z%s-%s�[r> )
r r$ r+ r, Z memory_mapsr r5 �addrr0 r Zrssr6 r ) r1 r �mapsr7 r! �fields�_�start�stopZ_perms�res�mapr r r �test_memory_maps� s z&FreeBSDPsutilTestCase.test_memory_mapsc C s< t d| j �}tj| j�j� |jd�d j� d ks8t�d S )Nzprocstat -b %sr r ���)r r$ r+ r, Zexer r0 )r1 r r r r �test_exe� s z FreeBSDPsutilTestCase.test_exec C sL t d| j �}djtj| j�j� �dj|jd�d j� dd � �ksHt�d S )Nzprocstat -c %s� r r r )r r$ �joinr+ r, r r r0 )r1 r r r r � test_cmdline� s z"FreeBSDPsutilTestCase.test_cmdlinec C s� t d| j �}|jd�d j� dd� \}}}}}}tj| j�}|j� } |j� }
| jt|�ksbt �| j
t|�kstt �| j t|�ks�t �|
jt|�ks�t �|
j
t|�ks�t �|
j t|�ks�t �d S )Nzprocstat -s %sr r r � ) r r$ r r+ r, �uids�gids�realr r0 � effective�saved) r1 r ZeuidZruidZsuidZegidZrgidZsgid�prn ro r r r �test_uids_gids� s & z$FreeBSDPsutilTestCase.test_uids_gidsc C s� g }t d| j �}tj| j�}x�|jd�D ]�}|j� j� }d|krtt|j� d �}|j� j }||ksht
�|j d � q*d|kr*t|j� d �}|j� j }||ks�t
�|j d � q*W t
|�dkr�td��d S )
Nzprocstat -r %sr z voluntary contextr z involuntary contextr z)couldn't find lines match in procstat outrh rh )r r$ r+ r, r �lowerr* r Znum_ctx_switches� voluntaryr0 �append� involuntary�lenrO )r1 �testedr rs r! � pstat_value� psutil_valuer r r �test_ctx_switches� s"
z'FreeBSDPsutilTestCase.test_ctx_switchesc C s� g }t d| j �}tj| j�}x�|jd�D ]�}|j� j� }d|kr�td|j� d
jd�d �}|j� j }||ksvt
�|j d � q*d|kr*td|j� d jd�d
�}|j� j }||ks�t
�|j d � q*W t
|�dkr�td ��d S )Nzprocstat -r %sr z user timez0.r �.z system timer z)couldn't find lines match in procstat outrh rh rh rh )r r$ r+ r, r ru r* �floatZ cpu_times�userr0 rw �systemry rO )r1 rz r rs r! r{ r| r r r �test_cpu_times� s"
z$FreeBSDPsutilTestCase.test_cpu_timesN)
rU rV rW rY r& r' r
rg ri rl rt r} r� r r r r r\ � s
r\ c @ s� e Zd Zedd� �Zdd� Ze� dd� �Ze� dd� �Ze� d d
� �Z e� d d � �Z
e� d
d� �Z e� dd� �Z e
jje dd�dd� �Ze
jje dd�e� dd� ��Ze
jje dd�e� dd� ��Ze
jje dd�e� dd� ��Ze
jje dd�e� dd � ��Ze
jje dd�e� d d � ��Ze
jje dd�e� d d � ��Zd!d"� Zd#d$� Zd%d&� Ze� d'd(� �Zd)d*� Z d+d,� Z d-d.� Z d/d0� Z e
jje d1d�d2d3� �Z!e
jje d1d�d4d5� �Z"e
jje d6d�d7d8� �Z#d9d:� Z$d;S )<�FreeBSDSystemTestCasec C sR t d�j� d } tjd| �}|s,td| ��dd� |dd� D �\}}}|||fS ) Nz swapinfo -kr z\s+zCan't parse swapinfo: %sc s s | ]}t |�d V qdS )i N)r )�.0rs r r r � <genexpr> s z7FreeBSDSystemTestCase.parse_swapinfo.<locals>.<genexpr>r4 rh )r �
splitlinesrQ r r )r2 �partsr9 r: r; r r r �parse_swapinfo� s z$FreeBSDSystemTestCase.parse_swapinfoc C s� d}yt t|��}W n tk
r2 tjd��Y nX tj� j|ksFt�d}t|�}t |j � d j d�d �}t |j � d j d�d �}tj� j
|ks�t�tj� j |ks�t�d S )Nzdev.cpu.0.freqz#frequencies not supported by kernelzdev.cpu.0.freq_levelsr �/r rh ) r r rO r �skipr+ Zcpu_freq�currentr0 r �max�min)r1 �sensor�
sysctl_resultZmax_freqZmin_freqr r r �!test_cpu_frequency_against_sysctl s z7FreeBSDSystemTestCase.test_cpu_frequency_against_sysctlc C s* t d�t }ttj� j| �tk s&t�d S )Nzvm.stats.vm.v_active_count)r �PAGESIZErB r+ rI �activer r0 )r1 rG r r r �test_vmem_active s z&FreeBSDSystemTestCase.test_vmem_activec C s* t d�t }ttj� j| �tk s&t�d S )Nz vm.stats.vm.v_inactive_count)r r� rB r+ rI �inactiver r0 )r1 rG r r r �test_vmem_inactive! s z(FreeBSDSystemTestCase.test_vmem_inactivec C s* t d�t }ttj� j| �tk s&t�d S )Nzvm.stats.vm.v_wire_count)r r� rB r+ rI �wiredr r0 )r1 rG r r r �test_vmem_wired& s z%FreeBSDSystemTestCase.test_vmem_wiredc C s* t d�t }ttj� j| �tk s&t�d S )Nzvm.stats.vm.v_cache_count)r r� rB r+ rI �cachedr r0 )r1 rG r r r �test_vmem_cached+ s z&FreeBSDSystemTestCase.test_vmem_cachedc C s* t d�t }ttj� j| �tk s&t�d S )Nzvm.stats.vm.v_free_count)r r� rB r+ rI r; r r0 )r1 rG r r r �test_vmem_free0 s z$FreeBSDSystemTestCase.test_vmem_freec C s& t d�}ttj� j| �tk s"t�d S )Nz vfs.bufspace)r rB r+ rI �buffersr r0 )r1 rG r r r �test_vmem_buffers5 s z'FreeBSDSystemTestCase.test_vmem_bufferszmuse not installed)r" c C s t d�}tj� j|kst�d S )NZTotal)r r+ rI r9 r0 )r1 rJ r r r �test_muse_vmem_total< s z*FreeBSDSystemTestCase.test_muse_vmem_totalc C s& t d�}ttj� j| �tk s"t�d S )NZActive)r rB r+ rI r� r r0 )r1 rJ r r r �test_muse_vmem_activeA s z+FreeBSDSystemTestCase.test_muse_vmem_activec C s& t d�}ttj� j| �tk s"t�d S )NZInactive)r rB r+ rI r� r r0 )r1 rJ r r r �test_muse_vmem_inactiveG s z-FreeBSDSystemTestCase.test_muse_vmem_inactivec C s& t d�}ttj� j| �tk s"t�d S )NZWired)r rB r+ rI r� r r0 )r1 rJ r r r �test_muse_vmem_wiredM s z*FreeBSDSystemTestCase.test_muse_vmem_wiredc C s& t d�}ttj� j| �tk s"t�d S )N�Cache)r rB r+ rI r� r r0 )r1 rJ r r r �test_muse_vmem_cachedS s z+FreeBSDSystemTestCase.test_muse_vmem_cachedc C s& t d�}ttj� j| �tk s"t�d S )NZFree)r rB r+ rI r; r r0 )r1 rJ r r r �test_muse_vmem_freeY s z)FreeBSDSystemTestCase.test_muse_vmem_freec C s& t d�}ttj� j| �tk s"t�d S )NZBuffer)r rB r+ rI r� r r0 )r1 rJ r r r �test_muse_vmem_buffers_ s z,FreeBSDSystemTestCase.test_muse_vmem_buffersc C s" t tj� jtd� �dk s t�d S )Nzvm.stats.sys.v_swtchi� )rB r+ � cpu_stats� ctx_switchesr r0 )r1 r r r �test_cpu_stats_ctx_switchese s
z1FreeBSDSystemTestCase.test_cpu_stats_ctx_switchesc C s" t tj� jtd� �dk s t�d S )Nzvm.stats.sys.v_intri� )rB r+ r� �
interruptsr r0 )r1 r r r �test_cpu_stats_interruptsn s z/FreeBSDSystemTestCase.test_cpu_stats_interruptsc C s" t tj� jtd� �dk s t�d S )Nzvm.stats.sys.v_softi� )rB r+ r� �soft_interruptsr r0 )r1 r r r � test_cpu_stats_soft_interruptst s
z4FreeBSDSystemTestCase.test_cpu_stats_soft_interruptsc C s" t tj� jtd� �dk s t�d S )Nzvm.stats.sys.v_syscalli@
)rB r+ r� �syscallsr r0 )r1 r r r �test_cpu_stats_syscalls} s z-FreeBSDSystemTestCase.test_cpu_stats_syscallsc C s, | j � \}}}ttj� j| �tk s(t�d S )N)r� rB r+ � swap_memoryr; r r0 )r1 �_total�_usedr; r r r �test_swapmem_free� s z'FreeBSDSystemTestCase.test_swapmem_freec C s, | j � \}}}ttj� j| �tk s(t�d S )N)r� rB r+ r� r: r r0 )r1 r� r: �_freer r r �test_swapmem_used� s z'FreeBSDSystemTestCase.test_swapmem_usedc C s, | j � \}}}ttj� j| �tk s(t�d S )N)r� rB r+ r� r9 r r0 )r1 r9 r� r� r r r �test_swapmem_total� s z(FreeBSDSystemTestCase.test_swapmem_totalc C sL t d�}||jd�d d � }|d |jd�� }t|�}|tj� ksHt�d S )Nzsysctl kern.boottimez sec = � �,)r r r r+ � boot_timer0 )r1 �sZbtimer r r �test_boot_time� s
z$FreeBSDSystemTestCase.test_boot_timez
no batteryc C s� dd� }t d�}tdd� |jd�D ��}tj� }t|d jdd ��}|d
}|j|ksZt�|d krt|j tj
ks�t�n||j �|ks�t�d S ) Nc S s( t | d�\}}t |d�\}}d||f S )N�< z%d:%02d)�divmod)Zsecs�mZ_s�hr r r �
secs2hours� s z>FreeBSDSystemTestCase.test_sensors_battery.<locals>.secs2hoursz
acpiconf -i 0c S s( g | ] }|j d �d |j d �d f�qS )� r r rh )r )r� �xr r r �
<listcomp>� s z>FreeBSDSystemTestCase.test_sensors_battery.<locals>.<listcomp>r zRemaining capacity:�%r( zRemaining time:�unknown) r �dictr r+ �sensors_batteryr r) �percentr0 �secsleft�POWER_TIME_UNLIMITED)r1 r� r ra �metricsr� Zremaining_timer r r �test_sensors_battery� s z*FreeBSDSystemTestCase.test_sensors_batteryc C sl t j� jtd�kst�t j� jtd�dkks0t�t j� j}|dk rTtd�dksht�n|td�d ksht�d S )Nzhw.acpi.battery.lifezhw.acpi.acliner r zhw.acpi.battery.timer� rh )r+ r� r� r r0 �
power_pluggedr� )r1 r� r r r �#test_sensors_battery_against_sysctl� s
z9FreeBSDSystemTestCase.test_sensors_battery_against_sysctlz has batteryc C sB t jt�� td� td� td� W d Q R X tj� d ks>t�d S )Nzhw.acpi.battery.lifezhw.acpi.battery.timezhw.acpi.acline)r ZraisesrO r r+ r� r0 )r1 r r r � test_sensors_battery_no_battery� s
z5FreeBSDSystemTestCase.test_sensors_battery_no_batteryc C s� t jd�}x�t|�D ]�}d| }y ttt|�d d� ��}W n tk
rZ tjd��Y nX t t j
� d | j | �dk s~t �d| }ttt|�d d � ��}t j
� d | j
|kst �qW d S )
NTzdev.cpu.%s.temperaturer z$temperatures not supported by kernelZcoretempr> zdev.cpu.%s.coretemp.tjmaxrh rh )r+ rF �ranger r r rO r r� rB Zsensors_temperaturesr� r0 �high)r1 Znum_cpusZcpur� r� r r r �(test_sensors_temperatures_against_sysctl� s
z>FreeBSDSystemTestCase.test_sensors_temperatures_against_sysctlN)%rU rV rW � staticmethodr� r� r
r� r� r� r� r� r� r rZ r[ �MUSE_AVAILABLEr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r r r r r� � sB
r� z OPENBSD onlyc @ s e Zd Zdd� ZdS )�OpenBSDTestCasec C s6 t d�}tjj|d�}tjjtj� �}||ks2t�d S )Nz
kern.boottimez%a %b %d %H:%M:%S %Y)r �datetime�strptime�
fromtimestampr+ r� r0 )r1 r� Zsys_btZ psutil_btr r r r� � s z OpenBSDTestCase.test_boot_timeN)rU rV rW r� r r r r r� � s r� z NETBSD onlyc @ sh e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� ZdS )�NetBSDTestCasec C sP t d��2}x*|D ]"}|j| �rt|j� d �d S qW W d Q R X td| ��d S )Nz
/proc/meminfor i z
can't find %s)�openr r r r )Zlook_for�fr! r r r �
parse_meminfo� s
"z NetBSDTestCase.parse_meminfoc C s t j� j| jd�kst�d S )Nz MemTotal:)r+ rI r9 r� r0 )r1 r r r �test_vmem_total s z NetBSDTestCase.test_vmem_totalc C s$ t tj� j| jd� �tk s t�d S )NzMemFree:)rB r+ rI r; r� r r0 )r1 r r r r� s z NetBSDTestCase.test_vmem_freec C s$ t tj� j| jd� �tk s t�d S )NzBuffers:)rB r+ rI r� r� r r0 )r1 r r r r� s z NetBSDTestCase.test_vmem_buffersc C s$ t tj� j| jd� �tk s t�d S )Nz
MemShared:)rB r+ rI �sharedr� r r0 )r1 r r r �test_vmem_shared s z NetBSDTestCase.test_vmem_sharedc C s$ t tj� j| jd� �tk s t�d S )NzCached:)rB r+ rI r� r� r r0 )r1 r r r r� $ s z NetBSDTestCase.test_vmem_cachedc C s$ t tj� j| jd� �tk s t�d S )Nz
SwapTotal:)rB r+ r� r9 r� r r0 )r1 r r r r� , s z!NetBSDTestCase.test_swapmem_totalc C s$ t tj� j| jd� �tk s t�d S )Nz SwapFree:)rB r+ r� r; r� r r0 )r1 r r r r� 2 s z NetBSDTestCase.test_swapmem_freec C s" t j� }|j|j|j ks t�d S )N)r+ r� r: r9 r; r0 )r1 �smemr r r r� 8 s z NetBSDTestCase.test_swapmem_usedc C sf t dd��8}x0|D ] }|jd�rt|j� d �}P qW td��W d Q R X ttj� j| �dk sbt �d S )Nz
/proc/stat�rbs intrr zcouldn't find linei� )
r� r r r r rB r+ r� r� r0 )r1 r� r! r� r r r r� > s
z(NetBSDTestCase.test_cpu_stats_interruptsc C sf t dd��8}x0|D ] }|jd�rt|j� d �}P qW td��W d Q R X ttj� j| �dk sbt �d S )Nz
/proc/statr� s ctxtr zcouldn't find linei� )
r� r r r r rB r+ r� r� r0 )r1 r� r! r� r r r r� H s
z*NetBSDTestCase.test_cpu_stats_ctx_switchesN)rU rV rW r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� )"rX r� �osrQ r- r+ r r r r Z psutil.testsr r r r r
r r r
r Zpsutil._psutil_posixr r� �getuidr� r r rZ r[ r# r\ r� r� r� r r r r �<module>
sJ QX x
|