U
|
Z ±d-O ã @ sf d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl
|
Z
|
d dl m Z ddl mZmZmZmZmZmZmZmZmZ ddlmZmZ ddlmZ dd l m Z m Z e j d
|
kZ e rÐd d lm!Z!m"Z"m#Z# n
|
d \Z!Z"Z#d d Z$e$dZ%e$dZ&e$dZ'e r
|
dd Z(ndd Z(dd Z)e dd Z*e dd Z+e
|
j,j-de*e+gd d gd Z.e.d d Z/e.d!d" Z0e.d#d$ Z1d%d& Z2d'd( Z3e$d)Z4e.d*d+ Z5e.d,d- Z6d.d/ Z7d0d1 Z8ed2d3 Z9e.d4d5 Z:d6d7 Z;e.d8d9 Z<e
|
j,j=e d:d;e.d<d= Z>ed>d? Z?d@dA Z@dBdC ZAe
|
j,j=e dDd;dEdF ZBdGdH ZCe
|
j,j=e dI D¡ dJd;dKdL ZEdMdN ZFdS )Oé N)Úpartial)ÚPath)Úasynccontextmanageré ) ÚClosedResourceErrorÚEventÚProcessÚ_coreÚ
|
fail_afterÚ move_on_afterÚ run_processÚsleepÚ sleep_forever)Úskip_if_fbsd_pipes_brokenÚslow)Ú open_process)Úassert_no_checkpointsÚwait_all_tasks_blockedÚposix)ÚSIGKILLÚSIGTERMÚSIGUSR1)NNNc C s t jddd| gS )Nz-uú-cz import sys; )ÚsysÚ
|
executable)Úcode© r úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_subprocess.pyÚpython% s r z sys.exit(0)z sys.exit(1)z0sys.stdout.buffer.write(sys.stdin.buffer.read())c C s dt | gS )Nz
|
/bin/sleep)Ústr©Úsecondsr r r Ú<lambda>. ó r" c C s t d | ¡S )Nzimport time; time.sleep({}))r Úformatr r r r r" 0 r# c C s t r| j| kS | jdkS d S )Nr )r Ú
|
returncode)ÚprocÚsigr r r Ú
|
got_signal3 s r( c O s8 t | |I d H }z
|
|V W 5 | ¡ | ¡ I d H X d S ©N)r ÚkillÚwait)ÚargsÚkwargsr& r r r Úopen_process_then_kill: s
|
|
r. c
|
O s^ t ¡ 4 I d H >}| dd¡ | ttf| |¡I d H }|V |j ¡ W 5 Q I d H R X d S )NÚcheckF)r Ú open_nurseryÚ
|
setdefaultÚstartr r Ú cancel_scopeÚcancel)r, r- Únurseryr& r r r Úrun_process_in_nurseryD s
|
r6 Úbackground_processr zrun_process in nursery)Úidsc
|
à sÌ | t 4 I d H }| ¡ I d H W 5 Q I d H R X t|ts>t|jd ksLt|jdksZtt|dt dksrt| t4 I d H }| ¡ I d H W 5 Q I d H R X |jdks°tt|d td¡ksÈtd S )Nr z<trio.Process z: exited with status 0>é ú<trio.Process {!r}: {}>úexited with status 1)
|
Ú EXIT_TRUEr+ Ú
|
isinstancer ÚAssertionErrorÚ_pidfdr% ÚreprÚ
|
EXIT_FALSEr$ )r7 r& r r r Ú
|
test_basicT s
|
ÿrB c
|
à s | t d4 I d H p}|jd ks$tdt|ks4t| ¡ |j ¡ |jd k sTtdt|ksdt|jd ksrt|jd k stW 5 Q I d H R X d S )Né' ZrunningZexited)ÚSLEEPr% r> r@ r* Ú_procr+ r? )r7 Úpr r r Útest_auto_update_returncodee s
|
rG c à sà | t d4 I d H ¼}t ¡ 4 I d H @}| |j¡ | |j¡ | |j¡ t I d H |j ¡ W 5 Q I d H R X t ¡ 4 I d H >}| |j¡ | |j¡ | |j¡ t I d H | ¡ W 5 Q I d H R X W 5 Q I d H R X d S )Né
|
) rD r r0 Ú
|
start_soonr+ r r3 r4 r* )r7 r& r5 r r r Útest_multi_waitr s rJ c
|
à s t ttjtjdI d H 4 I d H }W 5 Q I d H R X |jd k s@tt t¡ |j
|
d¡I d H W 5 Q R X t t¡ |j ¡ I d H W 5 Q R X d S )N)ÚstdinÚstdoutó x) r ÚCATÚ
|
subprocessÚPIPEr% r> ÚpytestÚraisesr rK Úsend_allrL Ú receive_some)Úrecwarnr& r r r Ú!test_async_with_basics_deprecated s ÿ rV c à s¤ t dV}ttdI d H 4 I d H *}| ¡ d ks6t| ¡ t I d H W 5 Q I d H R X W 5 Q R X |jsntt|t s|tt
|
|d tdt rdnd¡ks td S )Néd rH r: zexited with signal 9r; ) r r rD Úpollr> r4 r Úcancelled_caughtr( r r@ r$ r )rU Úscoper& r r r Ú test_kill_when_context_cancelled s
|
&
|
|
|
ÿr[ zbdata = sys.stdin.buffer.read(); sys.stdout.buffer.write(data); sys.stderr.buffer.write(data[::-1])c sÖ | t tjtjtjd4 I d H ¨d fdd}dd }t ¡ 4 I d H J}t ¡ d |j_| |¡ | |j ¡ | |j
|
d d d
¡ W 5 Q I d H R X |jj r¬t d ¡ I d H ksÂt W 5 Q I d H R X d S )
|
N©rK rL Ústderrs+ the quick brown fox jumps over the lazy dogc s& j ¡I d H j ¡ I d H d S r) )rK rS Úacloser ©Úmsgr& r r Ú
|
feed_input² s z test_pipes.<locals>.feed_inputc Ó s2 t }| 2 z3 d H W }||7 }q
|
6 ||ks.td S r) )Ú bytearrayr> )ÚstreamÚexpectedÚseenÚchunkr r r Ú check_output¶ s z test_pipes.<locals>.check_outputg >@éÿÿÿÿr )Ú+COPY_STDIN_TO_STDOUT_AND_BACKWARD_TO_STDERRrO rP r r0 Ú current_timer3 ÚdeadlinerI rL r] rY r> r+ )r7 ra rg r5 r r_ r Ú
|
test_pipes¨ s" ü
|
* rl c sÞ | t dtjtjtjd4 I d H tr.dnd fdd}tdj j d¡I d H td I d H j d
|
¡I d H |d d I d H j d d ¡I d H |ddI d H |ddI d H j d ¡I d H |dd I d H j d¡I d H td }|dd I d H W 5 Q R X |j s8t
|
j ¡I d H |ddI d H j ¡ I d H j d¡I d H dkst
|
j d¡I d H dks¢t
|
¡ I d H W 5 Q R X W 5 Q I d H R X jd ksÚt
|
d S )Nz×idx = 0
|
while True:
|
line = sys.stdin.readline()
|
if line == '': break
|
request = int(line.strip())
|
print(str(idx * 2) * request)
|
print(str(idx * 2 + 1) * request * 2, file=sys.stderr)
|
idx += 1
|
r\ ó
|
s
|
c
|
sf t ¡ 4 I d H F} fdd}| |j|| d ¡ | |j|d | d d ¡ W 5 Q I d H R X d S )Nc sr |dkrF| |¡I d H }|d |¡ d¡t| ks8t|t|8 }q |dksRt| t ¡I d H ksntd S )Nr z{}úutf-8)rT r$ ÚencodeÚlenr> )rc ÚcountÚdigitÚresult)Únewliner r Ú drain_oneé s ÿ z3test_interactive.<locals>.expect.<locals>.drain_oner r9 )r r0 rI rL r] )ÚidxÚrequestr5 ru ©rt r& r r Úexpectæ s
|
z test_interactive.<locals>.expecté s 12g¹?s 345r i90 s 100s 200r9 rW r éÈ ó 0é s 999999é i?B r# )r rO rP r r
|
rK rS r r rY r> r^ rL rT r] r+ r% )r7 ry rZ r rx r Útest_interactiveÇ s@ ÿ
|
ó
|
(r c à s t dd tdD } tt| dddI d H }|jtks:t|jdksHt|j| ksVt|jdksdtttddI d H }|jtkst|jdkst|jdks t|jd ks®ttt | dddI d H }|jt ksÒt|jdksàt|j| ksît|j| d d d
kstt
|
t ¡ ttd
|
d I d H W 5 Q R X t
|
t ¡ tttjd I d H W 5 Q R X t
|
t ¡ tttjd I d H W 5 Q R X t
|
t ¡ tttjd I d H W 5 Q R X t
|
t ¡ ttdtjdI d H W 5 Q R X t
|
t ¡ ttdd dI d H W 5 Q R X d S )Nc s s | ]}t d d¡V qdS )r éÿ N)ÚrandomÚrandint)Ú.0Ú_r r r Ú <genexpr> s ztest_run.<locals>.<genexpr>i T)rK Úcapture_stdoutÚcapture_stderrr r# )r rh zoh no, it's text©rK )rL )r] )r rL )r r] )ÚbytesÚranger rN r, r> r% rL r] ri rQ rR Ú UnicodeErrorÚ
|
ValueErrorrO rP ÚDEVNULL)Údatars r r r Útest_run sL ÿ ü r c à sÈ t d} t tj¡ }t| tjddI d H W 5 Q R X |jj| ksFt |jj
|
dksVt |jj dksft |jj d ksvt t| ddddI d H }|j | kst |j dks¨t |j dks¶t |j
|
dksÄt d S ) Nz/sys.stderr.buffer.write(b'test\n'); sys.exit(1)T)rK r r9 s test
|
F)r r r/ r# )r rQ rR rO ÚCalledProcessErrorr r ÚvalueÚcmdr> r% r] rL r, )r Úexcinfors r r r Útest_run_check9 s" ÿ r c à sN t tjddgdd dI d H } | jdks,t| j| j krDd ksJn td S )Nr z import sys; sys.stdin.close()rM i r r )r r r r% r> rL r] )rs r r r Útest_run_with_broken_pipeK s
|
ÿ r c
|
à sl | t tjtjtjd4 I d H }|jd k s.t|jd ks<t|j d¡I d H |j ¡ I d H g }|j
|
d¡I d H }|dkr~q| |¡ qbd |¡dkstW 5 Q I d H R X |j dksºttt ddtjdI d H }|j dksàt|jdksît|jd ksüt| ttjtjd 4 I d H D}|jd ks(t|jd ks8t|j ¡ I d H | ¡ I d H W 5 Q I d H R X |j dksvttrhzÞt ¡ \}}| t tj|tjd4 I d H ¦}t |¡ |jd ksÂt|jd ksÒt|jd ksât|j d¡I d H |j ¡ I d H | ¡ I d H dks tt |d
|
¡dks2tt |d
|
¡dksHtW 5 Q I d H R X W 5 t |¡ X d S ) Nr\ s 1234é r# s 12344321r T)rK r r] )rK r] i )ri rO rP ÚSTDOUTrL r> r] ZstdiorS Zsend_eofrT ÚappendÚjoinr% r rN rK r^ r+ r ÚosÚcloseÚpipeÚread)r7 r& Úoutputrf rs ÚrÚwr r r Útest_stderr_stdoutT st ü "ü ÿ ü
|
*r¡ c à s¨ t t¡} tdgddI d H W 5 Q R X dt| jks<tdt| jksNttr¤t t¡} tdgddI d H W 5 Q R X t t¡} tdddI d H W 5 Q R X d S ) NZlsrn )Úencodingzunbuffered byte streamsz&the 'encoding' option is not supportedT)ÚshellF)rQ rR Ú TypeErrorr r r r> r )r r r r Ú test_errors s r¥ c sL fdd}|t jtI d H |t jtI d H trH|dd tI d H d S )Nc s t dB} td4 I d H }| | | ¡ I d H W 5 Q I d H R X W 5 Q R X |jrZttrp|j| ks~tn|jdks~td S )Nç ð?é r )r rD r+ rY r> r r% )Zsend_itÚsignumrZ r& ©r7 r r Útest_one_signal¦ s
|
(
|
z%test_signals.<locals>.test_one_signalc S s
|
| t¡S r) )Ú send_signalr ©r& r r r r" » r# z test_signals.<locals>.<lambda>)r r* r Ú terminater r r )r7 rª r r© r Ú test_signals¤ s
|
r® zPOSIX specific)Úreasonc à s¼ t t jt j¡}z| td4 I d H r}t ¡ 4 I d H 6}| |j¡ t I d H | ¡ t
|
¡ d |j _ W 5 Q I d H R X |j j rt|jdkstW 5 Q I d H R X W 5 t t j|¡ X d S )Nr§ r¦ r )ÚsignalZSIGCHLDÚSIG_IGNrD r r0 rI r+ r r* rj r3 rk rY r> r% )r7 Z old_sigchldr& r5 r r r Útest_wait_reapable_fails¾ s "r² c
|
s¸ ddl m} | j d¡s"t d¡ ddlm} d t ddg¡ fd d
|
}t
|
|
t
|
j |¡}z,t
|
d ¡ |j jd d d kstW 5 j d kr¤ ¡ ¡ t
|
|
t
|
j |¡ X d S )Nr )Úwait_child_exitingZwaitidz waitid only)Úsync_wait_reapableFr Z3600c s d ¡ d S ©NT)r* )r' Úframe©Z got_alarmZsleeperr r Úon_alarmß s z#test_waitid_eintr.<locals>.on_alarmr9 )Útimeouti÷ÿÿÿ)Z_subprocess_platformr³ Ú
|
__module__ÚendswithrQ ÚskipZ_subprocess_platform.waitidr´ rO ÚPopenr° ZSIGALRMr% r* r+ ZalarmÚpidr> )r³ r´ r¸ Z old_sigalrmr r· r Útest_waitid_eintrÒ s
|
|
|
|
r¿ c
|
sl d fdd} t ¡ 4 I d H 4}| tttd| d¡ t I d H |j ¡ W 5 Q I d H R X sht d S )NFc s2 d | ¡ zt I d H W 5 | jd k s,tX d S rµ )r r% r> r r¬ ©Z custom_deliver_cancel_calledr r Úcustom_deliver_cancelõ s
|
z9test_custom_deliver_cancel.<locals>.custom_deliver_cancelrC )Zdeliver_cancel)
|
r r0 rI r r rD r r3 r4 r> )rÁ r5 r rÀ r Útest_custom_deliver_cancelò s ÿ r c s t j fdd}| t d|¡ tjtddN t ¡ 4 I d H ,}| t t
|
d¡ t I d H |j ¡ W 5 Q I d H R X W 5 Q R X d S )Nc s | t dd S )NZwhoops)ÚOSError)Úself©Zoriginal_terminater r Úbroken_terminate s z>test_warn_on_failed_cancel_terminate.<locals>.broken_terminater z
|
.*whoops.*©ÚmatchrC )r r ÚsetattrrQ ÚwarnsÚRuntimeWarningr r0 rI r rD r r3 r4 )Ú monkeypatchrÆ r5 r rÅ r Ú$test_warn_on_failed_cancel_terminate
|
s rÍ z
|
posix onlyc à sx | tddd ¡ tjtddN t ¡ 4 I d H ,}| tt d¡ t
|
I d H |j ¡ W 5 Q I d H R X W 5 Q R X d S )Nr c W s d S r) r )r, r r r r" r# z8test_warn_on_cancel_SIGKILL_escalation.<locals>.<lambda>z.*ignored SIGTERM.*rÇ rC ) rÉ r rQ rÊ rË r r0 rI r rD r r3 r4 )Zautojump_clockrÌ r5 r r r Ú&test_warn_on_cancel_SIGKILL_escalation s rÎ c à s^ t tj¡: t ¡ 4 I d H } | tt¡I d H }W 5 Q I d H R X W 5 Q R X |j dksZt
|
d S )Nr9 ) rQ rR rO r r r0 r2 r rA r% r> )r5 r& r r r Ú test_run_process_background_fail' s ,rÏ ú/dev/fdz,requires a way to iterate through open filesc à s¶ t td ¡ } ttI d H t td ¡ | ks6tt tj ¡ tt
|
I d H W 5 Q R X t td ¡ | ksttt t ¡ tdgI d H W 5 Q R X t td ¡ | ks²td S )NrÐ z /dev/fd/0) ÚsetÚSyncPathÚiterdirr r< r> rQ rR rO r rA ÚPermissionError)Z starting_fdsr r r Útest_for_leaking_fds. s rÕ c
|
sÒ d d d fdd } t tdI d H }t ¡ 4 I d H }| | |¡ t I d H t ts`t| ¡ t
|
|j ¡ | ¡ W 5 Q R X td ¡ I d H W 5 Q R X ¡ s¾tdW 5 Q I d H R X d S )N)Úreturnc s t | ¡ I d H ¡ d S r) )r r+ rÑ r¬ ©Z noticed_exitr r Ú wait_and_tellD s z7test_subprocess_pidfd_unnotified.<locals>.wait_and_tellrC rz z,child task wasn't woken after poll, DEADLOCK)r rD r r0 rI r r= r r> r r rE r+ rX r Úis_set)rØ r& r5 r r× r Ú test_subprocess_pidfd_unnotifiedA s
|
|
rÚ )Gr r r° rO r Ú functoolsr Úpathlibr rÒ rQ Úasync_generatorr Ú r r r r r
|
r r r r Z_core.tests.tutilr r Zlowlevelr Ztestingr r Únamer r r r r r< rA rN rD r( r. r6 ÚmarkZ parametrizeZbackground_process_paramrB rG rJ rV r[ ri rl r r r r r¡ r¥ r® Zskipifr² r¿ r rÍ rÎ rÏ ÚexistsrÕ rÚ r r r r Ú<module> s ,
|
|
|
|
|
ý
|
|
|
ÿ
|
|
F+
|
|
B
|
|
|
þ
|
|
