U
|
Z ±dÙ" ã @ s d dl Z d dlZd dlZd dlZddlmZ ddlmZm Z m
|
Z
|
ddl T ddlmZ dd Z d d
|
Zd d Zd d Zdd Zdd Zdd Zdd ZdS )é Né )Ú_core)Úcheck_half_closeable_streamÚwait_all_tasks_blockedÚassert_checkpoints)Ú*)Úsocketc à s t ¡ \} }| 4 |$ t t¡ t| W 5 Q R X W 5 Q R X W 5 Q R X tjtjd$}t t ¡ t| W 5 Q R X W 5 Q R X t ¡ \} }| , | t| }|j| ks°t
|
W 5 Q R X W 5 Q R X t ¡ °}| d¡I d H | d¡ t ¡ }| | ¡ ¡I d H t|}| tjtj¡s$t
|
| tjtjd¡ | tjtj¡rLt
|
| tjtjd¡}t|tsnt
|
W 5 Q R X W 5 Q R X d S )N©Útype©z 127.0.0.1r é F)Ú stdlib_socketÚ
|
socketpairÚpytestÚraisesÚ TypeErrorÚ SocketStreamÚtsocketr Ú
|
SOCK_DGRAMÚ
|
ValueErrorÚAssertionErrorÚbindÚlistenÚconnectÚ getsocknameÚ
|
getsockoptÚ IPPROTO_TCPÚ TCP_NODELAYÚ
|
setsockoptÚ
|
isinstanceÚbytes)ÚaÚbÚsockÚsÚ listen_sockÚ client_sock© r' úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_highlevel_socket.pyÚtest_SocketStream_basics s, & "
|
|
|
r) c sä d t ¡ \} }| Æ |¶ t| t| fdd} fdd}t ¡ 4 I d H }| |¡ | |¡ W 5 Q I d H R X d¡I d H d¡I d H dks¦t ¡ I d H d¡I d H dksÌtW 5 Q R X W 5 Q R X d S ) Ni c s( t } | ¡I d H | t 7 } d S ©N)Ú bytearrayÚsend_allr )Údata)ÚBIGr! r' r( ÚsenderB s z*test_SocketStream_send_all.<locals>.senderc sB t I d H d} | k r2| t ¡I d H 7 } q| ks>td S )Nr )r ÚlenÚ receive_somer )Únbytes)r. r" r' r( ÚreceiverZ s
|
z,test_SocketStream_send_all.<locals>.receiveró eé
|
ó )
|
r r r r Ú open_nurseryÚ
|
start_soonr, r1 r Zsend_eof)Za_sockZb_sockr/ r3 Únurseryr' )r. r! r" r( Útest_SocketStream_send_all7 s
|
r: c
|
sP fdd}dd }t ¡ 4 I d H }| |¡ | ||¡ W 5 Q I d H R X d S )Nc s dd ¡I d H q d S )Nó xi' )r, r' ©r$ r' r( r/ o s zfill_stream.<locals>.senderc Ó s t I d H | j ¡ d S r* )r Z cancel_scopeÚcancel)r9 r' r' r( Úwaiters s zfill_stream.<locals>.waiter)r r7 r8 )r$ r/ r> r9 r' r< r( Ú fill_streamn s
|
|
r? c s( dd fdd} t | I d H d S )Nc Ó s t ¡ \} }t| t|fS r* )r r r ©ÚleftÚrightr' r' r( Ú stream_maker} s z/test_SocketStream_generic.<locals>.stream_makerc s4 I d H \} }t | I d H t |I d H | |fS r* )r? r@ ©rC r' r( Úclogged_stream_maker s z7test_SocketStream_generic.<locals>.clogged_stream_maker)r )rE r' rD r( Útest_SocketStream_generic| s rF c
|
à s6 t ¡ 8} | d¡ | d¡ t t¡ t| W 5 Q R X W 5 Q R X tjtj d>} | d¡I d H t t
|
¡}t| W 5 Q R X | d¡ W 5 Q R X t j dkrît ¡ >} | d¡I d H t t
|
¡}t| W 5 Q R X | d¡ W 5 Q R X t ¡ }| d¡I d H | d¡ t|}|j|ks(tt ¡ }| | ¡ ¡I d H t | ¡ I d H }W 5 Q R X t|tstt|j ¡ | ¡ kst|j ¡ | ¡ ks¤tt | ¡ I d H W 5 Q R X t | ¡ I d H W 5 Q R X t , t tj¡ | ¡ I d H W 5 Q R X W 5 Q R X | ¡ | ¡ I d H d S )Nr r5 r z .*SOCK_STREAMÚdarwinz.*listen)r r r r r r r ÚSocketListenerr r r ÚmatchÚsysÚplatformr r r r Úacceptr r Ú getpeernameÚacloser ÚClosedResourceErrorÚclose)r$ Úexcinfor% Úlistenerr& Z server_streamr' r' r( Útest_SocketListener sJ
|
|
|
|
|
|
"rS c
|
à sn t ¡ } | d¡I d H | d¡ t| }| ¡ t , t t j
|
¡ | ¡ I d H W 5 Q R X W 5 Q R X d S )Nr r5 ) r r r r rH rP r r r r rO rL )r% rR r' r' r( Ú+test_SocketListener_socket_closed_underfoot¿ s
|
rT c à s( G dd dt j} | g }| ttjdttjdttjd|ttjdttjdttj d|g}t
|
|}t " | ¡ I d H }|j |kstW 5 Q R X tjtjtj fD ]J}t : t t¡}| ¡ I d H W 5 Q R X |jj|ksètW 5 Q R X q¨t $ | ¡ I d H }|j |kstW 5 Q R X d S ) Nc @ s2 e Zd Zdd ZejZdd Zdd Zdd Z d S )
|
z5test_SocketListener_accept_errors.<locals>.FakeSocketc S s t || _d S r* )ÚiterÚ_events)ÚselfÚeventsr' r' r( Ú__init__Ð s z>test_SocketListener_accept_errors.<locals>.FakeSocket.__init__c S s dS )NTr' )rW ÚlevelÚoptr' r' r( r Ö s z@test_SocketListener_accept_errors.<locals>.FakeSocket.getsockoptc S s d S r* r' )rW rZ r[ Úvaluer' r' r( r Ù s z@test_SocketListener_accept_errors.<locals>.FakeSocket.setsockoptc Ó s4 t ¡ I d H t| j}t|tr(|n|d fS d S r* )r Ú
|
checkpointÚnextrV r Ú BaseException)rW Úeventr' r' r( rL Ü s
|
|
|
z<test_SocketListener_accept_errors.<locals>.FakeSocket.acceptN)
|
Ú__name__Ú
|
__module__Ú __qualname__rY r Ú SOCK_STREAMr
|
r r rL r' r' r' r( Ú
|
FakeSocketÏ s
|
re zConnection abortedzPermission deniedz Bad protocolzOut of file descriptorsz$attempt to write to read-only memoryzout of buffers)r Ú
|
SocketTypeÚOSErrorÚerrnoÚ ECONNABORTEDÚEPERMZEPROTOZEMFILEZEFAULTZENOBUFSrH r rL r r r r r\ )re Zfake_server_sockZfake_listen_sockÚlr$ ÚcoderQ r' r' r( Ú!test_SocketListener_accept_errorsÎ s2
|
|
|
|
|
|
øÿ rm c
|
à s t ¡ \} }| f |V | d¡I d H | ¡ t| }| d¡I d H dksPt| d¡I d H dkshtW 5 Q R X W 5 Q R X d S )Nr; r r6 )r r ÚsendrP r r1 r )Zsock_aZsock_bÚstreamr' r' r( Ú5test_socket_stream_works_when_peer_has_already_closed s rp )r rJ r r rh Ú r Ztestingr r r Z_highlevel_socketr r) r: r? rF rS rT rm rp r' r' r' r( Ú<module> s &756
|
