U
|
Z ±d ã @ s d dl Z d dlZd dlZd dlmZ d dlmZ d dlZd dlmZ d dl m
|
Z
|
m Z m Z ej dd Zej dd Zej d
|
d Zd d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Z d$d% Z d&d' Z d(d) Z d*d+ Z dS ),é N)Úmock)Úsentinel)Ú_core)ÚAsyncIOWrapperÚ_FILE_SYNC_ATTRSÚ_FILE_ASYNC_METHODSc C s t | d¡¡S )NÚtest)ÚosÚfspathÚjoin)Ztmpdir© r úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/tests/test_file_io.pyÚpath s r c C s t jtjdS )N)Zspec_set)r ZMockÚioÚStringIOr r r r Úwrapped s r c C s
|
t | ¡S ©N)ÚtrioÚ wrap_file)r r r r Ú
|
async_file s r c C s& t t¡ t t ¡ W 5 Q R X d S r )ÚpytestÚraisesÚ TypeErrorr r Ústrr r r r Útest_wrap_invalid s r c C sf G dd d} | }t |tjr$tt |¡}t |ts<t| `t t
|
¡ t | ¡ W 5 Q R X d S )Nc @ s e Zd Zdd Zdd ZdS )z&test_wrap_non_iobase.<locals>.FakeFilec S s d S r r ©Úselfr r r Úclose# s z,test_wrap_non_iobase.<locals>.FakeFile.closec S s d S r r r r r r Úwrite& s z,test_wrap_non_iobase.<locals>.FakeFile.writeN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r ÚFakeFile" s r" ) Ú
|
isinstancer ÚIOBaseÚAssertionErrorr r r r r r r )r" r r r r r Útest_wrap_non_iobase! s
|
r& c C s | j |kstd S r )r r% ©r r r r r Útest_wrapped_property5 s r( c sF t t¡}t fdd|D s&tt fdd|D rBtd S )Nc 3 s& | ] }|t kr|t kV qd S r ©Údir©Ú.0Úattrr' r r Ú <genexpr>= s z+test_dir_matches_wrapped.<locals>.<genexpr>c 3 s& | ] }|t kr|t kV qd S r r) r+ r' r r r. ? s )r Úunionr Úallr% Úany)r r Úattrsr r' r Útest_dir_matches_wrapped9 s
|
|
ÿr3 c C sR G dd dt j} t | ¡}t|jds.tt t ¡ t
|
|d W 5 Q R X d S )Nc @ s e Zd Zdd ZdS )z0test_unsupported_not_forwarded.<locals>.FakeFilec S s d S r r r r r r Úunsupported_attrF s zAtest_unsupported_not_forwarded.<locals>.FakeFile.unsupported_attrN)r r r! r4 r r r r r" E s r" r4 ) r Ú RawIOBaser r Úhasattrr r% r r ÚAttributeErrorÚgetattr)r" r r r r Ú test_unsupported_not_forwardedD s
|
r9 c C s4 t D ]*}|t| krqt| |t||kstqd S r )r r* r8 r% ©r r Ú attr_namer r r Útest_sync_attrs_forwardedQ s r< c
|
C s\ t D ]R}|t| krqt t¡ t| | W 5 Q R X t t¡ t|| W 5 Q R X qd S r )r r* r r r7 r8 r: r r r Ú test_sync_attrs_match_wrapperY s r= c C s4 t D ]*}|t| krqt| |t| |kstqd S r )r r* r8 r% )r Ú meth_namer r r Ú!test_async_methods_generated_oncee s r? c C s4 | j jdkst| j jdks td| j jks0td S )NÚreadzAsyncIOWrapper.readzio.StringIO.read)r@ r r% r! Ú__doc__)r r r r Ú test_async_methods_signaturem s rB c à sn t D ]d}|t| krqt| |}t||}|tjtjdI d H }|jtjtjd || ks`t| ¡ qd S )N)Úkeyword) r r* r8 r ZargumentrC Zassert_called_once_withr% Z
|
reset_mock)r r r> ÚmethZ wrapped_methÚvaluer r r Útest_async_methods_wrapu s
|
|
ÿrF c
|
à s\ t D ]R}|t| krqt t¡ t| | W 5 Q R X t t¡ t|| W 5 Q R X qd S r )r r* r r r7 r8 )r r r> r r r Ú test_async_methods_match_wrapper s rG c à s2 t | d¡I d H }t|ts t| ¡ I d H d S ©NÚw)r Ú open_filer# r r% Úaclose©r Úfr r r Ú test_open s rN c
|
à sR t | d¡I d H 4 I d H }t|ts*t|jr4tW 5 Q I d H R X |jsNtd S rH )r rJ r# r r% ÚclosedrL r r r Útest_open_context_manager s rP c à sX t t d¡¡} t| j}g }| j d¡ | 2 z3 d H W }| |¡ q.6 ||ksTtd S )Nz test
|
foo
|
barr ) r r r r Úlistr ÚseekÚappendr% )r ÚexpectedÚresultÚliner r r Útest_async_iter£ s
|
rW c
|
à s t ¡ n}t | d¡I d H }| ¡ t t j¡ | d¡I d H W 5 Q R X t t j¡ | ¡ I d H W 5 Q R X W 5 Q R X |j
|
st d S )NrI Úa) r Z CancelScoper rJ Úcancelr r Z Cancelledr rK rO r% )r ZcscoperM r r r Útest_aclose_cancelled¯ s
|
"rZ c à sJ t ¡ } t | ¡}t |¡}| ¡ I d H }t|ts8t|j | ksFtd S r )
|
r ÚBytesIOÚBufferedReaderr r Údetachr# r r% r )ÚrawZbufferedr Zdetachedr r r Ú test_detach_rewraps_asynciobase½ s
|
|
r_ )!r r r Zunittestr Z unittest.mockr r r Z trio._file_ior r r Zfixturer r r r r& r( r3 r9 r< r= r? rB rF rG rN rP rW rZ r_ r r r r Ú<module> s:
|
|
|
|
