
    "j~                         d Z ddlZddlZddlZddlmZ ej                  j                  dd       ddlm	Z	 ddl
mZ ddlmZmZmZ  G d	 d
ej                         Zedk(  r ej&                          yy)z)Unit tests for `ConnectionAsync` class.

    N)patchz..)ConnectionAsync   )awaiter)FakeStreamReaderFakeStreamWriterasync_patchc                   <    e Zd Zed        Zed        Zed        Zy)TestConnectionAsyncc                   K   t        dt               t               f      5  t               }|j	                          d {    | j                  |j                         | j                  |j                         d d d        j                          d {    | j                  |j                         | j                  |j                         y 7 # 1 sw Y   ZxY w7 Iw)Nasyncio.open_connectionreturn_value)
r	   r   r   r   connectassertIsNotNonereaderwritercloseassertIsNoneselfconns     d/var/www/oz-panel.ru/ozpay-pub/venv/lib/python3.12/site-packages/test_async/test_connection_async.pytest_connect_closez&TestConnectionAsync.test_connect_close   s     2BRBTVfVhAij 	."$D,,.    -  -		. jjl$++&$++& !	. 	. 	s:    C"C C:C;C"C ;C"CCC"c                 :  K   t        dt               t               f      5  t               }|j	                          d {    | j                  |j                         | j                  |j                         d d d        t        dj                  t              t              5  j                          d {    | j                  |j                         | j                  |j                         d d d        y 7 # 1 sw Y   xY w7 Q# 1 sw Y   y xY ww)Nr   r   z{}.FakeStreamWriter.closeside_effect)r	   r   r   r   r   r   r   r   r   format__name__OSErrorr   r   r   s     r    test_connect_close_catch_oserrorz4TestConnectionAsync.test_connect_close_catch_oserror    s     2BRBTVfVhAij 	."$D,,.    -  -		. .55h?WU 	+**,dkk*dkk*	+ 	+	 !	. 	. 	+ 	+sX    DD C?:D;,D'D;D<:D6	D?DD
DDDDc                   K   | j                  t              5  t        dt        j                        5  t        d      }|j                          d {    d d d        d d d        y 7 # 1 sw Y   xY w# 1 sw Y   y xY ww)Nr   r   r   )timeout)assertRaisesRuntimeErrorr	   asyncioTimeoutErrorr   r   r   s     r   test_connect_with_timeoutz-TestConnectionAsync.test_connect_with_timeout-   sr     |, 	%6GDXDXY %&q1lln$$%	% 	% %% %	% 	%sE   BA8 A,A*A,A8!	B*A,,A5	1A88B=BN)r   
__module____qualname__r   r   r!   r(        r   r   r      s<    	' 	' 
+ 
+ % %r,   r   __main__)__doc__r&   sysunittestunittest.mockr   pathinsertppadb.connection_asyncr   async_wrapperr   patchersr   r   r	   TestCaser   r   mainr+   r,   r   <module>r9      sc   
  
   4  2 " E E%(++ %D zHMMO r,   