U
|
Z ±dG ã @ sN d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z d dl
|
Z
|
d dl Z d dl Z d dl Z d dlZd dlZd dlZddlmZmZmZ ddlmZ dd Zd
|
d Zd d Zdd Zdd Zdd Z dd Z e dd Z ddddZ d d Z d d Z!d d! Z"d"d# Z#e j$j%ed$d%e j$j&ej'j(d&kd'd%e d(d) Z)dS )*é N)Úpartial©Úinfé )Úgc_collect_harderÚbuggy_pypy_asyncgensÚrestore_unraisablehooké )Ú signal_raisec s¢ t ¡ t ¡ fdd} fdd}fdd}tjj| |f|||d| z> ¡ \}}|dkrt| qX|d kr| ¡ W ¢S d
|
sXt qXW 5 ~~X d S ) Nc s8 t ¡ kr&ttjd} d|f¡ d| f¡ d S )Nz0run_sync_soon_threadsafe called from host threadÚrun©Ú threadingÚcurrent_threadr ÚpytestZfailÚput©ÚfnZcrash©Z host_threadÚtodo© úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_guest_mode.pyÚrun_sync_soon_threadsafe s ÿz3trivial_guest_run.<locals>.run_sync_soon_threadsafec s8 t ¡ k r&ttjd} d|f¡ d| f¡ d S )Nz6run_sync_soon_not_threadsafe called from worker threadr r r r r r Ú run_sync_soon_not_threadsafe' s ÿz7trivial_guest_run.<locals>.run_sync_soon_not_threadsafec s d| f¡ d S )NÚunwrap)r )Zoutcome)r r r Ú done_callback/ s z(trivial_guest_run.<locals>.done_callback)r r r r r F)
|
ÚqueueÚQueuer r ÚtrioÚlowlevelÚstart_guest_runÚgetr ÚAssertionError)Útrio_fnÚstart_guest_run_kwargsr r r ÚopÚobjr r r Útrivial_guest_run s. þûú r& c C sF dd } t | dkstdd }tjtdd t | W 5 Q R X d S )Nc Ó s t d¡I d H dS )Nr Úok©r Úsleep©Úin_hostr r r Ú trio_returnL s z'test_guest_trivial.<locals>.trio_returnr' c Ó s t dd S )NÚ whoopsiedaisy)ÚKeyErrorr* r r r Ú trio_failR s z%test_guest_trivial.<locals>.trio_failr- ©Úmatch)r& r! r Úraisesr. )r, r/ r r r Útest_guest_trivialK s
|
r3 c C s dd } t | d S )Nc s g t j ¡ \ } p |` t ¡ 4 I d H >} fdd}| |¡ t j ¡ I d H | d¡I d H W 5 Q I d H R X W 5 Q R X W 5 Q R X dgkstd S )Nc s d¡I d H ¡ d S )Nr )ÚappendÚrecvr ©ÚaÚrecordr r Ú
|
do_receive` s z;test_guest_can_do_io.<locals>.trio_main.<locals>.do_receiveó x) r ÚsocketÚ
|
socketpairÚ open_nurseryÚ
|
start_soonÚtestingÚwait_all_tasks_blockedÚsendr! )r+ ÚbÚnurseryr9 r r6 r Ú trio_mainZ s
|
4z'test_guest_can_do_io.<locals>.trio_main)r& ©rD r r r Útest_guest_can_do_ioY s rF c C s dd } t | dkstd S )Nc Ó s$ t ¡ }| |j | ¡ I d H dS ©Nr' )r ÚEventÚsetÚwait)r+ Zevr r r rD n s
|
z8test_host_can_directly_wake_trio_task.<locals>.trio_mainr' ©r& r! rE r r r Ú%test_host_can_directly_wake_trio_taskm s rL c s( dd fdd} t | dks$td S )Nc S s
|
|| _ d S ©N)Údeadline©ÚcscopeZ new_deadliner r r Ú set_deadlinex s z@test_host_altering_deadlines_wakes_trio_up.<locals>.set_deadlinec s t ¡ & | fdd t ¡ I d H W 5 Q R X js>tt ¡ : | fdd | fdd t d¡I d H W 5 Q R X jstdS )Nc s t S rM r r ©rP rQ r r Ú<lambda>} ó zOtest_host_altering_deadlines_wakes_trio_up.<locals>.trio_main.<locals>.<lambda>c s
|
dS )Ng .Ar r rR r r rS rT c s t S rM r r rR r r rS
rT iç r' )r Ú CancelScopeÚ sleep_foreverÚcancelled_caughtr! r) r* ©rQ ©rP r rD { s
|
|
|
|
z=test_host_altering_deadlines_wakes_trio_up.<locals>.trio_mainr' rK rE r rX r Ú*test_host_altering_deadlines_wakes_trio_upw s rZ c sÚ t d¡dkstdd } t ¡ \ } ¤ | d¡ t ¡ ¡ z.tjt dd t
|
| dksptW 5 Q R X W 5 t d¡ ¡ kstX t ¡ ¡ z2tjt dd t
|
| dddksÊtW 5 Q R X W 5 t d¡ ¡ ksîtX t ¡ " t d ¡ t
|
| dkstW 5 Q R X t ¡ & t d ¡ t
|
| d
|
ddksJtW 5 Q R X t ¡ ¡ zD fd d }t ¡ & t d ¡ t
|
|d
|
ddkstW 5 Q R X W 5 t d¡ ¡ ksÀtX W 5 Q R X W 5 Q R X d S ) Néÿÿÿÿc Ó s dS rG r r* r r r rD s z4test_warn_set_wakeup_fd_overwrite.<locals>.trio_mainFz signal handling code.*collidedr0 r' ©Ú host_uses_signal_set_wakeup_fdÚerrorTc s( t d¡}| ¡ kstt |¡ dS )Nr[ r' )ÚsignalÚ set_wakeup_fdÚfilenor! )r+ Úfd©r7 r r Ú trio_check_wakeup_fd_unaltered» s
|
|
zItest_warn_set_wakeup_fd_overwrite.<locals>.trio_check_wakeup_fd_unaltered)r_ r` r! r; r< Ú setblockingra r ÚwarnsÚRuntimeWarningr& ÚwarningsÚcatch_warningsÚ simplefilter)rD rB rd r rc r Ú!test_warn_set_wakeup_fd_overwrite sP
|
|
ÿÿ
|
|
|
|
|
ÿÿ
|
|
þüÿrk c s( dd fdd} t | dks$td S )Nc S s t d| || _d S )Nzsetting deadline )ÚprintrN rO r r r rQ Õ s zLtest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.set_deadlinec
|
sl dd } fdd}t ¡ 4 I d H 6}t ¡ }| ||¡ t j ¡ I d H | ||¡ W 5 Q I d H R X dS )Nc Ó s: | " t jjddI d H ds"tW 5 Q R X | js6td S )Né' )ZcushionF)r r? r@ r! rW )Ú watb_cscoper r r Ú sit_in_wait_all_tasks_blockedÚ s zptest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.sit_in_wait_all_tasks_blockedc sd t ¡ @ td G fddd}t j | ¡ t ¡ I d H W 5 Q R X jsXt| ¡ d S )Nzscheduling stuff to happenc s2 e Zd Zdd Z fddZ fddZdS )ztest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadline.<locals>.InstrumentHelperc S s
|
d| _ d S )NF)Úprimed)Úselfr r r Ú__init__õ s ztest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadline.<locals>.InstrumentHelper.__init__c s> t d| d |dkr:| jr"t fdd d| _d S )Nzbefore_io_wait(ú)rm c s
|
dS )Ng eÍÍAr r rR r r rS ü rT z¨test_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadline.<locals>.InstrumentHelper.before_io_wait.<locals>.<lambda>T)rl rp r! ©rq Útimeout©rP r+ rQ r r Úbefore_io_waitø s
|
|
ztest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadline.<locals>.InstrumentHelper.before_io_waitc s. | j r*td fdd tj | ¡ d S )Nzinstrument triggeredc s ¡ S rM )Úcancelr rY r r rS rT z§test_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadline.<locals>.InstrumentHelper.after_io_wait.<locals>.<lambda>)rp rl r r Zremove_instrumentrt )rP r+ r r Ú after_io_waitÿ s ztest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadline.<locals>.InstrumentHelper.after_io_waitN)Ú__name__Ú
|
__module__Ú __qualname__rr rw ry r rv r r ÚInstrumentHelperô s r} ) r rU rl r Zadd_instrumentrV rW r! rx )rn r} )r+ rQ rY r Úget_woken_by_host_deadlineã s
|
|
zmtest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_main.<locals>.get_woken_by_host_deadliner' )r r= rU r> r? r@ )r+ ro r~ rC rn rX r* r rD Ù s ' zItest_host_wakeup_doesnt_trigger_wait_all_tasks_blocked.<locals>.trio_mainr' rK rE r rX r Ú6test_host_wakeup_doesnt_trigger_wait_all_tasks_blockedÎ s 9r c
|
C sP dd } t jtdd0 | t t t¡ t ¡ W 5 Q R X W 5 Q R X d S )Nc S s* dd } t t¡ t| W 5 Q R X d S )Nc Ó s" | dd t d¡I d H q d S )Nc S s dd S )Nr r r r r r r rS rT zgtest_guest_warns_if_abandoned.<locals>.do_abandoned_guest_run.<locals>.abandoned_main.<locals>.<lambda>r r( r* r r r Úabandoned_main s zUtest_guest_warns_if_abandoned.<locals>.do_abandoned_guest_run.<locals>.abandoned_main)r r2 ÚZeroDivisionErrorr& )r r r r Údo_abandoned_guest_run s z=test_guest_warns_if_abandoned.<locals>.do_abandoned_guest_runz Trio guest run got abandonedr0 )r rf rg r r2 Ú RuntimeErrorr Ú current_time)r r r r Ú test_guest_warns_if_abandoned s r
T)Úpass_not_threadsafec s: t ¡ fdd}z | ¡W ¢S ¡ X d S )Nc sL ¡ fdd} r"jd< tjjfj| d I d H ¡ S )Nc s t d| | ¡ d S )Nztrio_fn finished: )rl Z
|
set_result)Z main_outcome©Z trio_done_futr r Útrio_done_callbackE s z9aiotrio_run.<locals>.aio_main.<locals>.trio_done_callbackr )r r )Z create_futureZ call_soonr r r Zcall_soon_threadsafer )r ©Zloopr r# r" r r Úaio_mainB s
|
ÿýüz aiotrio_run.<locals>.aio_main)ÚasyncioZnew_event_loopÚcloseZrun_until_complete)r" r r# r r r r Ú aiotrio_run? s
|
r c sB fdd} dd t | dddks(tt | ddd dks>td S )
|
Nc s t d t td¡\} }t ¡ }t || ¡}t d¡I d H | d¡ |2 z>3 d H W }t d| | |d ¡ |dkrP| ¡ dS qP6 d S )Nz
|
trio_main!r r z
|
trio got: r é
|
útrio-main-done)
|
rl r Zopen_memory_channelÚfloatr r Ú ensure_futurer) Ú
|
put_nowaitrx )Úto_trioZfrom_aioÚ from_trioZaio_taskÚn©Ú aio_pingpongr r rD \ s
|
z-test_guest_mode_on_asyncio.<locals>.trio_mainc Ó sj t d z0| ¡ I d H }t d| | |d ¡ q
|
W n, tjk
|
rP Y n t ¡ Y nX d S )Nz aio_pingpong!z aio got: r )rl r Z send_nowaitr ZCancelledErrorÚ tracebackÚ print_exc)r r r r r r r q s z0test_guest_mode_on_asyncio.<locals>.aio_pingpongTr\ r F)r r] )r r! rE r r r Útest_guest_mode_on_asyncio[ s" ûùÿ ûùÿr c
|
sÊ | ¡ 2 fdd}t tj¡ t| W 5 Q R X W 5 Q R X | ¡ 2 fdd}t tj¡ t| W 5 Q R X W 5 Q R X | ¡ 2 fdd}t tj¡ t| W 5 Q R X W 5 Q R X t d S )Nc s dd¡ t d¡I d H d S )Nz.trio._core._run.GLOBAL_RUN_CONTEXT.runner.runqZHIr ©Úsetattrr r) r* ©Úmr r Úcrash_in_run_loop s z:test_guest_mode_internal_errors.<locals>.crash_in_run_loopc s dd ¡ t d¡I d H d S )Nú'trio._core._run.TheIOManager.get_eventsr r r* r r r Ú crash_in_io¢ s z4test_guest_mode_internal_errors.<locals>.crash_in_ioc sB t ¡ tjjjj fdd} d|¡ t d¡I d H d S )Nc s" t ¡ k rtdn | S d S )Nzoh no!)r r Ú
|
ValueError)Úargs©Zold_get_eventsÚtr r Úbad_get_events¯ s
|
zZtest_guest_mode_internal_errors.<locals>.crash_in_worker_thread_io.<locals>.bad_get_eventsr r ) r r r Z_coreZ_runZ TheIOManagerZ
|
get_eventsr r) )r+ r¦ r r¤ r Úcrash_in_worker_thread_io« s
|
zBtest_guest_mode_internal_errors.<locals>.crash_in_worker_thread_io)Úcontextr r2 r ZTrioInternalErrorr& r )Z monkeypatchZrecwarnr r¡ r§ r r r Ú test_guest_mode_internal_errors s
|
|
|
r© c s¾ t t j¡t jkstdd } t t¡}t| W 5 Q R X |j j
|
d ksLtt t j¡t jksbtt d fdd}t t¡}t| W 5 Q R X |j j
|
ks¤tt t j¡t jksºtd S )Nc Ó sD t t¡ ttj W 5 Q R X | tttj t d¡I d H d S )Nr ) r r2 ÚKeyboardInterruptr
|
r_ ÚSIGINTr r r) r* r r r rD Ã s z%test_guest_mode_ki.<locals>.trio_mainZwhoac s | t ttj d S rM )r r
|
r_ r« r* ©Z final_excr r Útrio_main_raisingÔ s z-test_guest_mode_ki.<locals>.trio_main_raising) r_ Ú getsignalr« Údefault_int_handlerr! r r2 rª r& ÚvalueÚ __context__r. )rD Úexcinfor r r¬ r Útest_guest_mode_ki¿ s r³ c sP t j ¡ d fdd} t ¡ }t| d t ¡ }|| d k sLtd S )Néx c sD t ¡ dkst| fdd t ¡I d H t ¡ ks@td S )Nr c s t ddS )NZautojump_thresholdr )r r ©Úclockr r rS ê rT zVtest_guest_mode_autojump_clock_threshold_changing.<locals>.trio_main.<locals>.<lambda>)r r r! r) r* ©ZDURATIONr¶ r r rD è s zDtest_guest_mode_autojump_clock_threshold_changing.<locals>.trio_mainrµ é )r r? Z MockClockÚtimeÚ monotonicr& r! )rD ÚstartÚendr r· r Ú1test_guest_mode_autojump_clock_threshold_changingß s
|
r½ zPyPy 7.2 is buggy)ÚreasonÚpypyz)async generator issue under investigationc sd dd l t fdd fdd fdd} t ¡ }|jt| dd d
|
d hks`td S ) Nr c sj ¡ | kstz
|
dV W 5 ¡ }ztj| d¡I d H W n tjk
|
rT Y nX | |f¡ X d S )Nr r )Zcurrent_async_libraryr! ÚsysÚmodulesr) r Z CancelledÚadd)ÚlabelZlibrary)r8 Úsniffior r Úagen s
|
z'test_guest_mode_asyncgens.<locals>.agenc s$ j d¡ d d ¡I d H d S )Nr )Zcurrent_async_library_cvarrI Úasendr )rÅ rÄ r r Úiterate_in_aio s z1test_guest_mode_asyncgens.<locals>.iterate_in_aioc sh t ¡} t ¡ | fdd¡ t d¡ ¡ I d H W 5 Q R X d d ¡I d H t d S )Nc s ¡ S rM )rI )Ú_©Zdone_evtr r rS rT z>test_guest_mode_asyncgens.<locals>.trio_main.<locals>.<lambda>r r ) r r r rH Zadd_done_callbackZ
|
fail_afterrJ rÆ r )Ztask)rÅ rÇ rÉ r rD s z,test_guest_mode_asyncgens.<locals>.trio_mainTr\ )r r )r r )rÄ rI Ú contextvarsZ copy_contextr r r! )rD r¨ r )rÅ rÇ r8 rÄ r Útest_guest_mode_asyncgensö s rË )*r r rÊ rÀ r r Ú functoolsr Úmathr r_ r; r r¹ rh r Z trio.testingZtutilr r r Z_utilr
|
r& r3 rF rL rZ rk r r
r r r© r³ r½ ÚmarkZskipifZxfailÚimplementationÚnamerË r r r r Ú<module> sJ 1
|
@G
|
) ;)
|
þ
|
