U
|
Z ±d~: ã @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ ddlm Z ddl
|
m Z m Z m Z d dlZdd Zdd Ze jd
|
d Zd d ZejjgZejjgZejjgZeeefD ]Zedd eD 7 Zqªe j j dedd dZ e j j dedd dZ e j j dedd dZ e e dd Z!e dd Z"e d d Z#e e e d d Z$e e d d! Z%e e d"d# Z&d$d% Z'd&d' Z(d(d) Z)d*d+ Z*dS ),é N)Úsuppressé )Ú_core)Úwait_all_tasks_blockedÚ SequencerÚassert_checkpointsc C s. z| dd ¡ qW n tk
|
r( Y nX d S )Nó xé )ÚsendÚBlockingIOError©Úsock© r úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_io.pyÚ fill_socket s r c C s* z| d¡ qW n tk
|
r$ Y nX d S )Nr )Úrecvr r r r r Ú drain_socket s r c c s8 t ¡ } | D ]}| d¡ q | V | D ] }| ¡ q&d S )NF)Ú stdlib_socketÚ
|
socketpairÚ setblockingÚclose)Úpairr r r r r s r c s( fdd}d j¡}| |_|_|S )Nc s | ¡ S ©N)Úfileno)Úfileobj©Úfnr r Úfileno_wrapper+ s z$using_fileno.<locals>.fileno_wrapperz<{} on fileno>)ÚformatÚ__name__Ú __qualname__)r r Únamer r r Ú using_fileno* s r" c C s g | ] }t |qS r )r" )Ú.0Úfr r r Ú
|
<listcomp>< s r% Ú wait_readablec C s | j S r ©r r r r r Ú<lambda>C ó r( )ÚidsÚ wait_writablec C s | j S r r' r r r r r( F r) Únotify_closingc C s | j S r r' r r r r r( I r) c
|
sÊ | \ }t I d H W 5 Q R X g fdd}t ¡ 4 I d H 2}| |¡ t I d H g kspt| d¡ W 5 Q I d H R X t t |I d H W 5 Q R X g fdd}t ¡ 4 I d H 0}| |¡ t I d H g ksútt| W 5 Q I d H R X g t ¡ 4 I d H &}| |¡ t I d H |j
|
¡ W 5 Q I d H R X dgkshtt g t ¡ 4 I d H &}| |¡ t I d H |j
|
¡ W 5 Q I d H R X dgksÆtd S )Nc sf z$t I d H W 5 Q R X W n tjk
|
rD d¡ Y n X d¡ d¡dksbtd S )NÚ cancelledÚreadableé
|
r )r r Ú CancelledÚappendr ÚAssertionErrorr )ÚaÚrecordr& r r Ú block_on_read\ s
|
z&test_wait_basic.<locals>.block_on_readr c sT z$t I d H W 5 Q R X W n tjk
|
rD d¡ Y n X d¡ d S )Nr- Úwritable)r r r0 r1 r )r3 r4 r+ r r Úblock_on_writes s z'test_wait_basic.<locals>.block_on_writer- ) r r Ú open_nurseryÚ
|
start_soonr r2 r
|
r r Ú cancel_scopeÚcancel)r r& r+ Úbr5 Únurseryr7 r )r3 r4 r& r+ r Útest_wait_basicP sB
|
|
|
|
|
r> c
|
à sv | \}}t ¡ 4 I d H N}| ||¡ t I d H t t j¡ ||I d H W 5 Q R X |j ¡ W 5 Q I d H R X d S r ) r r8 r9 r ÚpytestÚraisesÚBusyResourceErrorr: r; )r r& r3 r< r= r r r Útest_double_read s rB c
|
à s~ | \}}t | t ¡ 4 I d H N}| ||¡ t I d H t tj¡ ||I d H W 5 Q R X |j ¡ W 5 Q I d H R X d S r )
|
r r r8 r9 r r? r@ rA r: r; )r r+ r3 r< r= r r r Útest_double_write s rC c
|
sz | \ } fdd} fdd}t t ¡ 4 I d H .}| |¡ | |¡ t I d H | W 5 Q I d H R X d S )Nc s* t tj¡ I d H W 5 Q R X d S r ©r? r@ r ZClosedResourceErrorr )r3 r& r r Úreader¶ s z)test_interrupted_by_close.<locals>.readerc s* t tj¡ I d H W 5 Q R X d S r rD r )r3 r+ r r Úwriterº s z)test_interrupted_by_close.<locals>.writer)r r r8 r9 r )r r& r+ r, r< rE rF r= r )r3 r& r+ r Útest_interrupted_by_close® s
|
|
rG c
|
sÎ g fdd} fdd}| \}}t | t ¡ 4 I d H ~}| ||¡ | ||¡ t I d H g ksrt| d¡ t I d H dgkstt| t I d H ddgksºtW 5 Q I d H R X d S )Nc s | I d H d¡ d S )NÚr_task©r1 r )r4 r& r r rH Ì s z3test_socket_simultaneous_read_write.<locals>.r_taskc s | I d H d¡ d S )NÚw_taskrI r )r4 r+ r r rJ Ð s z3test_socket_simultaneous_read_write.<locals>.w_taskr rH rJ )r r r8 r9 r r2 r
|
r )r r& r+ rH rJ r3 r< r= r )r4 r& r+ r Ú#test_socket_simultaneous_read_writeÇ s
|
rK c
|
sÒ | \}}| tjtjd¡ dd i fdd} fdd}t ¡ 4 I d H B}| ||dd ¡ | ||d
|
d ¡ | ||d ¡ | ||d ¡ W 5 Q I d H R X d d ksºtd d ksÎtd S )Ni' i@B r c s t |¡}d}|k rrtd| t| ¡}|rt | I d H W 5 Q R X | |¡}||7 }|d |
= q.q| tj ¡ ||< d S )Nr Úsent)
|
ÚrandomÚRandomÚprintÚ bytearrayÚ randranger r
|
Úshutdownr ÚSHUT_WR)r ÚseedÚkeyÚrrL ÚchunkÚthis_chunk_size)Ú MAX_CHUNKÚNÚresultsr+ r r Úsenderñ s
|
|
|
z,test_socket_actual_streaming.<locals>.senderc sX d}t d| t | I d H W 5 Q R X t| ¡}|sBqL||7 }q||< d S )Nr Úreceived)rO r Úlenr )r rU r] rX )rY r[ r& r r Úreceiver s
|
|
z.test_socket_actual_streaming.<locals>.receiverr Zsend_aé Zsend_bZrecv_aZrecv_b)Ú
|
setsockoptr Ú
|
SOL_SOCKETÚ SO_SNDBUFr r8 r9 r2 )r r& r+ r3 r< r\ r_ r= r )rY rZ r[ r& r+ r Ú test_socket_actual_streamingã s rd c à sF d} d}zt j d¡ W n tk
|
r0 d} Y nX d}| sB|sBtd S )NFéÿÿÿÿT)ÚtrioÚlowlevelr, ÚOSErrorr2 )Z got_oserrorZ got_no_errorr r r Ú%test_notify_closing_on_invalid_object s
|
ri c
|
à sZ t jjt jjfD ]D} t ¡ }| ¡ }W 5 Q R X t t ¡ | |I d H W 5 Q R X qd S r )
|
rf rg r& r+ r Úsocketr r? r@ rh )ÚwaitÚsr r r r Útest_wait_on_invalid_object& s
|
|
rm c à sr dd } t ¡ \}}t ¡ \}}t ¡ \}}||||||fD ]}| d¡ q<| | |ô |ä |Ô |Ä t I d H | ddd t| t| t ¡ 4 I d H n}| tj|¡ | tj |¡ | tj |¡ | tj|¡ | tj |¡ t I d H | ddd |j
|
¡ W 5 Q I d H R X | ddd W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S ) Nc S sb t ¡ }t| |j}|jdkr>|j| ks.t|j|ks^tn |jdksLt|j| | ks^td S )N)ZepollZwindowsZkqueue) r Zcurrent_statisticsrO Z io_statisticsÚbackendZtasks_waiting_readr2 Ztasks_waiting_writeZ tasks_waiting)Úexpected_readersÚexpected_writersZ
|
statisticsZiostatsr r r Úcheck3 s
|
z)test_io_manager_statistics.<locals>.checkFr` r )ro rp é é ) r r r r r r r8 r9 r+ r& r: r; ) rq Za1Zb1Úa2Zb2Úa3Zb3r r= r r r Útest_io_manager_statistics2 s* ( rv c sè dd } t ¡ d}t ¡ 4 I d H B}| | tjj|¡ t I d H | ¡ t I d H |j
|
¡ W 5 Q I d H R X W 5 Q R X t ¡ x}| ¡ d}t ¡ 4 I d H B}| | tjj|¡ t I d H | ¡ t I d H |j
|
¡ W 5 Q I d H R X W 5 Q R X W 5 Q R X t ¡ \}}|ª | | ¡ | d¡ | d¡ t| t ¡ 4 I d H H}| | tjj|¡ | | tjj|¡ t I d H | ¡ |j
|
¡ W 5 Q I d H R X W 5 Q R X W 5 Q R X W 5 Q R X t ¡ \}}| |ü | ¡ è td | ¡ | ¡ ¡ ¡ | d¡ | d¡ t| t ¡ fdd}t ¡ 4 I d H v}| | tjj|¡ | | tjj|¡ | |¡ t I d H | ¡ | d¡ ¡ I d H t I d H |j
|
¡ W 5 Q I d H R X W 5 Q R X W 5 Q R X W 5 Q R X d S )Nc × s& t t | | I d H W 5 Q R X d S r )r rh )Z
|
async_funcÚargsr r r Ú allow_OSErrort s
|
z8test_can_survive_unnotified_close.<locals>.allow_OSErrorFza={}, b={}, a2={}c s t j ¡I d H ¡ d S r )rf rg r& Úsetr ©rt Úer r Úwait_readable_a2_then_set° s zDtest_can_survive_unnotified_close.<locals>.wait_readable_a2_then_setr )r rj rf r8 r9 rg r& r r r: r; Údupr r r r+ rO r r ÚEventr
|
rk )rx rl r= Ús2r3 r< r| r rz r Ú!test_can_survive_unnotified_closeb sX
|
$ .
|
|
8
|
|
|
|
r )+r? rj r ÚselectrM ÚerrnoÚ
|
contextlibr Ú r Ztestingr r r rf r r Zfixturer r" rg r& Zwait_readable_optionsr+ Zwait_writable_optionsr, Znotify_closing_optionsZ options_listÚmarkZ parametrizeZread_socket_testZwrite_socket_testZnotify_closing_testr> rB rC rG rK rd ri rm rv r r r r r Ú<module> sr
|
|
|
|
ý ÿ ÿ ÿ A
|
|
1 0
|
