File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/test_buffered_proto.cpython-37.pyc
B
��g� � @ s� d dl Z d dlZd dlmZ G dd� de j�ZG dd� dej�ZG dd� deej �Z
e� e e d �d
�G d d � d eej ��Z
ed
kr�e�� dS )� N)�
functionalc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ReceiveStuffProtoc C s || _ || _d S )N)�cb� con_lost_fut)�selfr r � r �A/usr/local/lib/python3.7/test/test_asyncio/test_buffered_proto.py�__init__ s zReceiveStuffProto.__init__c C s t d�| _| jS )N�d )� bytearray�buffer)r �sizehintr r r �
get_buffer s
z ReceiveStuffProto.get_bufferc C s | � | jd |� � d S )N)r r )r �nbytesr r r �buffer_updated s z ReceiveStuffProto.buffer_updatedc C s& |d kr| j �d � n | j �|� d S )N)r Z
set_resultZ
set_exception)r �excr r r �connection_lost s z!ReceiveStuffProto.connection_lostN)�__name__�
__module__� __qualname__r r r r r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )�BaseTestBufferedProtocolc C s t �d S )N)�NotImplementedError)r r r r �new_loop s z!BaseTestBufferedProtocol.new_loopc s� dd � � �fdd�}� fdd�}�j �t�|dd��}|jd �� }�j �tj||�d �j d
�� |�� �j �|�� � d S ) Ns 12345678+i c � sR d����fdd���j �� � �j j� �fdd�f| �� I d H \�}� I d H d S )N� c s �| 7 ��� kr�� d� d S )N� 1)�write)�buf)�NOISE�data�trr r �on_buf& s z^BaseTestBufferedProtocol.test_buffered_proto_create_connection.<locals>.client.<locals>.on_bufc s
t �� �S )N)r r )�
conn_lost_futr r r �<lambda>/ r z`BaseTestBufferedProtocol.test_buffered_proto_create_connection.<locals>.client.<locals>.<lambda>)�loopZ
create_futureZcreate_connection)�addrZpr)r r )r! r r r r �client# s
zNBaseTestBufferedProtocol.test_buffered_proto_create_connection.<locals>.clientc � s4 |� � � | �d�I d H |�� |�� I d H d S )N� )r Z readexactly�close� wait_closed)�reader�writer)r r r �on_server_client3 s
zXBaseTestBufferedProtocol.test_buffered_proto_create_connection.<locals>.on_server_clientz 127.0.0.1r � )r# ) r# Zrun_until_complete�asyncioZ start_serverZsocketsZ getsocknameZwait_forr' r( )r r% r+ Zsrvr$ r )r r r �%test_buffered_proto_create_connection s z>BaseTestBufferedProtocol.test_buffered_proto_create_connectionN)r r r r r. r r r r r s r c @ s e Zd Zdd� ZdS )� BufferedProtocolSelectorTestsc C s t �� S )N)r- ZSelectorEventLoop)r r r r r H s z&BufferedProtocolSelectorTests.new_loopN)r r r r r r r r r/ E s r/ �ProactorEventLoopz Windows onlyc @ s e Zd Zdd� ZdS )� BufferedProtocolProactorTestsc C s t �� S )N)r- r0 )r r r r r P s z&BufferedProtocolProactorTests.new_loopN)r r r r r r r r r1 L s r1 �__main__)r- ZunittestZtest.test_asyncior Z
func_testsZBufferedProtocolr ZFunctionalTestCaseMixinr ZTestCaser/ Z
skipUnless�hasattrr1 r �mainr r r r �<module> s +