Your IP : 18.191.171.72
3
��JZ�? � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm
Z
ddlm Z ddlm Z ddlm
Z
ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! dd� Z"d d � Z#d d � Z$e j%e d �G d!d"� d"e j&��Z'e j%e d �G d#d$� d$e j&��Z(e)d%k�r�ee*� dS )&zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�OPENBSD)�OSX)�POSIX)�SUNOS)�callable)�PY3)�APPVEYOR)�get_kernel_version)�get_test_subprocess)�mock)�
PYTHON_EXE)�
reap_children)�retry_before_failing)�run_test_module_by_name)�sh)�skip_on_access_denied)�TRAVIS)�unittest)� wait_for_pid)�whichc C sp t s| jdd�} tr | jdd�} tr0| jdd�} t| �}t sN|jd�d j� }yt|�S tk
rj |S X d S )
zkExpects a ps command with a -o argument and parse the result
returning only the value of interest.
z --no-headers � z-o startz-o stimez-o rssz -o rssize�
� N) r �replacer r r �split�strip�int�
ValueError)�cmd�output� r# �"/usr/lib64/python3.6/test_posix.py�ps+ s r% c C s& d}t r d}td|| f �jd�d S )N�commandZcommzps --no-headers -o %s -p %sr r )r r% r )�pid�fieldr# r# r$ �ps_nameI s r) c C s d}t s trd}td|| f �S )Nr& �argszps --no-headers -o %s -p %s)r r r% )r' r( r# r# r$ �ps_argsP s r+ z
POSIX onlyc @ 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� Zejep�ed �d d � �Zd d � Zd!d"� Zejed#�eje d$�d%d&� ��Z d'd(� Z d)S )*� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s& t tddgtjd�j| _t| j� d S )Nz-Ez-O)�stdin)r
r �
subprocess�PIPEr' r )�clsr# r# r$ �
setUpClass[ s
zTestProcess.setUpClassc C s
t � d S )N)r )r0 r# r# r$ �
tearDownClassa s zTestProcess.tearDownClassc C s. t d| j �}tj| j�j� }| j||� d S )Nz ps --no-headers -o ppid -p %s)r% r' �psutil�ProcessZppid� assertEqual)�selfZppid_psZ ppid_psutilr# r# r$ � test_ppide s zTestProcess.test_ppidc C s0 t d| j �}tj| j�j� j}| j||� d S )Nz ps --no-headers -o uid -p %s)r% r' r3 r4 �uids�realr5 )r6 Zuid_psZ
uid_psutilr# r# r$ �test_uidj s zTestProcess.test_uidc C s0 t d| j �}tj| j�j� j}| j||� d S )Nz ps --no-headers -o rgid -p %s)r% r' r3 r4 Zgidsr9 r5 )r6 Zgid_psZ
gid_psutilr# r# r$ �test_gido s zTestProcess.test_gidc C s. t d| j �}tj| j�j� }| j||� d S )Nz ps --no-headers -o user -p %s)r% r' r3 r4 �usernamer5 )r6 Z username_psZusername_psutilr# r# r$ �
test_usernamet s zTestProcess.test_usernamec C sJ t j� }tjdtd��*}| j|j� t|j� j �� |j
s<t �W d Q R X d S )Nzpsutil.pwd.getpwuid)� side_effect) r3 r4 r �patch�KeyErrorr5 r< �strr8 r9 �called�AssertionError)r6 �pZfunr# r# r$ �test_username_no_resolutiony s z'TestProcess.test_username_no_resolutionc C s@ t jd� td| j �}tj| j�j� d d }| j||� d S )Ng�������?z ps --no-headers -o rss -p %sr i )�time�sleepr% r' r3 r4 � memory_infor5 )r6 Zrss_psZ
rss_psutilr# r# r$ �test_rss_memory� s
zTestProcess.test_rss_memoryc C s@ t jd� td| j �}tj| j�j� d d }| j||� d S )Ng�������?z ps --no-headers -o vsz -p %sr i )rF rG r% r' r3 r4 rH r5 )r6 Zvsz_psZ
vsz_psutilr# r# r$ �test_vsz_memory� s
zTestProcess.test_vsz_memoryc C sZ t | j�}tjj|�j� }tj| j�j� j� }t j
dd|�}t j
dd|�}| j ||� d S )Nz\d.\d� ) r) r' �os�path�basename�lowerr3 r4 �name�re�subr5 )r6 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|j� d� 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? r3 r4 r5 rP )r6 rP �cmdlinerD r# r# r$ �test_name_long� s
zTestProcess.test_name_longc C s\ d}t jd|d��@ t jdtjdd�d�� tj� }| j|j� d� W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)rT z"psutil._psplatform.Process.cmdliner rK )r> )r r? r3 � AccessDeniedr4 r5 rP )r6 rP rD r# r# r$ � test_name_long_cmdline_ad_exc� s
z)TestProcess.test_name_long_cmdline_ad_excc C s\ d}t jd|d��@ t jdtjdd�d�� tj� }| jtj|j� W d Q R X W d Q R X d S )Nzlong-program-namez psutil._psplatform.Process.name)rT z"psutil._psplatform.Process.cmdliner rK )r> )r r? r3 Z
NoSuchProcessr4 � assertRaisesrP )r6 rP rD r# r# r$ � test_name_long_cmdline_nsp_exc� s
z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablec C sh t d| j �jd�d }tj| j�j� }tjj|�jd�}t |�}tjj|�jd�}| j
|||g� d S )Nz ps --no-headers -o start -p %sr r z%H:%M:%S) r% r' r r3 r4 Z create_time�datetimeZ
fromtimestampZstrftime�round�assertIn)r6 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| j||� W n0 tk
rZ |d t|�� }| j||� Y nX d S )N)r) r' r3 r4 Zexer5 rC �len)r6 Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer# r# r$ �test_exe� s
zTestProcess.test_exec C s0 t | j�}djtj| j�j� �}| j||� d S )Nr )r+ r' �joinr3 r4 rU r5 )r6 Z
ps_cmdlineZpsutil_cmdliner# r# r$ � test_cmdline� s
zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s* t d| j �}tj� j� }| j||� d S )Nz ps --no-headers -o nice -p %s)r% r' r3 r4 �nicer5 )r6 Zps_niceZ psutil_nicer# r# r$ � test_nice� s zTestProcess.test_nicec
s � fdd�}t jtj� �}g }ddddddd d
d d g
}trNt� dk rN|jd� trft� dk rf|jd� x�tt j�D ]�� � jd�sr� |kr�qrqry0|j � }xt
d
�D ]}||� � q�W |j � }W n t j k
r� Y qrX t || �dkrrd� ||f }|j|� qrW |�r| j
ddj|� � d S )Nc sB f }t | � d �}|d k r:t|�r:� dkr0tjf}||� n| d S )N�rlimit)�getattrr r3 Z
RLIMIT_NOFILE)rD �attrr* )rP r# r$ �call� s
z&TestProcess.test_num_fds.<locals>.callZ terminate�killZsuspendZresumerc Z send_signal�waitZchildrenZas_dictZmemory_info_ex� � �$ re � Znum_ctx_switches�_r z@failure while processing Process.%s method (before=%s, after=%s)r )rk rl rm )rk rl rn )r3 r4 rL �getpidr r �append�dir�
startswithZnum_fds�rangerW �abs�failra ) r6 rh rD ZfailuresZ
ignored_namesZnum1�xZnum2rv r# )rP r$ � test_num_fds� s4
zTestProcess.test_num_fdsN) �__name__�
__module__� __qualname__�__doc__� classmethodr1 r2 r7 r: r; r= rE r r rI rJ rS rV rX rZ r �skipIfr r r^ r` rb r r rd rx r# r# r# r$ r, W s,
r, c @ s� e Zd ZdZe� dd� �Zejed�eje d�eje
d� d�dd � ���Z eje p`e o`e
j� d
�e� d d � ��Zd
d� Zdd� Zdd� Zdd� Zejed�dd� �ZdS )�TestSystemAPIszTest some system APIs.c C s� t strddddg}n ddddg}t|tjd�}|j� d j� }|j� dksPt�t rbt
|t j j
�}g }x<|jd�d d � D ]&}|rzt|j� d j� �}|j|� qzW |j|j� tj� }|j� |j� ts�tr�d|kr�|jdd� | j||� d S )
Nr% z-Az-or' Zax)�stdoutr r r )r r r
r. r/ Z communicater ZpollrC r
rA �sysr� �encodingr r rq �remover' r3 Zpids�sortr r �insertr5 )r6 r! rD r" Zpids_ps�liner' Z pids_psutilr# r# r$ � test_pids s( zTestSystemAPIs.test_pidszunreliable on SUNOSzunreliable on TRAVISZifconfigzno ifconfig cmdc C sV t d�}xHtjdd�j� 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 r3 Znet_io_counters�keysr rs rv )r6 r" Znicr� r# r# r$ �test_nic_names? s
z TestSystemAPIs.test_nic_namesz unreliable on APPVEYOR or TRAVISc C sx t d�}|jd�}dd� |D �}dd� |D �}| jt|�ttj� �� x,tj� D ] }| j|j|� | j|j|� qPW d S )NZwhor c S s g | ]}|j � d �qS )r )r )�.0rw r# r# r$ �
<listcomp>T s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ]}|j � d �qS )r )r )r� rw r# r# r$ r� U s ) r r r5 r_ r3 �usersr] rP Zterminal)r6 �out�linesr� Z terminals�ur# r# r$ �
test_usersN s
zTestSystemAPIs.test_usersc C sF t jdttjd�d��&}| jttjjt j
� � |j s8t �W d Q R X d S )Nzpsutil._psposix.os.killrK )r> )
r r? �OSError�errno�EBADFrY r3 �_psposixZ
pid_existsrL rp rB rC )r6 �mr# r# r$ �test_pid_exists_let_raise[ s z(TestSystemAPIs.test_pid_exists_let_raisec C sF t jdttjd�d��&}| jttjjt j
� � |j s8t �W d Q R X d S )Nzpsutil._psposix.os.waitpidrK )r> )
r r? r� r� r� rY r3 r� �wait_pidrL rp rB rC )r6 r� r# r# r$ �test_os_waitpid_let_raised s z(TestSystemAPIs.test_os_waitpid_let_raisec
C sN t jdttjd�d��.}| jtjjtjj t
j � dd� |j s@t
�W d Q R X d S )Nzpsutil._psposix.os.waitpidrK )r> g{�G�z�?)Ztimeout)r r? r� r� ZEINTRrY r3 r� ZTimeoutExpiredr� rL rp rB rC )r6 r� r# r# r$ �test_os_waitpid_eintrl s z$TestSystemAPIs.test_os_waitpid_eintrc C s> t jddd��&}| jttjjtj� � |j s0t
�W d Q R X d S )Nzpsutil._psposix.os.waitpidr )rT ���)r r� ) r r? rY r r3 r� r� rL rp rB rC )r6 r� r# r# r$ � test_os_waitpid_bad_ret_statusu s
z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc
C s� dd� }d
}x�t jdd�D ]�}t j|j�}y||j�\}}}}W nL tk
r� } z0t| �j� } d| ksvd| ksvd | krzwn� W Y d d } ~ X qX | j|j ||d
� | j|j
||d
� | j|j ||d
� | j|j |d d
� qW d S )Nc S sv t d| �j� }|jd�d }|j� }t|d �d }t|d �d }t|d �d }t|d jdd ��}||||fS )
Nzdf -k %sr r i rk � � �%rK )r r r r �floatr )�devicer� r� Zfields�total�used�free�percentr# r# r$ �df� s z*TestSystemAPIs.test_disk_usage.<locals>.dfr� i F)�allzno such file or directoryzraw devices not supportedzpermission denied)Zdeltar i i @ )
r3 Zdisk_partitionsZ
disk_usageZ
mountpointr� � RuntimeErrorrA rO ZassertAlmostEqualr� r� r� r� )
r6 r� Z tolerance�partZusager� r� r� r� �errr# r# r$ �test_disk_usage~ s"
z TestSystemAPIs.test_disk_usageN)ry rz r{ r| r r� r r~ r r r r� r r3 r� r� r� r� r� r� r r� r# r# r# r$ r s
r �__main__)+r| r[ r� rL rQ r. r� rF r3 r r r r r r r Zpsutil._compatr r
Z psutil.testsr r r
r r r r r r r r r r r r% r) r+ r~ ZTestCaser, r ry �__file__r# r# r# r$ �<module> sT
E