U
|
Z ±dê% ã @ s> d dl Z d dlZd dlZd dlZd dlZd dlZddlmZmZ ddl m
|
Z
|
m Z ddl m Z mZ ejdkZejje ddZerdd lmZ n"e e¡ dd lmZ W 5 Q R X d
|
d d d Zdd Zdd Zdd Zdd Z dd Z dd Z dd Z d d Z d d Z!d d! Z"ejjej# $d"¡d#dd$d% Z%ed&d' Z&dS )(é Né )Úgc_collect_harderÚskip_if_fbsd_pipes_broken)Ú_coreÚ move_on_after)Úwait_all_tasks_blockedÚcheck_one_way_streamÚposixz
|
posix only)Úreason)ÚFdStreamzTuple[FdStream, FdStream])Úreturnc à s t ¡ \} }t|t| fS )zMakes a new pair of pipes.)ÚosÚpiper )ÚrÚw© r úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_unix_pipes.pyÚ make_pipe s r c à sX t I d H \} }z*ttdd}t | ¡ d| d ¡ qW n tk
|
rN Y nX | |fS )NZPIPE_BUFi ó xr )r ÚgetattrÚselectr ÚwriteÚfilenoÚBlockingIOError)Úsr Úbuf_sizer r r Úmake_clogged_pipe s r c
|
à sp t ¡ \} }t|4 I d H D}| ¡ |ks.t| d¡I d H t | d¡dksRtt | ¡ W 5 Q I d H R X d S ©Ns 123é )r r r r ÚAssertionErrorÚsend_allÚreadÚclose)r r Úsendr r r Útest_send_pipe7 s r$ c
|
à sp t ¡ \} }t| 4 I d H D}| ¡ | ks.tt |d¡ | d¡I d H dksRtt |¡ W 5 Q I d H R X d S r )r r r r r r Ú receive_somer" )r r Úrecvr r r Útest_receive_pipeA s r' c
|
s t I d H \d fdd} fdd}t ¡ 4 I d H }| | ¡ | |¡ W 5 Q I d H R X ¡ I d H ¡ I d H d S )Ni c s t } | ¡I d H d S ©N)Ú bytearrayr )Úbig)Úcountr r r ÚsenderO s z#test_pipes_combined.<locals>.senderc sB t I d H d} | k r2| t d¡I d H 7 } q| ks>td S )Nr i )r Úlenr% r )Zreceived)r+ r! r r ÚreaderS s
|
z#test_pipes_combined.<locals>.reader)r r Ú open_nurseryÚ
|
start_soonÚaclose)r, r. Únr )r+ r! r r Útest_pipes_combinedK s
|
r3 c
|
à s t t¡ td W 5 Q R X t ¡ \} }t |¡ t| 4 I d H ,}t t¡ | d¡I d H W 5 Q R X W 5 Q I d H R X d S )Nr ) ÚpytestÚraisesÚ TypeErrorr r r r" Ú
|
ValueErrorr% )r r r r r r Útest_pipe_errorsc s
|
r8 c à s t I d H \} }| ¡ | ¡ }}~ ~t t t¡}t |¡ W 5 Q R X |jj t j
|
ks^t t t¡}t |¡ W 5 Q R X |jj t j
|
kst d S r( ) r r r r4 r5 ÚOSErrorr r" ÚvalueÚerrnoÚEBADFr )r r Úf1Úf2Úexcinfor r r Útest_deln s r@ c à sÜ t I d H \} }| 4 I d H $ |4 I d H W 5 Q I d H R X W 5 Q I d H R X | ¡ dks\t| ¡ dksltt t¡}t | ¡ ¡ W 5 Q R X |jj t j
|
ks¢tt t¡}t | ¡ ¡ W 5 Q R X |jj t j
|
ksØtd S )Néÿÿÿÿ) r r r r4 r5 r9 r r" r: r; r< )r r r? r r r Útest_async_with} s rB c sü t I d H \} }| ¡ }| ¡ I d H | ¡ I d H t ¡ \}}||kr^t ||¡ t |¡ t|4 I d H ~ ¡ |kst fdd}t
|
¡ 4 I d H @}| |¡ t I d H | ¡ I d H t I d H t |¡ W 5 Q I d H R X W 5 Q I d H R X d S )Nc s d¡I d H dkstd S )Né
|
ó )r% r r ©Zr2r r Ú
|
expect_eof¢ s z6test_misdirected_aclose_regression.<locals>.expect_eof) r r r1 r r Údup2r" r r r r/ r0 r )r r Zold_r_fdZr2_fdZw2_fdrF Únurseryr rE r Ú"test_misdirected_aclose_regression s"
|
|
rI c sÎ fdd}t jjj fdd}| t jjd|¡ t I d H \}|4 I d H l 4 I d H N t ¡ 4 I d H ,}| |¡ t I d H | d¡I d H 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 )Nc s, t tj¡ d¡I d H W 5 Q R X d S )NrC )r4 r5 r ÚClosedResourceErrorr% r )r r r Úexpect_closedresourceerror¼ s zKtest_close_at_bad_time_for_receive_some.<locals>.expect_closedresourceerrorc s" | |I d H ¡ I d H d S r( ©r1 ©ÚargsÚkwargs©Zorig_wait_readabler r r Úpatched_wait_readable s zFtest_close_at_bad_time_for_receive_some.<locals>.patched_wait_readableÚ wait_readabler )
|
r Ú_runÚ TheIOManagerrR Úsetattrr r/ r0 r r )Ú monkeypatchrK rQ r rH r rP r Ú'test_close_at_bad_time_for_receive_some´ s
|
|
rW c sä fdd}t jjj fdd}| t jjd|¡ t I d H \}4 I d H |4 I d H d t ¡ 4 I d H B}| |¡ t I d H | d¡I d H }|r°| d¡I d H }qW 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 )Nc s, t tj¡ d¡I d H W 5 Q R X d S )Nsd xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)r4 r5 r rJ r r )r r r rK Ø s zGtest_close_at_bad_time_for_send_all.<locals>.expect_closedresourceerrorc s" | |I d H ¡ I d H d S r( rL rM ©Zorig_wait_writabler r r Úpatched_wait_writableÞ s zBtest_close_at_bad_time_for_send_all.<locals>.patched_wait_writableÚ wait_writablei' )
|
r rS rT rZ rU r r/ r0 r r% )rV rK rY r rH Z received_datar rX r Ú#test_close_at_bad_time_for_send_allÐ s
|
|
r[ Zfreebsdz7no way to make read() return a bizarro error on FreeBSDc à sª t I d H \} }| 4 I d H ~ |4 I d H ` t dtjd¡}z<t || ¡ ¡ t t j
|
¡ | d¡I d H W 5 Q R X W 5 t |¡ X W 5 Q I d H R X W 5 Q I d H R X d S )Nú/r rC ) r r ÚopenZ O_DIRECTORYr" rG r r4 r5 r ZBrokenResourceErrorr% )r r Údir_fdr r r Ú!test_bizarro_OSError_from_receiveû s r_ c à s t ttI d H d S r( )r r r r r r r Útest_pipe_fully s r` )'r; r r ÚtempfileÚsysr4 Z_core.tests.tutilr r Ú r r Ztestingr r Únamer ÚmarkZskipifZ
|
pytestmarkZ _unix_pipesr r5 Ú ImportErrorr r r$ r' r3 r8 r@ rB rI rW r[ ÚplatformÚ
|
startswithr_ r` r r r r Ú<module> s@
|
|
|
& +
|
þ
|
|
