U
|
Z ±dJ ã @ s d dl Z d dlZddlmZ ddlmZ ddlmZ ddlmZ ddl T ddl
|
m Z dd l m Z dd
|
lmZ dd lmZ d d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Z d"d# Z d$d% Z d&d' Z d(d) Z d*d+ Z!d,d- Z"d.d/ Z#dS )0é Né )Ú can_bind_ipv6)Úsleep)Ú_core)Úaclose_forcefully)Ú*)Ú_assert_raises)Ú_UnboundedByteQueue)Úsocket)ÚSocketListenerc
|
s´ g fdd} fdd}t ¡ 4 I d H $}| | ¡ | |¡ | |¡ W 5 Q I d H R X g fdd}t ¡ 4 I d H }| |¡ |j ¡ W 5 Q I d H R X dgks°td S )Nc s* t dD ]} t ¡ I d H q d¡ d S )Né
|
zbusy bee exhausted)Úranger Ú
|
checkpointÚappend)Ú_©Úrecord© úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_testing.pyÚbusy_bee s z-test_wait_all_tasks_blocked.<locals>.busy_beec s t I d H d¡ d S )Nzquiet at last!©Úwait_all_tasks_blockedr r r r r Úwaiting_for_bee_to_leave s z=test_wait_all_tasks_blocked.<locals>.waiting_for_bee_to_leavec s6 zt I d H W n tjk
|
r0 d¡ Y nX d S )NÚok)r r Ú Cancelledr r r r r Úcancelled_while_waiting& s z<test_wait_all_tasks_blocked.<locals>.cancelled_while_waitingr )r Ú open_nurseryÚ
|
start_soonÚ cancel_scopeÚcancelÚAssertionError)r r Únurseryr r r r Útest_wait_all_tasks_blocked s
|
|
|
r" c
|
s g fdd}t ¡ 4 I d H P}| |¡ t I d H dgksFt| d¡ t I d H ddgksltW 5 Q I d H R X d S )Nc s& d¡ tdI d H d¡ d S )Nútt starté ú tt finished©r r r r r r Ú timeout_task5 s
|
z?test_wait_all_tasks_blocked_with_timeouts.<locals>.timeout_taskr# r r% )r r r r r Zjump)Z
|
mock_clockr' r! r r r Ú)test_wait_all_tasks_blocked_with_timeouts2 s
|
|
r( c
|
s¤ g fdd} fdd} fdd} fdd}t ¡ 4 I d H 8}| | ¡ | |¡ | |¡ | |¡ | |¡ W 5 Q I d H R X d d
|
d d d d gks td S )Nc sB d¡ tdI d H tdI d H tdI d H d¡ d S )Nú blink startg{®Gáz?ú blink endr& r r r r ÚblinkF s
|
|
z7test_wait_all_tasks_blocked_with_cushion.<locals>.blinkc s t I d H d¡ d S )Núwait_no_cushion endr r r r r Úwait_no_cushionM s zAtest_wait_all_tasks_blocked_with_cushion.<locals>.wait_no_cushionc s t dI d H d¡ d S )Ng{®Gáz?úwait_small_cushion endr r r r r Úwait_small_cushionQ s zDtest_wait_all_tasks_blocked_with_cushion.<locals>.wait_small_cushionc s t dI d H d¡ d S )Ng¸
ëQ¸?úwait_big_cushion endr r r r r Úwait_big_cushionU s zBtest_wait_all_tasks_blocked_with_cushion.<locals>.wait_big_cushionr) r, r* r. r0 )r r r r )r+ r- r/ r1 r! r r r Ú(test_wait_all_tasks_blocked_with_cushionC s&
|
|
|
|
úr2 c à sÈ t t ¡ I d H W 5 Q R X t t¡ t W 5 Q R X W 5 Q R X tjtjfD ]@}t| t t¡$ t | I d H W 5 Q R X W 5 Q R X qTt " t ¡ I d H t ¡ I d H W 5 Q R X d S ©N) Úassert_checkpointsr r ÚpytestÚraisesr Úcheckpoint_if_cancelledÚcancel_shielded_checkpointÚprint©ZrecwarnZ partial_yieldr r r Útest_assert_checkpointsm s þ "r; c à sÞ t W 5 Q R X t t¡& t t ¡ I d H W 5 Q R X W 5 Q R X tjtjfD ]@}t| t t¡$ t | I d H W 5 Q R X W 5 Q R X qTt t¡4 t " t ¡ I d H t ¡ I d H W 5 Q R X W 5 Q R X d S r3 ) Zassert_no_checkpointsr5 r6 r r r r7 r8 r9 r: r r r Útest_assert_no_checkpoints s
|
"þ " r< c s g fddfdd} fdd}t }t ¡ 4 I d H b}| | |¡ | ||¡ |d4 I d H t I d H W 5 Q I d H R X dd d
|
d d gkstW 5 Q I d H R X t }|d 4 I d H W 5 Q I d H R X t t¡( |d 4 I d H W 5 Q I d H R X W 5 Q R X d S )Nc s t | | ¡ d S r3 )r9 r )Úvalr r r Út¨ s ztest_Sequencer.<locals>.tc
|
s | d4 I d H d W 5 Q I d H R X | d4 I d H d W 5 Q I d H R X | d4 I d H d W 5 Q I d H R X d S )Né ©Úf1r? é ©rA rB é ©rA rD r ©Úseq©r> r r rA ¬ s ztest_Sequencer.<locals>.f1c
|
sX | d4 I d H d W 5 Q I d H R X | d4 I d H d W 5 Q I d H R X d S )Nr ©Úf2r r ©rJ r r rF rH r r rJ ´ s ztest_Sequencer.<locals>.f2r$ rI r@ rK rC rE r ) Ú Sequencerr r r r r r5 r6 Ú RuntimeError)rA rJ rG r! r )r r> r Útest_Sequencer¥ s" & rN c sÀ g t fdd} t ¡ 4 I d H @}| | d¡ | | d¡ d4 I d H W 5 Q I d H R X W 5 Q I d H R X ddgkstt t¡( d4 I d H W 5 Q I d H R X W 5 Q R X d S ) Nc st t ¡ b}| dkr| ¡ z&| 4 I d H W 5 Q I d H R X W n$ tk
|
rd d | ¡¡ Y nX W 5 Q R X d S )Nr? zseq({}) RuntimeError)r Z CancelScoper rM r Úformat)ÚiZscope©r rG r r ÚchildÐ s
|
z$test_Sequencer_cancel.<locals>.childr? r r zseq(1) RuntimeErrorzseq(2) RuntimeErrorrB )rL r r r r r5 r6 rM )rR r! r rQ r Útest_Sequencer_cancelË s
|
rS c
|
à sx t t¡ tt W 5 Q R X W 5 Q R X t t¡" tt dd W 5 Q R X W 5 Q R X tt
|
tW 5 Q R X d S )NZfoor? )r5 r6 r r rM Ú TypeErrorr r r r Útest__assert_raisesë s
|
|
|
rU c sÔ t d¡ d¡ d¡dks,t d¡dks>t d¡ ¡ dksXtt tj¡ d¡ W 5 Q R X t tj¡ ¡ W 5 Q R X t t¡ d¡ W 5 Q R X d ¡ t
|
d¡I d H d ksätW 5 Q R X d
|
¡ d ¡ t
|
d¡I d H d ks$tW 5 Q R X t
|
¡ I d H d ksNtW 5 Q R X fdd} fdd}t ¡ 4 I d H }| |d¡ | | d¡ W 5 Q I d H R X t tj ¡@ t ¡ 4 I d H }| |d¡ | |d¡ W 5 Q I d H R X W 5 Q R X ¡ t tj¡ d¡ W 5 Q R X d¡dks:t ¡ dksLt
|
d¡I d H dksft
|
¡ I d H dks~t ¡ t fdd}t ¡ 4 I d H }| |d¡ | |¡ W 5 Q I d H R X d S )Nó 123ó 456r? ó 1r s 23456ó 789Ústringó abcó defs ghió ds efghic s t I d H | ¡ d S r3 )r Úput©Údata©Úubqr r Úputter s z'test__UnboundeByteQueue.<locals>.putterc s, t ¡ I d H | ks tW 5 Q R X d S r3 )r4 Úgetr )Úexpectra r r Úgetter s z'test__UnboundeByteQueue.<locals>.getteró xyzs asdfó ---ó c s t I d H ¡ d S r3 )r Úcloser )Úubq2r r Úcloser9 s z'test__UnboundeByteQueue.<locals>.closer)r r^ Ú
|
get_nowaitr r5 r6 r Ú
|
WouldBlockrT r4 rd r r ÚBusyResourceErrorrj ÚClosedResourceError)rc rf r! rl r )rb rk r Útest__UnboundeByteQueueú sX
|
|
|
|
"
|
|
$" & rq c
|
s t fdd dI d H d¡dks2t ¡ dksBtt ¡ I d H W 5 Q R X t tj¡ ¡ W 5 Q R X t tj¡ d¡ W 5 Q R X dI d H t ¡ I d H dksÐtW 5 Q R X d fd
|
d } t
|
¡ 4 I d H }| | ¡ | | ¡ W 5 Q I d H R X dks0tt ¡ I d H W 5 Q R X ¡ I d H d ksht ¡ I d H d kstt tj ¡ dI d H W 5 Q R X jd ks¶tjd ksÆtjd ksÖtg fdd}fdd}fdd}t |||j|ks tj|ks,tj|ks<t d¡I d H ¡ I d H tI d H ¡ ddddgkstd S )Nc s& t | ¡I d H W 5 Q R X d S r3 )r4 Úsend_allr_ )Úmssr r Ú do_send_allE s z*test_MemorySendStream.<locals>.do_send_allrV r? rX s 23r rW r c s6 z dI d H W n t jk
|
r0 d7 Y nX d S )Nó xxxr? )r ro r )rt Úresource_busy_countr r Ú do_send_all_count_resourcebusy_ s z=test_MemorySendStream.<locals>.do_send_all_count_resourcebusyru ri rh c s ¡ dkst d¡ d S )Nr[ Ú send_all_hook)Úget_data_nowaitr r r )Úmss2r r r rx | s z,test_MemorySendStream.<locals>.send_all_hookc s d¡ d S )NÚ"wait_send_all_might_not_block_hook©r r r r r r{ s zAtest_MemorySendStream.<locals>.wait_send_all_might_not_block_hookc s d¡ d S )NÚ
|
close_hookr| r r r r r} s z)test_MemorySendStream.<locals>.close_hookr[ rx r{ r} )ÚMemorySendStreamry r r4 Z wait_send_all_might_not_blockr5 r6 r rn Úget_datar r Úacloserp rx r{ r} rr r rj )rw r! rx r{ r} r )rt rs rz r rv r Útest_MemorySendStreamB sf
|
ÿür c sx t fdd} d¡ | dI d H dks2t| dI d H dksHt d¡ | d I d H dkshtt tj¡@ t ¡ 4 I d H }| | d¡ | | d¡ W 5 Q I d H R X W 5 Q R X j d ksÈt d¡
|
¡
|
¡ | dI d H dksøt| dI d H d kst| dI d H d ks(tt tj ¡ d
|
¡ W 5 Q R X fd d }g fd d}t ||j |kstj |kst d¡ d¡I d H dks´t d¡I d H dksÎt d¡I d H dksèt d¡ d _ d¡I d H dkst d¡ t ¡ I d H W 5 Q R X dgksLtt tj ¡ d¡I d H W 5 Q R X d S )Nc
|
s. t | ¡I d H W 5 Q R £ S Q R X d S r3 )r4 Ú receive_some©Z max_bytes)Úmrsr r Údo_receive_some s z1test_MemoryReceiveStream.<locals>.do_receive_somer[ r? ó ar s bcr\ ri rh c s d¡ d S )Nru )Úput_datar )Úmrs2r r Úreceive_some_hook» s z3test_MemoryReceiveStream.<locals>.receive_some_hookc s d¡ d S )NÚclosedr| r r r r r} À s z,test_MemoryReceiveStream.<locals>.close_hooks yyys yyyxxxru s zzzs lost on closer )ÚMemoryReceiveStreamr r r5 r6 r ro r r r Zput_eofrp r} r r4 r )r
r! r r} r )r r r r Útest_MemoryReceiveStream sN
|
|
&
|
|
|
|
|
r c à s¬ t } | ¡ t tj¡ | d¡I d H dks4tW 5 Q R X | ¡ t tj¡ | d¡ W 5 Q R X t }| d¡ | ¡ t tj¡ | d¡I d H W 5 Q R X d S )Nr ri rV rg ) r rj r5 r6 r rp r r r )r r r r r Ú test_MemoryRecvStream_closingÙ s "
|
r c à sþ t } t }t| | | d¡I d H t| | | d¡I d H dksHt| d¡I d H t| |ddsjt| d¡I d H dkstt| |ddstt| |dds¦tt| |ddr¸t| d¡I d H dksÐt| ¡ t| | | d¡I d H dksútd S ) NrV r rW r? r ó 4s 56ri )r~ r Zmemory_stream_pumprr r r rj )rs r r r r Útest_memory_stream_pumpí s
|
|
|
r c
|
sú t \} | jd k st| jd ks&t| jd k s4t jd ksBt| d¡I d H d¡I d H dksjt fdd}t ¡ 4 I d H .}|
|
|d¡ t I d H | d¡I d H W 5 Q I d H R X t ¡ 4 I d H ,}|
|
|d¡ t I d H | ¡ I d H W 5 Q I d H R X t \} t ¡ 4 I d H &}|
|
|d¡ t I d H | ¡ W 5 Q I d H R X t \} | j}d | _| d¡I d H dd } fd
|
d }t ¡ 4 I d H }|
|
||¡ |
|
|¡ W 5 Q I d H R X || _| d ¡I d H d¡I d H d ksötd S )NrV r c s d¡I d H | kstd S ©Nr ©r r )Úexpected©Úrr r Úreceiver s z1test_memory_stream_one_way_pair.<locals>.receiverr[ ri rW c Ó s t I d H | j ¡ d S r3 )r r r )r! r r r Úcancel_after_idle* s z:test_memory_stream_one_way_pair.<locals>.cancel_after_idlec s, t tj¡ d¡I d H W 5 Q R X d S r )r5 r6 r r r r r r r Úcheck_for_cancel. s z9test_memory_stream_one_way_pair.<locals>.check_for_cancelrY s 456789)Úmemory_stream_one_way_pairrx r r{ r} r rr r r r r r r rj )Úsr r! Úoldr r r r r Ú test_memory_stream_one_way_pair sB
|
|
|
r c
|
sÒ t \ d¡I d H d¡I d H d¡I d H dksBt d¡I d H dksZt ¡ I d H d¡I d H dkstfdd} fdd}t ¡ 4 I d H }| |¡ | | ¡ W 5 Q I d H R X d S ) NrV r[ r ri c s t I d H d¡I d H d S )Nrg )r rr r )Úbr r ÚsenderG s z'test_memory_stream_pair.<locals>.senderc s d¡I d H dkstd S )Nr rg r r )Úar r r K s z)test_memory_stream_pair.<locals>.receiver)Úmemory_stream_pairrr r r Zsend_eofr r r )r r r! r )r r r Útest_memory_stream_pair= s
|
|
r c à s4 dd } t | d I d H dd }t|d I d H d S )Nc Ó s t S r3 )r r r r r Úone_way_stream_makerT s zDtest_memory_streams_with_generic_tests.<locals>.one_way_stream_makerc Ó s t S r3 )r r r r r Úhalf_closeable_stream_makerY s zKtest_memory_streams_with_generic_tests.<locals>.half_closeable_stream_maker)Úcheck_one_way_streamZcheck_half_closeable_stream)r¡ r¢ r r r Ú&test_memory_streams_with_generic_testsS s r¤ c à s4 dd } t | | I d H dd }t||I d H d S )Nc Ó s t S r3 )Z lockstep_stream_one_way_pairr r r r r¡ ` s zFtest_lockstep_streams_with_generic_tests.<locals>.one_way_stream_makerc Ó s t S r3 )Zlockstep_stream_pairr r r r Útwo_way_stream_makere s zFtest_lockstep_streams_with_generic_tests.<locals>.two_way_stream_maker)r£ Zcheck_two_way_stream)r¡ r¥ r r r Ú(test_lockstep_streams_with_generic_tests_ s r¦ c à s dd } t ¡ }| d¡I d H | d¡ | t|I d H t ¡ }| d¡I d H | d¡ | t|I d H tr®t jt jd}| d¡I d H | d¡ | t|I d H tt drt jt jd}t
|
¡ <}d |¡}| |¡I d H | d¡ | t|I d H W 5 Q R X d S )
|
Nc Ó s | 4 I d H t | I d H }|4 I d H V | ¡ I d H }|4 I d H * | d¡I d H | d¡I d H dk W 5 Q I d H R X W 5 Q I d H R X W 5 Q I d H R X d S )Nó xr? )Z open_stream_to_socket_listenerÚacceptrr r )ZlistenerZ client_streamZ server_streamr r r Úcheckl s z2test_open_stream_to_socket_listener.<locals>.check)z 127.0.0.1r r )z0.0.0.0r )Úfamily)z::r ÚAF_UNIXz{}/sock) Útsocketr
|
ÚbindÚlistenr r ÚAF_INET6Úhasattrr« ÚtempfileÚTemporaryDirectoryrO )r© ÚsockZtmpdirÚpathr r r Ú#test_open_stream_to_socket_listenerk s*
|
|
|
|
|
|
|
rµ )$r± r5 Z_core.tests.tutilr Ú r r Z_highlevel_genericr ZtestingZtesting._check_streamsr Ztesting._memory_streamsr r
|
r¬ Z_highlevel_socketr r" r( r2 r; r< rN rS rU rq r r r r r r r¤ r¦ rµ r r r r Ú<module> s8 * & HZ=8
|
