File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_transports.cpython-310.opt-2.pyc
o
�i� � @ s\ d dl Z d dl mZ d dlZd dlmZ dd� ZG dd� de j�Zedkr,e �� dS dS ) � N)�mock)�
transportsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �>/usr/local/lib/python3.10/test/test_asyncio/test_transports.py�tearDownModule
s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�TransportTestsc C s t �� }| �|ji � d S r )r � Transport� assertEqualZ_extra��self� transportr r r �test_ctor_extra_is_none s z&TransportTests.test_ctor_extra_is_nonec C sN t �ddi�}| �d|�d�� | �|�d�� t� }| �||�d|�� d S )N�extra�info�unknown)r r
r Zget_extra_infoZ assertIsNone�objectZassertIs)r
r �defaultr r r �test_get_extra_info s
z"TransportTests.test_get_extra_infoc sX t �� � G � fdd�dtj�}|� }|�dtd�td�g� | �d� j� � � d� d S )Nc s e Zd Z� fdd�ZdS )z3TransportTests.test_writelines.<locals>.MyTransportc s � |� d S r r )r
�data��writerr r �write" s z9TransportTests.test_writelines.<locals>.MyTransport.writeN)�__name__�
__module__� __qualname__r r r r r � MyTransport! s r s line1s line2s line3� s line1line2line3)
r �Mockr r
�
writelines� bytearray�
memoryviewr Z
call_countZassert_called_with)r
r r r r r �test_writelines s �z TransportTests.test_writelinesc C s� t �� }| �t|j� | �t|j� | �t|jd� | �t|j� | �t|j� | �t|j � | �t|j
� | �t|j � | �t|j � | �t|j
� d S �Nr )r r
� assertRaises�NotImplementedError�set_write_buffer_limits�get_write_buffer_sizer Z write_eofZ
can_write_eofZ
pause_readingZresume_readingZ
is_reading�close�abortr r r r �test_not_implemented- s �z#TransportTests.test_not_implementedc C s* t �� }| �t|jd� | �t|j� d S r$ )r ZDatagramTransportr% r& �sendtor* r r r r �test_dgram_not_implemented< s z)TransportTests.test_dgram_not_implementedc C sd t �� }| �t|j� | �t|j� | �t|jd� | �t|jd� | �t|j� | �t|j � d S )Nr )
r ZSubprocessTransportr% r& Zget_pidZget_returncodeZget_pipe_transport� send_signal� terminate�killr r r r �)test_subprocess_transport_not_implementedB s z8TransportTests.test_subprocess_transport_not_implementedc C s� G dd� dt jt j�}t�� }||d�}t�� |_| �|j� | �t d�� |j
ddd� W d � n1 s7w Y |j
dd d� | �|j� | � |� � d
� |j
d d d� | �
|j� | � |� � d � d S )
Nc @ s e Zd Zdd� ZdS )zKTransportTests.test_flowcontrol_mixin_set_write_limits.<locals>.MyTransportc S s dS )Ni r )r
r r r r( Q s zaTransportTests.test_flowcontrol_mixin_set_write_limits.<locals>.MyTransport.get_write_buffer_sizeN)r r r r( r r r r r N s r )�loopzhigh.*must be >= lowr r )�high�low� � )r6 r5 � )r6 r7 )r Z_FlowControlMixinr
r r Z _protocolZ assertFalseZ_protocol_pausedZassertRaisesRegex�
ValueErrorr' r Zget_write_buffer_limitsZ
assertTrue)r
r r2 r r r r �'test_flowcontrol_mixin_set_write_limitsL s �
� z6TransportTests.test_flowcontrol_mixin_set_write_limitsN)
r r r r r r# r+ r- r1 r9 r r r r r s
r �__main__) Zunittestr r r r ZTestCaser r �mainr r r r �<module> s V �