U
|
Z ±dZb ã @ sÎ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
|
Z
|
m Z ddl m Z ddl mZmZmZ ddlmZ ddlmZ dd lmZmZmZmZ dd
|
lmZ d d Z d d Z dd Z dd Z dd Z dd Z!dd Z"ej# $dddd g¡ej# $d d d g¡ej# $d d d g¡d!d" Z%d#d$ Z&d%d& Z'd'd( Z(d)d* Z)d+d, Z*e +d-¡Z,d.d/ Z-d0d1 Z.d2d3 Z/d4d5 Z0d6d7 Z1d8d9 Z2d:d; Z3d<d= Z4d>d? Z5d@dA Z6ej#j7edBdCdDdE Z8dFdG Z9dHdI Z:dS )Jé N)Úcurrent_async_library_cvar)Ú TrioTokenÚcurrent_trio_tokené )Ú_core)ÚEventÚCapacityLimiterÚsleep)Úwait_all_tasks_blocked)Úbuggy_pypy_asyncgens)Úto_thread_run_syncÚ current_default_thread_limiterÚfrom_thread_runÚfrom_thread_run_sync)Úki_selfc s¢ t ¡ d fdd } t ¡ }dd }| t|d|dI d H dd }| t|dtf|dI d H d d }| t|d
|
|dI d H d d }| t|dtf|dI d H d S ) Nc sj g fdd}t j|dd}| ¡ | ¡ rLtd tdI d H q,d|fdf|gksftd S ) Nc
|
st z2 dt ¡ f¡ d} d| f¡ W n< tk
|
rn } z t| dt|f¡ W 5 d }~X Y nX d S )NÚstart©Ú
|
trio_tokenÚgotÚerror)ÚappendÚ threadingÚcurrent_threadÚ BaseExceptionÚprintÚtype)ÚxÚexc©Údo_in_trio_threadÚfnÚrecordr © úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_threads.pyÚthreadfn s z<test_do_in_trio_thread.<locals>.check_case.<locals>.threadfnT)ÚtargetÚdaemonZyawnç{®Gáz?r Úf)r ÚThreadr Úis_aliver r ÚAssertionError)r r Úexpectedr r$ Ú child_thread©Ú trio_threadr r# Ú
|
check_case s z*test_do_in_trio_thread.<locals>.check_casec S s" t ¡ r t| dt ¡ f¡ dS )Nr( r )r Úcurrently_ki_protectedr+ r r r ©r! r" r" r# r( 1 s z!test_do_in_trio_thread.<locals>.f)r r r c S s& t ¡ r t| dt ¡ f¡ td S ©Nr( )r r1 r+ r r r Ú
|
ValueErrorr2 r" r" r# r( 8 s r c Ó s0 t ¡ r tt ¡ I d H | dt ¡ f¡ dS )Nr( é )r r1 r+ Ú
|
checkpointr r r r2 r" r" r# r( ? s )r r5 c Ó s4 t ¡ r tt ¡ I d H | dt ¡ f¡ td S r3 )r r1 r+ r6 r r r ÚKeyErrorr2 r" r" r# r( G s )N)r r r r r r4 r r7 )r0 Útokenr( r" r. r# Útest_do_in_trio_thread s r9 c à sL t t¡ tdd W 5 Q R X dd } t t¡ t| W 5 Q R X d S )Nc S s d S ©Nr" r" r" r" r# Ú<lambda>R ó z9test_do_in_trio_thread_from_trio_thread.<locals>.<lambda>c Ó s d S r: r" r" r" r" r# ÚfooT s z4test_do_in_trio_thread_from_trio_thread.<locals>.foo)ÚpytestÚraisesÚ RuntimeErrorr r )r= r" r" r# Ú'test_do_in_trio_thread_from_trio_threadP s
|
rA c s0 t fdd} t | ¡ ddhks,td S )Nc s~ t ¡ dd fdd fdd} tj| d}| ¡ td | ¡ rbtd I d H qJtd
|
| ¡ td d S ) Nc
|
S sD t d t ¡ rtt d z
|
t W 5 dd l} t d| ¡ X d S )Nzin Trio threadr r Úfinally)r r r1 r+ ÚsysÚexc_infor )rC r" r" r# Útrio_thread_fnc s
|
zTtest_run_in_trio_thread_ki.<locals>.check_run_in_trio_thread.<locals>.trio_thread_fnc s
|
d S r: r" r" )rE r" r# Útrio_thread_afnn s zUtest_run_in_trio_thread_ki.<locals>.check_run_in_trio_thread.<locals>.trio_thread_afnc s| zt d td W n& tk
|
r> t d d¡ Y nX ztd W n& tk
|
rv t d d¡ Y nX d S )NÚrunningr Úok1Úok2)r r ÚKeyboardInterruptÚaddr r" )r! r8 rF rE r" r# Úexternal_thread_fnq s zXtest_run_in_trio_thread_ki.<locals>.check_run_in_trio_thread.<locals>.external_thread_fn)r% Zwaitingr' zwaited, joiningÚdone) r r r r) r r r* r Újoin)rL Úthreadr2 )r8 rF rE r# Úcheck_run_in_trio_thread` s z<test_run_in_trio_thread_ki.<locals>.check_run_in_trio_threadrH rI )Úsetr Úrunr+ )rP r" r2 r# Útest_run_in_trio_thread_ki[ s '
|
rS c s\ g t fddfdd fdd} t | ¡}| ¡ ddgksXtd S ) Nc s* d¡ ¡ t dd ¡I d H d S )NÚsleepingc S s t jjS r: )r ZAbortZ SUCCEEDED)Ú_r" r" r# r; r< zMtest_await_in_trio_thread_while_main_exits.<locals>.trio_fn.<locals>.<lambda>)r rQ r Zwait_task_rescheduledr" )Úevr! r" r# Útrio_fn s
|
z;test_await_in_trio_thread_while_main_exits.<locals>.trio_fnc s6 zt | d W n tjk
|
r0 d¡ Y nX d S )Nr Ú cancelled)r r Z Cancelledr )r8 )r! rW r" r# Ú thread_fn s z=test_await_in_trio_thread_while_main_exits.<locals>.thread_fnc s@ t ¡ } tj| fd}| ¡ ¡ I d H dgks<t|S )N)r% ÚargsrT )r r r r) r Úwaitr+ )r8 rO )rV r! rY r" r# Úmain s z8test_await_in_trio_thread_while_main_exits.<locals>.mainrT rX )r r rR rN r+ )r\ rO r" )rV r! rY rW r# Ú*test_await_in_trio_thread_while_main_exits s
|
r] c à s t ¡ } dd }t|dI d H \}}|dks0t|| ks<tdd }t t¡}t|I d H W 5 Q R X t|jj |jj d | kstd S )Nc S s | t ¡ fS r: )r r )r r" r" r# r( ª s z$test_run_in_worker_thread.<locals>.fé c S s t t ¡ d S r: )r4 r r r" r" r" r# Úg± s z$test_run_in_worker_thread.<locals>.gr )
|
r r r r+ r> r? r4 r ÚvaluerZ )r/ r( r r- r_ Úexcinfor" r" r# Útest_run_in_worker_thread§ s rb c
|
s d gfdd fdd} g t ¡ }t ¡ 4 I d H 8}| | |d¡ t I d H dgksft|j ¡ W 5 Q I d H R X d dkst| d ¡ d dkr²t
|
d ¡ qg d d< t ¡ 4 I d H z}| | |d
|
¡ t I d H |j ¡ tj dd $ t d D ]}t ¡ I d H q
|
W 5 Q R X dgks:t| d ¡ W 5 Q I d H R X t ¡ }| ¡ | |d
|
I d H W 5 Q R X |jstd S ) Nc s d d< | ¡ d d< d S )NÚblockingr Úfinished)Úget)Úq)Úregisterr" r# r( ½ s z1test_run_in_worker_thread_cancellation.<locals>.fc s4 d¡ zt | |dI d H W ¢S d¡ X d S )Nr Úexit©Ú cancellable)r r )rf rj )r( r! r" r# Úchildà s
|
z5test_run_in_worker_thread_cancellation.<locals>.childTr r rd r' F©Zshieldé
|
)Ú stdlib_queueÚQueuer Ú open_nurseryÚ
|
start_soonr
|
r+ Ú cancel_scopeÚcancelÚputÚtimer Ú CancelScopeÚranger6 Úcancelled_caught)rk rf ÚnurseryrU Zscoper" )r( r! rg r# Ú&test_run_in_worker_thread_cancellationº s:
|
|
|
rz c s | tjdd¡ t ¡ t ¡ fddfdd}t |¡ d ¡ ¡ }| ¡ rjt
|
d¡ qV| ¡ \}}d|kst d|kst d S )NZ IDLE_TIMEOUTr' c s ¡ t ¡ ¡ d S r: )re rt r r r" )Úq1Úq2r" r# rY ü s z6test_run_in_worker_thread_abandoned.<locals>.thread_fnc
|
sR fdd} t ¡ 4 I d H &}| | ¡ t I d H |j ¡ W 5 Q I d H R X d S )Nc s t ddI d H d S )NTri )r r" ©rY r" r# rk s z@test_run_in_worker_thread_abandoned.<locals>.main.<locals>.child)r rp rq r
|
rr rs )rk ry r} r" r# r\ s
|
|
z1test_run_in_worker_thread_abandoned.<locals>.mainzException in thread) Úsetattrr Ú _thread_cachern ro rR rt re r* ru r Z
|
readouterrr+ )ZcapfdÚ monkeypatchr\ rO ÚoutÚerrr" )r{ r| rY r# Ú#test_run_in_worker_thread_abandonedö s
|
|
r ÚMAXr5 é rm rs FTÚuse_default_limiterc
|
|
s d| }t ¡ t ¡ |r4t }|j}| |_d nt| }| }|znG dd dd_d_d_d_ t
|
¡ fdd fdd}t
|
¡ 4 I d H ª}t d g }t|D ],} | t ¡ | ||d
|
¡ t I d H q¾ rt d |d ¡ I d H t I d H j | ks4| ¡ j| krFtd I d H q ¡ W 5 Q I d H R X j| ksnt r|jdkrtd I d H qtj|ks¢tjdks²tW 5 ||_X d S ) Nr c @ s e Zd ZdS )z0test_run_in_worker_thread_limiter.<locals>.stateN)Ú__name__Ú
|
__module__Ú __qualname__r" r" r" r# Ústate9 s r r c s t d t| jd @ jd7 _ jd7 _tjj_ jd7 _W 5 Q R X ¡ " jd8 _ jd8 _W 5 Q R X t d d S )Nzthread_fn startr r^ zthread_fn exiting) r r rs ÚranrG ÚmaxÚ
|
high_waterÚparkedr[ )rr )ÚgateÚlockr r8 r" r# rY C s z4test_run_in_worker_thread_limiter.<locals>.thread_fnc sB t ¡ }t| dI d H W 5 Q R X td|j | ¡ d S )N)Úlimiterrj z run_thread finished, cancelled:)r rv r r rx rQ )Úeventrr )rs Ú limiter_argrY r" r# Ú
|
run_threadS s
|
ÿ z5test_run_in_worker_thread_limiter.<locals>.run_threadZspawningéÿÿÿÿz'waiting for first cancellation to clearr' )r r ÚLockr Z total_tokensr r r rG r r r rp r rw r rq r
|
r[ Z
|
statisticsZ tasks_waitingr rQ r+ Úborrowed_tokens)
|
r rs r ZCOUNTÚcZorig_total_tokensr ry ÚeventsÚir" )rs r r r r rY r8 r# Ú!test_run_in_worker_thread_limiter sR r c sB g G fddd} t dd | dI d H ddgks>td S )Nc s$ e Zd Z fddZ fddZdS )z?test_run_in_worker_thread_custom_limiter.<locals>.CustomLimiterc s d¡ || _d S ©NÚacquire)r Ú _borrower©ÚselfZborrowerr2 r" r# Úacquire_on_behalf_of s
|
zTtest_run_in_worker_thread_custom_limiter.<locals>.CustomLimiter.acquire_on_behalf_ofc s d¡ || jkstd S ©NÚrelease)r r r+ r r2 r" r# Úrelease_on_behalf_of s
|
zTtest_run_in_worker_thread_custom_limiter.<locals>.CustomLimiter.release_on_behalf_ofN©r r r r¡ r¤ r" r2 r" r# Ú CustomLimiter s r¦ c S s d S r: r" r" r" r" r# r; r< z:test_run_in_worker_thread_custom_limiter.<locals>.<lambda>©r r r£ )r r+ )r¦ r" r2 r# Ú(test_run_in_worker_thread_custom_limiter s r¨ c sÆ g G fddd} | }t t¡ }tdd |dI d H W 5 Q R X |jjd ksXtddgkshtg i t t¡ }t fdd|dI d H W 5 Q R X t|jjts²tddgksÂtd S ) Nc s$ e Zd Z fddZ fddZdS )zCtest_run_in_worker_thread_limiter_error.<locals>.BadCapacityLimiterc s d¡ d S r )r r r2 r" r# r¡ s zXtest_run_in_worker_thread_limiter_error.<locals>.BadCapacityLimiter.acquire_on_behalf_ofc s d¡ td S r¢ )r r4 r r2 r" r# r¤ s
|
zXtest_run_in_worker_thread_limiter_error.<locals>.BadCapacityLimiter.release_on_behalf_ofNr¥ r" r2 r" r# ÚBadCapacityLimiter s r© c S s d S r: r" r" r" r" r# r; ¤ r< z9test_run_in_worker_thread_limiter_error.<locals>.<lambda>r§ r r£ c s d S )Nr r" r" )Údr" r# r; r< ) r> r? r4 r r` Ú __context__r+ Ú
|
isinstancer7 )r© Úbsra r" )rª r! r# Ú'test_run_in_worker_thread_limiter_error s $r® c à sz dd }| tjjd|¡ t }|jdks.tt t ¡}t
|
dd I d H W 5 Q R X dt |j ksht|jdksvtd S )Nc W s t dd S )Nz!the engines canna take it captain)r@ )r rZ r" r" r# Ú bad_start´ s z:test_run_in_worker_thread_fail_to_spawn.<locals>.bad_startZstart_thread_soonr c S s d S r: r" r" r" r" r# r; ¾ r< z9test_run_in_worker_thread_fail_to_spawn.<locals>.<lambda>Zengines) r~ r r Z ThreadCacher r r+ r> r? r@ r Ústrr` )r r¯ r ra r" r" r# Ú'test_run_in_worker_thread_fail_to_spawn² s r± c à s. dd } t ¡ }t| I d H }||ks*td S )Nc S s t tj} | S r: ©r r r ©Ú callee_tokenr" r" r# rY Ç s
|
z5test_trio_to_thread_run_sync_token.<locals>.thread_fn©r r r r+ ©rY Z caller_tokenr´ r" r" r# Ú"test_trio_to_thread_run_sync_tokenÄ s r· c à s4 dd } t jtdd t| I d H W 5 Q R X d S )Nc Ó s d S r: r" r" r" r" r# Úasync_fnÒ s z=test_trio_to_thread_run_sync_expected_error.<locals>.async_fnúexpected a sync function©Úmatch)r> r? Ú TypeErrorr ©r¸ r" r" r# Ú+test_trio_to_thread_run_sync_expected_errorÐ s r¾ Útrio_test_contextvarc
|
à sÄ t ¡ } t d¡ dd }t|I d H \}}}|dks:t|d ksFt|| ksRtdd }t|I d H \}}}}t ¡ }t ¡ } |dkst|dkst|dks¨td|d ks´t| dksÀtd S ) Nr\ c S s t ¡ } t ¡ }| |t ¡ fS r: )r¿ re r r r )r` Úsniffio_cvar_valuer" r" r# r( à s z3test_trio_to_thread_run_sync_contextvars.<locals>.fc S s2 t ¡ } t d¡ t ¡ }t ¡ }| ||t ¡ fS )NÚworker)r¿ re rQ r r r )Ú parent_valueÚ inner_valuerÀ r" r" r# r_ ê s
|
üz3test_trio_to_thread_run_sync_contextvars.<locals>.grÁ zRThe contextvar value set on the worker would not propagate back to the main threadÚtrio)r r r¿ rQ r r+ re r )
|
r/ r( r` rÀ r- r_ r rÃ Ú current_valueZsniffio_outer_valuer" r" r# Ú(test_trio_to_thread_run_sync_contextvarsÜ s0
|
û ÿþ rÆ c sd dd } t | I d H }t|ts$tdd fdd} tjtdd t | I d H W 5 Q R X d S )Nc S s t tj} | S r: )r r Ú current_time)Ú trio_timer" r" r# rY
|
s
|
z1test_trio_from_thread_run_sync.<locals>.thread_fnc Ó s d S r: r" r" r" r" r# r¸ s z0test_trio_from_thread_run_sync.<locals>.async_fnc s t d S r: )r r" r½ r" r# rY s r¹ rº )r r¬ Úfloatr+ r> r? r¼ )rY rÈ r" r½ r# Ú test_trio_from_thread_run_sync s rÊ c sr g fdd fdd} t | I d H ddgks<tdd }tjtd d
|
t t|I d H W 5 Q R X d S ) Nc s t ¡ d¡ d S )Nú back in trio)r rÇ r r" r2 r" r# Úback_in_trio_fn! s z2test_trio_from_thread_run.<locals>.back_in_trio_fnc s d¡ t d S )Nú in thread)r r r" ©rÌ r! r" r# rY % s
|
z,test_trio_from_thread_run.<locals>.thread_fnrÍ rË c S s d S r: r" r" r" r" r# Úsync_fn- s z*test_trio_from_thread_run.<locals>.sync_fnzappears to be synchronousrº )r r+ r> r? r¼ r )rY rÏ r" rÎ r# Útest_trio_from_thread_run s rÐ c à s. dd } t ¡ }t| I d H }||ks*td S )Nc S s t tj} | S r: r² r³ r" r" r# rY 7 s
|
z.test_trio_from_thread_token.<locals>.thread_fnrµ r¶ r" r" r# Útest_trio_from_thread_token4 s rÑ c à s0 dd } t ¡ }t| |I d H }||ks,td S )Nc S s t tj| d}|S )Nr r² )r8 r´ r" r" r# rY C s z4test_trio_from_thread_token_kwarg.<locals>.thread_fnrµ r¶ r" r" r# Ú!test_trio_from_thread_token_kwarg@ s rÒ c à s$ t t¡ ttj W 5 Q R X d S r: ©r> r? r@ r r rÇ r" r" r" r# Útest_from_thread_no_tokenL s rÔ c à s t d¡ dd } t| I d H \}}}}}}}}t ¡ } t ¡ }
|
| | krTdksZn t|| krx| krxdks~n t|dkst|
|
| krdks¤n t|| kr¸d ks¾n td S )Nr\ c S s\ t ¡ } t d¡ t ¡ }t ¡ }dd }t|\}}}t ¡ }t ¡ }| |||||||fS )NrÁ c S s, t ¡ } t d¡ t ¡ }t ¡ }| ||fS ©NÚ back_in_main©r¿ re rQ r ©Úback_parent_valueÚback_current_valueÚsniffio_cvar_back_valuer" r" r# rÖ ] s
|
|
zStest_trio_from_thread_run_sync_contextvars.<locals>.thread_fn.<locals>.back_in_main)r¿ re rQ r r ) Úthread_parent_valueÚthread_current_valueÚ sniffio_cvar_thread_pre_valuerÖ rÙ rÚ rÛ Úthread_after_valueÚ sniffio_cvar_thread_after_valuer" r" r# rY W s*
|
üøz=test_trio_from_thread_run_sync_contextvars.<locals>.thread_fnrÁ rÖ rÄ )r¿ rQ r re r r+ ) rY rÜ rÝ rß rÞ rà rÙ rÚ rÛ rÅ Zsniffio_cvar_out_valuer" r" r# Ú*test_trio_from_thread_run_sync_contextvarsT s&
|
( ÷$ rá c
|
à s¬ t d¡ dd } t| I d H \}}}}}}}}t ¡ } | | krLdksRn t|| krp| krpdksvn t|dkst|| krd ksn t|dks¨td S )Nr\ c S s\ t ¡ } t d¡ t ¡ }t ¡ }dd }t|\}}}t ¡ }t ¡ }| |||||||fS )NrÁ c Ó s, t ¡ } t d¡ t ¡ }t ¡ }| ||fS rÕ r× rØ r" r" r# Úasync_back_in_main s
|
|
zTtest_trio_from_thread_run_contextvars.<locals>.thread_fn.<locals>.async_back_in_main)r¿ re rQ r r ) rÜ rÝ rÞ râ rÙ rÚ rÛ rß rà r" r" r# rY s*
|
üøz8test_trio_from_thread_run_contextvars.<locals>.thread_fnrÁ rÖ rÄ )r¿ rQ r re r+ )
|
rY rÜ rÝ rß rÞ rà rÙ rÚ rÛ rÅ r" r" r# Ú%test_trio_from_thread_run_contextvars s$
|
( ÷$ rã c C s( t t¡ ttjdd W 5 Q R X d S )NzNot TrioTokentyper rÓ r" r" r" r# Ú4test_run_fn_as_system_task_catched_badly_typed_token½ s rä c à s6 dd } t ¡ }t t¡ t| |d W 5 Q R X d S )Nc S s dst d S )NF)r+ r" r" r" r# Ú
|
not_calledà s z7test_from_thread_inside_trio_thread.<locals>.not_calledr )r r r> r? r@ r )rå r r" r" r# Ú#test_from_thread_inside_trio_thread s ræ zpypy 7.2.0 is buggy)Úreasonc s> g g fdd fdd} t | ¡ dgks:td S )Nc s^ z
|
d V W 5 t tj¡0 tjdd tttdI d H W 5 Q R X W 5 Q R X d¡ X d S )NTrl r Úok) r> r? r ZRunFinishedErrorrv r r r r r" r2 r" r# ÚagenÐ s
|
|
&z2test_from_thread_run_during_shutdown.<locals>.agenc s$ ¡ d d ¡I d H d S )Nr )r Úasendr" )ré Úsaver" r# r\ Ø s z2test_from_thread_run_during_shutdown.<locals>.mainrè )r rR r+ )r\ r" )ré r! rë r# Ú$test_from_thread_run_during_shutdownË s
|
rì c à s0 t } t| tstt | ¡}| | ks,td S r: )r r¬ r r+ ÚweakrefÚref)r8 Zweak_referencer" r" r# Ú"test_trio_token_weak_referenceableà s
|
rï c à s< G dd d} t t¡ tt| dI d H W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z.test_unsafe_cancellable_kwarg.<locals>.BadBoolc S s t d S r: )ÚNotImplementedError)r r" r" r# Ú__bool__ì s z7test_unsafe_cancellable_kwarg.<locals>.BadBool.__bool__N)r r r rñ r" r" r" r# ÚBadBoolë s rò ri )r> r? rð r Úint)rò r" r" r# Ú test_unsafe_cancellable_kwargç s rô );Z contextvarsr Úqueuern ru rí r> Zsniffior Z
|
trio._corer r Ú r r r r Ztestingr
|
Z_core.tests.tutilr Ú_threadsr r r r Z_core.tests.test_kir r9 rA rS r] rb rz r ÚmarkZ parametrizer r¨ r® r± r· r¾ Z
|
ContextVarr¿ rÆ rÊ rÐ rÑ rÒ rÔ rá rã rä ræ Zskipifrì rï rô r" r" r" r# Ú<module> sX 7 0 <#h
|
+ 54
|
|
