Cs[�? � @ s\ 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 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 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 � 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( d$ k rXe e) � d S)%zPOSIX specific tests.� N)�AIX)�BSD)�LINUX)�MACOS)�OPENBSD)�POSIX)�SUNOS)�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 s� t s | j d d � } t r0 | j d d � } t rH | j d d � } t | � } t ss | j d � d j � } y t | � SWn t k
r� | SYn Xd 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" �/test_posix.py�ps+ s
r$ c C s3 d } t r d } t d | | f � j d � 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 t r d } t d | | 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 @ sZ e Z d 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 e � e
� d d � � � Z e � e
� d d � � � Z d d � Z d d � Z d d � Z d d � Z e j e p� e d � d d � � Z d d � Z d! d" � Z e j e d# � e j e d$ � d% d& � � � Z d' d( � Z d) S)*� TestProcesszBCompare psutil results against 'ps' command line utility (mainly).c C s5 t t d d g d t j �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 )r/ r" r" r# �
tearDownClassa s zTestProcess.tearDownClassc C s? t d | j � } t j | 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 sB t d | j � } t j | j � j � j } | j | | � d S)Nz ps --no-headers -o uid -p %s)r$ r& r2 r3 �uids�realr4 )r5 Zuid_psZ
uid_psutilr" r" r# �test_uidj s zTestProcess.test_uidc C sB t d | j � } t j | j � j � j } | j | | � d S)Nz ps --no-headers -o rgid -p %s)r$ r& r2 r3 Zgidsr8 r4 )r5 Zgid_psZ
gid_psutilr" r" r# �test_gido s zTestProcess.test_gidc C s? t d | j � } t j | j � j � } | j | | � d S)Nz ps --no-headers -o user -p %s)r$ r& r2 r3 �usernamer4 )r5 Z username_psZusername_psutilr" r" r# �
test_usernamet s zTestProcess.test_usernamec C sc t j � } t j d d t ��; } | j | j � t | j � j � � | j
sX t � Wd QRXd S)Nzpsutil.pwd.getpwuid� side_effect) r2 r3 r
�patch�KeyErrorr4 r; �strr7 r8 �called�AssertionError)r5 �pZfunr" r" r# �test_username_no_resolutiony s %z'TestProcess.test_username_no_resolutionc C sT t j d � t d | j � } t j | j � j � d d } | j | | � d S)Ng�������?z ps --no-headers -o rss -p %sr i )�time�sleepr$ r& r2 r3 � memory_infor4 )r5 Zrss_psZ
rss_psutilr" r" r# �test_rss_memory� s
zTestProcess.test_rss_memoryc C sT t j d � t d | j � } t j | j � j � d d } | j | | � d S)Ng�������?z ps --no-headers -o vsz -p %sr i )rE rF r$ r& r2 r3 rG r4 )r5 Zvsz_psZ
vsz_psutilr" r" r# �test_vsz_memory� s
zTestProcess.test_vsz_memoryc C s� t | j � } t j j | � j � } t j | j � j � j � } t j
d d | � } t j
d d | � } | j | | � d S)Nz\d.\d� ) r( r& �os�path�basename�lowerr2 r3 �name�re�subr4 )r5 Zname_psZ name_psutilr" r" r# � test_name� s zTestProcess.test_namec C su d } d d d g } t j d d | ��D t j d d | ��' t j � } | j | j � d � Wd QRXWd QRXd S)Nzlong-program-namezlong-program-name-extendedZfooZbarz psutil._psplatform.Process.name� return_valuez"psutil._psplatform.Process.cmdline)r
r> r2 r3 r4 rO )r5 rO �cmdlinerC r" r" r# �test_name_long� s
zTestProcess.test_name_longc C sr d } t j d d | ��P t j d d t j d d � ��' t j � } | j | j � d � Wd QRXWd QRXd S)Nzlong-program-namez psutil._psplatform.Process.namerS z"psutil._psplatform.Process.cmdliner= r rJ )r
r> r2 � AccessDeniedr3 r4 rO )r5 rO rC r" r" r# � test_name_long_cmdline_ad_exc� s
z)TestProcess.test_name_long_cmdline_ad_excc C sr d } t j d d | ��P t j d d t j d d � ��' t j � } | j t j | j � Wd QRXWd QRXd S)Nzlong-program-namez psutil._psplatform.Process.namerS z"psutil._psplatform.Process.cmdliner= r rJ )r
r> r2 Z
NoSuchProcessr3 � assertRaisesrO )r5 rO rC r" r" r# � test_name_long_cmdline_nsp_exc� s
z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablec C s� t d | j � j d � d } t j | j � j � } t j j | � j d � } t | � } t j j | � j d � } | j
| | | g � d S)Nz ps --no-headers -o start -p %sr r z%H:%M:%S) r$ r& r r2 r3 Z create_time�datetimeZ
fromtimestamp�strftime�round�assertIn)r5 Ztime_psZ time_psutilZtime_psutil_tstampZround_time_psutilZround_time_psutil_tstampr" r" r# �test_create_time� s z TestProcess.test_create_timec C sz t | j � } t j | j � j � } y | j | | � Wn8 t k
ru | d t | � � } | j | | � Yn Xd S)N)r( r& r2 r3 Zexer4 rB �len)r5 Z ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer" r" r# �test_exe� s
zTestProcess.test_exec C sD t | j � } d j t j | j � j � � } | j | | � d S)Nr )r* r& �joinr2 r3 rT r4 )r5 Z
ps_cmdlineZpsutil_cmdliner" r" r# � test_cmdline� s !zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXc C s9 t d | j � } t j � j � } | j | | � d S)Nz ps --no-headers -o nice -p %s)r$ r& r2 r3 �nicer4 )r5 Zps_niceZ psutil_nicer" r" r# � test_nice� s zTestProcess.test_nicec
s� � f d d � } t j t j � � } g } d d d d d d d d
d d g
} t rs t � d k rs | j d � t r� t � d k r� | j d � x� t t j � D]� � � j d � s� � | k r� q� q� y@ | j � } x! t
d
� D] } | | � � q� W| j � } Wn t j k
r#Yq� Xt | | � d k r� d � | | f } | j | � q� W| r~| j
d d j | � � d S)Nc s] f } t | � d � } | d k rU t | � rU � d k rH t j f } | | � n | d S)N�rlimit)�getattr�callabler2 Z
RLIMIT_NOFILE)rC �attrr) )rO r" r# �call� s
z&TestProcess.test_num_fds.<locals>.call� terminate�killZsuspendZresumerc � send_signal�waitZchildrenZas_dictZmemory_info_ex� � �$ re � Znum_ctx_switches�_r z@failure while processing Process.%s method (before=%s, after=%s)r )rn ro rp )rn ro rq )r2 r3 rK �getpidr r �append�dir�
startswithZnum_fds�rangerV �abs�failra ) r5 ri rC Zfailures�
ignored_namesZnum1�xZnum2ry r" )rO r# � test_num_fds� s4
zTestProcess.test_num_fdsN) �__name__�
__module__� __qualname__�__doc__� classmethodr0 r1 r6 r9 r: r< rD r r rH rI rR rU rW rY r �skipIfr r r^ r` rb r r rd r| r" r" r" r# r+ W s,
$
!r+ c @ s� e Z d Z d Z e � d d � � Z e j e d � e j e d � e j e
d � d � d d � � � � Z e j e p� e o� e
j � d
� e � d d � � � Z d
d � Z d d � Z d d � Z d d � Z e j e d � d d � � Z d S)�TestSystemAPIszTest some system APIs.c C sV t s t r! d d d d g } n d d d d g } t | d t j �} | j � d j � } | j � d k sv t � t r� t
| t j j
� } g } xP | j d � d d � D]5 } | r� t | j � d j � � } | j | � q� W| j | j � t j � } | j � | j � t s2t rBd | k rB| j d d � | j | | � d S)
Nr$ z-Az-or& Zax�stdoutr r r )r r r r- r. � communicater �pollrB r r@ �sysr� �encodingr r rt �remover& r2 Zpids�sortr r �insertr4 )r5 r rC r! Zpids_ps�liner& Z pids_psutilr" r" r# � test_pids s(
zTestSystemAPIs.test_pidszunreliable on SUNOSzunreliable on TRAVISZifconfigzno ifconfig cmdc C sq t d � } x^ t j d d � j � D]D } x; | j � D] } | j | � r8 Pq8 W| j d | | f � q% Wd S)Nz ifconfig -aZpernicTz/couldn't find %s nic in 'ifconfig -a' output
%s)r r2 Znet_io_counters�keysr rv ry )r5 r! Znicr� r" r" r# �test_nic_names? s z TestSystemAPIs.test_nic_namesz unreliable on APPVEYOR or TRAVISc C s� t d � } | j d � } d d � | D� } d d � | D� } | j t | � t t j � � � x: t j � D], } | j | j | � | j | j | � qp Wd S)NZwhor c S s g | ] } | j � d � q S)r )r )�.0r{ r" r" r# �
<listcomp>T s z-TestSystemAPIs.test_users.<locals>.<listcomp>c S s g | ] } | j � d � q S)r )r )r� r{ r" r" r# r� U s ) r r r4 r_ r2 �usersr] rO Zterminal)r5 �out�linesr� Z terminals�ur" r" r# �
test_usersN s "zTestSystemAPIs.test_usersc C s] t j d d t t j d � ��5 } | j t t j j t j
� � | j sR t � Wd QRXd S)Nzpsutil._psposix.os.killr= rJ )
r
r> �OSError�errno�EBADFrX r2 �_psposixZ
pid_existsrK rs rA rB )r5 �mr" r" r# �test_pid_exists_let_raise[ s z(TestSystemAPIs.test_pid_exists_let_raisec C s] t j d d t t j d � ��5 } | j t t j j t j
� � | j sR t � Wd QRXd S)Nzpsutil._psposix.os.waitpidr= rJ )
r
r> r� r� r� rX r2 r� �wait_pidrK rs rA rB )r5 r� r" r" r# �test_os_waitpid_let_raised s z(TestSystemAPIs.test_os_waitpid_let_raisec
C si t j d d t t j d � ��A } | j t j j t j j t
j � d d �| j s^ t
� Wd QRXd S)Nzpsutil._psposix.os.waitpidr= rJ �timeoutg{�G�z�?)r
r> r� r� ZEINTRrX r2 r� �TimeoutExpiredr� rK rs rA rB )r5 r� r" r" r# �test_os_waitpid_eintrl s z$TestSystemAPIs.test_os_waitpid_eintrc C sQ t j d d d ��5 } | j t t j j t j � � | j sF t
� Wd QRXd S)Nzpsutil._psposix.os.waitpidrS r ���)r r� ) r
r> rX r r2 r� r� rK rs rA rB )r5 r� r" r" r# � test_os_waitpid_bad_ret_statusu s
z-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc
C s/ d d � } d
} xt j d d � D]} t j | j � } y | | j � \ } } } } Wnd t k
r� } zD t | � j � } d | k s� d | k s� d | k r� w% n � WYd d } ~ Xq% X| j | j | d
| �| j | j
| d
| �| j | j | d
| �| j | j | d
d �q% Wd S)Nc S s� t d | � j � } | j d � d } | j � } t | d � d } t | d � d } t | d � d } t | d j d d � � } | | | | f S)
Nzdf -k %sr r i rn � � �%rJ )r r r r �floatr )�devicer� r� Zfields�total�used�free�percentr" r" r# �df� s z*TestSystemAPIs.test_disk_usage.<locals>.dfr� i �allFzno such file or directoryzraw devices not supportedzpermission deniedZdeltar i i @ )
r2 Zdisk_partitions�
disk_usageZ
mountpointr� � RuntimeErrorr@ rN ZassertAlmostEqualr� r� r� r� )
r5 r� Z tolerance�part�usager� r� r� r� �errr" r" r# �test_disk_usage~ s"
z TestSystemAPIs.test_disk_usageN)r} r~ r r� r r� r r� r r r r� r
r2 r� r� r� r� r� r� r r� r" r" r" r# r� s +
r� �__main__)*r� rZ r� rK rP r- r� rE r2 r r r r r r r Zpsutil._compatr Z psutil.testsr
r r r
r r r r r r r r r r r$ r( r* r� ZTestCaser+ r� r} �__file__r" r" r" r# �<module> sN � �