File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/test_selector_events.cpython-38.pyc
U
/D�g�� � @ s2 d Z ddlZddlZddlZddlmZ z ddlZW n ek
rL dZY nX ddlZddlm Z ddlm
Z
ddlm Z ddlm Z ddl
mZ ejZd d
� ZG d d � d e �Zd dd�Zdd� ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�Ze d k�r.e� � dS ) z Tests for selector_events.py� N)�mock)�BaseSelectorEventLoop)�_SelectorTransport)�_SelectorSocketTransport)�_SelectorDatagramTransport)�utilsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r
r
�E/opt/cppython/lib/python3.8/test/test_asyncio/test_selector_events.py�tearDownModule s r c @ s e Zd Zdd� Zdd� ZdS )�TestBaseSelectorEventLoopc C s& t �� | _t �� | _| jd7 _d S �N� �r �Mock�_ssock�_csockZ
_internal_fds��selfr
r
r �_make_self_pipe s
z)TestBaseSelectorEventLoop._make_self_pipec C s d S r r
r r
r
r �_close_self_pipe" s z*TestBaseSelectorEventLoop._close_self_pipeN)�__name__�
__module__� __qualname__r r r
r
r
r r
s r
r
c C s t � �| �S r )� bytearray�join)�lr
r
r �list_to_buffer& s r c C s" | j d krd S | j �� d | _ d S r )�_sock�close)� transportr
r
r �close_transport* s
r" c s e Zd Z� fdd�Zdd� Ze�edkd�dd� �Ze �
d d�e �
d
d�d d � ��Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!� Z"S )9�BaseSelectorEventLoopTestsc s: t � �� t�� | _g | jj_t| j�| _| � | j� d S r )
�super�setUpr r �selector�select� return_valuer
�loop�set_event_loopr �� __class__r
r r% 5 s
z BaseSelectorEventLoopTests.setUpc C sZ t �� }t �� | j_d| jj_| j�|t�� �}| �|t � | j�
� t |� t |� d S )NF)
r r r) �
add_reader�
_is_coroutineZ_make_socket_transportr �Protocol�assertIsInstancer r �reprr" )r �mr! r
r
r �test_make_socket_transport< s
z5BaseSelectorEventLoopTests.test_make_socket_transportNz
No ssl modulec
C s, t �� }t �� | j_d| jj_t �� | j_t �� | j_t �� | j_| j�� }t �
� �H | j� |t �
� ||�}| �td�� |�� W 5 Q R X t �| j� W 5 Q R X | �|�� � |�� |�� | �|�� � |�� |�� | �|�� � |jj}| �d|�� � | �d|�� � |�� t �| j� d S )NFz SSL transport.*not.*initialized�sslr! )r r r) � _add_readerr. � _add_writer�_remove_reader�_remove_writer�
create_future�
test_utils�disable_logger�_make_ssl_transportr r/ ZassertRaisesRegex� RuntimeError�
is_reading� run_briefly�
assertTrue�
pause_reading� assertFalse�resume_readingr, r ZassertIn�lowerr )r r2 �waiterr! �
class_namer
r
r �test_make_ssl_transportI s>
��z2BaseSelectorEventLoopTests.test_make_ssl_transportzasyncio.selector_events.sslzasyncio.sslproto.sslc C sd t �� }t �� | j_t �� | j_t �� | j_t �� | j_| �t�� | j� ||||� W 5 Q R X d S r )
r r r) r- �
add_writer�
remove_reader�
remove_writer� assertRaisesr= r< )r r2 r
r
r �)test_make_ssl_transport_without_ssl_erroro s zDBaseSelectorEventLoopTests.test_make_ssl_transport_without_ssl_errorc C sh G dd� dt �}|| j�| _| �| j� | jj}d|j_| jj}d|j_t� � }| j_
| jj � � t� � | j_ }| �
| j�� � | j� � | �| j�� � | �| jj � | �| jj� | �| jj� |j �� |j �� |j �� |�d� | j� � | j� � | j�� }| �t| jj� | �t| jj|� d}dd� }| �t| jj||� | �t| jj||� d S )Nc @ s e Zd Zdd� ZdS )z8BaseSelectorEventLoopTests.test_close.<locals>.EventLoopc S s& t �� | _t �� | _| jd7 _d S r r r r
r
r r | s
zHBaseSelectorEventLoopTests.test_close.<locals>.EventLoop._make_self_pipeN)r r r r r
r
r
r � EventLoop{ s rM � r r c S s d S r r
r
r
r
r �callback� s z7BaseSelectorEventLoopTests.test_close.<locals>.callback)r r& r) r* r �filenor( r r r r7 � _selectorr rB Z is_closedr@ � assertIsNone�assert_called_withr9 rK r= Z run_forever�run_until_completer- rH ) r rM ZssockZcsockrI r& �f�fdrO r
r
r �
test_closez s:
z%BaseSelectorEventLoopTests.test_closec C s< t �� | j_| jj�� d | j_| j�� | �| jj� d S r )r r r) rI rQ r rR r r
r
r �test_close_no_selector� s
z1BaseSelectorEventLoopTests.test_close_no_selectorc C s t | jjj_| �| j�� � d S r )�BlockingIOErrorr) r �recv� side_effectrR �_read_from_selfr r
r
r � test_read_from_self_tryagain� s z7BaseSelectorEventLoopTests.test_read_from_self_tryagainc C s t | jjj_| �t | jj� d S r )�OSErrorr) r rZ r[ rK r\ r r
r
r � test_read_from_self_exception� s z8BaseSelectorEventLoopTests.test_read_from_self_exceptionc C s4 t | jjj_t�� � | �| j�� � W 5 Q R X d S r ) rY r) r �sendr[ r: r; rR �_write_to_selfr r
r
r �test_write_to_self_tryagain� s
z6BaseSelectorEventLoopTests.test_write_to_self_tryagainc C s" t � | jjj_| �t | jj� d S r )r= r) r r` r[ rK ra r r
r
r � test_write_to_self_exception� s z7BaseSelectorEventLoopTests.test_write_to_self_exceptionc C s� t | jjj_dd� }| j�d|� | �| jjjj� | jjjj d \}}\}}| �
d|� | �
t j |� | �
||j
� | �|� d S )Nc S s dS �NTr
r
r
r
r �<lambda>� � z<BaseSelectorEventLoopTests.test_add_reader.<locals>.<lambda>r r )�KeyErrorr) rQ �get_keyr[ r- r@ �register�called� call_args� assertEqual� selectors�
EVENT_READ� _callbackrR �r �cbrV �mask�r�wr
r
r �test_add_reader� s z*BaseSelectorEventLoopTests.test_add_readerc C s� t �� }t �� }t�ddtj||f�| jjj_dd� }| j� d|� | �
|j j � | �
| jjjj � | �
| jjjj � | jjjjd \}}\}}| �d|� | �tjtjB |� | �||j� | �||� d S )Nr c S s dS rd r
r
r
r
r re � rf zEBaseSelectorEventLoopTests.test_add_reader_existing.<locals>.<lambda>r )r r rm � SelectorKey� EVENT_WRITEr) rQ rh r( r- r@ �cancelrj rB ri �modifyrk rl rn ro �r �reader�writerrq rV rr rs rt r
r
r �test_add_reader_existing� s$ � z3BaseSelectorEventLoopTests.test_add_reader_existingc C s� t �� }t�ddtjd |f�| jjj_dd� }| j� d|� | �
| jjj j � | �
| jjjj � | jjjjd \}}\}}| �d|� | �tjtjB |� | �||j� | �||� d S )Nr c S s dS rd r
r
r
r
r re � rf zLBaseSelectorEventLoopTests.test_add_reader_existing_writer.<locals>.<lambda>r )r r rm rv rw r) rQ rh r( r- rB ri rj r@ ry rk rl rn ro )r r| rq rV rr rs rt r
r
r � test_add_reader_existing_writer� s � z:BaseSelectorEventLoopTests.test_add_reader_existing_writerc C sB t �ddt jd�| jjj_| �| j�d�� | � | jjj
j � d S �Nr )NN) rm rv rn r) rQ rh r( rB rI r@ �
unregisterrj r r
r
r �test_remove_reader� s � z-BaseSelectorEventLoopTests.test_remove_readerc C s� t �� }t �� }t�ddtjtjB ||f�| jjj_ | �
| j� d�� | � | jjj
j� | �dtjd |ff| jjjjd � d S �Nr r )r r rm rv rn rw r) rQ rh r( r@ rI rB r� rj rl ry rk �r r{ r| r
r
r � test_remove_reader_read_write� s
�
��z8BaseSelectorEventLoopTests.test_remove_reader_read_writec C s" t | jjj_| �| j�d�� d S r )rg r) rQ rh r[ rB rI r r
r
r �test_remove_reader_unknown s
�z5BaseSelectorEventLoopTests.test_remove_reader_unknownc C s� t | jjj_dd� }| j�d|� | �| jjjj� | jjjj d \}}\}}| �
d|� | �
t j |� | �
|� | �
||j� d S )Nc S s dS rd r
r
r
r
r re rf z<BaseSelectorEventLoopTests.test_add_writer.<locals>.<lambda>r r )rg r) rQ rh r[ rH r@ ri rj rk rl rm rw rR ro rp r
r
r �test_add_writer s
z*BaseSelectorEventLoopTests.test_add_writerc C s� t �� }t �� }t�ddtj||f�| jjj_dd� }| j� d|� | �
|j j � | �
| jjjj � | �
| jjjj � | jjjjd \}}\}}| �d|� | �tjtjB |� | �||� | �||j� d S )Nr c S s dS rd r
r
r
r
r re rf zEBaseSelectorEventLoopTests.test_add_writer_existing.<locals>.<lambda>r )r r rm rv rn r) rQ rh r( rH r@ rx rj rB ri ry rk rl rw ro rz r
r
r �test_add_writer_existing s$ � z3BaseSelectorEventLoopTests.test_add_writer_existingc C sB t �ddt jd�| jjj_| �| j�d�� | � | jjj
j � d S r ) rm rv rw r) rQ rh r( rB rJ r@ r� rj r r
r
r �test_remove_writer" s � z-BaseSelectorEventLoopTests.test_remove_writerc C s� t �� }t �� }t�ddtjtjB ||f�| jjj_ | �
| j� d�� | � | jjj
j� | �dtj|d ff| jjjjd � d S r� )r r rm rv rn rw r) rQ rh r( r@ rJ rB r� rj rl ry rk r� r
r
r � test_remove_writer_read_write) s
�
��z8BaseSelectorEventLoopTests.test_remove_writer_read_writec C s" t | jjj_| �| j�d�� d S r )rg r) rQ rh r[ rB rJ r r
r
r �test_remove_writer_unknown7 s
�z5BaseSelectorEventLoopTests.test_remove_writer_unknownc C sb t �� }d|_t �� | j_| j�t�ddtj|d f�tjfg� | � | jjj
� | jj� |� d S �NFr ) r r �
_cancelledr) �
_add_callback�_process_eventsrm rv rn r@ rj rS �r r{ r
r
r �test_process_events_read< s ���z3BaseSelectorEventLoopTests.test_process_events_readc C sR t �� }d|_t �� | j_| j�t�ddtj|d f�tjfg� | jj� d� d S �NTr )
r r � cancelledr) r7 r� rm rv rn rS r� r
r
r �"test_process_events_read_cancelledH s ���z=BaseSelectorEventLoopTests.test_process_events_read_cancelledc C sR t �� }d|_t �� | j_| j�t�ddtjd |f�tjfg� | jj� |� d S r� )
r r r� r) r� r� rm rv rw rS �r r| r
r
r �test_process_events_writeS s ���z4BaseSelectorEventLoopTests.test_process_events_writec C sR t �� }d|_t �� | j_| j�t�ddtjd |f�tjfg� | jj� d� d S r� )
r r r� r) r8 r� rm rv rw rS r� r
r
r �#test_process_events_write_cancelled^ s ���z>BaseSelectorEventLoopTests.test_process_events_write_cancelledc C s| t �� }t �� t �� f|j_d}t jj}|| jd�� }| jjt �� ||d� W 5 Q R X | j�t �
d�� | � |jj |� d S )N�d Z_accept_connection2)�backlogr )
r r �acceptr( �patch�objectr) Z_accept_connectionrT r �sleeprl �
call_count)r �sockr� Zmock_objZ accept2_mockr
r
r � test_accept_connection_multiplei s �z:BaseSelectorEventLoopTests.test_accept_connection_multiple)#r r r r% r3 �unittestZskipIfr4 rG r r� rL rW rX r] r_ rb rc ru r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r
r
r+ r r# 3 s:
%
, r# c s� e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
� d�dd� �Z e
� d�dd� �Z
dd� Zdd� Z� ZS )�SelectorTransportTestsc s> t � �� | �� | _t�tj�| _t �
t j �| _ d| j j
_d S �NrN )r$ r% �
new_test_loopr) r: �make_test_protocolr r/ �protocolr r �socketr� rP r( r r+ r
r r% } s
z SelectorTransportTests.setUpc C s$ t | j| j| jd �}| �t|� |S r )r r) r� r� �
addCleanupr" �r r! r
r
r �create_transport� s
� z'SelectorTransportTests.create_transportc C s: | � � }| �|j| j� | �|j| j� | �|jd� d S r� )r� ZassertIs�_loopr) r r� Z_sock_fd�r �trr
r
r � test_ctor� s z SelectorTransportTests.test_ctorc C s* | � � }t�� |_|�� |j�d � d S r )r� r r � _force_close�abortrS r� r
r
r �
test_abort� s
z!SelectorTransportTests.test_abortc C sz | � � }|�� | �|�� � | �d| jjd � | j�d � | �|j d� |�� | �|j d� | �d| jjd � d S )Nr rN )
r� r r@ �
is_closingrl r) �remove_reader_countr� �connection_lost�
_conn_lostr� r
r
r rW � s z!SelectorTransportTests.test_closec C sJ | � � }|j�d� |�� | �| jj� t�| j� | �| j j
j � d S �N� data) r� �_buffer�extendr rB r) �readersr: r? r� r� rj r� r
r
r �test_close_write_buffer� s z.SelectorTransportTests.test_close_write_bufferc C s� | � � }|j�d� | j�dtj� | j�dtj� |�d � | � |�
� � | � |jt � � | �
| jj� | �
| jj� |�d � | �
| jj� | � d| jjd � d S )N� 1rN r )r� r� r� r) r5 r �sentinelr6 r� r@ r� rl r rB r� �writersr� r� r
r
r �test_force_close� s
z'SelectorTransportTests.test_force_closezasyncio.log.logger.errorc C s: t � }| �� }t�� |_|�|� |�� |j�|� d S r )r^ r� r r r� � _fatal_error�assert_not_calledrS )r �m_exc�excr� r
r
r �test_fatal_error� s
z'SelectorTransportTests.test_fatal_errorc C s\ G dd� dt �}|� }| �� }t�� |_|�|� |jt�d�|t t fd� |j�|� d S )Nc @ s e Zd ZdS )zISelectorTransportTests.test_fatal_error_custom_exception.<locals>.MyErrorN�r r r r
r
r
r �MyError� s r� �1Fatal error on transport
protocol:.*
transport:.*��exc_info)
� Exceptionr� r r r� r� rS r: � MockPattern�MOCK_ANY)r r� r� r� r� r
r
r �!test_fatal_error_custom_exception� s
��z8SelectorTransportTests.test_fatal_error_custom_exceptionc C sr t � }| �� }| �|j� | �|j� |�|� | jj�|� | j j
�� | � |j � | � |j� | � |j� d S r )
r^ r� ZassertIsNotNoneZ _protocolr� Z_call_connection_lostr� r� rS r� r rR r )r r� r� r
r
r �test_connection_lost� s
z+SelectorTransportTests.test_connection_lostc C sv | � � }|j�d� |�dtj� | �| jj� |� d � | �|�
� � | � | jj� |�dtj� | � | jj� d S )Nr� rN ) r� r� r� r5 r r� r@ r) r� r� r� rB r� r
r
r �test__add_reader� s
z'SelectorTransportTests.test__add_reader)r r r r% r� r� r� rW r� r� r r� r� r� r� r� r� r
r
r+ r r� { s
r� c s� e Zd Z� fdd�ZdQdd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z e
�d�dd� �Ze
�d�dd� �Ze
�d�dd � �Ze
�d�d d � �Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z e
�d3�d4d5� �Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)e
�dN�dOdP� �Z*� Z+S )R� SelectorSocketTransportTestsc sD t � �� | �� | _t�tj�| _t �
t j �| _ d | _
| j j_d S r� )r$ r% r� r) r: r� r r/ r� r r r� r� �sock_fdrP r( r r+ r
r r% � s
z"SelectorSocketTransportTests.setUpNc C s& t | j| j| j|d�}| �t|� |S �N�rE �r r) r� r� r� r" �r rE r! r
r
r �socket_transport s
� z-SelectorSocketTransportTests.socket_transportc C sP | j �� }| j|d�}| j �|� | j �d|j� t�| j � | jj �
|� d S �Nr� rN � r) r9 r� rT �
assert_reader� _read_readyr: r? r� Zconnection_maderS �r rE r� r
r
r r� s
z&SelectorSocketTransportTests.test_ctorc C s4 | j �� }| j|d� | j �|� | �|�� � d S r� )r) r9 r� rT rR �result)r rE r
r
r �test_ctor_with_waiter s
z2SelectorSocketTransportTests.test_ctor_with_waiterc C s� | � � }t�| j� | �|j� | �|�� � | j�d|j � |�
� |�
� | �|j� | �|�� � | j� d� |� � |� � | �|j� | �|�� � | j�d|j � |�
� | �|�� � | j� d� d S r� )r� r: r? r) rB Z_pausedr@ r> r� r� rA Zassert_no_readerrC r r� r
r
r �test_pause_resume_reading s$ z6SelectorSocketTransportTests.test_pause_resume_readingc C sd | � � }t�� |_t�� |_t�� | j_t� | jj _
d| j j _
|�� | jj �� | �|jj� d S �Nrf )r� r r r r� r) �call_exception_handler� LookupErrorr� � eof_receivedr[ r� rZ r( r� rS r@ rj r� r
r
r � test_read_eof_received_error- s
z9SelectorSocketTransportTests.test_read_eof_received_errorc C s^ | � � }t�� |_t�� | j_t� | jj_ d| j
j _ |�
� | �|jj� | �| jjj� d S r� )r� r r r� r) r� r� r� �
data_receivedr[ r� rZ r( r� r@ rj r� r
r
r �test_data_received_error< s
z5SelectorSocketTransportTests.test_data_received_errorc C s, | � � }d| jj_|�� | jj�d� d S r� )r� r� rZ r( r� r� r� rS r� r
r
r �test_read_readyI s
z,SelectorSocketTransportTests.test_read_readyc C s> | � � }t�� |_d| jj_|�� | jj �
� |j�
� d S r� ) r� r r r r� rZ r( r� r� r� rS r� r
r
r �test_read_ready_eofQ s
z0SelectorSocketTransportTests.test_read_ready_eofc C sL | � � }t�� |_d| jj_d| jj_|� � | jj�
� | � |jj � d S )Nrf T)
r� r r r r� rZ r( r� r� r� rS rB rj r� r
r
r � test_read_ready_eof_keep_open[ s
z:SelectorSocketTransportTests.test_read_ready_eof_keep_open�logging.exceptionc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S r ) rY r� rZ r[ r� r r r� r� rB rj �r r� r! r
r
r �test_read_ready_tryagainf s
z5SelectorSocketTransportTests.test_read_ready_tryagainc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S r ) �InterruptedErrorr� rZ r[ r� r r r� r� rB rj r� r
r
r �$test_read_ready_tryagain_interruptedp s
zASelectorSocketTransportTests.test_read_ready_tryagain_interruptedc C sN t � }| jj_| �� }t�� |_t� � � |�
� W 5 Q R X |j� |� d S r ) �ConnectionResetErrorr� rZ r[ r� r r r� r: r; r� rS �r r� �errr! r
r
r �test_read_ready_conn_resetz s
z7SelectorSocketTransportTests.test_read_ready_conn_resetc C s< t � }| jj_| �� }t�� |_|�� |j� |d� d S �Nz$Fatal read error on socket transport)
r^ r� rZ r[ r� r r r� r� rS r� r
r
r �test_read_ready_err� s
�z0SelectorSocketTransportTests.test_read_ready_errc C s6 d}t |�| jj_| �� }|�|� | jj�|� d S r� )�lenr� r` r( r� �writerS �r �datar! r
r
r �
test_write� s
z'SelectorSocketTransportTests.test_writec C sJ t d�}t|�| jj_| �� }|�|� | jj�|� | �|t d�� d S r� ) r r� r� r` r( r� r� rS rl r� r
r
r �test_write_bytearray� s
z1SelectorSocketTransportTests.test_write_bytearrayc C s: t d�}t|�| jj_| �� }|�|� | jj�|� d S r� )�
memoryviewr� r� r` r( r� r� rS r� r
r
r �test_write_memoryview� s
z2SelectorSocketTransportTests.test_write_memoryviewc C sF | � � }|j�d� |�d� | �| jjj� | �t dg�|j� d S )Nr� rf �
r� r� r� r� rB r� r` rj rl r r� r
r
r �test_write_no_data� s
z/SelectorSocketTransportTests.test_write_no_datac C sH | � � }|j�d� |�d� | �| jjj� | �t ddg�|j� d S )N� data1� data2r� r� r
r
r �test_write_buffer� s
�z.SelectorSocketTransportTests.test_write_bufferc C sH d}d| j j_| �� }|�|� | j�d|j� | �t dg�|j
� d S �Nr� � rN s ta) r� r` r( r� r� r) �
assert_writer� _write_readyrl r r� r� r
r
r �test_write_partial� s
z/SelectorSocketTransportTests.test_write_partialc C s\ t d�}d| jj_| �� }|�|� | j�d|j� | � t
dg�|j � | � |t d�� d S r ) r r� r` r( r� r� r) r r rl r r� r� r
r
r � test_write_partial_bytearray� s
z9SelectorSocketTransportTests.test_write_partial_bytearrayc C sL t d�}d| jj_| �� }|�|� | j�d|j� | � t
dg�|j � d S r ) r� r� r` r( r� r� r) r r rl r r� r� r
r
r � test_write_partial_memoryview� s
z:SelectorSocketTransportTests.test_write_partial_memoryviewc C sR d}d| j j_d| j j_| �� }|�|� | j�d|j� | � t
dg�|j � d S �Nr� r rN ) r� r` r( rP r� r� r) r r rl r r� r� r
r
r �test_write_partial_none� s
z4SelectorSocketTransportTests.test_write_partial_nonec C sH t | jj_d}| �� }|�|� | j�d|j� | � t
dg�|j � d S �Nr� rN ) rY r� r` r[ r� r� r) r r rl r r� r� r
r
r �test_write_tryagain� s
z0SelectorSocketTransportTests.test_write_tryagain� asyncio.selector_events.loggerc C s� t � }| jj_d}| �� }t�� |_|�|� |j� |d� d|_
| j� � |�|� | � | jjj
� | �|j
d� |�|� |�|� |�|� |�|� |j� d� d S )Nr� �%Fatal write error on socket transportr r � socket.send() raised exception.)r^ r� r` r[ r� r r r� r� rS r� Z
reset_mockrB rj rl �warning)r �m_logr� r� r! r
r
r �test_write_exception� s&
�
z1SelectorSocketTransportTests.test_write_exceptionc C s | � � }| �t|jd� d S )N�str)r� rK � TypeErrorr� r� r
r
r �test_write_str s z+SelectorSocketTransportTests.test_write_strc C s: | � � }|�� | �|jd� |�d� | �|jd� d S )Nr r� r )r� r rl r� r� r� r
r
r �test_write_closing s
z/SelectorSocketTransportTests.test_write_closingc C s` d}t |�| jj_| �� }|j�|� | j�d|j � |� � | �
| jjj � | � | jj
� d S r
)r� r� r` r( r� r� r� r) r6 r r@ rj rB r� r� r
r
r �test_write_ready s z-SelectorSocketTransportTests.test_write_readyc C s� d}t |�| jj_| �� }d|_|j�|� | j� d|j
� |�
� | � | jjj � | �
| jj� | jj�� | jj�d � d S )Nr� TrN )r� r� r` r( r� �_closingr� r� r) r6 r r@ rj rB r� r rS r� r� r� r
r
r �test_write_ready_closing s z5SelectorSocketTransportTests.test_write_ready_closingc C s | � � }| �t|j� d S r )r� rK �AssertionErrorr r� r
r
r �test_write_ready_no_data' s z5SelectorSocketTransportTests.test_write_ready_no_datac C sb d}d| j j_| �� }|j�|� | j�d|j� |�� | j� d|j� | �
t dg�|j� d S r � r� r` r( r� r� r� r) r6 r r rl r r� r
r
r �test_write_ready_partial, s
z5SelectorSocketTransportTests.test_write_ready_partialc C sb d}d| j j_| �� }|j�|� | j�d|j� |�� | j� d|j� | �
t dg�|j� d S r r r� r
r
r � test_write_ready_partial_none7 s
z:SelectorSocketTransportTests.test_write_ready_partial_nonec C s` t | jj_| �� }tddg�|_| j�d|j � |� � | j�
d|j � | � tdg�|j� d S )Nr� r� rN s
data1data2) rY r� r` r[ r� r r� r) r6 r r rl r� r
r
r �test_write_ready_tryagainB s
z6SelectorSocketTransportTests.test_write_ready_tryagainc C sH t � }| jj_| �� }t�� |_|j� d� |�
� |j� |d� d S )Nr� r
) r^ r� r` r[ r� r r r� r� r� r rS �r r� r! r
r
r �test_write_ready_exceptionM s
�z7SelectorSocketTransportTests.test_write_ready_exceptionc C sT | � � }| �|�� � |�� | jj�tj� |�� | � | jjj
d� |� � d S r ) r� r@ Z
can_write_eof� write_eofr� �shutdownrS r� �SHUT_WRrl r� r r� r
r
r �test_write_eofX s z+SelectorSocketTransportTests.test_write_eofc C s� | � � }t| jj_|�d� |�� | �|jt dg�� | �
|j � | � | jj
j� dd� | jj_|�� | �
| jjj� | jj
�tj� |�� d S )Nr� c S s dS )N� r
)�_r
r
r re i rf zDSelectorSocketTransportTests.test_write_eof_buffer.<locals>.<lambda>)r� rY r� r` r[ r� r! rl r� r r@ �_eofrB r" rj r rS r� r# r r� r
r
r �test_write_eof_buffera s
z2SelectorSocketTransportTests.test_write_eof_bufferc C s. | � � }|�� | j�t�d�� |�� d S �Nr )r� r r) rT r r� r! r� r
r
r �test_write_eof_after_closeo s z7SelectorSocketTransportTests.test_write_eof_after_closezasyncio.base_events.loggerc C s0 t �� }| j_| �� }|�� |�| j� d S r )r r r) r8 r� r rS r� )r r rJ r! r
r
r �"test_transport_close_remove_writeru s z?SelectorSocketTransportTests.test_transport_close_remove_writer)N),r r r r% r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r r r r$ r( r* r+ r� r
r
r+ r r� � sX
r� c s� e Zd Z� fdd�Zd!dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
e�d�dd� �Ze�d�dd � �Ze�d�d d � �Ze�d�d d � �Z� ZS )"�,SelectorSocketTransportBufferedProtocolTestsc s` t � �� � �� � _t�tj�� _t d�� _
� fdd�� jj _ t
�tj�� _d � _� jj_d S )Nr c s � j S r ��buf�Zhintr r
r re � rf zDSelectorSocketTransportBufferedProtocolTests.setUp.<locals>.<lambda>rN )r$ r% r� r) r: r� r �BufferedProtocolr� r r. �
get_bufferr[ r r r� r� r� rP r( r r+ r r r% � s
z2SelectorSocketTransportBufferedProtocolTests.setUpNc C s& t | j| j| j|d�}| �t|� |S r� r� r� r
r
r r� � s
� z=SelectorSocketTransportBufferedProtocolTests.socket_transportc C sP | j �� }| j|d�}| j �|� | j �d|j� t�| j � | jj �
|� d S r� r� r� r
r
r r� � s
z6SelectorSocketTransportBufferedProtocolTests.test_ctorc C sd | � � }t�� |_t�� | j_t� | jj_ |�
� | � |jj � | � | jjj � | �
| jjj � d S r )r� r r r� r) r� r� r� r1 r[ r� r@ rj rB �buffer_updatedr� r
r
r �test_get_buffer_error� s
zBSelectorSocketTransportBufferedProtocolTests.test_get_buffer_errorc C sf | � � }t�� |_t�� | j_dd� | jj_|� � | �
|jj � | �
| jjj � | � | jj
j � d S )Nc S s t d�S r) )r r/ r
r
r re � rf zXSelectorSocketTransportBufferedProtocolTests.test_get_buffer_zerosized.<locals>.<lambda>)r� r r r� r) r� r� r1 r[ r� r@ rj rB r2 r� r
r
r �test_get_buffer_zerosized� s
zFSelectorSocketTransportBufferedProtocolTests.test_get_buffer_zerosizedc s� t �tj�| _| �� }d| jj_|� � | jj
� d� t �tj �}t
d�� � fdd�|j_|�|� d| jj_|� � |j� d� |j� d� d S )Nr� r% c s � S r r
r/ r- r
r re � rf zUSelectorSocketTransportBufferedProtocolTests.test_proto_type_switch.<locals>.<lambda>�
���)r: r� r r/ r� r� r� rZ r( r� r� rS r0 r r1 r[ Z set_protocol� recv_intor2 )r r! Z buf_protor
r- r �test_proto_type_switch� s
zCSelectorSocketTransportBufferedProtocolTests.test_proto_type_switchc C sn | � � }t�� |_t�� | j_t� | jj_ d| j
j _ |�
� | �|jj� | �| jjj� | �| jjj� d S )Nr5 )r� r r r� r) r� r� r� r2 r[ r� r7 r( r� r@ rj r1 r� r
r
r �test_buffer_updated_error� s
zFSelectorSocketTransportBufferedProtocolTests.test_buffer_updated_errorc C sd | � � }t�� |_t�� |_t�� | j_t� | jj _
d| j j _
|�� | jj �� | �|jj� d S r) )r� r r r r� r) r� r� r� r� r[ r� r7 r( r� rS r@ rj r� r
r
r r� � s
zISelectorSocketTransportBufferedProtocolTests.test_read_eof_received_errorc C s: | � � }d| jj_|�� | jj�d� | jj�d� d S )Nr5 r6 ) r� r� r7 r( r� r� r1 rS r2 r� r
r
r r� � s
z<SelectorSocketTransportBufferedProtocolTests.test_read_readyc C s> | � � }t�� |_d| jj_|�� | jj �
� |j�
� d S r) ) r� r r r r� r7 r( r� r� r� rS r� r
r
r r� � s
z@SelectorSocketTransportBufferedProtocolTests.test_read_ready_eofc C sL | � � }t�� |_d| jj_d| jj_|� � | jj�
� | � |jj � d S )Nr T)
r� r r r r� r7 r( r� r� r� rS rB rj r� r
r
r r� � s
zJSelectorSocketTransportBufferedProtocolTests.test_read_ready_eof_keep_openr� c C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S r ) rY r� r7 r[ r� r r r� r� rB rj r� r
r
r r� s
zESelectorSocketTransportBufferedProtocolTests.test_read_ready_tryagainc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S r ) r� r� r7 r[ r� r r r� r� rB rj r� r
r
r r� s
zQSelectorSocketTransportBufferedProtocolTests.test_read_ready_tryagain_interruptedc C sN t � }| jj_| �� }t�� |_t� � � |�
� W 5 Q R X |j� |� d S r ) r� r� r7 r[ r� r r r� r: r; r� rS r� r
r
r r� s
zGSelectorSocketTransportBufferedProtocolTests.test_read_ready_conn_resetc C s< t � }| jj_| �� }t�� |_|�� |j� |d� d S r� )
r^ r� r7 r[ r� r r r� r� rS r� r
r
r r� $ s
�z@SelectorSocketTransportBufferedProtocolTests.test_read_ready_err)N)r r r r% r� r� r3 r4 r8 r9 r� r� r� r� r r� r� r� r� r� r� r
r
r+ r r, ~ s&
r, c s e Zd Z� fdd�Zd>dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Ze�d �d d � �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z e�d9�d:d;� �Z e�d9�d<d=� �Z!� Z"S )?� SelectorDatagramTransportTestsc s@ t � �� | �� | _t�tj�| _t j
t j d�| _ d| j j
_d S )N)Zspec_setrN )r$ r% r� r) r: r� r ZDatagramProtocolr� r r r� r� rP r( r r+ r
r r% 3 s
z$SelectorDatagramTransportTests.setUpNc C s8 |rd nt | jj_t| j| j| j|d�}| �t|� |S )N��address) r^ r� � getpeernamer[ r r) r� r� r" )r r< r! r
r
r �datagram_transport: s
� z1SelectorDatagramTransportTests.datagram_transportc C s. | � � }d| jj_|�� | jj�dd� d S )N)r� ��0.0.0.0i� r� r? )r> r� �recvfromr( r� r� Zdatagram_receivedrS r� r
r
r r� B s
�z.SelectorDatagramTransportTests.test_read_readyc C s6 | � � }t| jj_t�� |_|�� | � |jj
� d S r ) r> rY r� rA r[ r r r� r� rB rj r� r
r
r r� K s
z7SelectorDatagramTransportTests.test_read_ready_tryagainc C s< | � � }t� }| jj_t�� |_|�� |j� |d� d S )Nz&Fatal read error on datagram transport)
r> r= r� rA r[ r r r� r� rS �r r! r� r
r
r r� T s
�z2SelectorDatagramTransportTests.test_read_ready_errc C sJ | � � }t� }| jj_t�� |_|�� | � |jj
� | j j �
|� d S r )r> r^ r� rA r[ r r r� r� rB rj r� �error_receivedrS rB r
r
r �test_read_ready_oserr_ s
z4SelectorDatagramTransportTests.test_read_ready_oserrc C sF d}| � � }|�|d� | �| jjj� | �| jjjd |df� d S �Nr� r? r )r> �sendtor@ r� rj rl rk r� r
r
r � test_sendtoi s �z*SelectorDatagramTransportTests.test_sendtoc C sJ t d�}| �� }|�|d� | �| jjj� | �| jjjd |df� d S rE )r r> rF r@ r� rj rl rk r� r
r
r �test_sendto_bytearrayq s �z4SelectorDatagramTransportTests.test_sendto_bytearrayc C sJ t d�}| �� }|�|d� | �| jjj� | �| jjjd |df� d S rE )r� r> rF r@ r� rj rl rk r� r
r
r �test_sendto_memoryviewy s �z5SelectorDatagramTransportTests.test_sendto_memoryviewc C sH | � � }|j�d� |�dd� | �| jjj� | �dgt|j�� d S )N�r� �r@ i90 rf r
� r> r� �appendrF rB r� rj rl �listr� r
r
r �test_sendto_no_data� s �z2SelectorDatagramTransportTests.test_sendto_no_datac C sJ | � � }|j�d� |�dd� | �| jjj� | �ddgt|j�� d S )N�r� rK r� rK �r� rK rL r� r
r
r �test_sendto_buffer� s ��z1SelectorDatagramTransportTests.test_sendto_bufferc C sh t d�}| �� }|j�d� |�|d� | �| jjj� | �ddgt |j�� | �
|jd d t � d S �Nr� rP rK rQ r r ) r r> r� rM rF rB r� rj rl rN r0 �bytes�r Zdata2r! r
r
r � test_sendto_buffer_bytearray� s ��z;SelectorDatagramTransportTests.test_sendto_buffer_bytearrayc C sh t d�}| �� }|j�d� |�|d� | �| jjj� | �ddgt |j�� | �
|jd d t � d S rS ) r� r> r� rM rF rB r� rj rl rN r0 rT rU r
r
r � test_sendto_buffer_memoryview� s ��z<SelectorDatagramTransportTests.test_sendto_buffer_memoryviewc C sJ d}t | jj_| �� }|�|d� | j�d|j� | �dgt |j
�� d S )Nr� rK rN rJ ) rY r� rF r[ r> r) r �
_sendto_readyrl rN r� r� r
r
r �test_sendto_tryagain� s
�z3SelectorDatagramTransportTests.test_sendto_tryagainr c C s� d}t � }| jj_| �� }t�� |_|�|d� | �|jj � |j�
|d� d|_ d|_ |�|� |�|� |�|� |�|� |�|� |j
�
d� d S )Nr� r
�'Fatal write error on datagram transportr )Z123r )r= r� rF r[ r> r r r� r@ rj rS r� �_addressr )r r r� r� r! r
r
r �test_sendto_exception� s$
�
z4SelectorDatagramTransportTests.test_sendto_exceptionc C sL d}t | jj_| �� }t�� |_|�|d� | �|j d� | �
|jj � d S )Nr� r
r ) �ConnectionRefusedErrorr� rF r[ r> r r r� rl r� rB rj r� r
r
r �test_sendto_error_received� s
z9SelectorDatagramTransportTests.test_sendto_error_receivedc C sP d}t | jj_| jdd�}t�� |_|�|� | � |jj
� | � | j j
j
� d S )Nr� �r@ r r; )r] r� r` r[ r> r r r� rF rB rj r@ r� rC r� r
r
r �$test_sendto_error_received_connected� s
zCSelectorDatagramTransportTests.test_sendto_error_received_connectedc C s | � � }| �t|jdd� d S )Nr r
)r> rK r rF r� r
r
r �test_sendto_str� s z.SelectorDatagramTransportTests.test_sendto_strc C s" | j dd�}| �t|jdd� d S )Nr_ r; s str)r@ r )r> rK �
ValueErrorrF r� r
r
r �test_sendto_connected_addr� s �z9SelectorDatagramTransportTests.test_sendto_connected_addrc C s@ | j dd�}|�� | �|jd� |�dd� | �|jd� d S )N)r r; r r� r )r> r rl r� rF r� r
r
r �test_sendto_closing� s
z2SelectorDatagramTransportTests.test_sendto_closingc C s~ d}t |�| jj_| �� }|j�|df� | j�d|j � |� � | �
| jjj � | � | jjj
d |df� | �| jj� d S )Nr� rK rN r )r� r� rF r( r> r� rM r) r6 rX r@ rj rl rk rB r� r� r
r
r �test_sendto_ready� s �z0SelectorDatagramTransportTests.test_sendto_readyc C s� d}t |�| jj_| �� }d|_|j�|df� | j� d|j
� |�
� | jj � |d� | �
| jj� | jj� � | jj� d � d S )Nr� Tr
rN )r� r� r` r( r> r r� rM r) r6 rX rF rS rB r� r r� r� r� r
r
r �test_sendto_ready_closing s z8SelectorDatagramTransportTests.test_sendto_ready_closingc C sB | � � }| j�d|j� |�� | �| jjj� | �| jj� d S r� ) r> r) r6 rX rB r� rF rj r� r� r
r
r �test_sendto_ready_no_data s
z8SelectorDatagramTransportTests.test_sendto_ready_no_datac C sd t | jj_| �� }|j�ddg� | j�d|j � |� � | j�
d|j � | � ddgt |j�� d S )N)r� r
)r� r
rN )
rY r� rF r[ r> r� r� r) r6 rX r rl rN r� r
r
r �test_sendto_ready_tryagain s
�z9SelectorDatagramTransportTests.test_sendto_ready_tryagainc C sH t � }| jj_| �� }t�� |_|j� d� |�
� |j� |d� d S )N�r� r
rZ ) r= r� rF r[ r> r r r� r� rM rX rS r r
r
r �test_sendto_ready_exception% s
�z:SelectorDatagramTransportTests.test_sendto_ready_exceptionc C sB t | jj_| �� }t�� |_|j� d� |�
� | � |jj � d S )Nri )
r] r� rF r[ r> r r r� r� rM rX rB rj r� r
r
r � test_sendto_ready_error_received1 s
z?SelectorDatagramTransportTests.test_sendto_ready_error_receivedc C sV t | jj_| jdd�}t�� |_|j� d� |�
� | � |jj � | �
| jjj � d S )Nr_ r; ri )r] r� r` r[ r> r r r� r� rM rX rB rj r@ r� rC r� r
r
r �+test_sendto_ready_error_received_connection; s
zJSelectorDatagramTransportTests.test_sendto_ready_error_received_connectionz asyncio.base_events.logger.errorc C s8 | j dd�}t� }|�|� | �| jjj� |�� d S )Nr_ r; )r> r] r� rB r� rC rj r� )r r� r! r� r
r
r �test_fatal_error_connectedF s
z9SelectorDatagramTransportTests.test_fatal_error_connectedc C sZ G dd� dt �}| jdd�}|� }|�|� | �| jjj� |jt� d�|t
t
fd� d S )Nc @ s e Zd ZdS )z[SelectorDatagramTransportTests.test_fatal_error_connected_custom_error.<locals>.MyExceptionNr� r
r
r
r � MyExceptionP s rn r_ r; r� r� ) r� r> r� rB r� rC rj rS r: r� r� )r r� rn r! r� r
r
r �'test_fatal_error_connected_custom_errorN s
��zFSelectorDatagramTransportTests.test_fatal_error_connected_custom_error)N)#r r r r% r> r� r� r� rD rG rH rI rO rR rV rW rY r r� r\ r^ r` ra rc rd re rf rg rh rj rk rl rm ro r� r
r
r+ r r: 1 s@
r: �__main__)r
) �__doc__rm r� r� r r4 � ImportErrorr Zasyncio.selector_eventsr r r r Ztest.test_asyncior r: �ANYr� r r
r r" ZTestCaser# r� r� r, r: r �mainr
r
r
r �<module> sD
J|
4 -