ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ U xeº_T8ã@s~dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlmZddlm Z ddlmZddlmZdd lmZdd lmZddlmZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddlm%Z%ddlm&Z&ddlm'Z'dd lm(Z(dd!lm)Z)dd"lm*Z*dd#lm+Z+ddlZddlZGd$d%„d%eƒZ,Gd&d'„d'eƒZ-Gd(d)„d)eƒZ.Gd*d+„d+eƒZ/Gd,d-„d-eƒZ0e#Gd.d/„d/e'ƒƒZ1Gd0d1„d1eƒZ2Gd2d3„d3eƒZ3e4d4krzdd5l5m6Z6e6e7ƒdS)6z3 Tests for testing utils (psutil.tests namespace). éN)ÚFREEBSD)ÚNETBSD)ÚPOSIX)Úopen_binary)Ú open_text)Ú supports_ipv6)Úbind_socket)Úbind_unix_socket)Ú call_until)Úchdir)Ú CI_TESTING)Úcreate_sockets)Ú get_free_port)ÚHAS_CONNECTIONS_UNIX)Ú is_namedtuple)Úmock)Úprocess_namespace)ÚPsutilTestCase)Ú PYTHON_EXE)Ú reap_children)Úretry)Úretry_on_failure)Ú safe_mkdir)Úsafe_rmpath)Ú serialrun)Úsystem_namespace)Útcp_socketpair)Ú terminate)ÚTestMemoryLeak)Úunittest)Úunix_socketpair)Ú wait_for_file)Úwait_for_pidc@sxeZdZe d¡dd„ƒZe d¡dd„ƒZe d¡dd„ƒZe d¡dd „ƒZe d¡d d„ƒZ e d¡dd „ƒZ dS)ÚTestRetryDecoratorz time.sleepcsFtdddd‡fdd„ƒ}ttdƒƒ‰| |ƒd¡| |jd¡dS)Néé©ÚretriesÚintervalZlogfuncsˆrˆ ¡ddqdS©Nr%r©Úpop©©Úqueuer,úN/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_testutils.pyÚfooCs z2TestRetryDecorator.test_retry_success..fooé)rÚlistÚrangeÚassertEqualÚ call_count©ÚselfÚsleepr0r,r-r/Útest_retry_success?s z%TestRetryDecorator.test_retry_successcsDtdddd‡fdd„ƒ}ttdƒƒ‰| t|¡| |jd¡dS)Nr$r%r&csˆrˆ ¡ddqdSr)r*r,r-r,r/r0Qs z2TestRetryDecorator.test_retry_failure..fooé)rr2r3ÚassertRaisesÚZeroDivisionErrorr4r5r6r,r-r/Útest_retry_failureNs z%TestRetryDecorator.test_retry_failurecCs2ttdddd„ƒ}| t|¡| |jd¡dS)Nr%)Ú exceptionr(cSst‚dS©N)Ú TypeErrorr,r,r,r/r0^sz2TestRetryDecorator.test_exception_arg..foor)rÚ ValueErrorr;r@r4r5r6r,r,r/Útest_exception_arg\s z%TestRetryDecorator.test_exception_argcCs4tdddddd„ƒ}| t|¡| |jd¡dS)Nr$r&cSsdddSr)r,r,r,r,r/r0isz4TestRetryDecorator.test_no_interval_arg..foor©rr;r<r4r5r6r,r,r/Útest_no_interval_arges z'TestRetryDecorator.test_no_interval_argcCs4tdddddd„ƒ}| t|¡| |jd¡dS)Nr$r%r&cSsdddSr)r,r,r,r,r/r0ssz0TestRetryDecorator.test_retries_arg..foorCr6r,r,r/Útest_retries_argps z#TestRetryDecorator.test_retries_argcCs|jttddddS)Nr$r%)r'Útimeout)r;rAr)r7r8r,r,r/Útest_retries_and_timeout_argszsz0TestRetryDecorator.test_retries_and_timeout_argsN)Ú__name__Ú __module__Ú__qualname__rÚpatchr9r=rBrDrErGr,r,r,r/r#=s r#c@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)ÚTestSyncTestUtilsc CsPtt ¡ƒtt ¡ƒd}tjdtdgƒd| tj t|¡W5QRXdS)NiŸ†úpsutil.tests.retry.__iter__r©Zreturn_value)r"ÚosÚgetpidÚmaxÚpsutilZpidsrrKÚiterr;Ú NoSuchProcess)r7Znopidr,r,r/Útest_wait_for_pidsz#TestSyncTestUtils.test_wait_for_pidc CsD| ¡}t|dƒ}| d¡W5QRXt|ƒtj |¡r@t‚dS)NÚwr0©Ú get_testfnÚopenÚwriter!rOÚpathÚexistsÚAssertionError©r7ÚtestfnÚfr,r,r/Útest_wait_for_file‡s z$TestSyncTestUtils.test_wait_for_filec Cs>| ¡}t|dƒW5QRXt|ddtj |¡r:t‚dS)NrVT)Úempty)rXrYr!rOr[r\r]©r7r_r,r,r/Útest_wait_for_file_emptyŽs z*TestSyncTestUtils.test_wait_for_file_emptyc Cs:| ¡}tjdtdgƒd| tt|¡W5QRXdS)NrMrrN)rXrrKrSr;ÚIOErrorr!rcr,r,r/Útest_wait_for_file_no_file•sz,TestSyncTestUtils.test_wait_for_file_no_filec CsH| ¡}t|dƒ}| d¡W5QRXt|ddtj |¡sDt‚dS)NrVr0F)ÚdeleterWr^r,r,r/Útest_wait_for_file_no_deletešs z.TestSyncTestUtils.test_wait_for_file_no_deletecCstdd„dƒ}| |d¡dS)NcSsdS)Nr%r,r,r,r,r/Ú¢óz3TestSyncTestUtils.test_call_until..zret == 1r%)r r4)r7Úretr,r,r/Útest_call_until¡sz!TestSyncTestUtils.test_call_untilN) rHrIrJrUrardrfrhrlr,r,r,r/rLsrLc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „ZdS)ÚTestFSTestUtilsc Cs&ttƒ}| |jd¡W5QRXdS)NÚrt)rÚ__file__r4Úmode©r7r`r,r,r/Útest_open_text¨s zTestFSTestUtils.test_open_textc Cs&ttƒ}| |jd¡W5QRXdS)NÚrb)rror4rprqr,r,r/Útest_open_binary¬s z TestFSTestUtils.test_open_binarycCs<| ¡}t|ƒtj |¡s t‚t|ƒtj |¡s8t‚dSr?)rXrrOr[Úisdirr]rcr,r,r/Útest_safe_mkdir°s zTestFSTestUtils.test_safe_mkdirc Cs¦| ¡}t|dƒ ¡t|ƒtj |¡r.t‚t|ƒt |¡t|ƒtj |¡rXt‚t j dttj dƒd.}| t¡t|ƒW5QRX|js˜t‚W5QRXdS)NrVzpsutil.tests.os.statÚ)Úside_effect)rXrYÚcloserrOr[r\r]ÚmkdirrrKÚOSErrorÚerrnoÚEINVALr;Úcalled)r7r_Úmr,r,r/Útest_safe_rmpath·s ÿz TestFSTestUtils.test_safe_rmpathc Cs\| ¡}t ¡}t |¡t|ƒ | t ¡tj ||¡¡W5QRX| t ¡|¡dSr?)rXrOÚgetcwdrzrr4r[Újoin)r7r_Úbaser,r,r/Ú test_chdirÊs $zTestFSTestUtils.test_chdirN)rHrIrJrrrtrvr€r„r,r,r,r/rm¦s rmc@s:eZdZdd„Zdd„Ze ed¡dd„ƒZdd „Z d S)ÚTestProcessUtilscCsN| ¡}t |j¡}| ¡s t‚tƒ| ¡r2t‚tjjr>t‚tjj rJt‚dSr?) Úspawn_testprocrRÚProcessÚpidÚ is_runningr]rÚtestsZ _pids_startedZ_subprocesses_started)r7ZsubpÚpr,r,r/Útest_reap_childrenÕsz#TestProcessUtils.test_reap_childrencCsä| ¡\}}| |j|j¡| ¡s(t‚| ¡s4t‚t ¡ ¡}| ||g¡t ¡jdd}| t |ƒd¡| ||¡| ||¡| | ¡t ¡¡| | ¡|j¡t|ƒ| ¡rÀt‚| ¡sÌt‚t|ƒ| ¡ràt‚dS)NT)Ú recursiveé)Zspawn_children_pairÚassertNotEqualrˆr‰r]rRr‡Úchildrenr4ÚlenZassertInÚppidrOrPr)r7ÚchildZ grandchildrr,r,r/Útest_spawn_children_pairÞs"z)TestProcessUtils.test_spawn_children_pairú POSIX onlycCs"| ¡\}}| | ¡tj¡dSr?)Úspawn_zombier4ÚstatusrRÚ STATUS_ZOMBIE)r7ÚparentÚzombier,r,r/Útest_spawn_zombieósz"TestProcessUtils.test_spawn_zombiecCsà| ¡}t|ƒ| |¡t|ƒt | ¡j¡}t|ƒ| |¡t|ƒtddg}tj|tj tj d}t|ƒ| |¡t|ƒ| ¡j}t|ƒ| |¡t|ƒt rÜ| ¡\}}t|ƒt|ƒ| |¡| |¡dS)Nz-czimport time; time.sleep(60);)ÚstdoutÚstderr)r†rZassertProcessGonerRr‡rˆrÚPopenÚ subprocessÚPIPErr–)r7r‹Úcmdrˆr™ršr,r,r/Útest_terminateøs. zTestProcessUtils.test_terminateN) rHrIrJrŒr”rÚskipIfrr›r¢r,r,r,r/r…Ós r…c@s`eZdZdd„Ze ed¡dd„ƒZdd„Ze ed¡e e pDe d¡d d „ƒƒZdd„Zd S)ÚTestNetUtilsc Cs>tƒ}t td|fd¡}| | ¡d|¡W5QRXdS)Nrw©Úaddrr%)rÚ contextlibÚclosingrr4Úgetsockname)r7ÚportÚsr,r,r/rszTestNetUtils.bind_socketr•c Cs¼| ¡}t|ƒ}t |¡\| |jtj¡| |jtj ¡| | ¡|¡tj |¡s\t‚t t |¡j¡srt‚W5QRX| ¡}t|tjd}t |¡| |jtj¡W5QRXdS)N)Útype)rXr r§r¨r4ÚfamilyÚsocketÚAF_UNIXr¬ÚSOCK_STREAMr©rOr[r\r]ÚstatÚS_ISSOCKÚst_modeÚ SOCK_DGRAM)r7ÚnameÚsockr,r,r/Útest_bind_unix_sockets z"TestNetUtils.test_bind_unix_socketc Cs|dtƒf}ttj|d\}}t |¡Lt |¡6| | ¡|¡| | ¡|¡| | ¡|¡W5QRXW5QRXdS)Nz 127.0.0.1r¥) rrr®ÚAF_INETr§r¨r4r©Úgetpeernamer)r7r¦ÚserverÚclientr,r,r/Útcp_tcp_socketpair.s zTestNetUtils.tcp_tcp_socketpairz*/var/run/log UNIX socket opened by defaultcCsÂt ¡}| ¡}|jddr t‚| ¡}t|ƒ\}}zvtj |¡sFt‚t t |¡j ¡s\t‚| | ¡|d¡| t|jddƒd¡| | ¡|¡| | ¡|¡W5| ¡| ¡XdS)NÚunix)ÚkindrŽ)rRr‡Únum_fdsZconnectionsr]rXr ryrOr[r\r±r²r³r4r‘r©r¹)r7r‹r¿rµrºr»r,r,r/Útest_unix_socketpair9sz!TestNetUtils.test_unix_socketpairc CsÊtƒº}t t¡}t t¡}|D]2}||jd7<|| tjtj¡d7<q | |tj d¡tƒr~| |tjd¡t r˜tr˜| |tjd¡| |tjd¡| |tjd¡W5QRXdS)Nr%rŽ)r ÚcollectionsÚdefaultdictÚintrÚ getsockoptr®Ú SOL_SOCKETÚSO_TYPEZassertGreaterEqualr¸rÚAF_INET6rrr¯r°r´)r7ZsocksZfamsÚtypesr«r,r,r/Útest_create_socketsMs z TestNetUtils.test_create_socketsN) rHrIrJrrr£rr·r¼rrrÀrÉr,r,r,r/r¤s ÿr¤c@sNeZdZdd„Zdd„Zeƒe ed¡dd„ƒƒZ dd „Z d d„Zdd „ZdS)ÚTestMemLeakClasscs8‡fdd„}ddi‰|j|ddd| ˆdd¡dS) Ncsˆdd7<dS)NÚcntr%r,r,©rËr,r/Úfunbsz(TestMemLeakClass.test_times..funrËré é)ÚtimesÚwarmup_timesé)Úexecuter4©r7rÍr,rÌr/Ú test_timesaszTestMemLeakClass.test_timescCs||jt|jdd„dd|jt|jdd„dd|jt|jdd„dd|jt|jd d„dd |jt|jdd„dddS) NcSsdS©Nrr,r,r,r,r/riirjz1TestMemLeakClass.test_param_err..r©rÐcSsdSrÖr,r,r,r,r/rijrjéÿÿÿÿcSsdSrÖr,r,r,r,r/rikrj)rÑcSsdSrÖr,r,r,r,r/rilrj)Ú tolerancecSsdSrÖr,r,r,r,r/rimrj)r')r;rArÓ©r7r,r,r/Útest_param_errhs zTestMemLeakClass.test_param_errz skipped on CIcCs4g}|fdd„}z|jtd|j|ddW5~XdS)NcSs| dd¡dS©NZxxxxxxxxxxxxxxxxxxxxxxxxi©Úappend©Úlsr,r,r/rÍtsz+TestMemLeakClass.test_leak_mem..funz extra-memé2r×)ÚassertRaisesRegexr]rÓ)r7ràrÍr,r,r/Ú test_leak_memosÿ zTestMemLeakClass.test_leak_memcs8‡‡fdd„}g‰trdnd}ˆ td|ˆj|¡dS)Ncs"ttƒ}ˆ |j¡ˆ |¡dSr?)rYroZ addCleanupryrÞ)r`©Zboxr7r,r/rÍsz1TestMemLeakClass.test_unclosed_files..funÚfdÚhandlez unclosed )rrâr]rÓ)r7rÍr¾r,rär/Útest_unclosed_files~sÿz$TestMemLeakClass.test_unclosed_filescs>‡fdd„}g‰d}|j||ddd| tˆƒ|d¡dS)Ncsˆ dd¡dSrÜrÝr,rßr,r/rÍŠsz,TestMemLeakClass.test_tolerance..funédri€)rÐrÑrÙr%)rÓr4r‘)r7rÍrÐr,rßr/Útest_tolerance‰s ÿzTestMemLeakClass.test_tolerancec Csddd„}| t|¡| t¡| t|¡W5QRXdd„}| t¡| t|¡W5QRXdS)NcSsdddSr)r,r,r,r,r/rÍ“sz0TestMemLeakClass.test_execute_w_exc..funcSsdSr?r,r,r,r,r/rÍ™s)Z execute_w_excr<r;r{r]rÔr,r,r/Útest_execute_w_exc’sz#TestMemLeakClass.test_execute_w_excN) rHrIrJrÕrÛrrr£rrãrçrérêr,r,r,r/rÊ^s rÊc@seZdZdd„Zdd„ZdS)ÚTestTestingUtilscCsLt ¡}t|ƒ}| ¡dd„| |j¡Dƒdd}| |ƒ| ¡¡dS)NcSsg|]}|ddkr|‘qS)r%r’r,©Ú.0Úxr,r,r/Ú ¥sz;TestTestingUtils.test_process_namespace..r)rRr‡rÚtestrSÚgettersr4r’)r7r‹ÚnsrÍr,r,r/Útest_process_namespace¡s z'TestTestingUtils.test_process_namespacecCs:tƒ}dd„| |j¡Dƒdd}| |ƒt ¡¡dS)NcSsg|]}|ddkr|‘qS)r%Únet_if_addrsr,rìr,r,r/rïªsz:TestTestingUtils.test_system_namespace..r)rrSrñr4rRrô)r7ròrÍr,r,r/Útest_system_namespace¨sz&TestTestingUtils.test_system_namespaceN)rHrIrJrórõr,r,r,r/rëŸsrëc@seZdZdd„ZdS)ÚTestOtherUtilscCs.tt dd¡dddƒƒst‚ttƒƒr*t‚dS)Nr0za b cr%rŽr1)rrÁÚ namedtupler]ÚtuplerÚr,r,r/Útest_is_namedtuple°sz!TestOtherUtils.test_is_namedtupleN)rHrIrJrùr,r,r,r/rö®sröÚ__main__)Ú run_from_name)8Ú__doc__rÁr§r|rOr®r±rŸrRrrrZpsutil._commonrrrZpsutil.testsrr r rrr rrrrrrrrrrrrrrrrrrr r!r"r#rLrmr…r¤rÊrërörHZpsutil.tests.runnerrûror,r,r,r/ÚslB'-DG@