
    daf                         d Z ddlZddlmZ ddlZddlmZ d Z G d dej                        Zedk(  r ej                          yy)	zTests for transports.py.    N)mock)
transportsc                  .    t        j                  d        y N)asyncioset_event_loop_policy     </root/Python-3.12.4/Lib/test/test_asyncio/test_transports.pytearDownModuler   
   s     !!$'r
   c                   6    e Zd Zd Zd Zd Zd Zd Zd Zd Z	y)	TransportTestsc                 d    t        j                         }| j                  |j                  i        y r   )r   	TransportassertEqual_extraself	transports     r   test_ctor_extra_is_nonez&TransportTests.test_ctor_extra_is_none   s&    %%'	))2.r
   c                    t        j                  ddi      }| j                  d|j                  d             | j	                  |j                  d             t               }| j                  ||j                  d|             y )Nextrainfounknown)r   r   r   get_extra_infoassertIsNoneobjectassertIs)r   r   defaults      r   test_get_extra_infoz"TransportTests.test_get_extra_info   sm    %%w&78	!9!9'!BC)229=>(gy77	7KLr
   c                    t        j                          G fddt        j                        } |       }|j	                  dt        d      t        d      g       | j                  dj                         j                  d       y )Nc                       e Zd Z fdZy)3TransportTests.test_writelines.<locals>.MyTransportc                      |       y r   r	   )r   datawriters     r   writez9TransportTests.test_writelines.<locals>.MyTransport.write"   s
    tr
   N)__name__
__module____qualname__r'   )r&   s   r   MyTransportr#   !   s    r
   r+   s   line1s   line2s   line3   s   line1line2line3)
r   Mockr   r   
writelines	bytearray
memoryviewr   
call_countassert_called_with)r   r+   r   r&   s      @r   test_writelineszTransportTests.test_writelines   su    	'++ 	  M	h'1(24 	5 	F--.!!"45r
   c                    t        j                         }| j                  t        |j                         | j                  t        |j
                         | j                  t        |j                  d       | j                  t        |j                         | j                  t        |j                         | j                  t        |j                         | j                  t        |j                         | j                  t        |j                         | j                  t        |j                         | j                  t        |j                         y Nr%   )r   r   assertRaisesNotImplementedErrorset_write_buffer_limitsget_write_buffer_sizer'   	write_eofcan_write_eofpause_readingresume_reading
is_readingcloseabortr   s     r   test_not_implementedz#TransportTests.test_not_implemented-   s    %%'	-#;;	=-y/N/NO-yG-y/B/BC-y/F/FG-y/F/FG-y/G/GH-y/C/CD-y?-y?r
   c                     t        j                         }| j                  t        |j                  d       | j                  t        |j
                         y r5   )r   DatagramTransportr6   r7   sendtor@   r   s     r   test_dgram_not_implementedz)TransportTests.test_dgram_not_implemented<   s>    --/	-y/?/?H-y?r
   c                    t        j                         }| j                  t        |j                         | j                  t        |j
                         | j                  t        |j                  d       | j                  t        |j                  d       | j                  t        |j                         | j                  t        |j                         y )Nr,   )
r   SubprocessTransportr6   r7   get_pidget_returncodeget_pipe_transportsend_signal	terminatekillr   s     r   )test_subprocess_transport_not_implementedz8TransportTests.test_subprocess_transport_not_implementedB   s    //1	-y/@/@A-y/G/GH-y/K/KQO-y/D/DaH-y/B/BC-y~~>r
   c                     G d dt         j                  t         j                        }t        j                         } ||      }t        j                         |_        | j                  |j                         | j                  t        d      5  |j                  dd       d d d        |j                  dd	       | j                  |j                         | j                  |j                         d
       |j                  dd	       | j                  |j                         | j                  |j                         d       y # 1 sw Y   xY w)Nc                       e Zd Zd Zy)KTransportTests.test_flowcontrol_mixin_set_write_limits.<locals>.MyTransportc                      y)Ni   r	   )r   s    r   r9   zaTransportTests.test_flowcontrol_mixin_set_write_limits.<locals>.MyTransport.get_write_buffer_sizeQ   s    r
   N)r(   r)   r*   r9   r	   r
   r   r+   rQ   N   s    r
   r+   )loopzhigh.*must be >= lowr   r,   )highlow      )rW   rV      )rW   rX   )r   _FlowControlMixinr   r   r-   	_protocolassertFalse_protocol_pausedassertRaisesRegex
ValueErrorr8   r   get_write_buffer_limits
assertTrue)r   r+   rS   r   s       r   'test_flowcontrol_mixin_set_write_limitsz6TransportTests.test_flowcontrol_mixin_set_write_limitsL   s   	*66$..	 yy{T*	"iik	334##J0FG 	=--1!-<	= 	))t)=334::<kJ))s)<	223::<jI	= 	=s   E		EN)
r(   r)   r*   r   r    r3   rA   rE   rN   ra   r	   r
   r   r   r      s)    /M6@@?Jr
   r   __main__)
__doc__unittestr   r   r   r   TestCaser   r(   mainr	   r
   r   <module>rg      sJ        (SJX&& SJl zHMMO r
   