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 splitstripint
ValueError)cmdoutput r" /test_posix.pyps+ s
r$ c C s3 d } t r d } t d | | f j d d S)NcommandZcommzps --no-headers -o %s -p %sr r )r r$ r )pidfieldr" 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-Ostdin)r r
subprocessPIPEr& 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& psutilProcessZppid 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 uidsrealr4 )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
patchKeyErrorr4 r; strr7 r8 calledAssertionError)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 )timesleepr$ 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& ospathbasenamelowerr2 r3 nameresubr4 )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_timedatetimeZ
fromtimestampstrftimeroundassertIn)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)Nrlimit)getattrcallabler2 Z
RLIMIT_NOFILE)rC attrr) )rO r" r# call s
z&TestProcess.test_num_fds.<locals>.call terminatekillZsuspendZresumerc send_signalwaitZchildrenZas_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 appenddir
startswithZnum_fdsrangerV absfailra ) r5 ri rC Zfailures
ignored_namesZnum1xZnum2ry 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& Zaxstdoutr r r )r r r r- r. communicater pollrB r r@ sysr encodingr r rt remover& r2 Zpidssortr r insertr4 )r5 r rC r! Zpids_psliner& 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_counterskeysr 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 outlinesr Z terminalsur" 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> OSErrorerrnoEBADFrX 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{Gz?)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 Zfieldstotalusedfreepercentr" 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 tolerancepartusager 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