­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ U =_';@sdZddlZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZe rBddlZddlZddlmZd*ddZ ddZ!ddZ"ddZ#dd Z$e%e d!Gd"d#d#eZ&e%e d!Gd$d%d%eZ'e%e d!Gd&d'd'eZ(e)d(krdd)l*m+Z+e+e,dS)+zPOSIX specific tests.N)AIX)BSD)LINUX)MACOS)OPENBSD)POSIX)SUNOS) CI_TESTING)spawn_testproc)HAS_NET_IO_COUNTERS)mock)PsutilTestCase) PYTHON_EXE)retry_on_failure)sh)skip_on_access_denied) terminate)unittest)which) getpagesizec Csdg}tr|d|dk r0|dt|gnts8trD|dn |dtrftd}|||}|d|gt|}tr| }n| d d}g}|D]8}| }z t |}Wnt k rYnX||q|dkr|S|d SdS) z| Wrapper for calling the ps command with a little bit of cross-platform support for a narrow range of features. psz --no-headersNz-pz-AZax)commandcommstartZstimez-or) rappendextendstrrrsetgetr splitlinesstripint ValueError)fmtpidcmdZfmt_mapoutputZ all_outputliner)J/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_posix.pyr-s6       rcCsd}tr d}t||dS)Nrrr)rrsplitr%fieldr)r)r*ps_nameisr.cCsd}ts trd}t||S)Nrargs)rrrr,r)r)r*ps_argspsr0cCsd}tr d}t||S)NZrssZrssizerrr,r)r)r*ps_rsswsr2cCsd}tr d}t||S)NZvszZvsizer1r,r)r)r*ps_vsz~sr3z POSIX onlyc@seZdZdZeddZeddZddZdd Zd d Z d d Z ddZ e e ddZe e ddZddZddZddZddZeepedddZdd Zd!d"Zeed#eed$d%d&Zd'S)( TestProcesszBCompare psutil results against 'ps' command line utility (mainly).cCsttddgtjdj|_dS)Nz-Ez-O)stdin)r r subprocessPIPEr%clsr)r)r* setUpClasss zTestProcess.setUpClasscCst|jdSN)rr%r8r)r)r* tearDownClassszTestProcess.tearDownClasscCs,td|j}t|j}|||dS)Nppid)rr%psutilProcessr= assertEqual)selfZppid_psZ ppid_psutilr)r)r* test_ppids zTestProcess.test_ppidcCs.td|j}t|jj}|||dS)Nuid)rr%r>r?uidsrealr@)rAZuid_psZ uid_psutilr)r)r*test_uids zTestProcess.test_uidcCs.td|j}t|jj}|||dS)NZrgid)rr%r>r?ZgidsrEr@)rAZgid_psZ gid_psutilr)r)r*test_gids zTestProcess.test_gidcCs,td|j}t|j}|||dS)Nuser)rr%r>r?usernamer@)rAZ username_psZusername_psutilr)r)r* test_usernames zTestProcess.test_usernamec CsJt}tjdtd*}||t|j |j sr?r patchKeyErrorr@rIrrDrEcalledAssertionError)rApfunr)r)r*test_username_no_resolutionsz'TestProcess.test_username_no_resolutioncCs<tdt|j}t|jdd}|||dS)N皙?r)timesleepr2r%r>r? memory_infor@)rAZrss_psZ rss_psutilr)r)r*test_rss_memorys  zTestProcess.test_rss_memorycCs<tdt|j}t|jdd}|||dS)NrTrrU)rVrWr3r%r>r?rXr@)rAZvsz_psZ vsz_psutilr)r)r*test_vsz_memorys  zTestProcess.test_vsz_memorycCsvt|j}tj|}t|j}t dd|}t dd|}t dd|}t dd|}| ||dS)Nz\d.\dz\d) r.r%ospathbasenamelowerr>r?nameresubr@)rAZname_psZ name_psutilr)r)r* test_names zTestProcess.test_namec Cs^d}dddg}tjd|d8tjd|dt}||dW5QRXW5QRXdS)Nlong-program-namezlong-program-name-extendedZfooZbarpsutil._psplatform.Process.nameZ return_value"psutil._psplatform.Process.cmdline)r rMr>r?r@r`)rAr`cmdlinerQr)r)r*test_name_longs zTestProcess.test_name_longc Cs\d}tjd|d@tjdtdddt}||dW5QRXW5QRXdSNrdrerfrgrr[rK)r rMr> AccessDeniedr?r@r`rAr`rQr)r)r*test_name_long_cmdline_ad_excs z)TestProcess.test_name_long_cmdline_ad_excc Cs\d}tjd|d@tjdtdddt}|tj|jW5QRXW5QRXdSrj)r rMr> NoSuchProcessr? assertRaisesr`rlr)r)r*test_name_long_cmdline_nsp_excs z*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablecCs\td|j}t|j}tj|d}t|}tj|d}| |||gdS)Nrz%H:%M:%S) rr%r>r?Z create_timedatetime fromtimestampstrftimeroundassertIn)rAZtime_psZ time_psutilZtime_psutil_tstampZround_time_psutilZround_time_psutil_tstampr)r)r*test_create_times zTestProcess.test_create_timecCs`t|j}t|j}z|||Wn0tk rZ|dt|}|||YnXdSr;)r.r%r>r?Zexer@rPlen)rAZ ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer)r)r*test_exes zTestProcess.test_execCs0t|j}dt|j}|||dS)N )r0r%joinr>r?rhr@)rAZ ps_cmdlineZpsutil_cmdliner)r)r* test_cmdlines zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXcCs(td|j}t}|||dS)Nnice)rr%r>r?r|r@)rAZps_niceZ psutil_nicer)r)r* test_nices  zTestProcess.test_niceN)__name__ __module__ __qualname____doc__ classmethodr:r<rBrFrGrJrSrrrYrZrcrirmrprskipIfrrrvrxr{rrr}r)r)r)r*r4s6           r4c@seZdZdZeddZeedee d dee ddd Z ee o^e  d ed d Zd dZddZddZddZeededdZdS)TestSystemAPIszTest some system APIs.cs~ttdtts$tr0dkr0ddttdkrzfddDfddD}|dt |dS)Nr%rrcsg|]}|kr|qSr)r).0x)pids_psr)r* 4sz,TestSystemAPIs.test_pids..csg|]}|kr|qSr)r)r) pids_psutilr)r*r5sz difference: ) sortedrr>Zpidsrrinsertrwfailr)rA differencer))rrr* test_pids's  zTestSystemAPIs.test_pidszunreliable on SUNOSifconfigzno ifconfig cmdz not supportedcCsPtd}tjddD]2}|D]}||r$qq$|d||fqdS)Nz ifconfig -aT)Zpernicz/couldn't find %s nic in 'ifconfig -a' output %s)rr>Znet_io_counterskeysr+ startswithr)rAr'Znicr(r)r)r*test_nic_names:s  zTestSystemAPIs.test_nic_nameszunreliable on CIcCstd}|s|d|d}dd|D}dd|D}|t|tttD] }||j |||j |q`dS)Nwhozno users on this system cSsg|]}|dqS)rr+rr)r)r*rOsz-TestSystemAPIs.test_users..cSsg|]}|dqS)rrrr)r)r*rPs) rr!ZskipTestr+r@rwr>usersrur`terminal)rAoutlinesrZ terminalsur)r)r* test_usersHs   zTestSystemAPIs.test_usersc CsFtjdttjdd&}|ttjjt |j s8t W5QRXdS)Nzpsutil._psposix.os.killr[rK) r rMOSErrorerrnoEBADFror>_psposixZ pid_existsr\getpidrOrPrAmr)r)r*test_pid_exists_let_raiseVs  z(TestSystemAPIs.test_pid_exists_let_raisec CsFtjdttjdd&}|ttjjt |j s8t W5QRXdS)Npsutil._psposix.os.waitpidr[rK) r rMrrrror>rwait_pidr\rrOrPrr)r)r*test_os_waitpid_let_raise_s  z(TestSystemAPIs.test_os_waitpid_let_raisec CsNtjdttjdd.}|jtjjtjj t dd|j s@t W5QRXdS)Nrr[rKg{Gz?)timeout)r rMrrZEINTRror>rTimeoutExpiredrr\rrOrPrr)r)r*test_os_waitpid_eintrgs z$TestSystemAPIs.test_os_waitpid_eintrc Cs>tjddd&}|ttjjt|j s0t W5QRXdS)Nr)rrf) r rMror#r>rrr\rrOrPrr)r)r*test_os_waitpid_bad_ret_statuspsz-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc Csdd}d}tjddD]}t|j}z||j\}}}}WnRtk r} z4t| } d| kstd| kstd| kr~WY qnW5d} ~ XYqX|j|j ||d |j|j ||d |j|j ||d |j|j |d d qdS) NcSsvtd|}|dd}|}t|dd}t|dd}t|dd}t|ddd }||||fS) Nzdf -k %srrrU%r[)rr!r+r"floatreplace)devicerr(fieldstotalusedfreepercentr)r)r*df|sz*TestSystemAPIs.test_disk_usage..dfi@F)allzno such file or directoryzraw devices not supportedzpermission denied)deltar) r>Zdisk_partitions disk_usage mountpointr RuntimeErrorrr_ZassertAlmostEqualrrrr) rArZ tolerancepartusagerrrrerrr)r)r*test_disk_usageys&    zTestSystemAPIs.test_disk_usageN)r~rrrrrrrrrr rr r>rrrrrrrrr)r)r)r*r#s"        rc@seZdZddZdS)TestMisccCs4t}||d||t||tjdS)Nr)rZ assertGreaterr@resourcemmapZPAGESIZE)rAZpagesizer)r)r*test_getpagesizes zTestMisc.test_getpagesizeN)r~rrrr)r)r)r*rsr__main__) run_from_name)N)-rrqrr\rar6rVr>rrrrrrrZ psutil.testsr r r r r rrrrrrrrrZpsutil._psutil_posixrrr.r0r2r3rr4rrr~Zpsutil.tests.runnerr__file__r)r)r)r*sZ                     <  y