3
�[g D � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z e
�rNddl Z ddl Z ddl m Z d,dd�Z!dd� Z"d d � Z#d d � Z$d d!� Z%d"d#� Z&ej'j(e
d$d%�G d&d'� d'e��Z)ej'j(e
d$d%�G d(d)� d)e��Z*ej'j(e
d$d%�G d*d+� d+e��Z+dS )-zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�AARCH64)�HAS_NET_IO_COUNTERS)�
PYTHON_EXE)� QEMU_USER)�PsutilTestCase)�mock)�pytest)�retry_on_failure)�sh)�skip_on_access_denied)�spawn_testproc)� terminate)�which)� getpagesizec C s� dg}t r|jd� |dk r0|jdt|�g� n ts8trD|jd� n
|jd� tr^| jdd�} |jd | g� t|�}t r�|j� n|j� d
d� }g }x@|D ]8}|j � }y t
|�}W n t k
r� Y nX |j|� q�W |dkr�|S |d S dS ) zwWrapper for calling the ps command with a little bit of cross-platform
support for a narrow range of features.
�psz --no-headersNz-pz-AZax�startZstimez-o� r ) r �append�extend�strr r �replacer �
splitlines�strip�int�
ValueError)�fmt�pid�cmd�outputZ
all_output�line� r' �C/usr/local/lib64/python3.6/site-packages/psutil/tests/test_posix.pyr / s0
r c C s> d}t r d}t|| �j� }tr6d|d ks.t�|d S |d S )N�commandZcommz
/bin/qemu-r r )r r �splitr �AssertionError)r# �fieldr) r' r' r( �ps_nameg s r- c C s0 d}t s trd}t|| �}tjdd|�}|j� S )Nr) �argsz\(python.*?\)$� )r r r �re�subr )r# r, �outr' r' r( �ps_argsr s
r3 c C s d}t r d}t|| �S )NZrssZrssize)r r )r# r, r' r' r( �ps_rss| s r4 c C s d}t r d}t|| �S )NZvszZvsize)r r )r# r, r' r' r( �ps_vsz� s r5 c C s� yt d| �j� }W n> tk
rR } z"dt|�j� kr@tjd��� W Y d d }~X nX |jd�d }|j� }t|d �d }t|d �d }t|d �d }t |d j
d
d ��}||||fS ) Nzdf -k %sz device busyzdf returned EBUSY�
r i � � � �%r/ ) r r � RuntimeErrorr �lowerr �skipr* r �floatr ) �devicer2 �errr& �fields� sys_total�sys_used�sys_free� sys_percentr' r' r( �df� s
rF z
POSIX only)�reasonc @ s� e Zd ZdZedd� �Zedd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z e � e
� dd� ��Ze � e
� dd� ��Zdd� Zdd� Zdd� Zdd� Zejjep�ed d �d d � �Zd d!� Ze
� d"d#� �Zejje d$d �ejje d%d �d&d'� ��Z d(S ))� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s t tddgtjd�j| _d S )Nz-Ez-O)�stdin)r r �
subprocess�PIPEr# )�clsr' r' r( �
setUpClass� s zTestProcess.setUpClassc C s t | j� d S )N)r r# )rL r' r' r( �
tearDownClass� s zTestProcess.tearDownClassc C s, t d| j�}tj| j�j� }||ks(t�d S )N�ppid)r r# �psutil�ProcessrO r+ )�selfZppid_psZ ppid_psutilr' r' r( � test_ppid� s zTestProcess.test_ppidc C s. t d| j�}tj| j�j� j}||ks*t�d S )N�uid)r r# rP rQ �uids�realr+ )rR Zuid_psZ
uid_psutilr' r' r( �test_uid� s zTestProcess.test_uidc C s. t d| j�}tj| j�j� j}||ks*t�d S )NZrgid)r r# rP rQ ZgidsrV r+ )rR Zgid_psZ
gid_psutilr' r' r( �test_gid� s zTestProcess.test_gidc C s, t d| j�}tj| j�j� }||ks(t�d S )N�user)r r# rP rQ �usernamer+ )rR Z username_psZusername_psutilr' r' r( �
test_username� s zTestProcess.test_usernamec
C sJ t j� }tjdtd��*}|j� t|j� j�ks2t �|j
s<t �W d Q R X d S )Nzpsutil.pwd.getpwuid)� side_effect) rP rQ r �patch�KeyErrorrZ r rU rV r+ �called)rR �p�funr' r' r( �test_username_no_resolution� s z'TestProcess.test_username_no_resolutionc C s< t jd� t| j�}tj| j�j� d d }||ks8t�d S )Ng�������?r i )�time�sleepr4 r# rP rQ � memory_infor+ )rR Zrss_psZ
rss_psutilr' r' r( �test_rss_memory� s
zTestProcess.test_rss_memoryc C s< t jd� t| j�}tj| j�j� d d }||ks8t�d S )Ng�������?r i )rc rd r5 r# rP rQ re r+ )rR Zvsz_psZ
vsz_psutilr' r' r( �test_vsz_memory� s
zTestProcess.test_vsz_memoryc C sv t | j�}tjj|�j� }tj| j�j� j� }t j
dd|�}t j
dd|�}t j
dd|�}t j
dd|�}||ksrt �d S )Nz\d.\dr/ z\d) r- r# �os�path�basenamer< rP rQ �namer0 r1 r+ )rR Zname_psZ name_psutilr' r' r( � test_name� s
zTestProcess.test_namec C s^ d}dddg}t jd|d��8 t jd|d�� tj� }|j� dksFt�W d Q R X W d Q R X d S )Nzlong-program-namezlong-program-name-extendedZfooZbarz psutil._psplatform.Process.name)� return_valuez"psutil._psplatform.Process.cmdline)r r] rP rQ rk r+ )rR rk �cmdliner` r' r' r( �test_name_long� s
zTestProcess.test_name_longc C s\ d}t jd|d��@ t jdtjdd�d�� tj� }|j� dksDt�W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)rm z"psutil._psplatform.Process.cmdliner r/ )r\ )r r] rP � AccessDeniedrQ rk r+ )rR rk r` r' r' r( � test_name_long_cmdline_ad_exc� s z)TestProcess.test_name_long_cmdline_ad_excc C sl d}t jd|d��P t jdtjdd�d��. tj� }tjtj�� |j� W d Q R X W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)rm z"psutil._psplatform.Process.cmdliner r/ )r\ )r r] rP �
NoSuchProcessrQ r �raisesrk )rR rk r` r' r' r( � test_name_long_cmdline_nsp_exc s z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not available)rG c C s\ t d| j�}tj| j�j� }tjj|�jd�}t|�}tjj|�jd�}|||gksXt �d S )Nr z%H:%M:%S)
r r# rP rQ Z create_time�datetime�
fromtimestamp�strftime�roundr+ )rR Ztime_psZ time_psutilZtime_psutil_tstampZround_time_psutilZround_time_psutil_tstampr' r' r( �test_create_time s z TestProcess.test_create_timec C s` t | j�}tj| j�j� }y||ks(t�W n0 tk
rZ |d t|�� }||ksVt�Y nX d S )N)r- r# rP rQ Zexer+ �len)rR Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer' r' r( �test_exe s
zTestProcess.test_exec C sT t | j�}djtj| j�j� �}trDt|�t|�k rD|j|�sPt �n ||ksPt �d S )N� )
r3 r# �joinrP rQ rn r rz �
startswithr+ )rR Z
ps_cmdlineZpsutil_cmdliner' r' r( � test_cmdline1 s
zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s( t d| j�}tj� j� }||ks$t�d S )N�nice)r r# rP rQ r� r+ )rR Zps_niceZ psutil_nicer' r' r( � test_nice? s zTestProcess.test_niceN) �__name__�
__module__� __qualname__�__doc__� classmethodrM rN rS rW rX r[ rb r r rf rg rl ro rq rt r �mark�skipifr r ry r{ r r r r� r' r' r' r( rH � s*
rH c @ s� e Zd ZdZe� dd� �Zejje dd�ejje
d� dd�ejje dd�d d
� ���Z e� d d � �Z
e� d
d� �Zdd� Zdd� Zdd� Zdd� Zejjedd�e� dd� ��ZdS )�TestSystemAPIszTest some system APIs.c s~ t td��� tj� �ts$tr0d� kr0� jdd� t� �t�� dkrz� fdd��D ��fdd�� D � }| jdt |� ��d S )Nr# r r c s g | ]}|� kr|�qS r' r' )�.0�x)�pids_psr' r( �
<listcomp>X s z,TestSystemAPIs.test_pids.<locals>.<listcomp>c s g | ]}|� kr|�qS r' r' )r� r� )� pids_psutilr' r( r� Y s z difference: )
�sortedr rP Zpidsr r �insertrz �failr )rR �
differencer' )r� r� r( � test_pidsK s zTestSystemAPIs.test_pidszunreliable on SUNOS)rG �ifconfigzno ifconfig cmdz
not supportedc C sR t d�}xDtjdd�D ]4}x.|j� D ]}|j|�r$P q$W | jd||f ��qW d S )Nz ifconfig -aT)Zpernicz/couldn't find %s nic in 'ifconfig -a' output
%s)r rP Znet_io_countersr* r~ r� )rR r% Znicr& r' r' r( �test_nic_names_ s
z TestSystemAPIs.test_nic_namesc C s� t d�}|j� stjd��|jd�}dd� |D �}dd� |D �}t|�ttj� �ksXt�| j tj� |d��Z xRt
tj� �D ]B\}}|j || ks�t�|j || ks�t�|j
d k rztj|j
� qzW W d Q R X d S )Nzwho -uzno users on this systemr6 c S s g | ]}|j � d �qS )r )r* )r� r� r' r' r( r� v s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|j � d �qS )r )r* )r� r� r' r' r( r� w s )rP �who)r r r r= r* rz rP �usersr+ �subTest� enumeraterk �terminalr# rQ )rR r2 �linesr� Z terminals�idx�ur' r' r( �
test_usersp s
zTestSystemAPIs.test_usersc C s� t d�}|j� stjd��d }tjd|�}|r4d}nNtjd|�}|rJd}n8tjd|�}|r`d}n"tjd |�}|r�d}d
d � |D �}|s�tjd | ��| jtj� |d
��D x<t tj� �D ],\}}t
j
j |j �j
|�}||| ks�t�q�W W d Q R X d S )Nzwho -uzno users on this systemz \d\d\d\d-\d\d-\d\d \d\d:\d\dz%Y-%m-%d %H:%Mz [A-Z][a-z][a-z] \d\d \d\d:\d\dz %b %d %H:%Mz[A-Z][a-z][a-z] \d\dz%b %dz[a-z][a-z][a-z] \d\dc S s g | ] }|j � �qS r' )�
capitalize)r� r� r' r' r( r� � s z5TestSystemAPIs.test_users_started.<locals>.<listcomp>z(cannot interpret tstamp in who output
%s)rP r� )r r r r= r0 �findallr� rP r� r� ru rv �startedrw r+ )rR r2 Ztstampr� r� r� Z psutil_valuer' r' r( �test_users_started� s4
z!TestSystemAPIs.test_users_startedc C sV t jdttjd�d��6}tjt�� tjj t
j � � W d Q R X |j sHt
�W d Q R X d S )Nzpsutil._psposix.os.killr/ )r\ )r r] �OSError�errno�EBADFr rs rP �_psposixZ
pid_existsrh �getpidr_ r+ )rR �mr' r' r( �test_pid_exists_let_raise� s z(TestSystemAPIs.test_pid_exists_let_raisec C sV t jdttjd�d��6}tjt�� tjj t
j � � W d Q R X |j sHt
�W d Q R X d S )Nzpsutil._psposix.os.waitpidr/ )r\ )r r] r� r� r� r rs rP r� �wait_pidrh r� r_ r+ )rR r� r' r' r( �test_os_waitpid_let_raise� s z(TestSystemAPIs.test_os_waitpid_let_raisec C s^ t jdttjd�d��>}tjtjj �� tjj
t j � dd� W d Q R X |j
sPt�W d Q R X d S )Nzpsutil._psposix.os.waitpidr/ )r\ g{�G�z�?)�timeout)r r] r� r� ZEINTRr rs rP r� �TimeoutExpiredr� rh r� r_ r+ )rR r� r' r' r( �test_os_waitpid_eintr� s z$TestSystemAPIs.test_os_waitpid_eintrc C sN t jddd��6}tjt�� tjjtj � � W d Q R X |j
s@t �W d Q R X d S )Nzpsutil._psposix.os.waitpidr )rm ���)r r� ) r r] r rs r! rP r� r� rh r� r_ r+ )rR r� r' r' r( � test_os_waitpid_bad_ret_status� s
z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc C s� d
}x�t jdd�D ]�}t j|j�}yt|j�\}}}}W nJ tk
r� } z.t|�j� }d|ksnd|ksnd|krpw� W Y d d }~X qX t |j
| �|k s�t �t |j | �|k s�t �t |j
| �|k s�t �t |j| �dkst �qW d S ) Nr9 i F)�allzno such file or directoryzraw devices not supportedzpermission deniedr i i @ )rP Zdisk_partitions�
disk_usage�
mountpointrF r? r; r r< �abs�totalr+ �used�free�percent) rR Z tolerance�part�usagerB rC rD rE r@ r' r' r( �test_disk_usage� s z TestSystemAPIs.test_disk_usageN)r� r� r� r� r r� r r� r� r r r
r� r� r� r� r� r� r� r r� r' r' r' r( r� G s '
r� c @ s e Zd Zdd� ZdS )�TestMiscc C s4 t � }|dkst�|tj � ks"t�|tjks0t�d S )Nr )r r+ �resource�mmapZPAGESIZE)rR Zpagesizer' r' r( �test_getpagesize� s zTestMisc.test_getpagesizeN)r� r� r� r� r' r' r' r( r� � s r� )N),r� ru r� rh r0 rJ rc rP r r r r r r r Z psutil.testsr r
r r r
r r r r r r r r r� r� Zpsutil._psutil_posixr r r- r3 r4 r5 rF r� r� rH r� r� r' r' r' r( �<module> sZ
8
- # |