U (Ûc‰<ã@sädZddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZer>ddlZddlZddlmZd)dd„Z dd„Z!dd„Z"dd„Z#dd„Z$e %e d ¡Gd!d"„d"eƒƒZ&e %e d ¡Gd#d$„d$eƒƒZ'e %e d ¡Gd%d&„d&eƒƒZ(e)d'kràdd(l*m+Z+e+e,ƒdS)*zPOSIX specific tests.éN)ÚAIX)ÚBSD)ÚLINUX)ÚMACOS)ÚOPENBSD)ÚPOSIX)ÚSUNOS)Ú CI_TESTING)ÚHAS_NET_IO_COUNTERS)Ú PYTHON_EXE)ÚPsutilTestCase)Úmock)Úretry_on_failure)Úsh)Úskip_on_access_denied)Úspawn_testproc)Ú terminate)Úwhich)Ú getpagesizec Csôdg}tr| d¡|dk r0| dt|ƒg¡nts8trD| d¡n | d¡trftdƒ}| ||¡}| d|g¡t|ƒ}trŠ|  ¡}n|  ¡d d…}g}|D]8}|  ¡}z t |ƒ}Wnt k rÎYnX| |¡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)ÚcommandÚcommÚstartZstimez-oér) rÚappendÚextendÚstrrrÚsetÚgetrÚ splitlinesÚstripÚintÚ ValueError)ÚfmtÚpidÚcmdZfmt_mapÚoutputÚ all_outputÚline©r)ú;/tmp/pip-unpacked-wheel-l83rk7k5/psutil/tests/test_posix.pyr.s6       rcCsd}tr d}t||ƒ ¡dS)Nrrr)rrÚsplit©r$Úfieldr)r)r*Úps_namejsr.cCsd}ts trd}t||ƒS)NrÚargs)rrrr,r)r)r*Úps_argsqsr0cCsd}tr d}t||ƒS)NZrssZrssize©rrr,r)r)r*Úps_rssxsr2cCsd}tr d}t||ƒS)NZvszZvsizer1r,r)r)r*Úps_vszsr3z POSIX onlyc@sæeZdZdZedd„ƒZedd„ƒZdd„Zdd „Zd d „Z d d „Z dd„Z e ƒe ƒdd„ƒƒZe ƒe ƒdd„ƒƒZdd„Zdd„Zdd„Zdd„Ze epžed¡dd„ƒZdd „Ze ƒd!d"„ƒZe ed#¡e ed$¡d%d&„ƒƒZd'S)(Ú TestProcesszBCompare psutil results against 'ps' command line utility (mainly).cCsttddgtjdj|_dS)Nz-Ez-O)Ústdin)rr Ú subprocessÚPIPEr$©Úclsr)r)r*Ú setUpClassŠs ÿzTestProcess.setUpClasscCst|jƒdS©N)rr$r8r)r)r*Ú tearDownClassszTestProcess.tearDownClasscCs,td|jƒ}t |j¡ ¡}| ||¡dS)NÚppid)rr$ÚpsutilÚProcessr=Ú assertEqual)ÚselfZppid_psZ ppid_psutilr)r)r*Ú test_ppid“s zTestProcess.test_ppidcCs.td|jƒ}t |j¡ ¡j}| ||¡dS)NÚuid)rr$r>r?ÚuidsÚrealr@)rAZuid_psZ uid_psutilr)r)r*Útest_uid˜s zTestProcess.test_uidcCs.td|jƒ}t |j¡ ¡j}| ||¡dS)NZrgid)rr$r>r?ZgidsrEr@)rAZgid_psZ gid_psutilr)r)r*Útest_gids zTestProcess.test_gidcCs,td|jƒ}t |j¡ ¡}| ||¡dS)NÚuser)rr$r>r?Úusernamer@)rAZ username_psZusername_psutilr)r)r*Ú test_username¢s zTestProcess.test_usernamec CsJt ¡}tjdtd*}| | ¡t| ¡j ƒ¡|j sr?r ÚpatchÚKeyErrorr@rIrrDrEÚcalledÚAssertionError)rAÚpZfunr)r)r*Útest_username_no_resolution§sz'TestProcess.test_username_no_resolutioncCs<t d¡t|jƒ}t |j¡ ¡dd}| ||¡dS)Nçš™™™™™¹?ré)ÚtimeÚsleepr2r$r>r?Ú memory_infor@)rAZrss_psZ rss_psutilr)r)r*Útest_rss_memory°s  zTestProcess.test_rss_memorycCs<t d¡t|jƒ}t |j¡ ¡dd}| ||¡dS)NrRrrS)rTrUr3r$r>r?rVr@)rAZvsz_psZ vsz_psutilr)r)r*Útest_vsz_memoryºs  zTestProcess.test_vsz_memorycCsvt|jƒ}tj |¡ ¡}t |j¡ ¡ ¡}t   dd|¡}t   dd|¡}t   dd|¡}t   dd|¡}|  ||¡dS)Nz\d.\dÚz\d) r.r$ÚosÚpathÚbasenameÚlowerr>r?ÚnameÚreÚsubr@)rAZname_psZ name_psutilr)r)r*Ú test_nameÄs zTestProcess.test_namec Cs^d}dddg}tjd|d8tjd|dt ¡}| | ¡d¡W5QRXW5QRXdS)Núlong-program-namezlong-program-name-extendedZfooÚbarúpsutil._psplatform.Process.name©Z return_valueú"psutil._psplatform.Process.cmdline)r rLr>r?r@r^)rAr^ÚcmdlinerPr)r)r*Útest_name_longÒs ÿÿzTestProcess.test_name_longc Cs\d}tjd|d@tjdt dd¡dt ¡}| | ¡d¡W5QRXW5QRXdS©NrbrdrerfrrYrK)r rLr>Z AccessDeniedr?r@r^©rAr^rPr)r)r*Útest_name_long_cmdline_ad_excßsÿ ÿz)TestProcess.test_name_long_cmdline_ad_excc Cs\d}tjd|d@tjdt dd¡dt ¡}| tj|j¡W5QRXW5QRXdSri)r rLr>Z NoSuchProcessr?Ú assertRaisesr^rjr)r)r*Útest_name_long_cmdline_nsp_excësÿ ÿz*TestProcess.test_name_long_cmdline_nsp_exczps -o start not availablecCs\td|jƒ}t |j¡ ¡}tj |¡ d¡}t|ƒ}tj |¡ d¡}|  |||g¡dS)Nrz%H:%M:%S) rr$r>r?Z create_timeÚdatetimeÚ fromtimestampÚstrftimeÚroundÚassertIn)rAZtime_psZ time_psutilZtime_psutil_tstampZround_time_psutilZround_time_psutil_tstampr)r)r*Útest_create_timeös ÿÿÿÿzTestProcess.test_create_timecCs`t|jƒ}t |j¡ ¡}z| ||¡Wn0tk rZ|dt|ƒ…}| ||¡YnXdSr;)r.r$r>r?Zexer@rOÚlen)rAZ ps_pathnameZpsutil_pathnameZadjusted_ps_pathnamer)r)r*Útest_exes zTestProcess.test_execCs0t|jƒ}d t |j¡ ¡¡}| ||¡dS)Nú )r0r$Újoinr>r?rgr@)rAZ ps_cmdlineZpsutil_cmdliner)r)r*Ú test_cmdlines zTestProcess.test_cmdlineznot reliable on SUNOSznot reliable on AIXcCs(td|jƒ}t ¡ ¡}| ||¡dS)NÚnice)rr$r>r?ryr@)rAZps_niceZ psutil_nicer)r)r*Ú test_nice"s  zTestProcess.test_niceN)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú classmethodr:r<rBrFrGrJrQrrrWrXrarhrkrmÚunittestÚskipIfrrrsrurxrrrzr)r)r)r*r4†s8           r4c@s°eZdZdZeƒdd„ƒZe ed¡e e dƒ d¡e e d¡dd „ƒƒƒZ e e o^e  ¡ d ¡eƒd d „ƒƒZd d„Zdd„Zdd„Zdd„Ze ed¡eƒdd„ƒƒZdS)ÚTestSystemAPIszTest some system APIs.cs~ttdƒƒ‰t ¡‰ts$tr0dˆkr0ˆ dd¡tˆƒtˆƒdkrz‡fdd„ˆDƒ‡fdd„ˆDƒ}| dt |ƒ¡‚dS)Nr$rrcsg|]}|ˆkr|‘qSr)r)©Ú.0Úx)Úpids_psr)r*Ú ;sz,TestSystemAPIs.test_pids..csg|]}|ˆkr|‘qSr)r)rƒ)Ú pids_psutilr)r*r‡<sz difference: ) Úsortedrr>ZpidsrrÚinsertrtÚfailr)rAÚ differencer))r†rˆr*Ú test_pids.s  ÿzTestSystemAPIs.test_pidszunreliable on SUNOSZifconfigzno ifconfig cmdz not supportedcCsPtdƒ}tjdd ¡D]2}| ¡D]}| |¡r$qq$| d||f¡‚qdS)Nz ifconfig -aT)Zpernicz/couldn't find %s nic in 'ifconfig -a' output %s)rr>Znet_io_countersÚkeysr+Ú startswithr‹)rAr&Znicr(r)r)r*Útest_nic_namesAs  ÿÿzTestSystemAPIs.test_nic_nameszunreliable on CIcCs†tdƒ}| ¡s| d¡‚| d¡}dd„|Dƒ}dd„|Dƒ}| t|ƒtt ¡ƒ¡t ¡D] }| |j |¡| |j |¡q`dS)NZwhozno users on this systemÚ cSsg|]}| ¡d‘qS)r©r+rƒr)r)r*r‡Vsz-TestSystemAPIs.test_users..cSsg|]}| ¡d‘qS)rr’rƒr)r)r*r‡Ws) rr ZskipTestr+r@rtr>Úusersrrr^Zterminal)rAÚoutÚlinesr“Z terminalsÚur)r)r*Ú test_usersOs   zTestSystemAPIs.test_usersc CsFtjdttjdƒd&}| ttjjt   ¡¡|j s8t ‚W5QRXdS)Nzpsutil._psposix.os.killrYrK) r rLÚOSErrorÚerrnoÚEBADFrlr>Ú_psposixZ pid_existsrZÚgetpidrNrO©rAÚmr)r)r*Útest_pid_exists_let_raise]s  ÿz(TestSystemAPIs.test_pid_exists_let_raisec CsFtjdttjdƒd&}| ttjjt   ¡¡|j s8t ‚W5QRXdS)Núpsutil._psposix.os.waitpidrYrK) r rLr˜r™ršrlr>r›Úwait_pidrZrœrNrOrr)r)r*Útest_os_waitpid_let_raisefs  ÿz(TestSystemAPIs.test_os_waitpid_let_raisec CsNtjdttjdƒd.}|jtjjtjj t   ¡dd|j s@t ‚W5QRXdS)Nr rYrKg{®Gáz„?)Útimeout)r rLr˜r™ZEINTRrlr>r›ÚTimeoutExpiredr¡rZrœrNrOrr)r)r*Útest_os_waitpid_eintrns ÿþz$TestSystemAPIs.test_os_waitpid_eintrc Cs>tjddd&}| ttjjt ¡¡|j s0t ‚W5QRXdS)Nr )réÿÿÿÿre) r rLrlr"r>r›r¡rZrœrNrOrr)r)r*Útest_os_waitpid_bad_ret_statuswsÿÿz-TestSystemAPIs.test_os_waitpid_bad_ret_statuszunreliable on AIXc Csàdd„}d}tjddD]Â}t |j¡}z||jƒ\}}}}WnRtk r} z4t| ƒ ¡} d| kstd| kstd| kr~WY¢ qn‚W5d} ~ XYqX|j|j ||d |j|j ||d |j|j ||d |j|j |d d qdS) NcSsvtd|ƒ ¡}| d¡d}| ¡}t|dƒd}t|dƒd}t|dƒd}t|d dd ¡ƒ}||||fS) Nzdf -k %sr‘rrSéééú%rY)rr r+r!ÚfloatÚreplace)Údevicer”r(ÚfieldsÚtotalÚusedÚfreeÚpercentr)r)r*Údfƒsz*TestSystemAPIs.test_disk_usage..dfi@F)Úallzno such file or directoryzraw devices not supportedzpermission denied)Údeltar) r>Zdisk_partitionsÚ disk_usageZ mountpointr®Ú RuntimeErrorrr]ZassertAlmostEqualr°r±r²r³) rAr´Z toleranceÚpartÚusager°r±r²r³Úerrr)r)r*Útest_disk_usage€s&   ÿþ zTestSystemAPIs.test_disk_usageN)r{r|r}r~rrr€rrrr rr r>r“r—rŸr¢r¥r§rr¼r)r)r)r*r‚*s"        r‚c@seZdZdd„ZdS)ÚTestMisccCs4tƒ}| |d¡| |t ¡¡| |tj¡dS)Nr)rZ assertGreaterr@ÚresourceÚmmapZPAGESIZE)rAZpagesizer)r)r*Útest_getpagesize§s zTestMisc.test_getpagesizeN)r{r|r}rÀr)r)r)r*r½¤sr½Ú__main__)Ú run_from_name)N)-r~rnr™rZr_r6rTr€r>rrrrrrrZ psutil.testsr r r r r rrrrrrr¿r¾Zpsutil._psutil_posixrrr.r0r2r3rr4r‚r½r{Zpsutil.tests.runnerrÂÚ__file__r)r)r)r*ÚsZ                    < $ y