U
|
Z ±d[
|
ã @ sl d dl Z d dlZddlmZmZ ddlmZ ejG dd deZejG dd deZ d d
|
Z
|
d d Z dS ) é Né )Ú
|
SendStreamÚ ReceiveStream)Ú StapledStreamc @ s0 e Zd ZejedZdd Zdd Zdd Z dS ) ÚRecordSendStream©Úfactoryc à s | j d|f¡ d S )NÚsend_all©ÚrecordÚappend)ÚselfÚdata© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_highlevel_generic.pyr s zRecordSendStream.send_allc à s | j d¡ d S )NÚ wait_send_all_might_not_blockr
|
©r r r r r s z.RecordSendStream.wait_send_all_might_not_blockc à s | j d¡ d S ©NÚacloser
|
r r r r r s zRecordSendStream.acloseN)
|
Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibÚlistr r r r r r r r r s r c @ s* e Zd ZejedZdddZdd ZdS )ÚRecordReceiveStreamr Nc à s | j d|f¡ d S )NÚ receive_somer
|
)r Z max_bytesr r r r s z RecordReceiveStream.receive_somec à s | j d¡ d S r r
|
r r r r r s zRecordReceiveStream.aclose)N) r r r r r r r r r r r r r r s
|
r c sL t t } t | }|j ks$t|j| ks2t| d¡I d H | ¡ I d H jddgksbt j ¡ |
|
¡ I d H jdgkst j ¡ fdd}| _
|
|
|
¡ I d H jdgksÄt j ¡ | jg ksÜt| d¡I d H | jd gksüt jg ks t| j ¡ | ¡ I d H | jdgks6t jdgksHtd S )
|
Nó foo)r r r r c s j d¡ d S )NÚsend_eofr
|
r ©Ú send_streamr r Ú fake_send_eof6 s z)test_StapledStream.<locals>.fake_send_eofr éÒ )r r" ) r r r r ÚAssertionErrorÚreceive_streamr r r Úclearr r r )r$ Ústapledr! r r r Útest_StapledStream" s8
|
þ
|
|
|
|
|
r' c à s G dd dt } G dd dt}t| | }t t¡}| ¡ I d H W 5 Q R X t|jj tsdt
|
|j j dgksvt
|
|j j dgkst
|
d S )Nc s e Zd Z fddZ ZS )z@test_StapledStream_with_erroring_close.<locals>.BrokenSendStreamc s t ¡ I d H td S ©N©Úsuperr Ú
|
ValueErrorr ©Ú __class__r r r N s zGtest_StapledStream_with_erroring_close.<locals>.BrokenSendStream.aclose©r r r r Ú __classcell__r r r, r ÚBrokenSendStreamM s r0 c s e Zd Z fddZ ZS )zCtest_StapledStream_with_erroring_close.<locals>.BrokenReceiveStreamc s t ¡ I d H td S r( r) r r, r r r S s zJtest_StapledStream_with_erroring_close.<locals>.BrokenReceiveStream.acloser. r r r, r ÚBrokenReceiveStreamR s r1 r )r r r ÚpytestZraisesr+ r Ú
|
isinstanceÚvalueÚ __context__r# r r r$ )r0 r1 r& Úexcinfor r r Ú&test_StapledStream_with_erroring_closeJ s r7 ) r2 r Úabcr r Z_highlevel_genericr Úsr r r' r7 r r r r Ú<module> s
|
(
|
