U
|
Z ±d61 ã @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
|
Z
|
m Z d dl m Z d dlmZ d dlZd dlZd dlZd dlZd dlZddlmZmZmZmZmZmZm Z ddl m Z dd l m Z m!Z! d
|
d l"m#Z# dd l$m%Z% dd l&m'Z'm(Z( ddl)m*Z*m+Z+m,Z, ej-dk r,d dl.m/Z/ dd Z0dd Z1dd Z2dd Z3dd Z4dd Z5d d Z6d d Z7d!d" Z8d#d$ Z9d%d& Z:d'd( Z;d)d* Z<d+d, Z=d-d. Z>d/d0 Z?d1d2 Z@d3d4 ZAd5d6 ZBd7d8 ZCd9d: ZDd;d< ZEd=d> ZFd?d@ ZGdAdB ZHdCdD ZIdEdF ZJdGdH ZKdIdJ ZLdKdL ZMdMdN ZNdOdP ZOdQdR ZPedSdT ZQdUdV ZRe dWdX ZSe dYdZ ZTd[d\ ZUd]d^ ZVd_d` ZWdadb ZXdcdd ZYdedf ZZdgdh Z[didj Z\dkdl Z]dmdn Z^dodp Z_dqdr Z`dsdt Zadudv Zbdwdx Zcdydz Zdd{d| Zed}d~ Zfdd Zgdd Zhdd Ziejjjked
ddd Zldd Zmdd Zndd Zodd Zpdd Zqdd Zrdd Zsdd Ztdd Zudd Zvdd Zwdd Zxd¡d¢ Zyd£d¤ Zzd¥d¦ Z{d§d¨ Z|d©dª Z}d«d¬ Z~dd® Zd¯d° Zd±d² Zd³d´ Zdµd¶ Zd·d¸ Zd¹dº Z
d»d¼ Zd½d¾ Zd¿dÀ ZdÁd ZdÃdÄ ZejdÅdÆ ZdÇdÈ ZdÉdÊ ZdËdÌ Ze dÍdÎ ZdÏdÐ ZdÑdÒ ZejjjkejjdÓkdÔddÕdÖ ZejjjkejjdÓkdÔdd×dØ ZejjjkejjdÓkdÔddÙdÚ ZejjjkejjdÓkdÔddÛdÜ ZdÝdÞ Zdßdà Zdádâ Zdãdä Zdådæ Zdçdè ZdS )éé N)ÚcontextmanagerÚ ExitStack)Úinf)Údedenté )ÚslowÚcheck_sequence_matchesÚgc_collect_harderÚ'ignore_coroutine_never_awaited_warningsÚbuggy_pypy_asyncgensÚrestore_unraisablehookÚ!create_asyncio_future_in_new_loopé )Ú_core)Ú
|
MultiErrorÚNonBaseMultiErroré )Ú!DEADLINE_HEAP_MIN_PRUNE_THRESHOLD)Úto_thread_run_sync)ÚsleepÚ
|
fail_after)Úwait_all_tasks_blockedÚ SequencerÚassert_checkpoints)r é )ÚExceptionGroupc à s t dd ¡I d H S )Nc S s t jjS ©N)r ÚAbortZ SUCCEEDED©Ú_© r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_run.pyÚ<lambda>1 ó z sleep_forever.<locals>.<lambda>)r Úwait_task_rescheduledr r r r! Ú sleep_forever0 s r% c C s dd } t | d¡dks tt t¡ t | ¡ W 5 Q R X t t¡ t dd ¡ W 5 Q R X dd }t |d¡dks|td S ) Nc Ó s | S r r ©Úxr r r! Útrivial5 s ztest_basic.<locals>.trivialé c S s d S r r r r r r! r" @ r# z test_basic.<locals>.<lambda>c Ó s t ¡ I d H | S r )r Ú
|
checkpointr& r r r! Útrivial2B s z test_basic.<locals>.trivial2r )r ÚrunÚAssertionErrorÚpytestÚraisesÚ TypeError)r( r+ r r r! Ú
|
test_basic4 s r1 c C s> dd } t t¡}t | d¡ W 5 Q R X |jjdks:td S )Nc Ó s t | d S r ©Ú
|
ValueErrorr& r r r! ÚmainJ s z%test_initial_task_error.<locals>.mainé )r5 )r. r/ r3 r r, ÚvalueÚargsr- ©r4 Úexcinfor r r! Útest_initial_task_errorI s r: c C s> dd } t t¡}t | ¡ W 5 Q R X dt|jks:td S )Nc Ó s dd } t | ¡S )Nc Ó s d S r r r r r r! r4 T s z1test_run_nesting.<locals>.inception.<locals>.main©r r, ©r4 r r r! Ú inceptionS s z#test_run_nesting.<locals>.inceptionz from inside)r. r/ Ú RuntimeErrorr r, Ústrr6 r- )r= r9 r r r! Útest_run_nestingR s r@ c
|
à sZ t t¡ } t ¡ }| W 5 Q R X W 5 Q R X | d¡ |4 I d H W 5 Q I d H R X d S )NzDuse 'async with open_nursery\(...\)', not 'with open_nursery\(...\)')r. r/ r> r Ú open_nurseryÚmatch)r9 Úonr r r! Ú test_nursery_warn_use_async_with^ s ÿrD c à sV t d} t t ¡,}t ¡ 4 I d H
|
| W 5 Q I d H R X W 5 Q R X |j| ksRtd S )NZwhoops)r3 r. r/ r rA r6 r- )Úexcr9 r r r! Ú#test_nursery_main_block_error_basicl s
|
rF c
|
sx t d fdd} z0t ¡ 4 I d H }| | ¡ W 5 Q I d H R X W n. t k
|
rr } z| ksbtW 5 d }~X Y nX d S )Núuh ohc s d S r r r ©rE r r! Úerroringx s z(test_child_crash_basic.<locals>.erroring)r3 r rA Ú
|
start_soonr- )rI ÚnurseryÚer rH r! Útest_child_crash_basicu s rM c
|
à sj dd } g }t ¡ 4 I d H "}| | d|¡ | | d|¡ W 5 Q I d H R X t|ddhddhd d
|
hg d S ) Nc Ó s. t dD ] }| | |f¡ t ¡ I d H qd S )Nr )ÚrangeÚappendr r* )ZwhoamiÚrecordÚir r r! Úlooper s z%test_basic_interleave.<locals>.looperÚaÚb)rS r )rT r )rS r )rT r )rS r )rT r )r rA rJ r )rR rP rK r r r! Útest_basic_interleave s ÿrU c sh g fdddd fdd} t t¡}t | ¡ W 5 Q R X dgksTt|jjdksdtd S ) Nc sB zt ¡ I d H qW n( t jk
|
r< td d¡ Y nX d S )Nzlooper cancelledÚ cancelled)r r* Ú CancelledÚprintrO r )Ú looper_recordr r! rR s
|
z+test_task_crash_propagation.<locals>.looperc Ó s t dd S )NÚarghr2 r r r r! Úcrasher s z,test_task_crash_propagation.<locals>.crasherc
|
s: t ¡ 4 I d H } | ¡ | ¡ W 5 Q I d H R X d S r ©r rA rJ ©rK )r[ rR r r! r4 ¡ s
|
z)test_task_crash_propagation.<locals>.mainrV )rZ )r. r/ r3 r r, r- r6 r7 r8 r )r[ rR rY r! Útest_task_crash_propagation s r^ c s` dd fdd} t t¡}t | ¡ W 5 Q R X t|j dd |jjD tt hks\t
|
d S )Nc Ó s t d S r r2 r r r r! r[ ° s z.test_main_and_task_both_crash.<locals>.crasherc
|
s4 t ¡ 4 I d H } | ¡ tW 5 Q I d H R X d S r )r rA rJ ÚKeyErrorr] ©r[ r r! r4 ³ s
|
z+test_main_and_task_both_crash.<locals>.mainc S s h | ] }t |qS r ©Útype©Ú.0rE r r r! Ú <setcomp>» s z0test_main_and_task_both_crash.<locals>.<setcomp>) r. r/ r r r, rX r6 Ú
|
exceptionsr3 r_ r- r8 r r` r! Ú test_main_and_task_both_crash s
|
þrg c sV dd fdd} t t¡}t | ¡ W 5 Q R X dd |jjD tthksRt d S )Nc Ó s | d S r r )Úetyper r r! r[ Â s z'test_two_child_crashes.<locals>.crasherc
|
s> t ¡ 4 I d H } | t¡ | t¡ W 5 Q I d H R X d S r )r rA rJ r_ r3 r] r` r r! r4 Å s z$test_two_child_crashes.<locals>.mainc S s h | ] }t |qS r ra rc r r r! re Ì s z)test_two_child_crashes.<locals>.<setcomp>)
|
r. r/ r r r, r6 rf r3 r_ r- r8 r r` r! Útest_two_child_crashesÁ s þri c à sZ dd } t t¡> t ¡ 4 I d H }| | ¡ t I d H W 5 Q I d H R X W 5 Q R X d S )Nc Ó s t d S r r2 r r r r! r[ Ó s z.test_child_crash_wakes_parent.<locals>.crasher)r. r/ r3 r rA rJ r% )r[ rK r r r! Ú test_child_crash_wakes_parentÒ s
|
|
rj c
|
sl d d fdd} fdd}t ¡ 4 I d H (}| | ¡ t ¡ I d H | |¡ W 5 Q I d H R X d S )Nc sX t ¡ td t I d H } td | dks0ttd t t t ¡¡ td d S )Nz child1 startz child1 woker zchild1 rescheduling t2z child1 exit) r Ú current_taskrX r% r- Ú
|
rescheduleÚoutcomeÚErrorr3 r& ©Út1Út2r r! Úchild1à s z test_reschedule.<locals>.child1c sX t d t ¡ t t d¡¡ t d t t¡ t I d H W 5 Q R X t d d S )Nz child2 startr z child2 sleepzchild2 successful exit)
|
rX r rk rl rm ÚValuer. r/ r3 r% r ro r r! Úchild2ë s z test_reschedule.<locals>.child2)r rA rJ r* ©rr rt rK r ro r! Útest_rescheduleÜ s
|
|
rv c à s2 t ¡ } t t d¡j¡ t ¡ }| |k s.td S )NÚ perf_counter)r Ú current_timeÚtimer Úget_clock_infoÚ
|
resolutionr- ro r r r! Útest_current_timeü s r| c à sd | ¡ }| ¡ t ¡ ks t| ¡ t ¡ ks0t| d¡ |d | ¡ krZt ¡ ks`n td S )Ng
ëQ¸ @)rx r r- Újump)Ú
|
mock_clockÚstartr r r! Ú!test_current_time_with_mock_clock s
|
|
r c à s | t ¡ kstd S r )r Z current_clockr- ©r~ r r r! Útest_current_clock s r c
|
sD t ¡ fdd} t ¡ 4 I d H }| | ¡ W 5 Q I d H R X d S )Nc s t ¡ jj kstd S r )r rk Úparent_nurseryÚ parent_taskr- r ©r r r! Úchild s z test_current_task.<locals>.child)r rk rA rJ ©r rK r r
r! Útest_current_task s r c à s* t ¡ } | j| j kr d ks&n td S r )r Zcurrent_root_taskr Úeventual_parent_nurseryr- )Úrootr r r! Útest_root_task s r c C s@ t t¡ t ¡ W 5 Q R X t t¡ t ¡ W 5 Q R X d S r )r. r/ r> r rk rx r r r r! Útest_out_of_context s r c
|
à s t I d H dd }t ¡ }t| |jdks2t|jdks@tt ¡ 4 I d H ¦}| |¡ t I d H t ¡ }|
|
dd ¡ |j
|
dd dd t ¡ }t| |jd
|
ks®t|j dks¼t|jd ksÊt|j ¡ t ¡ }t| |j d ksòtW 5 Q I d H R X t ¡ I d H t ¡ I d H tjt ¡ d d& t ¡ }t| |jd ksTtW 5 Q R X t ¡ }t| |jtks~td S )Nc Ó s, zt I d H W n tjk
|
r& Y nX d S r )r% r rW r r r r! r , s z&test_current_statistics.<locals>.childr r c S s d S r r r r r r! r" < r# z)test_current_statistics.<locals>.<lambda>c S s d S r r r r r r! r" = r# T©Z
|
idempotenté r r é ©Údeadline)r r Zcurrent_statisticsrX Z tasks_livingr- Zrun_sync_soon_queue_sizerA rJ Úcurrent_trio_tokenÚ run_sync_soonZtasks_runnableÚ cancel_scopeÚcancelr* Ú CancelScoperx Zseconds_to_next_deadliner )r~ r ÚstatsrK Útokenr r r! Útest_current_statistics' s<
|
|
r c à s¸ t ¡ }dt|kst| dt|ks.tt ¡ d |_dt|ksLtt ¡ d |_dt|ksjtdtt|I d H kst| ¡ dt|kstW 5 Q R X d t|ks´td S )
|
NÚunboundÚactiver z deadline is 1.00 seconds agoé
|
z"deadline is 10.00 seconds from nowr rV Zexited)r r Úreprr- rx r r r )r~ Úscoper r r! Útest_cancel_scope_repr[ s r c C sL dd } t | ¡ dd }t |¡ dd }t |¡ dd }t |¡ d S ) Nc
|
Ó sT t ¡ B} t ¡ I d H | ¡ t t j¡ t ¡ I d H W 5 Q R X W 5 Q R X d S r )r r Úcheckpoint_if_cancelledr r. r/ rW ©r r r r! Úmain1l s
|
|
z!test_cancel_points.<locals>.main1c
|
Ó sT t ¡ B} t ¡ I d H | ¡ t t j¡ t ¡ I d H W 5 Q R X W 5 Q R X d S r )r r r* r r. r/ rW r¡ r r r! Úmain2u s
|
|
z!test_cancel_points.<locals>.main2c
|
Ó sD t ¡ 2} | ¡ t t j¡ t I d H W 5 Q R X W 5 Q R X d S r )r r r r. r/ rW r% r¡ r r r! Úmain3~ s
|
z!test_cancel_points.<locals>.main3c
|
Ó sb t ¡ P} | ¡ t ¡ I d H t ¡ I d H t t j¡ t ¡ I d H W 5 Q R X W 5 Q R X d S r )r r r Úcancel_shielded_checkpointr. r/ rW r* r¡ r r r! Úmain4 s
|
z!test_cancel_points.<locals>.main4r; )r¢ r£ r¤ r¦ r r r! Útest_cancel_pointsk s
|
|
|
r§ c
|
à s® t ¡ $} | ¡ | ¡ t ¡ I d H W 5 Q R X | js<t| jsFtt ¡ V} | ¡ t t j ¡ t
|
I d H W 5 Q R X t t j ¡ t
|
I d H W 5 Q R X W 5 Q R X d S r ) r r r r* Ú cancel_calledr- Úcancelled_caughtr. r/ rW r% r¡ r r r! Útest_cancel_edge_cases s
|
|
|
|
rª c à s dd } z$t ¡ }zt ¡ 4 I d H d}| t¡ | t¡ |j ¡ t jdd t I d H W 5 Q R X | t¡ | ¡ | | ¡ W 5 Q I d H R X W n tk
|
r } zdt |j
|
dksÂt i }|j
|
D ](}| t |d¡ |t | d7 < qÌ|t jdtdikst W 5 d }~X Y nX W 5 Q R X W nL t k
|
rF Y n@ tk
|
rz } zt |tksjt W 5 d }~X Y n X d st d S )
|
Nc Ó s t d S r ©r_ r r r r! r[ ¤ s z7test_cancel_scope_multierror_filtering.<locals>.crasherT©Úshieldr r r r F)r r rA rJ r% r r r r Úlenrf r- Ú
|
setdefaultrb rW r_ Ú BaseException)r[ ÚouterrK Z multi_excÚsummaryrE r r r! Ú&test_cancel_scope_multierror_filtering£ s4
|
|
|
|
|
"$r³ c
|
sX g fdd} t ¡ 4 I d H }|j ¡ | | ¡ W 5 Q I d H R X dgksTtd S )Nc s d¡ t I d H d S )NÚstarted)rO r% r ©rP r r! ÚblockerÔ s
|
z'test_precancelled_task.<locals>.blockerr´ )r rA r r rJ r- )r¶ rK r rµ r! Útest_precancelled_taskÎ s
|
r· c à sx t ¡ d} t ¡ B}t ¡ I d H | ¡ t t j¡ t ¡ I d H W 5 Q R X |jdksbtt t ¡ d|_W 5 Q R X |jdkstd|_|jdks tt ¡ I d H t ¡ 4}| ¡ t t j¡ t ¡ I d H W 5 Q R X W 5 Q R X t ¡ I d H d|_t t j¡ t ¡ I d H W 5 Q R X d|_t ¡ I d H | ¡ t ¡ I d H W 5 Q R X |j
|
sjtW 5 Q R X d S )NFZhelloT) r r r* r r. r/ rW r r- r0 r© )r± ÚinnerZinnerestr r r! Útest_cancel_shieldingÞ s4
|
"r¹ c
|
sz t fdd fdd} t ¡ 4 I d H (}| | d¡ | | d¡ |j ¡ W 5 Q I d H R X ddd d
|
hksvtd S ) Nc s6 zt I d H W n tjk
|
r0 | ¡ Y nX d S r )r% r rW Úadd)Úidentrµ r r! Úleaf
|
s z%test_cancel_inheritance.<locals>.leafc
|
sF t ¡ 4 I d H &}| | d ¡ | | d ¡ W 5 Q I d H R X d S )Nz-l1z-l2r\ )r» rK )r¼ r r! Úworker s z'test_cancel_inheritance.<locals>.workerZw1Zw2zw1-l1zw1-l2zw2-l1zw2-l2)Úsetr rA rJ r r r- )r½ rK r )r¼ rP r! Útest_cancel_inheritance s r¿ c sº t ¡ ¨} t ¡ 4 I d H }| ¡ d|j_g fdd}| |¡ t I d H dgks`td|j_t jdd" t I d H ddgkstW 5 Q R X W 5 Q I d H R X W 5 Q R X d S )NTc s@ d¡ zt I d H W n tjk
|
r: d¡ Y nX d S )NÚsleepingrV )rO r% r rW r rµ r r! Úsleeper& s
|
|
z)test_cancel_shield_abort.<locals>.sleeperrÀ Fr¬ rV ) r r rA r r r rJ r r- )r± rK rÁ r rµ r! Útest_cancel_shield_abort s
|
|
r c
|
à s t ¡ }t ¡ 0}|jtks t|d |_|j|d ks<tW 5 Q R X |jrPt| d¡ t ¡ I d H t ¡ I d H t ¡ I d H |jrtt ¡ }t j|d d }| d¡ t I d H W 5 Q R X |jsÒt|j
|
sÜtt ¡ }t ¡ }t ¡ I d H |d |_t ¡ I d H | d¡ t ¡ I d H |d |_t t j ¡ t ¡ I d H W 5 Q R X t t j ¡ t ¡ I d H W 5 Q R X W 5 Q R X d S )Nr r r r r )r rx r r r r- r¨ r} r* r% r© r. r/ rW )r~ r r r r r! Útest_basic_timeout< s:
|
|
|
|
|
|
|
|
|
|
|
|
rà c à s t ¡ J} t ¡ 6}t ¡ "}| ¡ | ¡ t I d H W 5 Q R X W 5 Q R X W 5 Q R X |jsbt|jrlt|jsvt|jst| jrt| jrtt ¡ °} t ¡ }| ¡ t t j ¡ t
|
¡ I d H W 5 Q R X t t j ¡ t
|
¡ I d H W 5 Q R X d|_ t
|
¡ I d H | ¡ t t j ¡ t
|
¡ I d H W 5 Q R X W 5 Q R X W 5 Q R X t ¡ }| ¡ t ¡ I d H W 5 Q R X t
|
¡ I d H |jrtd S ©NT) r r r r% r¨ r- r© r. r/ rW r* r r¥ )Úscope1Úscope2Zscope3r r r r! Útest_cancel_scope_nestinga s:
|
|
|
*
|
|
|
|
|
|
|
|
,
|
rÇ c
|
à s` t ¡ >} t ¡ *}| ¡ zt ¡ I d H W 5 d|_X W 5 Q R X W 5 Q R X | jrX|jr\td S rÄ )r r r r r* r© r- )r± r¸ r r r! Ú&test_unshield_while_cancel_propagating s
|
|
rÈ c sÈ dd } t ¡ ¡ t ¡ 4 I d H }| | ¡ W 5 Q I d H R X t ¡ t ¡ 4 I d H &}| | ¡ t I d H ¡ W 5 Q I d H R X t ¡ d _t ¡ X} H | ¡ t ¡ I d H d _t t j
|
¡ t ¡ I d H W 5 Q R X W 5 Q R X W 5 Q R X t ¡ t ¡ I d H W 5 Q R X ¡ t ¡ I d H j sHt j rTt t t¡} W 5 Q R X W 5 Q R X dt|jkst t ¡ @ t t¡} W 5 Q R X W 5 Q R X dt|jksÒt W 5 Q R X t ¡ fdd}t ¡ 4 I d H d}|j|dd t I d H t t¡} W 5 Q R X W 5 Q R X dt|jksVt |j ¡ W 5 Q I d H R X t jt ¡ d
|
d j rt jd
|
8 _ j sªt jd
|
7 _ j sÄt d S ) Nc
|
Ó s4 | & t d t I d H W 5 Q R X W 5 Q R X d S ©Nr )r r% r¡ r r r! Úsleep_until_cancelled s z2test_cancel_unbound.<locals>.sleep_until_cancelledTFzsingle 'with' blockc s t I d H W 5 Q R X d S r ©r% r r¡ r r! Ú enter_scopeÅ s z(test_cancel_unbound.<locals>.enter_scopezthis one©Únamer r )r r r rA rJ r r r* r. r/ rW r¨ r- r© r> r? r6 r rx r )rÊ rK r± Úexc_inforÌ r r¡ r! Útest_cancel_unbound sb ,
|
|
rÐ c à sx t ¡ } t ¡ }t B}| | ¡ |( tjtdd | ¡ W 5 Q R X W 5 Q R X W 5 Q R X dd }dd }t x}| t ¡ ¡ t ¡ 4 I d H H}| |¡ | |¡ t
|
I d H tjtdd | ¡ W 5 Q R X W 5 Q I d H R X W 5 Q R X t ¡ }| ¡ I d H }z>| |¡ | |¡ | t¡ t
|
I d H |j d d d ¡ W 5 t t¡}|j t ¡ I d H W 5 Q R X dt|jkstt|jjtkstt|jjjdks´td }|jjjD ]:} t| tsÖtd
|
t| ksèt|t| jt jO }qÂ|stX d d }
|
t ¡ 4 I d H F}| |
|
¡I d H } tjtd d | d d d ¡ W 5 Q R X | ¡ W 5 Q I d H R X d S )Nzstill within its child©rB c Ó s( t tj¡ t I d H W 5 Q R X d S r )r. r/ r rW r% r r r r! Útask1é s z+test_cancel_scope_misnesting.<locals>.task1c
|
Ó s< t ¡ * t t j¡ t I d H W 5 Q R X W 5 Q R X d S r )r r r. r/ rW r% r r r r! Útask2î s
|
z+test_cancel_scope_misnesting.<locals>.task2z which had already been exitedr Fz closed before the task exitedc Ó s. t ¡ }| |¡ t I d H W 5 Q R X d S r )r r r´ r% )Ú task_statusr r r r! Útask3 s
|
|
z+test_cancel_scope_misnesting.<locals>.task3zfrom unrelated) r r r Ú enter_contextr. r/ r> ÚcloserA rJ r Ú
|
__aenter__Ú __aexit__ÚsysrÏ r? r6 r- rb Ú __context__r r® rf Ú
|
isinstancerW r% r Ú__exit__r r ) r± r¸ ÚstackrÒ rÓ rK Z nursery_mgrrÏ Zcancelled_in_contextrE rÕ r r r r! Ú test_cancel_scope_misnestingÝ sT
|
&
|
|
,
|
|
|
rß c
|
à s d} t dD ]r}t ¡ }t ¡ }t ¡ | |_t I d H W 5 Q R X t ¡ | }|| }t| d| krvdk r n q qq dst d S )Ng ð?r r F)
|
rN ry rw r r rx r r% rX r- )ZTARGETr Z
|
real_startr Z real_durationZaccuracyr r r! Útest_timekeeping! s
|
rà c
|
s¸ d gd gg fdd} t ¡ 4 I d H f}| | ¡ t I d H dgksVtd ¡ t I d H dgks|tt d t d¡¡ W 5 Q I d H R X dddgks´td S )Nc s t ¡ d< t ¡ v} | d< d¡ t dd ¡I d H }|dksHt d¡ zt ¡ I d H W n t jk
|
r d¡ Y nX W 5 Q R X d S )Nr r c S s t jjS r ©r r ÚFAILEDr r r r! r" @ r# z=test_failed_abort.<locals>.stubborn_sleeper.<locals>.<lambda>r ÚwokerV )r rk r rO r$ r- r rW )r r' ©rP Zstubborn_scopeZ stubborn_taskr r! Ústubborn_sleeper; s
|
|
|
z+test_failed_abort.<locals>.stubborn_sleeperr r r rã rV ) r rA rJ r r- r rl rm rs )rå rK r rä r! Útest_failed_abort6 s
|
&ræ c C s4 dd } t tj¡ t | ¡ W 5 Q R X t d S )Nc Ó sP t ¡ I d H t ¡ I d H t ¡ "} | ¡ t dd ¡I d H W 5 Q R X d S )Nc S s d S r r r r r r! r" e r# z1test_broken_abort.<locals>.main.<locals>.<lambda>)r r* r r r$ r¡ r r r! r4 W s
|
|
z test_broken_abort.<locals>.main)r. r/ r ÚTrioInternalErrorr, r r< r r r! Útest_broken_abortU s rè c
|
C s@ dd } t ( t tj¡ t | ¡ W 5 Q R X W 5 Q R X d S )Nc Ó s t ¡ } d| _t ¡ I d H d S )Nzhello!)r rk Z_schedule_pointsr* ©Útaskr r r! r4 s s z$test_error_in_run_loop.<locals>.main)r
|
r. r/ r rç r, r< r r r! Útest_error_in_run_loopp s rë c s< g fdd} t | d¡ t I d H ddgks8td S )Nc s2 d| f¡ dt ¡ f¡ t ¡ I d H d S )Nr' Úki)rO r Úcurrently_ki_protectedr* r& rµ r r! Ú system_task s z+test_spawn_system_task.<locals>.system_taskr )r' r )rì T)r Úspawn_system_taskr r- ©rî r rµ r! Útest_spawn_system_task} s
|
rñ c s: dd fdd} t tj¡ t | ¡ W 5 Q R X d S )Nc Ó s t d S r r« r r r r! r[ s z'test_system_task_crash.<locals>.crasherc s t ¡ t I d H d S r ©r rï r% r r` r r! r4 s
|
z$test_system_task_crash.<locals>.main)r. r/ r rç r, r< r r` r! Útest_system_task_crash s ró c s dd dd fddfdd} t tj¡}t | ¡ W 5 Q R X |jj}t|tsbt t
|
|j d kstt |j D ]}t|t t fszt qzd S )
|
Nc Ó s t d S r r« r r r r! Úcrasher1 s z3test_system_task_crash_MultiError.<locals>.crasher1c Ó s t d S r r2 r r r r! Úcrasher2 s z3test_system_task_crash_MultiError.<locals>.crasher2c
|
s: t ¡ 4 I d H } | ¡ | ¡ W 5 Q I d H R X d S r r\ r] )rô rõ r r! rî s
|
z6test_system_task_crash_MultiError.<locals>.system_taskc s t ¡ t I d H d S r rò r rð r r! r4 £ s
|
z/test_system_task_crash_MultiError.<locals>.mainr )r. r/ r rç r, r6 Ú __cause__rÜ r r- r® rf r_ r3 )r4 r9 ÚmerE r )rô rõ rî r! Ú!test_system_task_crash_MultiError s
|
rø c sd dd dd fddfdd} t tj¡}t | ¡ W 5 Q R X t|jjtks`t d S ) Nc Ó s0 zt I d H W n tjk
|
r* tY nX d S r )r% r rW r3 r r r r! r[ ´ s z6test_system_task_crash_plus_Cancelled.<locals>.crasherc Ó s t I d H d S r rË r r r r! Úcancelmeº s z7test_system_task_crash_plus_Cancelled.<locals>.cancelmec
|
s: t ¡ 4 I d H } | ¡ | ¡ W 5 Q I d H R X d S r r\ r] )rù r[ r r! rî ½ s
|
z:test_system_task_crash_plus_Cancelled.<locals>.system_taskc s t ¡ d S r )r rï r rð r r! r4 Â s z3test_system_task_crash_plus_Cancelled.<locals>.main)
|
r. r/ r rç r, rb r6 rö r3 r- r8 r )rù r[ rî r! Ú%test_system_task_crash_plus_Cancelled± s rú c sL dd fdd} t tj¡}t | ¡ W 5 Q R X t|jjtsHt d S )Nc Ó s t d S r )ÚKeyboardInterruptr r r r! rì Ì s z4test_system_task_crash_KeyboardInterrupt.<locals>.kic s t ¡ t I d H d S r rò r ©rì r r! r4 Ï s
|
z6test_system_task_crash_KeyboardInterrupt.<locals>.main)
|
r. r/ r rç r, rÜ r6 rö rû r- r8 r rü r! Ú(test_system_task_crash_KeyboardInterruptË s
|
rý c
|
à sb t jt ¡ d dD t ¡ I d H t t j¡ | d¡ t ¡ I d H W 5 Q R X W 5 Q R X d S )Nr r r )r r rx r* r. r/ rW r} r r r r! Ú%test_yield_briefly_checks_for_timeoutá s
|
|
rþ c
|
s| g t fdd} fdd}t ¡ 4 I d H }| | ¡ | |¡ W 5 Q I d H R X ddddd d
|
d d gksxtd S ) Nc sÆ t t¡} z8d4 I d H W 5 Q I d H R X d¡ tdW nX tk
|
r d¡ d4 I d H W 5 Q I d H R X d kst d¡ Y nX W 5 Q R X | jjd ks¸t d¡ d S ) Nr ú child1 raiserr ú child1 sleepr ú child2 wakeúchild1 re-raiseúchild1 success)r. r/ r3 rO r- r6 rÛ ©r9 ©rP Úseqr r! rr ó s
|
|
|
z test_exc_info.<locals>.child1c sâ t t¡´} d4 I d H W 5 Q I d H R X d ks:t d¡ t ¡ dksTtz tdW nX tk
|
r¸ d¡ d4 I d H W 5 Q I d H R X d ks¨t d ¡ Y nX W 5 Q R X | jjd ksÔt d
|
¡ d S ) Nr r r )NNNrt úchild2 sleep againr r úchild2 re-raiseúchild2 success) r. r/ r_ r- rO rÚ rÏ r6 rÛ r r r r! rt s"
|
|
|
z test_exc_info.<locals>.child2rÿ r r r r r r r )r r rA rJ r- ru r r r! Ú test_exc_infoï s"
|
ør
|
c sv d fdd} t t¡R t ¡ 4 I d H 0}| | ¡ t I d H t t t
|
¡¡ W 5 Q I d H R X W 5 Q R X d S )Nc sR t ¡ ztW n< tk
|
rL zt I d H W n tk
|
rD Y nX Y nX d S r ©r rk r_ Ú Exceptionr% r ©Z
|
child_taskr r! r / s z.test_exc_info_after_yield_error.<locals>.child) r. r/ r_ r rA rJ r rl rm rn r3 r r r r! Ú test_exc_info_after_yield_error, s
|
r c s d fdd} t t¡R}t ¡ 4 I d H 0}| | ¡ t I d H t t t ¡¡ W 5 Q I d H R X W 5 Q R X t
|
|j j t std S )Nc s6 t ¡ ztW n tk
|
r0 t I d H Y nX d S r r r r r r! r H s
|
z8test_exception_chaining_after_yield_error.<locals>.child)r. r/ r3 r rA rJ r rl rm rn rÜ r6 rÛ r_ r- )r r9 rK r r r! Ú)test_exception_chaining_after_yield_errorE s
|
.r c à sb dd } t t¡6}t ¡ 4 I d H }| | ¡ tW 5 Q I d H R X W 5 Q R X |jjd ks^t d S )Nc Ó s t d S r r« r r r r! r[ [ s zJtest_nursery_exception_chaining_doesnt_make_context_loops.<locals>.crasher)
|
r. r/ r r rA rJ r3 r6 rÛ r- )r[ r9 rK r r r! Ú9test_nursery_exception_chaining_doesnt_make_context_loopsZ s
|
r c C sL dd } t | ¡}t | ¡}||k s(t||ks4tt|t|ksHtd S )Nc Ó s t ¡ } | t ¡ kst| S r )r r r- ©r r r r! Úget_and_check_tokeng s z4test_TrioToken_identity.<locals>.get_and_check_token)r r, r- Úhash)r rp rq r r r! Útest_TrioToken_identityf s
|
|
r c sJ g fdd} t ¡ }| | d¡ r,tt I d H dgksFtd S )Nc s d| f¡ d S )NÚcb©rO r& rµ r r! r w s z.test_TrioToken_run_sync_soon_basic.<locals>.cbr )r r )r r r r- r )r r r rµ r! Ú"test_TrioToken_run_sync_soon_basict s r c sP d fdd} t | ¡ d k s&tt t j¡ dd ¡ W 5 Q R X d S )Nc s t ¡ d S r )r r r r r r! r4 s z3test_TrioToken_run_sync_soon_too_late.<locals>.mainc S s d S r r r r r r! r" r# z7test_TrioToken_run_sync_soon_too_late.<locals>.<lambda>)r r, r- r. r/ ÚRunFinishedErrorr r< r r r! Ú%test_TrioToken_run_sync_soon_too_late s
|
r c sú g fdd} t ¡ }| | d¡ |j| ddd |j| ddd |j| ddd |j| ddd |j| ddd t I d H t dkstt dddgks¦tg tdD ]"}tdD ]}|j| |dd q¾q²t I d H ttdksötd S ) Nc s | ¡ d S r r r& rµ r r! r s z3test_TrioToken_run_sync_soon_idempotent.<locals>.cbr Tr r r éd ) r r r r r® r- ÚsortedrN Úlist)r r r rQ r rµ r! Ú'test_TrioToken_run_sync_soon_idempotent s$ r c s< g fddfdd} t | ¡ t dks8td S )Nc s: d ¡ z| j| dd W n tjk
|
r4 Y nX d S ©NTr )rO r r r r ©rP Úredor r! r ° s
|
|
z=test_TrioToken_run_sync_soon_idempotent_requeue.<locals>.redoc sF t ¡ } | j | dd t ¡ I d H t ¡ I d H t ¡ I d H d S r )r r r r* r )r r r! r4 · s
|
z=test_TrioToken_run_sync_soon_idempotent_requeue.<locals>.mainr ©r r, r® r- r< r r r! Ú/test_TrioToken_run_sync_soon_idempotent_requeue© s
|
|
r" c sB g fdd} t t¡ t | ¡ W 5 Q R X dgks>td S )Nc s" t ¡ } | fdd¡ td S )Nc s
|
d¡S )Núsync-cbr r rµ r r! r" Ê r# zMtest_TrioToken_run_sync_soon_after_main_crash.<locals>.main.<locals>.<lambda>)r r r r3 r rµ r r! r4 Æ s z;test_TrioToken_run_sync_soon_after_main_crash.<locals>.mainr# )r. r/ r3 r r, r- r< r rµ r! Ú-test_TrioToken_run_sync_soon_after_main_crashà s
|
r$ c s\ t fdd} t tj¡}t | ¡ W 5 Q R X t|jjt ksHt
|
ddhksXt
|
d S )Nc s^ t ¡ } | dd ¡ | fdd¡ zt I d H W n t jk
|
rX d¡ Y nX d S )Nc S s
|
t d S ©NZnope)Údictr r r r! r" Ø r# zDtest_TrioToken_run_sync_soon_crashes.<locals>.main.<locals>.<lambda>c s
|
d¡S )Nú2nd run_sync_soon ran)rº r rµ r r! r" Û r# ú
|
cancelled!)r r r r% rW rº r rµ r r! r4 Ö s z2test_TrioToken_run_sync_soon_crashes.<locals>.mainr' r( ) r¾ r. r/ r rç r, rb r6 rö r_ r- r8 r rµ r! Ú$test_TrioToken_run_sync_soon_crashesÓ s r) c sV d} g t ¡ }t| D ]}| fdd|¡ qt I d H tt| ksRtd S )Nr c s
|
| ¡S r r )Újrµ r r! r" í r# z3test_TrioToken_run_sync_soon_FIFO.<locals>.<lambda>)r r rN r r r r- )ÚNr rQ r rµ r! Ú!test_TrioToken_run_sync_soon_FIFOè s r, c s~ d g fdd fdd} t | ¡ tdksBtd dksRtd d d ksftd d d
|
ksztd S ) Nc s> z | d ¡ W n$ tjk
|
r8 d| f¡ Y nX d S )Nr ú run finished)r r r rO ©rQ ©Ú
|
naughty_cbrP r r r! r0 ù s zFtest_TrioToken_run_sync_soon_starvation_resistance.<locals>.naughty_cbc s> t ¡ d¡ d¡ tdD ]} t ¡ I d H q&d S )Nr Ústartingé )r r r rO rN r* r r/ r r! r4 s
|
|
z@test_TrioToken_run_sync_soon_starvation_resistance.<locals>.mainr r r1 r r- é r! r< r r/ r! Ú2test_TrioToken_run_sync_soon_starvation_resistanceò s
|
r4 c s@ dfdd fddfdd} t | ¡ t d S )Nr c s d7 d S rÉ r r )Ú
|
cb_counterr r! r s z=test_TrioToken_run_sync_soon_threaded_stress_test.<locals>.cbc s6 z| ¡ t d¡ qW n tjk
|
r0 Y nX d S ©Nr )r ry r r r r )r r r! Ú stress_thread s
|
|
zHtest_TrioToken_run_sync_soon_threaded_stress_test.<locals>.stress_threadc sN t ¡ } tj| fd}| ¡ tdD ] } } |kr(tdI d H q0q(d S )N)Útargetr7 r ç{®Gáz?)r r Ú threadingÚThreadr rN r )r Úthreadr Z start_value)r5 r7 r r! r4 s z?test_TrioToken_run_sync_soon_threaded_stress_test.<locals>.main)r r, rX r< r )r r5 r7 r! Ú1test_TrioToken_run_sync_soon_threaded_stress_test s
|
r= c sX d} t ¡ }dg fdd}t| D ]}| ||¡ q&t I d H d | ksTtd S )Niè r c s$ d | kst d d7 < d S ©Nr r )r- r. ©Úcounterr r! r 6 s z6test_TrioToken_run_sync_soon_massive_queue.<locals>.cb)r r rN r r r- )ZCOUNTr r rQ r r? r! Ú*test_TrioToken_run_sync_soon_massive_queue+ s rA zPyPy 7.2 is buggy)Úreasonc sn g g fdd fdd} t tj¡}t | ¡ W 5 Q R X t|jjtksZt ddgksjt d S )Nc s: t ¡ } z
|
dV W 5 | dd ¡ | fdd¡ X d S )Nc S s i d S r% r r r r r! r" M r# zGtest_TrioToken_run_sync_soon_late_crash.<locals>.agen.<locals>.<lambda>c s
|
d¡S )Nú2nd ranr r rµ r r! r" N r# r )r r r r rµ r r! ÚagenH s
|
|
z5test_TrioToken_run_sync_soon_late_crash.<locals>.agenc s. ¡ d d ¡I d H d¡ d S )Néÿÿÿÿú main exiting)rO Úasendr ©rD rP Zsavedr r! r4 P s z5test_TrioToken_run_sync_soon_late_crash.<locals>.mainrF rC )
|
r. r/ r rç r, rb r6 rö r_ r- r8 r rH r! Ú'test_TrioToken_run_sync_soon_late_crashA s rI c
|
sf t ¡ T} | ¡ t t j¡4 t ¡ t ¡ fdd}t |¡I d H W 5 Q R X W 5 Q R X d S )Nc s" t | ¡} tj |¡ tjjS r )rm Úcapturer r rl r râ ©Z raise_cancelÚresult©rê r r r! Ú
|
slow_abortc s
|
z)test_slow_abort_basic.<locals>.slow_abort) r r r r. r/ rW rk r r$ )r rN r rM r! Útest_slow_abort_basic\ s
|
rO c sÎ g fdd} t ¡ ¬}t ¡ }t ¡ 4 I d H b}| | ¡ t I d H dgksZt| ¡ ddgksrt| ¡ ddgkstd|j_W 5 Q I d H R X ddddgks¶tW 5 Q R X W 5 Q R X d S )Nc sx t ¡ t ¡ fdd} t t j¡ d¡ t | ¡I d H W 5 Q R X d¡ t ¡ I d H d¡ d S )Nc s, d¡ t | ¡} tj|¡ tjjS )Nú abort-called)rO rm rJ r r rl r râ rK )rP rê r r r! rN r s
|
|
zDtest_slow_abort_edge_cases.<locals>.slow_aborter.<locals>.slow_abortrÀ rV Údone) r rk r r. r/ rW rO r$ r* )rN rµ rM r! Ú slow_abortern s
|
|
z0test_slow_abort_edge_cases.<locals>.slow_aborterrÀ rP TrV rQ ) r r rA rJ r r- r r r )rR Zouter1Zouter2rK r rµ r! Útest_slow_abort_edge_casesk s
|
|
|
rS c
|
s i i t jf fdd } fddt jffdd t ¡ 4 I d H }| | ¡ W 5 Q I d H R X ¡ D ]}|jd ksvtqvd S )Nc sH t ¡ d< d jg ks tt ¡ 4 I d H >}t ¡ 4 I d H }d j||gksXtW 5 Q I d H R X W 5 Q I d H R X d jg kstt ¡ 4 I d H }|d< | ¡I d H W 5 Q I d H R X d jd ksÚtd jd ksðtd jd kstd jd ks tt ¡ j}|d k rD|j}|j}q*d S )NÚparentrr rt )r rk Úchild_nurseriesr- rA r r r )rÔ Únursery1Únursery2rK Út)rr Ú nurseriesÚtasksr r! rT s" 6
|
|
z,test_task_tree_introspection.<locals>.parentc s t ¡ d< d j d gks$t d jtd hks@td j d gksXt d jtd hksttd jg kstd S )Nrt rT rr )r rk rU r- Ú child_tasksÚ frozensetr )rY rZ r r! rt ¶ s z,test_task_tree_introspection.<locals>.child2c
|
s¶ t ¡ }d< |jjd ks$t|jd k s6t|jd ksHt| ¡ |jd ksbt|jd ksptt ¡ I d H t ¡ 4 I d H }|d< | ¡ W 5 Q I d H R X d S )Nrr rT )
|
r rk r r r- r r´ r rA rJ )rÔ r÷ rK )rt rY rZ r r! rr ¾ s z,test_task_tree_introspection.<locals>.child1)r ÚTASK_STATUS_IGNOREDrA rJ Úvaluesr r- )rT rK rê r )rr rt rY rZ r! Ú test_task_tree_introspection s r_ c
|
sf fdd} dd t ¡ 4 I d H }| | |¡ W 5 Q I d H R X t t¡ | ¡ W 5 Q R X d S )Nc s | ¡ d S r )rJ r] ©rt r r! rr Ù s z$test_nursery_closure.<locals>.child1c Ó s d S r r r r r r! rt Þ s z$test_nursery_closure.<locals>.child2)r rA rJ r. r/ r> )rr rK r r` r! Útest_nursery_closureØ s ra c
|
à s dd } dd }t ¡ 4 I d H ^}|jt jfD ]J}|| d || d|d || ddd |t | d¡ || d t d q.W 5 Q I d H R X d S )
|
Nc Ó s t ¡ }| |jkstd S r )r rk rÎ r- )Úexpectedrê r r r! Úfunc1ê s z test_spawn_name.<locals>.func1c Ó s d S r r r r r r! Úfunc2î s z test_spawn_name.<locals>.func2rc rd rÍ Zfunc3Úobject)r rA rJ rï Ú functoolsÚpartialre )rc rd rK Zspawn_fnr r r! Útest_spawn_nameé s
|
rh c
|
à sþ t ¡ tkstt jddÈ}t jdd}t ¡ dks<td|_t ¡ dksRtd|_t ¡ dkshtd|_t ¡ dks~td|_t ¡ dkst| ¡ t ¡ t ks®td|_t ¡ dksÄtW 5 Q R X t ¡ t ksàtW 5 Q R X t ¡ tksútd S )Nr r r r TF)r Zcurrent_effective_deadliner r- r r r r )r~ rÅ rÆ r r r! Ú test_current_effective_deadlineú s$ ri c
|
C sz dd } dd }| |fD ]\}dd }t jtdd || W 5 Q R X d d
|
}t jtd d ||d W 5 Q R X qd S ) Nc W s t j| d S r r; ©r7 r r r! Ú bad_call_run s zBtest_nice_error_on_bad_calls_to_run_or_spawn.<locals>.bad_call_runc s fdd}t |¡ d S )Nc
|
s0 t ¡ 4 I d H } | j W 5 Q I d H R X d S r r\ r] rj r r! r4 s zRtest_nice_error_on_bad_calls_to_run_or_spawn.<locals>.bad_call_spawn.<locals>.mainr; )r7 r4 r rj r! Úbad_call_spawn s zDtest_nice_error_on_bad_calls_to_run_or_spawn.<locals>.bad_call_spawnc Ó s d S r r r r r r! Úf s z7test_nice_error_on_bad_calls_to_run_or_spawn.<locals>.fzexpecting an async functionrÑ c S s
|
| V d S r r )Úargr r r! Ú async_gen# s z?test_nice_error_on_bad_calls_to_run_or_spawn.<locals>.async_genz5expected an async function but got an async generatorr )r. r/ r0 )rk rl Zbad_callrm ro r r r! Ú,test_nice_error_on_bad_calls_to_run_or_spawn s ÿrp c sb dd fdd} t t¡}t | ¡ W 5 Q R X dt|jksFttdd |j D s^td S )Nc Ó s t I d H d S r )r r r r r! Ú child_xyzzy- s zCtest_calling_asyncio_function_gives_nice_error.<locals>.child_xyzzyc s I d H d S r r r ©rq r r! Ú misguided0 s zAtest_calling_asyncio_function_gives_nice_error.<locals>.misguidedÚasyncioc s s | ]}|j d kV qdS )rq NrÍ )rd Úentryr r r! Ú <genexpr>8 s zAtest_calling_asyncio_function_gives_nice_error.<locals>.<genexpr>)
|
r. r/ r0 r r, r? r6 r- ÚanyÚ traceback)rs r9 r rr r! Ú.test_calling_asyncio_function_gives_nice_error, s ÿry c à sl t t¡F} t ¡ 4 I d H $}dd l}| t¡ t I d H W 5 Q I d H R X W 5 Q R X dt | j
|
ksht d S )Nr rt ) r. r/ r0 r rA rt rJ r% r r? r6 r- )r9 rK rt r r r! Ú5test_asyncio_function_inside_nursery_does_not_explode= s
|
&rz c à s t t ¡ I d H W 5 Q R X t " t ¡ I d H t ¡ I d H W 5 Q R X t ( t ¡ 4 I d H W 5 Q I d H R X W 5 Q R X t ¡ } | ¡ t t
|
¡,}t ¡ 4 I d H
|
t W 5 Q I d H R X W 5 Q R X t |j jdksätdd |j jD t tjhkstW 5 Q R X d S )Nr c S s h | ] }t |qS r ra ©rd rL r r r! re Z s z&test_trivial_yields.<locals>.<setcomp>)r r r* r r¥ rA r r r. r/ r r_ r® r6 rf r- rW )r r9 r r r! Útest_trivial_yieldsH s$
|
þr| c à s dd }t ¡ 4 I d H ,}t t¡ | |¡I d H W 5 Q R X W 5 Q I d H R X t jddd}dD ]}t ¡ 4 I d H `}t|jdkst t
|
¡ }| ||¡I d H |ks®t t
|
¡ | |ksÂt t|jdksÔt W 5 Q I d H R X t
|
¡ | d | ksdt qdt
|
¡ }|d
|
I d H t
|
¡ | d ks*t t jfd d }t ¡ 4 I d H }| |¡I d H W 5 Q I d H R X t jfdd}t ¡ 4 I d H ,}t t ¡ | |¡I d H W 5 Q R X W 5 Q I d H R X t jfdd}t ¡ 4 I d H @}t t ¡} | |¡I d H W 5 Q R X dt | jkst W 5 Q I d H R X t jfdd}
|
t ¡ 4 I d H J}t ¡ 6} | ¡ t t j¡ | |
|
¡I d H W 5 Q R X W 5 Q R X W 5 Q I d H R X t jfdd} t ¡ 4 I d H H}t ¡ 4} | ¡ t t¡} | | ¡I d H W 5 Q R X W 5 Q R X W 5 Q I d H R X dd | jjD t jt hks2t t ¡ 4 I d H } W 5 Q I d H R X t
|
¡ }t t ¡ | |d¡I d H W 5 Q R X t
|
¡ |kst d S )Nc Ó s d S r r r r r r! Úno_argsa s z#test_nursery_start.<locals>.no_args©rÔ c Ó s2 t | t| I d H | | ¡ t| I d H d S r )r r r´ )ÚsecondsrÔ r r r! Úsleep_then_starti s
|
z,test_nursery_start.<locals>.sleep_then_start)r r r r r r é c Ó s* | ¡ t t¡ | ¡ W 5 Q R X d S r )r´ r. r/ r> r~ r r r! Údouble_started s z*test_nursery_start.<locals>.double_startedc Ó s t dd S )NZoopsr« r~ r r r! Úraise_keyerror s z*test_nursery_start.<locals>.raise_keyerrorc Ó s d S r r r~ r r r! Únothing s z#test_nursery_start.<locals>.nothingzexited without callingc Ó s | d¡ d S )NÚhi)r´ r~ r r r! Ú just_started s z(test_nursery_start.<locals>.just_startedc Ó s | ¡ tdd S )NZ whoopsiedaisy)r´ r_ r~ r r r! Ú raise_keyerror_after_started© s z8test_nursery_start.<locals>.raise_keyerror_after_startedc S s h | ] }t |qS r ra r{ r r r! re ² s z%test_nursery_start.<locals>.<setcomp>é )r rA r. r/ r0 r r] r® r[ r- rx r_ r> r? r6 r r rW r rf )Úautojump_clockr} rK r r Út0r r r r9 r Úcsr Zclosed_nurseryr r r! Útest_nursery_start` sd * " * $
|
4
|
4þ r c à s¸ t ¡ } | jg kstt ¡ 4 I d H t}| j|gks8tt t¡> t ¡ 4 I d H }| j||gkshttW 5 Q I d H R X W 5 Q R X | j|gkstW 5 Q I d H R X | jg ks´td S r )r rk Z_child_nurseriesr- rA r. r/ r_ )rê rV rW r r r! Útest_task_nursery_stackÀ s r c
|
à s¼ t jfdd} t jddd}t ¡ 4 I d H ,}| | ¡I d H }| ||jj¡I d H W 5 Q I d H R X t ¡ 4 I d H 6}| | ¡I d H }| |dd ¡I d H |j ¡ W 5 Q I d H R X d S )Nc
|
Ó s< t ¡ 4 I d H }| |¡ t I d H W 5 Q I d H R X d S r )r rA r´ r% )rÔ rK r r r! Ú setup_nurseryÐ s
|
z@test_nursery_start_with_cancelled_nursery.<locals>.setup_nurseryr~ c
|
Ó sT t ¡ 4 I d H 4}| t¡ | t¡ t I d H | | ¡ W 5 Q I d H R X d S r )r rA rJ r% r r´ )ÚfnrÔ rK r r r! Úsleeping_childrenØ s
|
|
zDtest_nursery_start_with_cancelled_nursery.<locals>.sleeping_childrenc S s d S r r r r r r! r" ê r# z;test_nursery_start_with_cancelled_nursery.<locals>.<lambda>)r r] rA r r r )r r rK Ztarget_nurseryr r r! Ú)test_nursery_start_with_cancelled_nurseryÍ s ÿr c s0 t jfdd}t ¡ 4 I d H n}t ¡ }t ¡ 4 I d H 0}| |j|¡ tdI d H | |j|¡ W 5 Q I d H R X t ¡ | dkstW 5 Q I d H R X t jfdd fdd}t ¡ 4 I d H ^}t ¡ }t ¡ 4 I d H }| ||¡ t I d H W 5 Q I d H R X t ¡ | d ks tW 5 Q I d H R X d S )
|
Nc Ó s( t dI d H | ¡ t dI d H d S )Nr r )r r´ r~ r r r! Ú sleep_a_bitï s z:test_nursery_start_keeps_nursery_open.<locals>.sleep_a_bitr r c Ó s t dI d H td S )Nr )r r_ r~ r r r! Úsleep_then_crash s z?test_nursery_start_keeps_nursery_open.<locals>.sleep_then_crashc s* t t¡ | ¡I d H W 5 Q R X d S r )r. r/ r_ r r] ©r r r! Ústart_sleep_then_crash s zEtest_nursery_start_keeps_nursery_open.<locals>.start_sleep_then_crashr ) r r] rA rx rJ r r r- r )r r rV r rW r r r r! Ú%test_nursery_start_keeps_nursery_openî s $ r c à sB t t¡. t ¡ 4 I d H t W 5 Q I d H R X W 5 Q R X d S r )r. r/ r_ r rA r r r r! Ú test_nursery_explicit_exception s r c
|
à s dd } z4t ¡ 4 I d H }| | ¡ tW 5 Q I d H R X W n> tk
|
rz } z ttt|jtt fksjt
|
W 5 d }~X Y nX d S )Nc Ó s t d S r r2 r r r r! Úfail s z)test_nursery_stop_iteration.<locals>.fail) r rA rJ Ú StopIterationr ÚtupleÚmaprb rf r3 r- )r rK rL r r r! Útest_nursery_stop_iteration s
|
r c à sh G dd d} G dd d}g }|| d| d2 z3 d H W }| |¡ q26 |ddgddggksdtd S ) Nc @ s e Zd Zdd Zdd ZdS )z-test_nursery_stop_async_iteration.<locals>.itc S s || _ d| _d S r6 )ÚcountÚval)Úselfr r r r! Ú__init__) s z6test_nursery_stop_async_iteration.<locals>.it.__init__c Ó s4 t dI d H | j}|| jkr"t| jd7 _|S r> )r r r ÚStopAsyncIteration)r r r r r! Ú __anext__- s
|
z7test_nursery_stop_async_iteration.<locals>.it.__anext__N)Ú__name__Ú
|
__module__Ú __qualname__r r¢ r r r r! Úit( s r¦ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
z4test_nursery_stop_async_iteration.<locals>.async_zipc W s dd |D | _ d S )Nc S s g | ]
|
}|j qS r )r¢ )rd Úobjr r r! Ú
|
<listcomp>7 s zQtest_nursery_stop_async_iteration.<locals>.async_zip.__init__.<locals>.<listcomp>)Únexts)r Úlargsr r r! r 6 s z=test_nursery_stop_async_iteration.<locals>.async_zip.__init__c Ó s | I d H ||< d S r r )r rm ÚitemsrQ r r r! Ú _accumulate9 s z@test_nursery_stop_async_iteration.<locals>.async_zip._accumulatec S s | S r r )r r r r! Ú __aiter__< s z>test_nursery_stop_async_iteration.<locals>.async_zip.__aiter__c
|
Ó s^ | j }d gt| }t ¡ 4 I d H *}t|D ]\}}| | j|||¡ q.W 5 Q I d H R X |S r )r© r® r rA Ú enumeraterJ r¬ )r r© r« rK rQ rm r r r! r¢ ? s $z>test_nursery_stop_async_iteration.<locals>.async_zip.__anext__N)r£ r¤ r¥ r r¬ r r¢ r r r r! Ú async_zip5 s r¯ r r r r )rO r- )r¦ r¯ rL Úvalsr r r! Ú!test_nursery_stop_async_iteration' s r± c
|
à s dd } z:t ¡ 4 I d H }| | ¡ | | ¡ W 5 Q I d H R X W nR tk
|
r } z4|jd }t|tslt|jj }|j
|
| j kstW 5 d }~X Y nX d S )Nc Ó s
|
t d S r r« r r r r! Ú my_child_taskP s z3test_traceback_frame_removal.<locals>.my_child_taskr ) r rA rJ r rf rÜ r_ r- Ú __traceback__Útb_frameÚf_codeÚ__code__)r² rK rE Z first_excÚframer r r! Ú test_traceback_frame_removalO s
|
|
r¸ c sN t d¡ d¡ ¡ dks$t fdd} t | ¡ ¡ dksJtd S )NÚtestÚbeforec sl t ¡ } | j ¡dks t ¡ dks,t d¡ ¡ dksFt | jksTt| j ¡dkshtd S )Nrº Úafter)r rk ÚcontextÚgetr- r¾ ré ©Úvarr r! r¸ m s
|
z&test_contextvar_support.<locals>.inner)Ú contextvarsÚ
|
ContextVarr¾ r½ r- r r, ©r¸ r r¾ r! Útest_contextvar_supportg s
|
|
|
rà c
|
s t jddd fdd} fdd}t ¡ 4 I d H L}| | ¡ t I d H ¡ dks^t d¡ | |¡ t I d H W 5 Q I d H R X d S ) Nr¹ Úhmmm)Údefaultc s. ¡ dkst d¡ ¡ dks*td S )NrÄ Úhmmmm)r½ r- r¾ r r¾ r r! rp } s
|
z%test_contextvar_multitask.<locals>.t1c s ¡ dkstd S )NrÆ ©r½ r- r r¾ r r! rq s z%test_contextvar_multitask.<locals>.t2rÆ ) rÀ rÁ r rA rJ r r½ r- r¾ )rp rq Únr r¾ r! Útest_contextvar_multitaskz s
|
|
|
rÉ c sJ t d¡ d¡ fdd fdd fdd} t | ¡ d S ) NZqwilfishÚwaterc s ¡ dkstd S )NrÊ rÇ r ©Úcvarr r! rî s z.test_system_task_contexts.<locals>.system_taskc s ¡ dkstd S ©NZpoisonrÇ r rË r r! Ú regular_task s z/test_system_task_contexts.<locals>.regular_taskc
|
sP t ¡ 4 I d H 0} d¡ | ¡ t ¡ t I d H W 5 Q I d H R X d S rÍ )r rA r¾ rJ rï r r] ©rÌ rÎ rî r r! r¸ s
|
|
|
|
z(test_system_task_contexts.<locals>.inner)rÀ rÁ r¾ r r, r r rÏ r! Útest_system_task_contexts s
|
|
rÐ c C s( t t¡ tj d d ¡ W 5 Q R X d S r ©r. r/ r0 r Ú_runZNurseryr r r r! Útest_Nursery_init¢ s rÓ c
|
à s4 t ¡ 4 I d H } d| jks tW 5 Q I d H R X d S )NF)r rA Ú_closedr- r] r r r! Útest_Nursery_private_init§ s rÕ c C s. t t¡ G dd dtjj} W 5 Q R X d S )Nc @ s e Zd ZdS )z'test_Nursery_subclass.<locals>.SubclassN©r£ r¤ r¥ r r r r! ÚSubclass° s r× rÑ ©r× r r r! Útest_Nursery_subclass s rÙ c C sH t t¡ tjW 5 Q R X t t¡ t ¡ W 5 Q R X tj ¡ d S r )r. r/ r0 r rW Ú_creater r r r! Útest_Cancelled_init´ s
|
rÛ c C s t j ¡ } t| dkstd S )NrW )r rW rÚ r? r- )rV r r r! Útest_Cancelled_str¿ s
|
rÜ c C s, t t¡ G dd dtj} W 5 Q R X d S )Nc @ s e Zd ZdS )z)test_Cancelled_subclass.<locals>.SubclassNrÖ r r r r! r× Ç s r× )r. r/ r0 r rW rØ r r r! Útest_Cancelled_subclassÄ s rÝ c C s, t t¡ G dd dtj} W 5 Q R X d S )Nc @ s e Zd ZdS )z+test_CancelScope_subclass.<locals>.SubclassNrÖ r r r r! r× Î s r× )r. r/ r0 r r rØ r r r! Útest_CancelScope_subclassË s rÞ c C sV t tj¡ t ¡ W 5 Q R X dd } t | ¡ t tj¡ t ¡ W 5 Q R X d S )Nc Ó s t ¡ dkstd S )NZtrio)ÚsniffioÚcurrent_async_libraryr- r r r r! Úcheck_inside_trioÖ s z3test_sniffio_integration.<locals>.check_inside_trio)r. r/ rß ZAsyncLibraryNotFoundErrorrà r r, )rá r r r! Útest_sniffio_integrationÒ s
|
râ c à sJ t ¡ } | jd kstd| _| jdks*tt ¡ I d H | jd ksFtd S rÉ )r rk Zcustom_sleep_datar- r* ré r r r! Útest_Task_custom_sleep_dataß s rã c c s
|
| V d S r r )r6 r r r! Ú async_yieldè s rä c s d d fdd} t ¡ 4 I d H }| | t d ¡d¡ W 5 Q I d H R X d ksXtj d¡dkslt t d¡ks~tt t
|
¡ j d ¡ W 5 Q R X d d t t ¡> t ¡ 4 I d H }| | t t ¡d¡ W 5 Q I d H R X W 5 Q R X t }j |¡dkst t |¡ks&tt t
|
¡ j d ¡ W 5 Q R X dd }t ¡ 4 I d H }| |¡ W 5 Q I d H R X d S )Nc s< t dI d H t ¡ t tj| ¡I d H t|I d H d S r6 )r r rk rm ZacaptureÚ#permanently_detach_coroutine_objecträ )Z task_outcomeZ yield_value©Z pdco_outcomerê r r! Údetachable_coroutineñ s ÿ
|
zFtest_permanently_detach_coroutine_object.<locals>.detachable_coroutinez I'm free!zbe free!rG c
|
Ó sd t ¡ 4 I d H D t t¡ } t t d ¡¡I d H W 5 Q R X dt| j ksPt
|
W 5 Q I d H R X d S )Nz open nurser) r rA r. r/ r> rå rm rs r? r6 r- r r r r! Ú
|
bad_detach s z<test_permanently_detach_coroutine_object.<locals>.bad_detach)r rA rJ rm rs r- ÚcoroÚsendr. r/ r r_ rn r3 Úthrow)rç rK Zthrow_inrè r ræ r! Ú(test_permanently_detach_coroutine_objectí s. $ 0 rì c
|
s¼ d d fdd} fdd}t ¡ 4 I d H z}| | ¡ | |¡ t I d H d k s`t d k slt j d¡dkst j d ¡dkst j d ¡dks¨tW 5 Q I d H R X d S ) Nc s t ¡ d S r )r rk r )Úunrelated_taskr r! Úunrelated_coroutine s zFtest_detach_and_reattach_coroutine_object.<locals>.unrelated_coroutinec s´ t dI d H t ¡ dd } t | ¡I d H }|dks:ttdI d H tdI d H t t¡}t d ¡I d H W 5 Q R X dt
|
|j kstt d¡I d H t dI d H d S ) Nr c S s t jjS r rá r r r r! Úabort_fn) s z[test_detach_and_reattach_coroutine_object.<locals>.reattachable_coroutine.<locals>.abort_fnú not trio!r r zdoes not matchÚbyebye) r r rk Ú#temporarily_detach_coroutine_objectr- rä r. r/ r> Ú"reattach_detached_coroutine_objectr? r6 )rï Úgotr9 ©rê rí r r! Úreattachable_coroutine# s zItest_detach_and_reattach_coroutine_object.<locals>.reattachable_coroutinerð r r rñ )r rA rJ r r- ré rê )rî rö rK r rõ r! Ú)test_detach_and_reattach_coroutine_object s
|
|
r÷ c
|
s| d d fdd} t ¡ 4 I d H >}| | ¡ t I d H d k sJt|j ¡ j d ¡ W 5 Q I d H R X sxtd S )NFc sn t dI d H t ¡ fdd} t | ¡I d H t d ¡I d H t tj¡ t dI d H W 5 Q R X d S )Nr c s d t jjS rÄ rá r )Úabort_fn_calledr r! rï S s zVtest_detached_coroutine_cancellation.<locals>.reattachable_coroutine.<locals>.abort_fn)r r rk rò ró r. r/ rW )rï ©rø rê r r! rö M s zDtest_detached_coroutine_cancellation.<locals>.reattachable_coroutine) r rA rJ r r- r r ré rê )rö rK r rù r! Ú$test_detached_coroutine_cancellationI s
|
|
rú c sH dd g } | }t |j¡ | dgks.t fdd}t |¡ d S )Nc S s t ¡ r t| d¡ d V d S ©Núthe generator ran)r rí r- rO rµ r r r! Úagen_fnl s
|
z5test_async_function_implemented_in_C.<locals>.agen_fnrü c
|
sL g } | }t ¡ 4 I d H }| |j¡ W 5 Q I d H R X | dgksHtd S rû )r rA rJ r¢ r- )Zstart_soon_recordrD rK ©rý r r! r4 v s
|
z2test_async_function_implemented_in_C.<locals>.main)r r, r¢ r- )Z
|
run_recordrD r4 r rþ r! Ú$test_async_function_implemented_in_Cg s rÿ c à sL t <} t ¡ }| |¡ tdD ]}| t ¡ ¡ q"| ¡ W 5 Q R X d S )Ni )r r r rÖ rN r )Z
|
exit_stackZoutermost_scoper r r r! Ú#test_very_deep_cancel_scope_nesting s
|
r c à sT t ¡ } t ¡ :}ttd D ]}| d |_| d |_q tdI d H W 5 Q R X d S )Nr i' i' r9 )r rx r rN r r r )ÚnowÚcscoper r r r! Ú%test_cancel_scope_deadline_duplicates s
|
|
r Úcpythonz$Only makes sense with refcounting GCc à sÀ t ¡ dd } dd }t ¡ }zt ¡ t t j¡ | I d H | I d H t t ¡2 t
|
¡ 4 I d H }| |¡ W 5 Q I d H R X W 5 Q R X t ¡ t jr¢t W 5 t |¡ t j ¡ X d S )Nc Ó s, t ¡ } | ¡ t I d H W 5 Q R X d S r )r r r r% )r r r r! Ú do_a_cancel s
|
zPtest_simple_cancel_scope_usage_doesnt_create_cyclic_garbage.<locals>.do_a_cancelc Ó s t d S r r2 r r r r! r[ ¡ s zLtest_simple_cancel_scope_usage_doesnt_create_cyclic_garbage.<locals>.crasher)ÚgcÚcollectÚ get_debugÚ set_debugÚgarbageÚclearÚ DEBUG_SAVEALLr. r/ r3 r rA rJ r- )r r[ Ú old_flagsrK r r r! Ú;test_simple_cancel_scope_usage_doesnt_create_cyclic_garbage s $
|
r c à sÆ t ¡ dd } t ¡ }zt t¡l t
|
¡ X}t ¡ 4 I d H 6}t ¡ t t j ¡ | t¡ | ¡ | | ¡ W 5 Q I d H R X W 5 Q R X W 5 Q R X t ¡ t jr¨tW 5 t |¡ t j ¡ X d S )Nc Ó s t d S r r2 r r r r! r[ Ã s zDtest_cancel_scope_exit_doesnt_create_cyclic_garbage.<locals>.crasher)r r r r r
|
r r. r/ r3 r r rA r rJ r% r r- )r[ r r± rK r r r! Ú3test_cancel_scope_exit_doesnt_create_cyclic_garbage¼ s
|
.
|
r c
|
sÀ fdd} d t ¡ t ¡ }zt d¡ t ¡ t t j¡ t ¡ 4 I d H }|j
|
¡ W 5 Q I d H R X t || ¡ ~t ¡ I d H stt ¡ t jr¢tW 5 t |¡ t j ¡ X d S )Nc s d d S rÄ r r ©Z collectedr r! Útoggle_collectedâ s zJtest_nursery_cancel_doesnt_create_cyclic_garbage.<locals>.toggle_collectedFr )r r r r r
|
r r r rA r r ÚweakrefÚfinalizer* r- )r r rK r r r! Ú0test_nursery_cancel_doesnt_create_cyclic_garbageÝ s$
|
|
r c
|
sn dfddG dd d fdd} t ¡ 4 I d H }| | ¡ |j ¡ W 5 Q I d H R X sjtd S )NFc s d d S rÄ r r )Ú destroyedr r! Ú finalizer s z;test_locals_destroyed_promptly_on_cancel.<locals>.finalizerc @ s e Zd ZdS )z3test_locals_destroyed_promptly_on_cancel.<locals>.ANrÖ r r r r! ÚA
|
s r c s$ } t | ¡ t ¡ I d H d S r )r r r r* )rS )r r r r! rê s z6test_locals_destroyed_promptly_on_cancel.<locals>.task)r rA rJ r r r- )rê rK r )r r r r! Ú(test_locals_destroyed_promptly_on_cancel s
|
r c C sr dd } t t¡}tj| dd W 5 Q R X t|jjdks@tt |jjd t
|
ksXt|jjd j dksntdS ) z\
|
Test that nurseries respect the global context setting of strict_exception_groups.
|
c
|
Ó s. t ¡ 4 I d H tdW 5 Q I d H R X d S )NÚfoo©r rA r r r r r! r4 s z.test_run_strict_exception_groups.<locals>.mainT©Zstrict_exception_groupsr r ©r N) r. r/ r r r, r® r6 rf r- rb r r7 )r4 rE r r r! Ú test_run_strict_exception_groups s r c C s4 dd } t jtdd tj| dd W 5 Q R X dS )ze
|
Test that a nursery can override the global context setting of
|
strict_exception_groups.
|
c
|
Ó s2 t jdd4 I d H tdW 5 Q I d H R X d S )NFr r r r r r r! r4 / s z?test_run_strict_exception_groups_nursery_override.<locals>.mainr rÑ Tr N)r. r/ r r r, r< r r r! Ú1test_run_strict_exception_groups_nursery_override) s r c à s t t¡4} tjdd4 I dH tdW 5 Q I dH R X W 5 Q R X t| jjdksXt t
|
| jjd tkspt | jjd j dkst dS )zHTest that strict exception groups can be enabled on a per-nursery basis.Tr Nr r r r ) r. r/ r r rA r r® r6 rf r- rb r7 rH r r r! Ú$test_nursery_strict_exception_groups7 s "r c à s dd } t t¡}t ¡ 4 I dH ^}| t¡ | | ¡ tjdd4 I dH $}| t¡ | | ¡ |j ¡ W 5 Q I dH R X W 5 Q I dH R X W 5 Q R X |j j
|
}t |dks°t t |d tsÂt t |d tsÔt t |d j
|
dksêt t |d j
|
d tst dS ) z
|
Test that a single exception from a nested nursery with strict semantics doesn't get
|
collapsed when CancelledErrors are stripped from it.
|
c Ó s t dd S ©Nz
|
test error©r> r r r r! Ú raise_errorH s z1test_nursery_collapse_strict.<locals>.raise_errorNTr r r r ©r. r/ r r rA rJ r% r r r6 rf r® r- rÜ r> ©r" rE rK rW rf r r r! Ú test_nursery_collapse_strictB s
|
|
|
|
4r% c à sÔ dd } t t¡|}t ¡ 4 I dH Z}| t¡ | | ¡ t ¡ 4 I dH $}| t¡ | | ¡ |j ¡ W 5 Q I dH R X W 5 Q I dH R X W 5 Q R X |j j
|
}t |dks¬t t |d ts¾t t |d tsÐt dS )z
|
Test that a single exception from a nested nursery with loose semantics gets
|
collapsed when CancelledErrors are stripped from it.
|
c Ó s t dd S r r! r r r r! r" b s z0test_nursery_collapse_loose.<locals>.raise_errorNr r r r# r$ r r r! Útest_nursery_collapse_loose\ s
|
|
|
|
4r& c
|
à sR t t¡4 t ¡ } | ¡ tdt t gW 5 Q R X W 5 Q R X | jrNtdS )z¥
|
Test that when a cancel scope encounters an exception group that does NOT contain
|
a Cancelled exception, it will NOT set the ``cancelled_caught`` flag.
|
r¹ N) r. r/ r r r r r> r© r- r¡ r r r! Ú#test_cancel_scope_no_cancellederrort s
|
|
&r' )rÀ rf ÚplatformrÚ r: ry ÚtypesÚwarningsr Ú
|
contextlibr r Úmathr Útextwrapr r Úattrrm rß r. Ztutilr r r r
|
r r r Ú r Z_core._multierrorr r rÒ r Ú_threadsr Z _timeoutsr r Ztestingr r r Ú version_infoZexceptiongroupr r% r1 r: r@ rD rF rM rU r^ rg ri rj rv 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 r r r r r r r" r$ r) r, r4 r= rA ÚmarkZskipifrI rO rS r_ ra rh ri rp ry rz r| r r r r r r r± r¸ rà rÉ rÐ rÓ rÕ rÙ rÛ rÜ rÝ rÞ râ rã Ú coroutinerä rì r÷ rú rÿ r r ÚimplementationrÎ r r r r r r r r% r& r' r r r r! Ú<module> s2 $
|
|
|
4&+) %' JD
|
|
|
=
|
|
+B ` !' (
|
..
|
|
|
ÿ
|
$
|
ÿ
|
|
ÿ
|
|
ÿ
|
|
