U
|
Z ±d8 ã @ sô d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
|
Z
|
m Z ddl m Z ddlmZ ddlmZmZ ddlmZ dd lmZ d
|
d Zd d Zdd Zdd Zdd Zdd Z dd Z dd Z dd Z d d Z d d Z!dS ) é N)Úasync_generatorÚyield_ÚisasyncgenfunctionÚasynccontextmanageré )Ú_core)Úwait_all_tasks_blocked)Ú signal_raiseÚis_main_thread)Úsleepé )Úslowc C s t tj d S ©N)r ÚsignalÚSIGINT© r r úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_ki.pyÚki_self s r c C s t t¡ t W 5 Q R X d S r )ÚpytestÚraisesÚKeyboardInterruptr r r r r Ú test_ki_self s r c
|
s t ¡ r tt ¡ } g fdd}| |¡ t I d H dgksHtt jfdd}t jdd | t j fdd }t jd
|
d | I d H t ¡ 4 I d H }| |¡ | ¡ W 5 Q I d H R X t jd d }| D ]}qät jdd }| D ]}qþd S )Nc s t ¡ ¡ d S r )Úappendr Úcurrently_ki_protectedr ©Úrecordr r Úcheck) s z test_ki_enabled.<locals>.checkTc s t ¡ s t d S r ©r r ÚAssertionErrorr )Ú unprotectedr r Ú protected0 s z"test_ki_enabled.<locals>.protectedc S s t ¡ r td S r r r r r r r 5 s z$test_ki_enabled.<locals>.unprotectedc s t ¡ s t I d H d S r r r )Ú aunprotectedr r Ú
|
aprotected; s z#test_ki_enabled.<locals>.aprotectedc Ó s t ¡ r td S r r r r r r r! @ s z%test_ki_enabled.<locals>.aunprotectedc s s t ¡ s td V d S r r r r r r Ú gen_protectedL s z&test_ki_enabled.<locals>.gen_protectedc s s t ¡ r td V d S r r r r r r Úgen_unprotectedT s z(test_ki_enabled.<locals>.gen_unprotected)
|
r r r Úcurrent_trio_tokenÚ run_sync_soonr Úenable_ki_protectionÚdisable_ki_protectionÚ open_nurseryÚ
|
start_soon)Útokenr r r" Únurseryr# Ú_r$ r )r! r r r Útest_ki_enabled! s8
|
|
|
|
|
|
|
|
r. c sH t j fdd} t j fdd}dd | I d H | I d H d S )Nc s dI d H d S )NTr r ©Úchildr r r g s z6test_ki_enabled_after_yield_briefly.<locals>.protectedc s dI d H d S )NFr r r/ r r r k s z8test_ki_enabled_after_yield_briefly.<locals>.unprotectedc Ó sJ dd l }| ¡ t ¡ | ks tt ¡ I d H | ¡ t ¡ | ksFtd S )Nr )Ú tracebackÚ print_stackr r r Ú
|
checkpoint)Úexpectedr1 r r r r0 o s z2test_ki_enabled_after_yield_briefly.<locals>.child)r r' r( )r r r r/ r Ú#test_ki_enabled_after_yield_brieflyf s r5 c
|
à sb t jtjdd } | t ¡ r(tW 5 Q R X t t¡ |
|
tW 5 Q R X W 5 Q R X d S )Nc s s* t ¡ s tz
|
d V W 5 t ¡ s$tX d S r r r r r r Úprotected_manager s
|
zEtest_generator_based_context_manager_throw.<locals>.protected_manager) Ú
|
contextlibÚcontextmanagerr r' r r r r ÚKeyError)r6 r r r Ú*test_generator_based_context_manager_throw~ s r: c à s t jtdd } t jtdd }tt jdd }tt jdd }t jd d
|
}t jd d }| |||||fD ]}| 2 z3 d H W }t ¡ r~tq~6 t|rtt| 4 I d H t ¡ rÂtW 5 Q I d H R X t t
|
¡. t| 4 I d H
|
t
|
W 5 Q I d H R X W 5 Q R X qtd S ) Nc Ó s0 t ¡ s tzt I d H W 5 t ¡ s*tX d S r ©r r r r r r r r Úagen_protected1 s z-test_agen_protection.<locals>.agen_protected1c Ó s0 t ¡ r tzt I d H W 5 t ¡ r*tX d S r r; r r r r Úagen_unprotected1 s z/test_agen_protection.<locals>.agen_unprotected1c Ó s0 t ¡ s tzt I d H W 5 t ¡ s*tX d S r r; r r r r Úagen_protected2¥ s z-test_agen_protection.<locals>.agen_protected2c Ó s0 t ¡ r tzt I d H W 5 t ¡ r*tX d S r r; r r r r Úagen_unprotected2® s z/test_agen_protection.<locals>.agen_unprotected2c S s* t ¡ s tz
|
d V W 5 t ¡ s$tX d S r r r r r r Úagen_protected3¸ s
|
z-test_agen_protection.<locals>.agen_protected3c S s* t ¡ r tz
|
d V W 5 t ¡ r$tX d S r r r r r r Úagen_unprotected3À s
|
z/test_agen_protection.<locals>.agen_unprotected3) r r' r r( r r r r r r r9 )r< r= r> r? r@ rA Zagen_fnr- r r r Útest_agen_protection s>
|
|
ú rB c C s t ¡ s td S r r r r r r Ú test_ki_disabled_out_of_contextá s rC c sB dd fdd t j fdd} | s>td S )Nc S s t ¡ S r )r r r r r r Únestedfunctionæ s z/test_ki_disabled_in_del.<locals>.nestedfunctionc s t ¡ s t std S r r r )rD r r Ú__del__é s z(test_ki_disabled_in_del.<locals>.__del__c s t ¡ r t rt d S r r r ©rE rD r r Ú outerfunctioní s
|
z.test_ki_disabled_in_del.<locals>.outerfunction)r r( r )rG r rF r Útest_ki_disabled_in_delå s rH c sh dd dd t d t fdd} t t¡ t | ¡ W 5 Q R X dd d
|
hks`tt d t fd d }t t¡ t |¡ W 5 Q R X dd dhks°tt d dd }t t¡ t |¡ W 5 Q R X t d G dd d}dd }t t¡ tj|| gd W 5 Q R X t d tjdd }t |¡ t d tjdd }t |¡ t d tjd d }t |¡ t d tjd d }t |¡ t d tjd!d }t t¡ t |¡ W 5 Q R X t d" g fd#d}tj|d$d% d&gkstg t t¡ t |¡ W 5 Q R X g ksDtt d' tjd(d }t |¡ d S ))Nc Ó s> zt ¡ I d H qW n$ t jk
|
r8 | | d ¡ Y nX d S )Nz ok)r r3 Ú CancelledÚadd©Únamer r r r Úsleeperù s z)test_ki_protection_works.<locals>.sleeperc Ó s zt dt ¡ t W n: tk
|
rR t d t ¡ I d H | | d ¡ Y nHX t d zt dd ¡I d H W n$ tjk
|
r | | d ¡ Y nX d S )Nzkilling, protection =zraised!z raise okz didn't raise!c S s t jjS r )r ÚAbortZ SUCCEEDED©r- r r r Ú<lambda> ó z:test_ki_protection_works.<locals>.raiser.<locals>.<lambda>z
|
cancel ok) Úprintr r r r r3 rJ Úwait_task_rescheduledrI rK r r r Úraiser s
|
z(test_ki_protection_works.<locals>.raiserzcheck 1c
|
sP t ¡ 4 I d H 0} | d¡ | d¡ | d¡ W 5 Q I d H R X d S ©NÚs1Ús2Zr1)r r) r* ©r, ©rT r rM r r Úcheck_unprotected_kill s z8test_ki_protection_works.<locals>.check_unprotected_killzs1 okzs2 okz r1 raise okzcheck 2c
|
sV t ¡ 4 I d H 6} | d¡ | d¡ | t ¡d¡ W 5 Q I d H R X d S rU )r r) r* r' rX rY r r Úcheck_protected_kill( s z6test_ki_protection_works.<locals>.check_protected_killz r1 cancel okzcheck 3c s$ t ¡ fdd ¡ d S )Nc sD t ¡ s tz ¡ W n$ t jk
|
r> td t Y nX d S )Nz kill! kill!)r r r r& ZRunFinishedErrorrR r r ©Úkill_during_shutdownr+ r r r] : s zZtest_ki_protection_works.<locals>.check_kill_during_shutdown.<locals>.kill_during_shutdown)r r% r& r r r\ r Úcheck_kill_during_shutdown7 s z<test_ki_protection_works.<locals>.check_kill_during_shutdownzcheck 4c @ s e Zd Zdd ZdS )z3test_ki_protection_works.<locals>.InstrumentOfDeathc S s
|
t d S r ©r )Úselfr r r Ú
|
before_runL s z>test_ki_protection_works.<locals>.InstrumentOfDeath.before_runN)Ú__name__Ú
|
__module__Ú __qualname__ra r r r r ÚInstrumentOfDeathK s re c Ó s t ¡ I d H d S r )r r3 r r r r ÚmainO s z&test_ki_protection_works.<locals>.main)Z instrumentszcheck 5c Ó s: t ¡ s tt t t¡ t ¡ I d H W 5 Q R X d S r )r r r r r r r Zcheckpoint_if_cancelledr r r r rf X s zcheck 6c Ó sd t ¡ s tt t ¡ I d H t ¡ I d H t ¡ I d H t t¡ t ¡ I d H W 5 Q R X d S r ) r r r r Zcancel_shielded_checkpointr r r r3 r r r r rf d s zcheck 7c sf t ¡ s tt t ¡ fdd} t | ¡I d H dks>tt t¡ t ¡ I d H W 5 Q R X d S )Nc s t t d¡¡ t jjS )Nr )r Ú
|
rescheduleÚoutcomeÚValuerN ÚFAILEDrO ©Ztaskr r Úaborty s ú5test_ki_protection_works.<locals>.main.<locals>.abortr )
|
r r r r Ú current_taskrS r r r r3 ©rl r rk r rf s s zcheck 8c sb t ¡ s tt t ¡ fdd} t t¡ t | ¡I d H sFtW 5 Q R X t ¡ I d H d S )Nc s t | ¡}t |¡ tjjS r )rh Úcapturer rg rN rj )Z raise_cancelÚresultrk r r rl s
|
rm )
|
r r r r rn r r r rS r3 ro r rk r rf s zcheck 9c Ó s
|
t d S r r_ r r r r rf s zcheck 10c sD t ¡ r tt d¡ t t¡ tdI d H W 5 Q R X d S )NÚoké
|
) r r r r r r r r r r r r r rf © s
|
|
T)Z*restrict_keyboard_interrupt_to_checkpointsrr zcheck 11c
|
Ó s¢ t ¡ s tt ¡ } | ¡ t t j¡ t ¡ I d H W 5 Q R X t t t
|
¡ 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 Z CancelScopeÚcancelr r rI r3 r r )Z cancel_scoper r r rf À s
|
) rR Úsetr r r r Úrunr r' )rZ r[ r^ re rf r rY r Útest_ki_protection_worksø sr
|
|
|
|
|
|
|
|
|
|
|
rw c
|
sX zBt t j¡} dd fdd}t |¡ t t j¡ ks@tW 5 t t j| ¡ X d S )Nc S s d S r r )ÚsignumÚframer r r Ú
|
my_handlerÖ s z,test_ki_is_good_neighbor.<locals>.my_handlerc s t t j ¡ d S r )r r r ©rz r r rf Ù s z&test_ki_is_good_neighbor.<locals>.main)r r Ú getsignalr rv r )Úorigrf r r{ r Útest_ki_is_good_neighborÐ s
|
r~ c C sN t ¡ } t j| j }z&t j| j= tjdd }t |¡ W 5 |t j| j< X d S )Nc Ó s t t j¡t jkstd S r )r r| r Údefault_int_handlerr r r r r Úinnerï s z*test_ki_with_broken_threads.<locals>.inner)Ú threadingÚ main_threadÚ_activeÚidentr r' rv )ÚthreadÚoriginalr r r r Útest_ki_with_broken_threadsä s
|
|
r )"rh r ÚsysÚosr r r7 Útimer r r r Ú r Ztestingr Z_utilr r
|
Z _timeoutsr Ztutilr r r r. r5 r: rB rC rH rw r~ r r r r r Ú<module> s2 EP Y
|
