U
Ã|ýf†: ã @ sž d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z ddl m Z ddl m Z ddl m
Z
ddlmZ ddlmZ ddlmZ dd l
mZ dd
l
mZ dd l
mZ dd l
mZ dd
l
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
mZ ddl
m Z ddl
m Z ddl
m Z ddl
m Z ddl
m Z ddl
m!Z! ddl
m"Z" ddl
m#Z# ddl
m$Z$ dd l
m%Z% dd l
m&Z& dd l
m'Z' dd l
m(Z( dd l
m)Z) dd!l
m*Z* dd"l
m+Z+ dd#l
m,Z, dd$l
m-Z- dd%l
m.Z. G d&d'„ d'eƒZ/G d(d)„ d)eƒZ0G d*d+„ d+eƒZ1G d,d-„ d-eƒZ2G d.d/„ d/eƒZ3e(G d0d1„ d1eƒƒZ4G d2d3„ d3eƒZ5G d4d5„ d5eƒZ6e7d6kršdd7l8m9Z9 e9e:ƒ dS )8z1Tests for testing utils (psutil.tests namespace).é N)ÚFREEBSD)ÚNETBSD)ÚPOSIX)Ú open_binary)Ú open_text)Ú
supports_ipv6)Ú
CI_TESTING)ÚCOVERAGE)ÚHAS_NET_CONNECTIONS_UNIX)Ú
PYTHON_EXE)ÚPYTHON_EXE_ENV)ÚPsutilTestCase)ÚTestMemoryLeak)Ú bind_socket)Úbind_unix_socket)Ú
call_until)Úchdir)Úcreate_sockets)Úfilter_proc_net_connections)Ú
get_free_port)Ú
is_namedtuple)Úmock)Úprocess_namespace)Ú
reap_children)Úretry)Úretry_on_failure)Ú
safe_mkdir)Ú safe_rmpath)Ú serialrun)Úsystem_namespace)Útcp_socketpair)Ú terminate)Úunix_socketpair)Ú
wait_for_file)Ú wait_for_pidc @ sx e Zd Ze d¡dd„ ƒZe d¡dd„ ƒZe d¡dd„ ƒZe d¡dd „ ƒZe d¡d
d „ ƒZ e d¡d d
„ ƒZ
dS )ÚTestRetryDecoratorz
time.sleepc sF t ddd d‡ fdd„ƒ}ttdƒƒ‰ | |ƒ d¡ | |jd¡ d S )Né é ©ÚretriesÚintervalZlogfunc s ˆ rˆ ¡ dd q dS ©Nr' r ©Úpop© ©Úqueuer. úN/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_testutils.pyÚfooD s
z2TestRetryDecorator.test_retry_success..fooé )r ÚlistÚrangeÚ assertEqualÚ
call_count©ÚselfÚsleepr2 r. r/ r1 Útest_retry_success@ s
z%TestRetryDecorator.test_retry_successc sD t ddd d‡ fdd„ƒ}ttdƒƒ‰ | t|¡ | |jd¡ d S )Nr& r' r( c s ˆ rˆ ¡ dd q dS r+ r, r. r/ r. r1 r2 R s
z2TestRetryDecorator.test_retry_failure..fooé )r r4 r5 Ú assertRaisesÚZeroDivisionErrorr6 r7 r8 r. r/ r1 Útest_retry_failureO s
z%TestRetryDecorator.test_retry_failurec C s2 t tdddd„ ƒ}| t|¡ | |jd¡ d S )Nr' )Ú exceptionr* c S s t ‚d S ©N)Ú TypeErrorr. r. r. r1 r2 _ s z2TestRetryDecorator.test_exception_arg..foor )r Ú
ValueErrorr= rB r6 r7 r8 r. r. r1 Útest_exception_arg] s
z%TestRetryDecorator.test_exception_argc C s4 t dd d ddd„ ƒ}| t|¡ | |jd¡ d S )Nr& r( c S s dd d S r+ r. r. r. r. r1 r2 j s z4TestRetryDecorator.test_no_interval_arg..foor ©r r= r> r6 r7 r8 r. r. r1 Útest_no_interval_argf s
z'TestRetryDecorator.test_no_interval_argc C s4 t ddd ddd„ ƒ}| t|¡ | |jd¡ d S )Nr& r' r( c S s dd d S r+ r. r. r. r. r1 r2 s s z0TestRetryDecorator.test_retries_arg..foorE r8 r. r. r1 Útest_retries_argq s
z#TestRetryDecorator.test_retries_argc C s | j ttddd d S )Nr& r' )r) Útimeout)r= rC r )r9 r: r. r. r1 Ú test_retries_and_timeout_argsz s z0TestRetryDecorator.test_retries_and_timeout_argsN) Ú__name__Ú
__module__Ú __qualname__r Úpatchr; r? rD rF rG rI r. r. r. r1 r% ? s
r% c @ s< e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
S )ÚTestSyncTestUtilsc C sP t t ¡ ƒ tt ¡ ƒd }tjdtdgƒd | tj
t |¡ W 5 Q R X d S )NiŸ† úpsutil.tests.retry.__iter__r ©Z return_value) r$ ÚosÚgetpidÚmaxÚpsutilZpidsr rM Úiterr= Z
NoSuchProcess)r9 Znopidr. r. r1 Útest_wait_for_pid€ s z#TestSyncTestUtils.test_wait_for_pidc C sD | ¡ }t|dƒ}| d¡ W 5 Q R X t|ƒ tj |¡r@t‚d S )NÚwr2 ©Ú
get_testfnÚopenÚwriter# rQ ÚpathÚexistsÚAssertionError©r9 ÚtestfnÚfr. r. r1 Útest_wait_for_file† s
z$TestSyncTestUtils.test_wait_for_filec C s> | ¡ }t|dƒ W 5 Q R X t|dd tj |¡r:t‚d S )NrW T)Úempty)rY rZ r# rQ r\ r] r^ ©r9 r` r. r. r1 Útest_wait_for_file_empty s
z*TestSyncTestUtils.test_wait_for_file_emptyc C s: | ¡ }tjdtdgƒd | tt|¡ W 5 Q R X d S )NrO r rP )rY r rM rU r= ÚIOErrorr# rd r. r. r1 Útest_wait_for_file_no_file” s z,TestSyncTestUtils.test_wait_for_file_no_filec C sH | ¡ }t|dƒ}| d¡ W 5 Q R X t|dd tj |¡sDt‚d S )NrW r2 F)ÚdeleterX r_ r. r. r1 Ú test_wait_for_file_no_delete™ s
z.TestSyncTestUtils.test_wait_for_file_no_deletec C s t dd„ dƒ}| |d¡ d S )Nc S s dS )Nr' r. r. r. r. r1 Ú¡ ó z3TestSyncTestUtils.test_call_until..zret == 1r' )r r6 )r9 Úretr. r. r1 Útest_call_until s z!TestSyncTestUtils.test_call_untilN) rJ rK rL rV rb re rg ri rm r. r. r. r1 rN s rN c @ s4 e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd S ) ÚTestFSTestUtilsc C s& t tƒ}| |jd¡ W 5 Q R X d S )NÚr)r Ú__file__r6 Úmode©r9 ra r. r. r1 Útest_open_text¦ s
z TestFSTestUtils.test_open_textc C s& t tƒ}| |jd¡ W 5 Q R X d S )NÚrb)r rp r6 rq rr r. r. r1 Útest_open_binaryª s
z TestFSTestUtils.test_open_binaryc C s< | ¡ }t|ƒ tj |¡s t‚t|ƒ tj |¡s8t‚d S rA )rY r rQ r\ Úisdirr^ rd r. r. r1 Útest_safe_mkdir® s
z TestFSTestUtils.test_safe_mkdirc
C s¦ | ¡ }t|dƒ ¡ t|ƒ tj |¡r.t‚t|ƒ t |¡ t|ƒ tj |¡rXt‚t j
dt t j
dƒd.}| t ¡ t|ƒ W 5 Q R X |js˜t‚W 5 Q R X d S )NrW zpsutil.tests.os.statÚ )Z side_effect)rY rZ Úcloser rQ r\ r] r^ Úmkdirr rM ÚOSErrorÚerrnoÚEINVALr= Úcalled)r9 r` Úmr. r. r1 Útest_safe_rmpathµ s
ÿ z TestFSTestUtils.test_safe_rmpathc C s\ | ¡ }t ¡ }t |¡ t|ƒ | t ¡ tj ||¡¡ W 5 Q R X | t ¡ |¡ d S rA )rY rQ Úgetcwdrz r r6 r\ Újoin)r9 r` Úbaser. r. r1 Ú
test_chdirÉ s
$zTestFSTestUtils.test_chdirN)rJ rK rL rs ru rw r€ r„ r. r. r. r1 rn ¥ s
rn c @ s: e Zd Zdd„ Zdd„ Ze e d¡dd„ ƒZdd „ Z d
S ) ÚTestProcessUtilsc C sN | ¡ }t |j¡}| ¡ s t‚tƒ | ¡ r2t‚tjjr>t‚tjj rJt‚d S rA )
Úspawn_testprocrT ÚProcessÚpidÚ
is_runningr^ r ÚtestsZ
_pids_startedZ_subprocesses_started)r9 ZsubpÚpr. r. r1 Útest_reap_childrenÓ s z#TestProcessUtils.test_reap_childrenc C sä | ¡ \}}| |j|j¡ | ¡ s(t‚| ¡ s4t‚t ¡ ¡ }| ||g¡ t ¡ jdd}| t |ƒd¡ |
||¡ |
||¡ | | ¡ t
¡ ¡ | | ¡ |j¡ t|ƒ | ¡ rÀt‚| ¡ sÌt‚t|ƒ | ¡ ràt‚d S )NT)Ú recursiveé )Zspawn_children_pairÚassertNotEqualrˆ r‰ r^ rT r‡ Úchildrenr6 ÚlenZassertInÚppidrQ rR r! )r9 ÚchildZ
grandchildr r. r. r1 Útest_spawn_children_pairÜ s" z)TestProcessUtils.test_spawn_children_pairú
POSIX onlyc C s" | ¡ \}}| | ¡ tj¡ d S rA )Ú spawn_zombier6 ÚstatusrT Z
STATUS_ZOMBIE)r9 Ú_parentÚzombier. r. r1 Útest_spawn_zombieñ s z"TestProcessUtils.test_spawn_zombiec C sî | ¡ }t|ƒ | |j¡ t|ƒ t | ¡ j¡}t|ƒ | |j¡ t|ƒ tddg}tj|tj tj t
d}t|ƒ | |j¡ t|ƒ | ¡ j}t|ƒ | |j¡ t|ƒ t rê| ¡ \}}t|ƒ t|ƒ | |j¡ | |j¡ d S )Nz-cz3import time; [time.sleep(0.1) for x in range(100)];)ÚstdoutÚstderrÚenv)
r† r! Z
assertPidGonerˆ rT r‡ r ÚPopenÚ
subprocessÚPIPEr r r– )r9 r‹ Úcmdrˆ Úparentr™ r. r. r1 Útest_terminateö s> ýü
z TestProcessUtils.test_terminateN)
rJ rK rL rŒ r” ÚunittestÚskipIfr rš r£ r. r. r. r1 r… Ò s
r… c @ s` e Zd Zdd„ Ze e d¡dd„ ƒZdd„ Ze e d¡e e pDe
d¡d d
„ ƒƒZ d d „ Z d
S )Ú TestNetUtilsc C s> t ƒ }t td|fd¡}| | ¡ d |¡ W 5 Q R X d S )Nrx ©Úaddrr' )r Ú
contextlibÚclosingr r6 Ú getsockname)r9 ÚportÚsr. r. r1 r s zTestNetUtils.bind_socketr• c C s¼ | ¡ }t|ƒ}t |¡\ | |jtj¡ | |jtj ¡ | |
¡ |¡ t j
|¡s\t‚t t |¡j¡srt‚W 5 Q R X | ¡ }t|tjd}t |¡ | |jtj¡ W 5 Q R X d S )N)Útype)rY r r© rª r6 ÚfamilyÚsocketÚAF_UNIXr® Ú SOCK_STREAMr« rQ r\ r] r^ ÚstatÚS_ISSOCKÚst_modeÚ
SOCK_DGRAM)r9 ÚnameÚsockr. r. r1 Útest_bind_unix_socket$ s z"TestNetUtils.test_bind_unix_socketc
C s| dt ƒ f}ttj|d\}}t |¡L t |¡6 | | ¡ |¡ | | ¡ |¡ | | ¡ |¡ W 5 Q R X W 5 Q R X d S )Nz 127.0.0.1r§ )
r r r° ÚAF_INETr© rª r6 r« Ú getpeernamer )r9 r¨ ÚserverÚclientr. r. r1 Útcp_tcp_socketpair4 s
z TestNetUtils.tcp_tcp_socketpairz*/var/run/log UNIX socket opened by defaultc C sÎ t ¡ }| ¡ }| t|jddƒg ¡ | ¡ }t|ƒ\}}zzt j
|¡sNt ‚t
t
|¡j¡sdt ‚| | ¡ | d¡ | tt|jddƒƒd¡ | | ¡ |¡ | | ¡ |¡ W 5 | ¡ | ¡ X d S )NÚunix)ÚkindrŽ )rT r‡ Únum_fdsr6 r Znet_connectionsrY r" ry rQ r\ r] r^ r³ r´ rµ r‘ r« r» )r9 r‹ rÁ r· r¼ r½ r. r. r1 Útest_unix_socketpair? s, ÿ ÿüz!TestNetUtils.test_unix_socketpairc C sÊ t ƒ º}t t¡}t t¡}|D ]2}||j d7 < || tjtj¡ d7 < q | |tj
d¡ t ƒ r~| |tj d¡ t
r˜tr˜| |tj d¡ | |tj d¡ | |tj d¡ W 5 Q R X d S )Nr' rŽ )r Ú collectionsÚ defaultdictÚintr¯ Ú
getsockoptr° Ú
SOL_SOCKETÚSO_TYPEZassertGreaterEqualrº r ÚAF_INET6r r
r± r² r¶ )r9 ÚsocksZfamsÚtypesr r. r. r1 Útest_create_sockets[ s
z TestNetUtils.test_create_socketsN)
rJ rK rL r r¤ r¥ r r¹ r¾ r r r rÌ r. r. r. r1 r¦ s
ÿ r¦ c @ s` e Zd Zeƒ dd„ ƒZdd„ Zeƒ e ed¡e e d¡dd„ ƒƒƒZ
d d
„ Z d d „ Z d
d„ Z
dS )ÚTestMemLeakClassc s8 ‡ fdd„}ddi‰ | j |ddd | ˆ d d¡ d S ) Nc s ˆ d d7 < d S )NÚcntr' r. r. ©rÎ r. r1 Úfunp s z(TestMemLeakClass.test_times..funrÎ r é
é )ÚtimesÚ warmup_timesé )Úexecuter6 )r9 rÐ r. rÏ r1 Ú
test_timesn s zTestMemLeakClass.test_timesc C s| | j t| jdd„ dd | j t| jdd„ dd | j t| jdd„ dd | j t| jd d„ dd
| j t| jd d„ dd d S )
Nc S s dS ©Nr r. r. r. r. r1 rj x rk z1TestMemLeakClass.test_param_err..r ©rÓ c S s dS rØ r. r. r. r. r1 rj y rk éÿÿÿÿc S s dS rØ r. r. r. r. r1 rj z rk )rÔ c S s dS rØ r. r. r. r. r1 rj { rk )Ú tolerancec S s dS rØ r. r. r. r. r1 rj | rk )r) )r= rC rÖ ©r9 r. r. r1 Útest_param_errw s
z TestMemLeakClass.test_param_errz
skipped on CIz skipped during test coveragec C s4 g }|fdd„}z| j td| j|dd W 5 ~X d S )Nc S s | dd ¡ d S )NZ|xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxé ©Úappend©Úlsr. r. r1 rÐ „ s z+TestMemLeakClass.test_leak_mem..funz extra-memé2 rÙ )ÚassertRaisesRegexr^ rÖ )r9 râ rÐ r. r. r1 Ú
test_leak_mem~ s ÿ
z TestMemLeakClass.test_leak_memc s8 ‡ ‡fdd„}g ‰ t rdnd}ˆ td| ˆj|¡ d S )Nc s" t tƒ} ˆ | j¡ ˆ | ¡ d S rA )rZ rp Z
addCleanupry rà )ra ©Úboxr9 r. r1 rÐ s z1TestMemLeakClass.test_unclosed_files..funÚfdÚhandlez unclosed )r rä r^ rÖ )r9 rÐ rÀ r. ræ r1 Útest_unclosed_files s ÿz$TestMemLeakClass.test_unclosed_filesc s> ‡ fdd„}g ‰ d}| j ||ddd | tˆ ƒ|d ¡ d S )Nc s ˆ dd ¡ d S )NZxxxxxxxxxxxxxxxxxxxxxxxxrÞ rß r. rá r. r1 rÐ œ s z,TestMemLeakClass.test_tolerance..funéd r i € )rÓ rÔ rÛ r' )rÖ r6 r‘ )r9 rÐ rÓ r. rá r1 Útest_tolerance› s ÿz TestMemLeakClass.test_tolerancec C sd dd„ }| t|¡ | t¡ | t|¡ W 5 Q R X dd„ }| t¡ | t|¡ W 5 Q R X d S )Nc S s dd d S r+ r. r. r. r. r1 Úfun_1§ s z2TestMemLeakClass.test_execute_w_exc..fun_1c S s d S rA r. r. r. r. r1 Úfun_2® s z2TestMemLeakClass.test_execute_w_exc..fun_2)Z
execute_w_excr> r= r{ r^ )r9 rí rî r. r. r1 Útest_execute_w_exc¦ s z#TestMemLeakClass.test_execute_w_excN)rJ rK rL r r× rÝ r¤ r¥ r r rå rê rì rï r. r. r. r1 rÍ l s
rÍ c @ s e Zd Zdd„ Zdd„ ZdS )ÚTestTestingUtilsc C sL t ¡ }t|ƒ}| ¡ dd„ | |j¡D ƒd d }| |ƒ | ¡ ¡ d S )Nc S s g | ]}|d dkr|‘qS )r' r’ r. ©Ú.0Úxr. r. r1 Ú
º s z;TestTestingUtils.test_process_namespace..r )rT r‡ r ÚtestrU Úgettersr6 r’ )r9 r‹ ÚnsrÐ r. r. r1 Útest_process_namespace¶ s
z'TestTestingUtils.test_process_namespacec C s: t ƒ }dd„ | |j¡D ƒd d }| |ƒ t ¡ ¡ d S )Nc S s g | ]}|d dkr|‘qS )r' Ú net_if_addrsr. rñ r. r. r1 rô ¿ s z:TestTestingUtils.test_system_namespace..r )r rU rö r6 rT rù )r9 r÷ rÐ r. r. r1 Útest_system_namespace½ s z&TestTestingUtils.test_system_namespaceN)rJ rK rL rø rú r. r. r. r1 rð µ s rð c @ s e Zd Zdd„ ZdS )ÚTestOtherUtilsc C s. t t dd¡dddƒƒs t‚t tƒ ƒr*t‚d S )Nr2 za b cr' rŽ r3 )r rà Ú
namedtupler^ ÚtuplerÜ r. r. r1 Útest_is_namedtupleÄ s z!TestOtherUtils.test_is_namedtupleN)rJ rK rL rþ r. r. r. r1 rû Ã s rû Ú__main__)Ú
run_from_name);Ú__doc__rà r© r| rQ r° r³ rŸ r¤ rT Z psutil.testsr r r Zpsutil._commonr r r r r r
r r r
r r r r r r r r r r r r r r r r r r r r! r" r# r$ r% rN rn r… r¦ rÍ rð rû rJ Zpsutil.tests.runnerr rp r. r. r. r1 Ú sr @&-LNH