File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/test_selector_events.cpython-37.pyc
B
��g� � @ s2 d Z ddlZddlZddlZddlZddlmZ y ddlZW n ek
rT dZY nX ddlZddl m
Z
ddl m Z ddl m Z ddl m
Z
ddlmZ ejZG d d
� d
e
�Zdd d
�Zdd� ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�Ze dk�r.e� � dS ) z Tests for selector_events.py� N)�mock)�BaseSelectorEventLoop)�_SelectorTransport)�_SelectorSocketTransport)�_SelectorDatagramTransport)�utilsc @ s e Zd Zdd� Zdd� ZdS )�TestBaseSelectorEventLoopc C s& t �� | _t �� | _| jd7 _d S )N� )r �Mock�_ssock�_csock�
_internal_fds)�self� r �B/usr/local/lib/python3.7/test/test_asyncio/test_selector_events.py�_make_self_pipe s
z)TestBaseSelectorEventLoop._make_self_pipec C s d S )Nr )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 )N)� bytearray�join)�lr r r �list_to_buffer# s r c C s" | j d krd S | j �� d | _ d S )N)�_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!d9d:� Z"d;d<� Z#e �
d=�d>d?� �Z$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)dJdK� Z*dLdM� Z+dNdO� Z,dPdQ� Z-dRdS� Z.dTdU� Z/dVdW� Z0dXdY� Z1dZd[� Z2d\d]� Z3d^d_� Z4d`da� Z5dbdc� Z6ddde� Z7dfdg� Z8dhdi� Z9djdk� Z:dldm� Z;dndo� Z<dpdq� Z=drds� Z>dtdu� Z?� Z@S )v�BaseSelectorEventLoopTestsc s: t � �� t�� | _g | jj_t| j�| _| � | j� d S )N)
�super�setUpr r
�selectorZselect� return_valuer �loop�set_event_loop)r )� __class__r r r 2 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_transport�asyncio�Protocol�assertIsInstancer r �reprr )r �mr r r r �test_make_socket_transport9 s
z5BaseSelectorEventLoopTests.test_make_socket_transportNz
No ssl modulec
C s0 t �� }t �� | j_d| jj_t �� | j_t �� | j_t �� | j_tj | jd�}t
� � �H | j� |t�
� ||�}| �td�� |�� W d Q R X t
�| j� W d Q R X | �|�� � |�� |�� | �|�� � |�� |�� | �|�� � |jj}| �d|�� � | �d|�� � |�� t
�| j� d S )NF)r# z SSL transport.*not.*initialized�sslr )r r
r# � _add_readerr' � _add_writer�_remove_reader�_remove_writerr( �Future�
test_utils�disable_logger�_make_ssl_transportr) ZassertRaisesRegex� RuntimeError�
is_reading� run_briefly�
assertTrue�
pause_reading� assertFalse�resume_readingr% r ZassertIn�lowerr )r r, �waiterr �
class_namer r r �test_make_ssl_transportF s4
z2BaseSelectorEventLoopTests.test_make_ssl_transportzasyncio.selector_events.sslzasyncio.sslproto.sslc C sd t �� }t �� | j_t �� | j_t �� | j_t �� | j_| �t�� | j� ||||� W d Q R X d S )N)
r r
r# r& �
add_writer�
remove_reader�
remove_writer� assertRaisesr7 r6 )r r, r r r �)test_make_ssl_transport_without_ssl_errorl s zDBaseSelectorEventLoopTests.test_make_ssl_transport_without_ssl_errorc C sl 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� � tj| jd�}| �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 )Nr )r r
r r r
)r r r r r y s
zHBaseSelectorEventLoopTests.test_close.<locals>.EventLoop._make_self_pipeN)r r r r r r r r � EventLoopx s rG � r )r# r c S s d S )Nr r r r r �callback� s z7BaseSelectorEventLoopTests.test_close.<locals>.callback)r r! r# r$ r �filenor" r r r
r1 � _selectorr r< Z is_closedr: � assertIsNone�assert_called_withr( r3 rE r7 Z run_forever�run_until_completer& rB ) r rG ZssockZcsockrC r! �f�fdrI r r r �
test_closew s:
z%BaseSelectorEventLoopTests.test_closec C s< t �� | j_| jj�� d | j_| j�� | �| jj� d S )N)r r
r# rC rK r rL )r r r r �test_close_no_selector� s
z1BaseSelectorEventLoopTests.test_close_no_selectorc C s t | jjj_| �| j�� � d S )N)�BlockingIOErrorr# r �recv� side_effectrL �_read_from_self)r r r r � test_read_from_self_tryagain� s z7BaseSelectorEventLoopTests.test_read_from_self_tryagainc C s t | jjj_| �t | jj� d S )N)�OSErrorr# r rT rU rE rV )r r r r � test_read_from_self_exception� s z8BaseSelectorEventLoopTests.test_read_from_self_exceptionc C s4 t | jjj_t�� � | �| j�� � W d Q R X d S )N) rS r# r �sendrU r4 r5 rL �_write_to_self)r r r r �test_write_to_self_tryagain� s
z6BaseSelectorEventLoopTests.test_write_to_self_tryagainc C s" t � | jjj_| �t | jj� d S )N)r7 r# r rZ rU rE r[ )r r r r � test_write_to_self_exception� s z7BaseSelectorEventLoopTests.test_write_to_self_exceptionc C s� t �� }t�� | j_| j�| j�|d��}| j�t j
d| jd�� | � | jjj d dd � d |df� |�
� | �t j�� | j�|� W d Q R X d S )Ni g{�G�z�?)r# r r )r4 �mock_nonblocking_socketr r
r# �
_sock_recv� create_task� sock_recvrN r( �sleep� assertEqual� call_args�cancelrE �CancelledError)r �sockrO r r r �test_sock_recv� s z)BaseSelectorEventLoopTests.test_sock_recvc C s� t �� }d|j_t|j_d|j_t �� | j_ t �� | j_
| j� | j� |d��}| j�
tjd| jd�� | jj jd d }| jj jd dd � }d |j_td
�|j_||� | j�
tjd| jd�� | �|�� t� | �d | jj
jd � d S ) N�
g i g{�G�z�?)r# r r � ���� )ri )r r
rJ r" rS rT rU �
gettimeoutr# r& rC r` ra rN r( rb rd rX r* � exceptionrc )r rg �futrI �paramsr r r �test_sock_recv_reconnection� s" z6BaseSelectorEventLoopTests.test_sock_recv_reconnectionc C sB t �� }tj| jd�}|�� | j�|d |d� | �|jj � d S )N)r# i )
r r
r( r3 r# re r_ r< rT �called)r rg rO r r r � test__sock_recv_canceled_fut� s
z7BaseSelectorEventLoopTests.test__sock_recv_canceled_futc C s^ t �� }d|j_tj| jd�}|�� t �� | j_| j� |d|d� | �
d| jjj d � d S )Nri )r# i )ri r ) r r
rJ r" r( r3 r# re rC r_ rc rd )r rg rO r r r �test__sock_recv_unregister� s z5BaseSelectorEventLoopTests.test__sock_recv_unregisterc C sn t j| jd�}t�� }d|j_t|j_ t�� | j_
| j� |d |d� | � d| jj |d|df| jj
j
d � d S )N)r# ri i r )r( r3 r# r r
rJ r" rS rT rU r& r_ rc rd )r rO rg r r r �test__sock_recv_tryagain� s z3BaseSelectorEventLoopTests.test__sock_recv_tryagainc C sR t j| jd�}t�� }d|j_t� }|j_ | j�
|d |d� | � ||� � � d S )N)r# ri i )
r( r3 r# r r
rJ r" rX rT rU r_ �assertIsrn )r rO rg �errr r r �test__sock_recv_exception s z4BaseSelectorEventLoopTests.test__sock_recv_exceptionc C s� t �� }t�� | j_| j�| j�|d��}| j�t j
d| jd�� | � d |df| jjj d dd � � |�
� | �t j�� | j�|� W d Q R X d S )Ns datag{�G�z�?)r# r r )r4 r^ r r
r# �
_sock_sendallr` � sock_sendallrN r( rb rc rd re rE rf )r rg rO r r r �test_sock_sendall
s z,BaseSelectorEventLoopTests.test_sock_sendallc C sr t �� }t�� | j_| j�| j�|d��}| j�t j
d| jd�� | � |� � � | �
|�� � | �| jjj� d S )N� r )r# )r4 r^ r r
r# ry r` rz rN r( rb r: �donerL �resultr< rr )r rg rO r r r �test_sock_sendall_nodata s z3BaseSelectorEventLoopTests.test_sock_sendall_nodatac C s� t �� }d|j_t|j_d|j_t �� | j_ t �� | j_
| j� | j� |d��}| j�
tjd| jd�� | jj jd d }| jj jd dd � }d |j_td
�|j_||� | j�
tjd| jd�� | �|�� t� | �d | jj
jd � d S ) Nri g s datag{�G�z�?)r# r r rj rk rl )ri )r r
rJ r" rS rZ rU rm r# rB rD r` rz rN r( rb rd rX r* rn rc )r rg ro rI rp r r r � test_sock_sendall_reconnection) s z9BaseSelectorEventLoopTests.test_sock_sendall_reconnectionc C sB t �� }tj| jd�}|�� | j�|d |d� | �|jj � d S )N)r# s data)
r r
r( r3 r# re ry r< rZ rr )r rg rO r r r � test__sock_sendall_canceled_futC s
z:BaseSelectorEventLoopTests.test__sock_sendall_canceled_futc C s^ t �� }d|j_tj| jd�}|�� t �� | j_| j� |d|d� | �
d| jjj d � d S )Nri )r# s data)ri r ) r r
rJ r" r( r3 r# re rD ry rc rd )r rg rO r r r � test__sock_sendall_unregisterL s z8BaseSelectorEventLoopTests.test__sock_sendall_unregisterc C sn t j| jd�}t�� }d|j_t|j_ t�� | j_
| j� |d |d� | � d| jj |d|df| jj
j
d � d S )N)r# ri s datar )r( r3 r# r r
rJ r" rS rZ rU rB ry rc rd )r rO rg r r r �test__sock_sendall_tryagainW s z6BaseSelectorEventLoopTests.test__sock_sendall_tryagainc C sn t j| jd�}t�� }d|j_t|j_ t�� | j_
| j� |d |d� | � d| jj |d|df| jj
j
d � d S )N)r# ri s datar )r( r3 r# r r
rJ r" �InterruptedErrorrZ rU rB ry rc rd )r rO rg r r r � test__sock_sendall_interruptedc s z9BaseSelectorEventLoopTests.test__sock_sendall_interruptedc C sR t j| jd�}t�� }d|j_t� }|j_ | j�
|d |d� | � |� � |� d S )N)r# ri s data)
r( r3 r# r r
rJ r" rX rZ rU ry rv rn )r rO rg rw r r r � test__sock_sendall_exceptiono s z7BaseSelectorEventLoopTests.test__sock_sendall_exceptionc C sX t �� }tj| jd�}d|j_d|j_| j�|d |d� | � |�
� � | � |� � � d S )N)r# ri � s data)
r r
r( r3 r# rJ r" rZ ry r: r} rL r~ )r rg rO r r r �test__sock_sendallx s z-BaseSelectorEventLoopTests.test__sock_sendallc C s| t �� }tj| jd�}d|j_d|j_t �� | j_| j� |d |d� | �
|� � � | � d| jj |d|df| jjj
d � d S )N)r# ri rj s datas tar )r r
r( r3 r# rJ r" rZ rB ry r< r} rc rd )r rg rO r r r �test__sock_sendall_partial� s z5BaseSelectorEventLoopTests.test__sock_sendall_partialc C s| t �� }tj| jd�}d|j_d|j_t �� | j_| j� |d |d� | �
|� � � | � d| jj |d|df| jjj
d � d S )N)r# ri r s data)r r
r( r3 r# rJ r" rZ rB ry r< r} rc rd )r rg rO r r r �test__sock_sendall_none� s z2BaseSelectorEventLoopTests.test__sock_sendall_nonec C s� t �� | j_t �� | j_t�� }t|j_ | j�
| j� |d��}| j� � | �
|jj� | �
| jjj� |j�� |�� | �tj�� | j�|� W d Q R X | �
| jjj� d S )N)z 127.0.0.1�P )r r
r# rB rD r4 r^ rS �connectrU r` � sock_connect� _run_oncer: rr �
reset_mockre rE r( rf rN )r rg ro r r r �test_sock_connect_timeout� s
z4BaseSelectorEventLoopTests.test_sock_connect_timeoutzsocket.getaddrinfoc C sv d}t �� }dd� |_| j�| j�||��}| j�|� |�|d |d |j|j |j
d� | j�|� |j �d� d S )N)zneed-resolution.comi� c W s dgS )N)NNNN)z 127.0.0.1r r )�argsr r r �<lambda>� r| zZBaseSelectorEventLoopTests.test_sock_connect_resolve_using_socket_params.<locals>.<lambda>r r )z 127.0.0.1r ) r4 r^ rU r# r` r� rN rM Zfamily�type�protor� )r Zm_gaiZaddrrg Zconr r r �-test_sock_connect_resolve_using_socket_params� s
zHBaseSelectorEventLoopTests.test_sock_connect_resolve_using_socket_paramsc C s� t j| jd�}t�� }d|j_| j�� }|�t j
t j t j ddfg� | j�
|||� | �|�� � | �|�� � | �|jj� d S )N)r# ri � )z 127.0.0.1i� )r( r3 r# r r
rJ r" �
create_future�
set_result�socket�AF_INET� SOCK_STREAM� IPPROTO_TCP�
_sock_connectr: r} rL r~ r� rr )r rO rg �resolvedr r r �test__sock_connect� s
z-BaseSelectorEventLoopTests.test__sock_connectc C sL t �� }t �� | j_tj| jd�}|�� | j�||d� | �|j j
� d S )N)r# )z 127.0.0.1i� ) r r
r# rD r( r3 re �_sock_connect_cbr< �
getsockoptrr )r rg rO r r r �#test__sock_connect_cb_cancelled_fut� s z>BaseSelectorEventLoopTests.test__sock_connect_cb_cancelled_futc C s� t �� | j_t �� | j_t �� | j_t �� }d|j_t|j _
d|j _d}| j� � }|�
tjtjtjd|fg� tj| jd�}| j�|||� | j�� | �| jjj� | �d| jjjd d � | j�|||� | j�|� | �d| jjjd � d S )Nri r )z 127.0.0.1i� r� )r# )ri ) r r
r# �_process_eventsrB rD rJ r" rS r� rU r� r� r� r� r� r� r� r( r3 r� r� r: rr rc rd r� rN )r rg �addressr� rO r r r �test__sock_connect_writer� s&
z4BaseSelectorEventLoopTests.test__sock_connect_writerc C s< t j| jd�}t�� }d|j_tj|j _| j�
||d� d S )N)r# ri )z 127.0.0.1i� ) r( r3 r# r r
rJ r" �errnoZEAGAINr� r� )r rO rg r r r � test__sock_connect_cb_tryagain� s
z9BaseSelectorEventLoopTests.test__sock_connect_cb_tryagainc C sX t j| jd�}t�� }d|j_tj|j _t�� | j_
| j� ||d� | � |�
� t� d S )N)r# ri )z 127.0.0.1i� )r( r3 r# r r
rJ r" r� ZENOTCONNr� rD r� r* rn rX )r rO rg r r r � test__sock_connect_cb_exception s
z:BaseSelectorEventLoopTests.test__sock_connect_cb_exceptionc C s� t �� }t�� | j_| j�| j�|��}| j�t j
d| jd�� | � | jjj d d � | �
| jjj d d |� |�� | �t j�� | j�|� W d Q R X d S )Ng{�G�z�?)r# r r rj )r4 r^ r r
r# � _sock_acceptr` Z sock_acceptrN r( rb r< rd rv re rE rf )r rg rO r r r �test_sock_accept s z+BaseSelectorEventLoopTests.test_sock_acceptc C s| t j| jd�}t�� }t�� }d|j_|df|j_| j�|d|� | � |�
� � | � |df|� � � | � d|j
jd � d S )N)r# ri )z 127.0.0.1i� F)Fr )r( r3 r# r r
rJ r" �acceptr� r: r} rc r~ Z setblockingrd )r rO Zconnrg r r r �test__sock_accept s z,BaseSelectorEventLoopTests.test__sock_acceptc C s@ t �� }tj| jd�}|�� | j�|d|� | �|jj � d S )N)r# F)
r r
r( r3 r# re r� r< r� rr )r rg rO r r r � test__sock_accept_canceled_fut( s
z9BaseSelectorEventLoopTests.test__sock_accept_canceled_futc C s\ t �� }d|j_tj| jd�}|�� t �� | j_| j� |d|� | �
d| jjj d � d S )Nri )r# T)ri r ) r r
rJ r" r( r3 r# re rC r� rc rd )r rg rO r r r � test__sock_accept_unregister1 s z7BaseSelectorEventLoopTests.test__sock_accept_unregisterc C sj t j| jd�}t�� }d|j_t|j_ t�� | j_
| j� |d|� | � d| jj |d|f| jj
j
d � d S )N)r# ri FTr )r( r3 r# r r
rJ r" rS r� rU r& r� rc rd )r rO rg r r r �test__sock_accept_tryagain< s z5BaseSelectorEventLoopTests.test__sock_accept_tryagainc C sP t j| jd�}t�� }d|j_t� }|j_ | j�
|d|� | � ||� � � d S )N)r# ri F)
r( r3 r# r r
rJ r" rX r� rU r� rv rn )r rO rg rw r r r �test__sock_accept_exceptionH s z6BaseSelectorEventLoopTests.test__sock_accept_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 r� S r| z<BaseSelectorEventLoopTests.test_add_reader.<locals>.<lambda>r r )�KeyErrorr# rK �get_keyrU r& r: �registerrr rd rc � selectors�
EVENT_READ� _callbackrL )r �cbrP �mask�r�wr r r �test_add_readerQ 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 )NTr r r r r r� b r| zEBaseSelectorEventLoopTests.test_add_reader_existing.<locals>.<lambda>r )r r
r� � SelectorKey� EVENT_WRITEr# rK r� r" r& r: re rr r< r� �modifyrd rc r� r� )r �reader�writerr� rP r� r� r� 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 )NTr r r r r r� r r| zLBaseSelectorEventLoopTests.test_add_reader_existing_writer.<locals>.<lambda>r )r r
r� r� r� r# rK r� r" r& r< r� rr r: r� rd rc r� r� )r r� r� rP r� r� r� r r r � test_add_reader_existing_writern s z:BaseSelectorEventLoopTests.test_add_reader_existing_writerc C sB t �ddt jd�| jjj_| �| j�d�� | � | jjj
j � d S )Nr )NN) r� r� r� r# rK r� r" r< rC r: �
unregisterrr )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
r� r� r� r� r# rK r� r" r: rC r< r� rr rc r� rd )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 )Nr )r� r# rK r� rU r< rC )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 )NTr r r r r r� � r| z<BaseSelectorEventLoopTests.test_add_writer.<locals>.<lambda>r r )r� r# rK r� rU rB r: r� rr rd rc r� r� rL r� )r r� rP r� r� r� 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 )NTr r r r r r� � r| zEBaseSelectorEventLoopTests.test_add_writer_existing.<locals>.<lambda>r )r r
r� r� r� r# rK r� r" rB r: re rr r< r� r� rd rc r� r� )r r� r� r� rP r� r� r� 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 )Nr )NN) r� r� r� r# rK r� r" r< rD r: r� rr )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 )Nr r )r r
r� r� r� r� r# rK r� r" r: rD r< r� rr rc r� rd )r r� r� r r r � test_remove_writer_read_write� s z8BaseSelectorEventLoopTests.test_remove_writer_read_writec C s" t | jjj_| �| j�d�� d S )Nr )r� r# rK r� rU r< rD )r r r r �test_remove_writer_unknown� 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_callbackr� r� r� r� r: rr rM )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# r1 r� r� r� r� rM )r r� r r r �"test_process_events_read_cancelled� s z=BaseSelectorEventLoopTests.test_process_events_read_cancelledc C sR t �� }d|_t �� | j_| j�t�ddtjd |f�tjfg� | jj� |� d S )NFr )
r r
r� r# r� r� r� r� r� rM )r r� r r r �test_process_events_write� s z4BaseSelectorEventLoopTests.test_process_events_writec C sR t �� }d|_t �� | j_| j�t�ddtjd |f�tjfg� | jj� d� d S )NTr )
r r
r� r# r2 r� r� r� r� rM )r 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��@}d |_|| jd��"}d |_| jjt �� ||d� W d Q R X W d Q R X | �|jj |� d S )N�d Z_accept_connection2r` )�backlog)
r r
r� r" �patch�objectr# Z_accept_connectionrc �
call_count)r rg r� Zmock_objZ accept2_mockZ task_mockr r r � test_accept_connection_multiple� s $z:BaseSelectorEventLoopTests.test_accept_connection_multiple)Ar r r r r- �unittestZskipIfr. rA r r� rF rQ rR rW rY r\ r] rh rq rs rt ru rx 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� �
__classcell__r r )r% r r 0 sr
&
,
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 )NrH )r r �
new_test_loopr# r4 �make_test_protocolr( r) �protocolr r
r� rg rJ r" )r )r% r r r
s
z SelectorTransportTests.setUpc C s$ t | j| j| jd �}| �t|� |S )N)r r# rg r� �
addCleanupr )r r r r r �create_transport s z'SelectorTransportTests.create_transportc C s: | � � }| �|j| j� | �|j| j� | �|jd� d S )NrH )r� rv �_loopr# r rg Z_sock_fd)r �trr r r � test_ctor s z SelectorTransportTests.test_ctorc C s* | � � }t�� |_|�� |j�d � d S )N)r� r r
� _force_close�abortrM )r 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 rH )
r� r r: �
is_closingrc r# �remove_reader_countr� �connection_lost�
_conn_lost)r r� r r r rQ ' s z!SelectorTransportTests.test_closec C sJ | � � }|j�d� |�� | �| jj� t�| j� | �| j j
j � d S )Ns data) r� �_buffer�extendr r< r# �readersr4 r9 r� r� rr )r r� r r r �test_close_write_buffer4 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� 1rH r )r� r� r� r# r/ r �sentinelr0 r� r: r� rc r r< r� �writersr� )r r� r r r �test_force_close= s
z'SelectorTransportTests.test_force_closezasyncio.log.logger.errorc C s: t � }| �� }t�� |_|�|� |�� |j�|� d S )N)rX r� r r
r� � _fatal_error�assert_not_calledrM )r �m_exc�excr� r r r �test_fatal_errorN 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� z1Fatal error on transport
protocol:.*
transport:.*)�exc_info)
� Exceptionr� r r
r� r� rM r4 � MockPattern�MOCK_ANY)r r� r� r� r� r r r �!test_fatal_error_custom_exceptionY s
z8SelectorTransportTests.test_fatal_error_custom_exceptionc C sr t � }| �� }| �|j� | �|j� |�|� | jj�|� | j j
�� | � |j � | � |j� | � |j� d S )N)
rX r� ZassertIsNotNoneZ _protocolr� Z_call_connection_lostr� r� rM rg r rL r )r r� r� r r r �test_connection_losti s
z+SelectorTransportTests.test_connection_lostc C sv | � � }|j�d� |�dtj� | �| jj� |� d � | �|�
� � | � | jj� |�dtj� | � | jj� d S )Nr� rH ) r� r� r� r/ r r� r: r# r� r� r� r< )r r� r r r �test__add_readerw s
z'SelectorTransportTests.test__add_reader)r r r r r� r� r� rQ 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 )NrH )r r r� r# r4 r� r( r) r� r r
r� rg �sock_fdrJ r" )r )r% r r r � s
z"SelectorSocketTransportTests.setUpNc C s& t | j| j| j|d�}| �t|� |S )N)r? )r r# rg r� r� r )r r? r r r r �socket_transport� s
z-SelectorSocketTransportTests.socket_transportc C sT t j| jd�}| j|d�}| j�|� | j�d|j� t�| j� | j j
� |� d S )N)r# )r? rH ) r( r3 r# r rN �
assert_reader� _read_readyr4 r9 r� �connection_maderM )r r? r� r r r r� � s z&SelectorSocketTransportTests.test_ctorc C s8 t j| jd�}| j|d� | j�|� | �|�� � d S )N)r# )r? )r( r3 r# r rN rL r~ )r r? 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 )NrH )r r4 r9 r# r< Z_pausedr: r8 r r r; Zassert_no_readerr= r )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 )Nr| )r r r
r r� r# �call_exception_handler� LookupErrorr� � eof_receivedrU rg rT r" r rM r: rr )r 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 )Ns data)r r r
r� r# r r r� �
data_receivedrU rg rT r" r r: rr )r r r r r �test_data_received_error� s
z5SelectorSocketTransportTests.test_data_received_errorc C s, | � � }d| jj_|�� | jj�d� d S )Ns data)r rg rT r" r r� r rM )r r r r r �test_read_ready� s
z,SelectorSocketTransportTests.test_read_readyc C s> | � � }t�� |_d| jj_|�� | jj �
� |j�
� d S )Nr| ) r r r
r rg rT r" r r� r
rM )r r r r r �test_read_ready_eof� s
z0SelectorSocketTransportTests.test_read_ready_eofc C sL | � � }t�� |_d| jj_d| jj_|� � | jj�
� | � |jj � d S )Nr| T)
r r r
r rg rT r" r� r
r rM r< rr )r r r r r � test_read_ready_eof_keep_open� s
z:SelectorSocketTransportTests.test_read_ready_eof_keep_openzlogging.exceptionc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S )N) rS rg rT rU r r r
r� r r< rr )r r� r r r r �test_read_ready_tryagain� s
z5SelectorSocketTransportTests.test_read_ready_tryagainc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S )N) r� rg rT rU r r r
r� r r< rr )r r� r r r r �$test_read_ready_tryagain_interrupted s
zASelectorSocketTransportTests.test_read_ready_tryagain_interruptedc C sN t � }| jj_| �� }t�� |_t� � � |�
� W d Q R X |j� |� d S )N) �ConnectionResetErrorrg rT rU r r r
r� r4 r5 r rM )r r� rw r r r r �test_read_ready_conn_reset
s
z7SelectorSocketTransportTests.test_read_ready_conn_resetc C s< t � }| jj_| �� }t�� |_|�� |j� |d� d S )Nz$Fatal read error on socket transport)
rX rg rT rU r r r
r� r rM )r r� rw r r r r �test_read_ready_err s
z0SelectorSocketTransportTests.test_read_ready_errc C s6 d}t |�| jj_| �� }|�|� | jj�|� d S )Ns data)�lenrg rZ r" r �writerM )r �datar r r r �
test_write s
z'SelectorSocketTransportTests.test_writec C sJ t d�}t|�| jj_| �� }|�|� | jj�|� | �|t d�� d S )Ns data) r r rg rZ r" r r rM rc )r r r r r r �test_write_bytearray( s
z1SelectorSocketTransportTests.test_write_bytearrayc C s: t d�}t|�| jj_| �� }|�|� | jj�|� d S )Ns data)�
memoryviewr rg rZ r" r r rM )r r r r r r �test_write_memoryview1 s
z2SelectorSocketTransportTests.test_write_memoryviewc C sF | � � }|j�d� |�d� | �| jjj� | �t dg�|j� d S )Ns datar| )
r r� r� r r< rg rZ rr rc r )r r r r r �test_write_no_data9 s
z/SelectorSocketTransportTests.test_write_no_datac C sH | � � }|j�d� |�d� | �| jjj� | �t ddg�|j� d S )Ns data1s data2)
r r� r� r r< rg rZ rr rc r )r 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 )Ns datarj rH s ta) rg rZ r" r r r# �
assert_writer� _write_readyrc r r� )r r r r r r �test_write_partialH s
z/SelectorSocketTransportTests.test_write_partialc C s\ t d�}d| jj_| �� }|�|� | j�d|j� | � t
dg�|j � | � |t d�� d S )Ns datarj rH s ta) r rg rZ r" r r r# r r rc r r� )r r r r r r � test_write_partial_bytearrayR s
z9SelectorSocketTransportTests.test_write_partial_bytearrayc C sL t d�}d| jj_| �� }|�|� | j�d|j� | � t
dg�|j � d S )Ns datarj rH s ta) r rg rZ r" r r r# r r rc r r� )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 )Ns datar rH ) rg rZ r" rJ r r r# r r rc r r� )r r r r r r �test_write_partial_noneg s
z4SelectorSocketTransportTests.test_write_partial_nonec C sH t | jj_d}| �� }|�|� | j�d|j� | � t
dg�|j � d S )Ns datarH ) rS rg rZ rU r r r# r r rc r r� )r r r r r r �test_write_tryagainr s
z0SelectorSocketTransportTests.test_write_tryagainz asyncio.selector_events.loggerc C s� t � }| jj_d}| �� }t�� |_|�|� |j� |d� d|_
| j� � |�|� | � | jjj
� | �|j
d� |�|� |�|� |�|� |�|� |j� d� d S )Ns dataz%Fatal write error on socket transportr rj z socket.send() raised exception.)rX rg rZ rU r r r
r� r rM r� r� r< rr rc �warning)r �m_logrw r r r r r �test_write_exception| s$
z1SelectorSocketTransportTests.test_write_exceptionc C s | � � }| �t|jd� d S )N�str)r rE � TypeErrorr )r r r r r �test_write_str� s z+SelectorSocketTransportTests.test_write_strc C s: | � � }|�� | �|jd� |�d� | �|jd� d S )Nr s datarj )r r rc r� 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 )Ns datarH )r rg rZ r" r r� r� r# r0 r r: rr r< r� )r 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 )Ns dataTrH )r rg rZ r" r �_closingr� r� r# r0 r r: rr r< r� r rM r� r� )r r r r r r �test_write_ready_closing� s z5SelectorSocketTransportTests.test_write_ready_closingc C s | � � }| �t|j� d S )N)r rE �AssertionErrorr )r 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 )Ns datarj rH s ta) rg rZ r" r r� r� r# r0 r r rc r )r 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 )Ns datar rH ) rg rZ r" r r� r� r# r0 r r rc r )r r r r r r � test_write_ready_partial_none� s
z:SelectorSocketTransportTests.test_write_ready_partial_nonec C s` t | jj_| �� }tddg�|_| j�d|j � |� � | j�
d|j � | � tdg�|j� d S )Ns data1s data2rH s
data1data2) rS rg rZ rU r r r� r# r0 r r rc )r r r r r �test_write_ready_tryagain� s
z6SelectorSocketTransportTests.test_write_ready_tryagainc C sH t � }| jj_| �� }t�� |_|j� d� |�
� |j� |d� d S )Ns dataz%Fatal write error on socket transport) rX rg rZ rU r r r
r� r� r� r rM )r rw r r r r �test_write_ready_exception� s
z7SelectorSocketTransportTests.test_write_ready_exceptionc C sT | � � }| �|�� � |�� | jj�tj� |�� | � | jjj
d� |� � d S )Nr ) r r: Z
can_write_eof� write_eofrg �shutdownrM r� �SHUT_WRrc r� r )r r� r r r �test_write_eof� s z+SelectorSocketTransportTests.test_write_eofc C s� | � � }t| jj_|�d� |�� | �|jt dg�� | �
|j � | � | jj
j� dd� | jj_|�� | �
| jjj� | jj
�tj� |�� d S )Ns datac S s dS )Nr� r )�_r r r r� � r| zDSelectorSocketTransportTests.test_write_eof_buffer.<locals>.<lambda>)r rS rg rZ rU r r6 rc r� r r: Z_eofr< r7 rr r rM r� r8 r )r r� r r r �test_write_eof_buffer� s
z2SelectorSocketTransportTests.test_write_eof_bufferc C s. | � � }|�� | j�t�d�� |�� d S )Nr )r r r# rN r( rb r6 )r r� r r r �test_write_eof_after_close� s z7SelectorSocketTransportTests.test_write_eof_after_closezasyncio.base_events.loggerc C s0 t �� }| j_| �� }|�� |�| j� d S )N)r r
r# r2 r r rM r )r r' rD r r r r �"test_transport_close_remove_writer 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/ r1 r2 r3 r4 r5 r9 r; r< r= r� r r )r% r r � sL
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 )N)�buf)�hint)r r r r� r| zDSelectorSocketTransportBufferedProtocolTests.setUp.<locals>.<lambda>rH )r r r� r# r4 r� r( �BufferedProtocolr� r r? �
get_bufferrU r r
r� rg r rJ r" )r )r% )r r r s
z2SelectorSocketTransportBufferedProtocolTests.setUpNc C s& t | j| j| j|d�}| �t|� |S )N)r? )r r# rg r� r� r )r r? r r r r r s
z=SelectorSocketTransportBufferedProtocolTests.socket_transportc C sT t j| jd�}| j|d�}| j�|� | j�d|j� t�| j� | j j
� |� d S )N)r# )r? rH ) r( r3 r# r rN r r r4 r9 r� r rM )r r? r� r r r r� ! s z6SelectorSocketTransportBufferedProtocolTests.test_ctorc C sd | � � }t�� |_t�� | j_t� | jj_ |�
� | � |jj � | � | jjj � | �
| jjj � d S )N)r r r
r� r# r r r� rB rU r r: rr r< �buffer_updated)r r 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 )Nr )r )r@ r r r r� < r| zXSelectorSocketTransportBufferedProtocolTests.test_get_buffer_zerosized.<locals>.<lambda>)r r r
r� r# r r� rB rU r r: rr r< rC )r r r r r �test_get_buffer_zerosized7 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 )Ns datar� c s � S )Nr )r@ )r? r r r� Q r| zUSelectorSocketTransportBufferedProtocolTests.test_proto_type_switch.<locals>.<lambda>ri rk )r4 r� r( r) r� r rg rT r" r r rM rA r rB rU Z set_protocol� recv_intorC )r r Z buf_protor )r? r �test_proto_type_switchD s
zCSelectorSocketTransportBufferedProtocolTests.test_proto_type_switchc C sn | � � }t�� |_t�� | j_t� | jj_ d| j
j _ |�
� | �|jj� | �| jjj� | �| jjj� d S )Nri )r r r
r� r# r r r� rC rU rg rF r" r r: rr rB )r 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 )Nr )r r r
r r� r# r r r� r
rU rg rF r" r rM r: rr )r r r r r r i s
zISelectorSocketTransportBufferedProtocolTests.test_read_eof_received_errorc C s: | � � }d| jj_|�� | jj�d� | jj�d� d S )Nri rk ) r rg rF r" r r� rB rM rC )r r r r r r x s
z<SelectorSocketTransportBufferedProtocolTests.test_read_readyc C s> | � � }t�� |_d| jj_|�� | jj �
� |j�
� d S )Nr ) r r r
r rg rF r" r r� r
rM )r 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 rg rF r" r� r
r rM r< rr )r r r r r r � s
zJSelectorSocketTransportBufferedProtocolTests.test_read_ready_eof_keep_openzlogging.exceptionc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S )N) rS rg rF rU r r r
r� r r< rr )r r� r r r r r � s
zESelectorSocketTransportBufferedProtocolTests.test_read_ready_tryagainc C s6 t | jj_| �� }t�� |_|�� | � |jj
� d S )N) r� rg rF rU r r r
r� r r< rr )r r� r r r r r � s
zQSelectorSocketTransportBufferedProtocolTests.test_read_ready_tryagain_interruptedc C sN t � }| jj_| �� }t�� |_t� � � |�
� W d Q R X |j� |� d S )N) r rg rF rU r r r
r� r4 r5 r rM )r r� rw r r r r r � s
zGSelectorSocketTransportBufferedProtocolTests.test_read_ready_conn_resetc C s< t � }| jj_| �� }t�� |_|�� |j� |d� d S )Nz$Fatal read error on socket transport)
rX rg rF rU r r r
r� r rM )r r� rw r r r r r � s
z@SelectorSocketTransportBufferedProtocolTests.test_read_ready_err)N)r r r r r r� rD rE rG rH 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_setrH )r r r� r# r4 r� r( ZDatagramProtocolr� r r
r� rg rJ r" )r )r% r r r � s
z$SelectorDatagramTransportTests.setUpNc C s8 |rd nt | jj_t| j| j| j|d�}| �t|� |S )N)r� ) rX rg Z getpeernamerU 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)s data)z0.0.0.0i� s data)z0.0.0.0i� )rJ rg �recvfromr" r r� Zdatagram_receivedrM )r r r r r r � s
z.SelectorDatagramTransportTests.test_read_readyc C s6 | � � }t| jj_t�� |_|�� | � |jj
� d S )N) rJ rS rg rK rU r r
r� r r< rr )r r r r r r � s
z7SelectorDatagramTransportTests.test_read_ready_tryagainc C s< | � � }t� }| jj_t�� |_|�� |j� |d� d S )Nz&Fatal read error on datagram transport)
rJ r7 rg rK rU r r
r� r rM )r r rw r r r r � s
z2SelectorDatagramTransportTests.test_read_ready_errc C sJ | � � }t� }| jj_t�� |_|�� | � |jj
� | j j �
|� d S )N)rJ rX rg rK rU r r
r� r r< rr r� �error_receivedrM )r r rw r r r �test_read_ready_oserr� s
z4SelectorDatagramTransportTests.test_read_ready_oserrc C sF d}| � � }|�|d� | �| jjj� | �| jjjd |df� d S )Ns data)z0.0.0.0i� r )rJ �sendtor: rg rr rc rd )r r r r r r � test_sendto� s z*SelectorDatagramTransportTests.test_sendtoc C sJ t d�}| �� }|�|d� | �| jjj� | �| jjjd |df� d S )Ns data)z0.0.0.0i� r )r rJ rN r: rg rr rc rd )r r r r r r �test_sendto_bytearray s z4SelectorDatagramTransportTests.test_sendto_bytearrayc C sJ t d�}| �� }|�|d� | �| jjj� | �| jjjd |df� d S )Ns data)z0.0.0.0i� r )r rJ rN r: rg rr rc rd )r r r r r r �test_sendto_memoryview s z5SelectorDatagramTransportTests.test_sendto_memoryviewc C sH | � � }|j�d� |�dd� | �| jjj� | �dgt|j�� d S )N)s data)z0.0.0.0i90 r| r ) rJ r� �appendrN r< rg rr rc �list)r r r r r �test_sendto_no_data s z2SelectorDatagramTransportTests.test_sendto_no_datac C sJ | � � }|j�d� |�dd� | �| jjj� | �ddgt|j�� d S )N)s data1)z0.0.0.0i90 s data2)z0.0.0.0i90 )s data2)z0.0.0.0i90 ) rJ r� rR rN r< rg rr rc rS )r 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 )Ns data2)s data1)z0.0.0.0i90 )z0.0.0.0i90 )s data2)z0.0.0.0i90 r r ) r rJ r� rR rN r< rg rr rc rS r* �bytes)r �data2r 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 )Ns data2)s data1)z0.0.0.0i90 )z0.0.0.0i90 )s data2)z0.0.0.0i90 r r ) r rJ r� rR rN r< rg rr rc rS r* rV )r rW r 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 )Ns data)z0.0.0.0i90 rH )s data)z0.0.0.0i90 ) rS rg rN rU rJ r# r �
_sendto_readyrc rS r� )r r r r r r �test_sendto_tryagain; s
z3SelectorDatagramTransportTests.test_sendto_tryagainz asyncio.selector_events.loggerc C s� d}t � }| jj_| �� }t�� |_|�|d� | �|jj � |j�
|d� d|_ d|_ |�|� |�|� |�|� |�|� |�|� |j
�
d� d S )Ns datar z'Fatal write error on datagram transportr )Z123z socket.send() raised exception.)r7 rg rN rU rJ r r
r� r: rr rM r� Z_addressr& )r r' r rw r r r r �test_sendto_exceptionG s"
z4SelectorDatagramTransportTests.test_sendto_exceptionc C sL d}t | jj_| �� }t�� |_|�|d� | �|j d� | �
|jj � d S )Ns datar r ) �ConnectionRefusedErrorrg rN rU rJ r r
r� rc r� r< rr )r r 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 )Ns data)z0.0.0.0r )r� )r] rg rZ rU rJ r r
r� rN r< rr r: r� rL )r r r r r r �$test_sendto_error_received_connectedj s
zCSelectorDatagramTransportTests.test_sendto_error_received_connectedc C s | � � }| �t|jdd� d S )Nr) r )rJ rE r* rN )r r r r r �test_sendto_strv s z.SelectorDatagramTransportTests.test_sendto_strc C s" | j dd�}| �t|jdd� d S )N)z0.0.0.0r )r� s str)z0.0.0.0rj )rJ rE �
ValueErrorrN )r r r r r �test_sendto_connected_addrz s z9SelectorDatagramTransportTests.test_sendto_connected_addrc C s@ | j dd�}|�� | �|jd� |�dd� | �|jd� d S )N)r )r� r s datarj )rJ r rc r� rN )r 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 )Ns data)z0.0.0.0i90 rH r )r rg rN r" rJ r� rR r# r0 rZ r: rr rc rd r< r� )r 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 )Ns dataTr rH )r rg rZ r" rJ r. r� rR r# r0 rZ rN rM r< r� r 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 )NrH ) rJ r# r0 rZ r< rg rN rr r� )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)s data1r )s data2r rH )
rS rg rN rU rJ r� r� r# r0 rZ r rc rS )r 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)s datar z'Fatal write error on datagram transport) r7 rg rN rU rJ r r
r� r� rR rZ rM )r rw 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 )N)s datar )
r] rg rN rU rJ r r
r� r� rR rZ r< rr )r r r r r � test_sendto_ready_error_received� s
z?SelectorDatagramTransportTests.test_sendto_ready_error_receivedc C sV t | jj_| jdd�}t�� |_|j� d� |�
� | � |jj � | �
| jjj � d S )N)z0.0.0.0r )r� )s datar )r] rg rZ rU rJ r r
r� r� rR rZ r< rr r: r� rL )r 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 )N)z0.0.0.0r )r� )rJ r] r� r< r� rL rr r� )r r� r rw r r r �test_fatal_error_connected� 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>.MyExceptionN)r r r r r r r � MyException� s rl )z0.0.0.0r )r� z1Fatal error on transport
protocol:.*
transport:.*)r� ) r� rJ r� r< r� rL rr rM r4 r� r� )r r� rl r rw r r r �'test_fatal_error_connected_custom_error� s
zFSelectorDatagramTransportTests.test_fatal_error_connected_custom_error)N)#r r r r rJ r r r rM rO rP rQ rT rU rX rY r[ r r� r\ r^ r_ r` rb rc rd re rf rg rh ri rj rk rm r� r r )r% r rI � s:
rI �__main__)r ) �__doc__r� r� r� r� r r. � ImportErrorr( Zasyncio.selector_eventsr r r r Ztest.test_asyncior r4 �ANYr� r r r ZTestCaser r� r r> rI r �mainr r r r �<module> sJ
`|
4 -