File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_selector_events.cpython-310.pyc
o
�iܿ � @ s2 d Z ddlZddlZddlZddlmZ zddlZW n ey% dZY nw 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 kr�e� � dS 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
�C/usr/local/lib/python3.10/test/test_asyncio/test_selector_events.py�tearDownModule s r c @ s e Zd Zdd� Zdd� ZdS )�TestBaseSelectorEventLoopc C �& t �� | _t �� | _| jd7 _d S �N� �r �Mock�_ssock�_csockZ
_internal_fds��selfr
r
r �_make_self_pipe �
z)TestBaseSelectorEventLoop._make_self_pipec C � d S r r
r r
r
r �_close_self_pipe" � 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 u rd S | j �� d | _ d S r )�_sock�close)� transportr
r
r �close_transport* s
r& c s e Zd Z� fdd�Zdd� Ze�edu d�dd� �Ze �
d d�e �
d
d�d d � ��Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Ze �
d�dd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"� Z#S )<�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 sT t �� }t �� | j_d| jj_t �� | j_t �� | j_t �� | j_| j�� }t �
� �3 | j� |t �
� ||�}| �td�� |�� W d � n1 sKw Y t �| j� W d � n1 s`w Y | �|�� � |�� |�� | �|�� � |�� |�� | �|�� � |jj}| �d|�� � | �d|�� � |�� t �| j� d S )NFz SSL transport.*not.*initialized�sslr% )r r r- � _add_readerr2 � _add_writer�_remove_reader�_remove_writer�
create_future�
test_utils�disable_logger�_make_ssl_transportr r3 ZassertRaisesRegex� RuntimeError�
is_reading� run_briefly�
assertTrue�
pause_reading� assertFalse�resume_readingr0 r ZassertIn�lowerr$ )r r6 �waiterr% �
class_namer
r
r �test_make_ssl_transportI s<
��
� � z2BaseSelectorEventLoopTests.test_make_ssl_transportzasyncio.selector_events.sslzasyncio.sslproto.sslc C sz t �� }t �� | j_t �� | j_t �� | j_t �� | j_| �t�� | j� ||||� W d � d S 1 s6w Y d S r )
r r r- r1 �
add_writer�
remove_reader�
remove_writer� assertRaisesrA r@ )r r6 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 r r r r r
r
r r | r zHBaseSelectorEventLoopTests.test_close.<locals>.EventLoop._make_self_pipeN)r r r r r
r
r
r � EventLoop{ s rQ � r r c S r r r
r
r
r
r �callback� r z7BaseSelectorEventLoopTests.test_close.<locals>.callback)r r* r- r. r �filenor, r r r r; � _selectorr$ rF Z is_closedrD � assertIsNone�assert_called_withr= rO rA Z run_forever�run_until_completer1 rL ) r rQ ZssockZcsockrM r* �f�fdrS 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- rM rU r$ rV 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_effectrV �_read_from_selfr r
r
r � test_read_from_self_tryagain� � z7BaseSelectorEventLoopTests.test_read_from_self_tryagainc C s t | jjj_| �t | jj� d S r )�OSErrorr- r r^ r_ rO r` r r
r
r � test_read_from_self_exception� rb z8BaseSelectorEventLoopTests.test_read_from_self_exceptionc C sJ t | jjj_t�� � | �| j�� � W d � d S 1 s w Y d S r ) r] r- r �sendr_ r> r? rV �_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 )rA r- r re r_ rO rf r r
r
r � test_write_to_self_exception� s z7BaseSelectorEventLoopTests.test_write_to_self_exceptionzsocket.getaddrinfoc
C s� d}t jt jfD ]P}| �|��A tj|d�}dd� |_| j�| j� ||��}| j�
|� |� |d |d |j |j
|jd� | j�
|� |j� d� W d � n1 sSw Y qd S )N)zneed-resolution.comi� )�typec W s dgS )N)NNNN�z 127.0.0.1r r
)�argsr
r
r �<lambda>� � zZBaseSelectorEventLoopTests.test_sock_connect_resolve_using_socket_params.<locals>.<lambda>r r rj )�socket� SOCK_STREAM�
SOCK_DGRAMZsubTestr> Zmock_nonblocking_socketr_ r- Z create_taskZ sock_connectrX rW �familyri �proto�connect)r Zm_gai�addrZ sock_type�sockZconr
r
r �-test_sock_connect_resolve_using_socket_params� s � � ���zHBaseSelectorEventLoopTests.test_sock_connect_resolve_using_socket_paramsc C s� t | jjj_dd� }| j�d|� | �| jjjj� | jjjj d \}}\}}| �
d|� | �
t j |� | �
||j
� | �|� d S )Nc S � dS �NTr
r
r
r
r rl � � z<BaseSelectorEventLoopTests.test_add_reader.<locals>.<lambda>r r )�KeyErrorr- rU �get_keyr_ r1 rD �register�called� call_args� assertEqual� selectors�
EVENT_READ� _callbackrV �r �cbrZ �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 rw rx r
r
r
r
r rl � ry zEBaseSelectorEventLoopTests.test_add_reader_existing.<locals>.<lambda>r )r r r� � SelectorKey� EVENT_WRITEr- rU r{ r, r1 rD �cancelr} rF r| �modifyr~ r r� r� �r �reader�writerr� rZ 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 rw rx r
r
r
r
r rl � ry zLBaseSelectorEventLoopTests.test_add_reader_existing_writer.<locals>.<lambda>r )r r r� r� r� r- rU r{ r, r1 rF r| r} rD r� r~ r r� r� )r r� r� rZ r� r� r� r
r
r � test_add_reader_existing_writer� s � z:BaseSelectorEventLoopTests.test_add_reader_existing_writerc C �B t �ddt jd�| jjj_| �| j�d�� | � | jjj
j � d S �Nr )NN) r� r� r� r- rU r{ r, rF rM rD �
unregisterr} r r
r
r �test_remove_reader� �
�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- rU r{ r, rD rM rF r� r} r r� r~ �r r� r� r
r
r � test_remove_reader_read_write � �
��z8BaseSelectorEventLoopTests.test_remove_reader_read_writec C �" t | jjj_| �| j�d�� d S r )rz r- rU r{ r_ rF rM r r
r
r �test_remove_reader_unknown �
�z5BaseSelectorEventLoopTests.test_remove_reader_unknownc C s� t | jjj_dd� }| j�d|� | �| jjjj� | jjjj d \}}\}}| �
d|� | �
t j |� | �
|� | �
||j� d S )Nc S rw rx r
r
r
r
r rl ry z<BaseSelectorEventLoopTests.test_add_writer.<locals>.<lambda>r r )rz r- rU r{ r_ rL rD r| r} r~ r r� r� rV 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 rw rx r
r
r
r
r rl ( ry zEBaseSelectorEventLoopTests.test_add_writer_existing.<locals>.<lambda>r )r r r� r� r� r- rU r{ r, rL rD r� r} rF r| r� r~ r r� r� r� r
r
r �test_add_writer_existing# s � z3BaseSelectorEventLoopTests.test_add_writer_existingc C r� r� ) r� r� r� r- rU r{ r, rF rN rD r� r} r r
r
r �test_remove_writer4 r� 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 r� r� r� r� r- rU r{ r, rD rN rF r� r} r r� r~ r� r
r
r � test_remove_writer_read_write; r� z8BaseSelectorEventLoopTests.test_remove_writer_read_writec C r� r )rz r- rU r{ r_ rF rN r r
r
r �test_remove_writer_unknownI r� 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_eventsr� r� r� rD r} rW �r r� r
r
r �test_process_events_readN 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- r; r� r� r� r� rW r� r
r
r �"test_process_events_read_cancelledZ 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� r� r� r� rW �r r� r
r
r �test_process_events_writee 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- r< r� r� r� r� rW r� r
r
r �#test_process_events_write_cancelledp s ���z>BaseSelectorEventLoopTests.test_process_events_write_cancelledc C s� t �� }t �� t �� f|j_d}t jj}|| jd��}| jjt �� ||d� W d � n1 s0w Y | j�t �
d�� | � |jj |� d S )N�d Z_accept_connection2)�backlogr )
r r �acceptr, �patch�objectr- Z_accept_connectionrX r �sleepr �
call_count)r ru r� Zmock_objZ accept2_mockr
r
r � test_accept_connection_multiple{ s
� �z:BaseSelectorEventLoopTests.test_accept_connection_multiple)$r r r r) r7 �unittestZskipIfr8 rK r r� rP r[ r\ ra rd rg rh rv r� 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 �NrR )r( r) �
new_test_loopr- r> �make_test_protocolr r3 �protocolr r rn ru rT r, r r/ r
r r) � s
z SelectorTransportTests.setUpc C s$ t | j| j| jd �}| �t|� |S r )r r- ru 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# ru 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�abortrW 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 rR )
r� r$ rD �
is_closingr r- �remove_reader_countr� �connection_lost�
_conn_lostr� r
r
r r[ � s z!SelectorTransportTests.test_closec C sJ | � � }|j�d� |�� | �| jj� t�| j� | �| j j
j � d S �N� data) r� �_buffer�extendr$ rF r- �readersr> rC r� r� r} 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� 1rR r )r� r� r� r- r9 r �sentinelr: r� rD r� r r" rF 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 )rc r� r r r� � _fatal_error�assert_not_calledrW )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 @ � e Zd ZdS )zISelectorTransportTests.test_fatal_error_custom_exception.<locals>.MyErrorN�r r r r
r
r
r �MyError� � r� �1Fatal error on transport
protocol:.*
transport:.*��exc_info)
� Exceptionr� r r r� r� rW 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 )
rc r� ZassertIsNotNoneZ _protocolr� Z_call_connection_lostr� r� rW ru r$ rV 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� rR ) r� r� r� r9 r r� rD r- r� r� r� rF r� r
r
r �test__add_reader� s
z'SelectorTransportTests.test__add_reader)r r r r) r� r� r� r[ 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 r3 r� r r rn ru �sock_fdrT r, r r/ r
r r) s
z"SelectorSocketTransportTests.setUpNc C �& t | j| j| j|d�}| �t|� |S �N�rI �r r- ru r� r� r& �r rI r% r
r
r �socket_transport �
� z-SelectorSocketTransportTests.socket_transportc C �P | j �� }| j|d�}| j �|� | j �d|j� t�| j � | jj �
|� d S �Nr� rR � r- r= r� rX �
assert_reader� _read_readyr> rC r� Zconnection_maderW �r rI r� r
r
r r� �
z&SelectorSocketTransportTests.test_ctorc C s4 | j �� }| j|d� | j �|� | �|�� � d S r� )r- r= r� rX rV �result)r rI 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> rC r- rF Z_pausedrD rB r� r� rE Zassert_no_readerrG r$ r� r
r
r �test_pause_resume_reading( s$ z6SelectorSocketTransportTests.test_pause_resume_readingc C �d | � � }t�� |_t�� |_t�� | j_t� | jj _
d| j j _
|�� | jj �� | �|jj� d S �N� )r� r r r$ r� r- �call_exception_handler� LookupErrorr� � eof_receivedr_ ru r^ r, r� rW rD r} r� r
r
r � test_read_eof_received_error? �
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_ ru r^ r, r� rD r} r� r
r
r �test_data_received_errorN s
z5SelectorSocketTransportTests.test_data_received_errorc C s, | � � }d| jj_|�� | jj�d� d S r� )r� ru r^ r, r� r� r rW r� r
r
r �test_read_ready[ s
z,SelectorSocketTransportTests.test_read_readyc C �> | � � }t�� |_d| jj_|�� | jj �
� |j�
� d S r� ) r� r r r$ ru r^ r, r� r� r rW r� r
r
r �test_read_ready_eofc �
z0SelectorSocketTransportTests.test_read_ready_eofc C �L | � � }t�� |_d| jj_d| jj_|� � | jj�
� | � |jj � d S )Nr� T)
r� r r r$ ru r^ r, r� r r� rW rF r} r� r
r
r � test_read_ready_eof_keep_openm �
z:SelectorSocketTransportTests.test_read_ready_eof_keep_open�logging.exceptionc C �6 t | jj_| �� }t�� |_|�� | � |jj
� d S r ) r] ru r^ r_ r� r r r� r� rF r} �r r� r% r
r
r �test_read_ready_tryagainx �
z5SelectorSocketTransportTests.test_read_ready_tryagainc C r
r ) �InterruptedErrorru r^ r_ r� r r r� r� rF r} r r
r
r �$test_read_ready_tryagain_interrupted� r zASelectorSocketTransportTests.test_read_ready_tryagain_interruptedc C �b t � }| jj_| �� }t�� |_t� � � |�
� W d � n1 s$w Y |j� |� d S r ) �ConnectionResetErrorru r^ r_ r� r r r� r> r? r� rW �r r� �errr% r
r
r �test_read_ready_conn_reset� �
�z7SelectorSocketTransportTests.test_read_ready_conn_resetc C �< t � }| jj_| �� }t�� |_|�� |j� |d� d S �Nz$Fatal read error on socket transport)
rc ru r^ r_ r� r r r� r� rW r r
r
r �test_read_ready_err� �
�z0SelectorSocketTransportTests.test_read_ready_errc C s6 d}t |�| jj_| �� }|�|� | jj�|� d S r� )�lenru re r, r� �writerW �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 ru re r, r� r rW r r r
r
r �test_write_bytearray� s
z1SelectorSocketTransportTests.test_write_bytearrayc C s: t d�}t|�| jj_| �� }|�|� | jj�|� d S r� )�
memoryviewr ru re r, r� r rW r r
r
r �test_write_memoryview� s
z2SelectorSocketTransportTests.test_write_memoryviewc C sF | � � }|j�d� |�d� | �| jjj� | �t dg�|j� d S )Nr� r� �
r� r� r� r rF ru re r} r 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� � rR s ta) ru re r, r� r r- �
assert_writer� _write_readyr 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 ru re r, r� r r- r, r- r 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# ru re r, r� r r- r, 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 �Nr� r rR ) ru re r, rT r� r r- r, r- r 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� rR ) r] ru re r_ r� r r- r, r- r 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.)rc ru re r_ r� r r r� r rW r� Z
reset_mockrF r} r �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� rO � 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$ 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 r3 )r ru re r, r� r� r� r- r: r- rD r} rF 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� TrR )r ru re r, r� �_closingr� r� r- r: r- rD r} rF r� r$ rW r� r� r r
r
r �test_write_ready_closing+ s z5SelectorSocketTransportTests.test_write_ready_closingc C s | � � }| �t|j� d S r )r� rO �AssertionErrorr- r� r
r
r �test_write_ready_no_data9 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* � ru re r, r� r� r� r- r: r- r, r r" r r
r
r �test_write_ready_partial> �
z5SelectorSocketTransportTests.test_write_ready_partialc C sb d}d| j j_| �� }|j�|� | j�d|j� |�� | j� d|j� | �
t dg�|j� d S r1 rD r r
r
r � test_write_ready_partial_noneI rF z:SelectorSocketTransportTests.test_write_ready_partial_nonec C s` t | jj_| �� }tddg�|_| j�d|j � |� � | j�
d|j � | � tdg�|j� d S )Nr' r( rR s
data1data2) r] ru re r_ r� r" r� r- r: r- r, r r� r
r
r �test_write_ready_tryagainT s
z6SelectorSocketTransportTests.test_write_ready_tryagainc C �H t � }| jj_| �� }t�� |_|j� d� |�
� |j� |d� d S )Nr� r6 ) rc ru re r_ r� r r r� r� r� r- rW �r r r% r
r
r �test_write_ready_exception_ s
�z7SelectorSocketTransportTests.test_write_ready_exceptionc C sT | � � }| �|�� � |�� | jj�tj� |�� | � | jjj
d� |� � d S r ) r� rD Z
can_write_eof� write_eofru �shutdownrW rn �SHUT_WRr r� r$ r� r
r
r �test_write_eofj 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 rw )N� r
)�_r
r
r rl { ry zDSelectorSocketTransportTests.test_write_eof_buffer.<locals>.<lambda>)r� r] ru re r_ r rL r r� r" rD �_eofrF rM r} r- rW rn rN r$ r� r
r
r �test_write_eof_buffers s
z2SelectorSocketTransportTests.test_write_eof_bufferc C s. | � � }|�� | j�t�d�� |�� d S �Nr )r� r$ r- rX r r� rL 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 r )r r r- r< r� r$ rW r� )r r9 rN r% r
r
r �"test_transport_close_remove_writer� s z?SelectorSocketTransportTests.test_transport_close_remove_writerr ),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/ r0 r2 r4 r: r= r> r? rA rC rE rG rH rK rO rS rU rV r� r
r
r/ r r� sZ
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 rl � rm zDSelectorSocketTransportBufferedProtocolTests.setUp.<locals>.<lambda>rR )r( r) r� r- r> r� r �BufferedProtocolr� r rY �
get_bufferr_ r r rn ru r� rT r, r r/ r r r) � s
z2SelectorSocketTransportBufferedProtocolTests.setUpNc C r� r� r� r� r
r
r r� � r� z=SelectorSocketTransportBufferedProtocolTests.socket_transportc C r� r� r� r� r
r
r r� � r� z6SelectorSocketTransportBufferedProtocolTests.test_ctorc C sd | � � }t�� |_t�� | j_t� | jj_ |�
� | � |jj � | � | jjj � | �
| jjj � d S r )r� r r r� r- r� r� r� r\ r_ r� rD r} rF �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 rT )r rZ r
r
r rl � s zXSelectorSocketTransportBufferedProtocolTests.test_get_buffer_zerosized.<locals>.<lambda>)r� r r r� r- r� r� r\ r_ r� rD r} rF r] 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� rP c s � S r r
rZ rX r
r rl � ry zUSelectorSocketTransportBufferedProtocolTests.test_proto_type_switch.<locals>.<lambda>�
���)r> r� r r3 r� r� ru r^ r, r� r rW r[ r r\ r_ Z set_protocol� recv_intor] )r r% Z buf_protor
rX 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 )Nr` )r� r r r� r- r� r� r� r] r_ ru rb r, r� rD r} r\ r� r
r
r �test_buffer_updated_error� s
zFSelectorSocketTransportBufferedProtocolTests.test_buffer_updated_errorc C r� rT )r� r r r$ r� r- r� r� r� r r_ ru rb r, r� rW rD r} r� r
r
r r � r zISelectorSocketTransportBufferedProtocolTests.test_read_eof_received_errorc C s: | � � }d| jj_|�� | jj�d� | jj�d� d S )Nr` ra ) r� ru rb r, r� r� r\ rW r] r� r
r
r r � s
z<SelectorSocketTransportBufferedProtocolTests.test_read_readyc C r rT ) r� r r r$ ru rb r, r� r� r rW r� r
r
r r r z@SelectorSocketTransportBufferedProtocolTests.test_read_ready_eofc C r )Nr T)
r� r r r$ ru rb r, r� r r� rW rF r} r� r
r
r r
r zJSelectorSocketTransportBufferedProtocolTests.test_read_ready_eof_keep_openr c C r
r ) r] ru rb r_ r� r r r� r� rF r} r r
r
r r r zESelectorSocketTransportBufferedProtocolTests.test_read_ready_tryagainc C r
r ) r ru rb r_ r� r r r� r� rF r} r r
r
r r " r zQSelectorSocketTransportBufferedProtocolTests.test_read_ready_tryagain_interruptedc C r r ) r ru rb r_ r� r r r� r> r? r� rW r r
r
r r , r zGSelectorSocketTransportBufferedProtocolTests.test_read_ready_conn_resetc C r r )
rc ru rb r_ r� r r r� r� rW r r
r
r r 6 r z@SelectorSocketTransportBufferedProtocolTests.test_read_ready_errr )r r r r) r� r� r^ r_ rc rd r r r r
r r� r r r r r� r
r
r/ r rW � s(
rW 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_setrR )r( r) r� r- r> r� r ZDatagramProtocolr� r r rn ru rT r, r r/ r
r r) E s
z$SelectorDatagramTransportTests.setUpNc C s8 |rd nt | jj_t| j| j| j|d�}| �t|� |S )N��address) rc ru � getpeernamer_ r r- r� r� r& )r rg r% r
r
r �datagram_transportL s
� z1SelectorDatagramTransportTests.datagram_transportc C s. | � � }d| jj_|�� | jj�dd� d S )N)r� ��0.0.0.0i� r� rj )ri ru �recvfromr, r� r� Zdatagram_receivedrW r� r
r
r r T s
�z.SelectorDatagramTransportTests.test_read_readyc C s6 | � � }t| jj_t�� |_|�� | � |jj
� d S r ) ri r] ru rl r_ r r r� r� rF 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)
ri rA ru rl r_ r r r� r� rW �r r% r r
r
r r f s
�z2SelectorDatagramTransportTests.test_read_ready_errc C sJ | � � }t� }| jj_t�� |_|�� | � |jj
� | j j �
|� d S r )ri rc ru rl r_ r r r� r� rF r} r� �error_receivedrW rm r
r
r �test_read_ready_oserrq s
z4SelectorDatagramTransportTests.test_read_ready_oserrc C sF d}| � � }|�|d� | �| jjj� | �| jjjd |df� d S �Nr� rj r )ri �sendtorD ru r} r r~ r r
r
r � test_sendto{ s �z*SelectorDatagramTransportTests.test_sendtoc C �J t d�}| �� }|�|d� | �| jjj� | �| jjjd |df� d S rp )r ri rq rD ru r} r r~ r r
r
r �test_sendto_bytearray� � �z4SelectorDatagramTransportTests.test_sendto_bytearrayc C rs rp )r# ri rq rD ru r} r r~ r r
r
r �test_sendto_memoryview� ru z5SelectorDatagramTransportTests.test_sendto_memoryviewc C sH | � � }|j�d� |�dd� | �| jjj� | �dgt|j�� d S )N�r� �rk i90 r� r
� ri r� �appendrq rF ru r} r �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' rx r( rx �r( rx ry r� r
r
r �test_sendto_buffer� s ��z1SelectorDatagramTransportTests.test_sendto_bufferc C �h t d�}| �� }|j�d� |�|d� | �| jjj� | �ddgt |j�� | �
|jd d t � d S �Nr( r} rx r~ r r ) r ri r� rz rq rF ru r} r r{ r4 �bytes�r Zdata2r% r
r
r � test_sendto_buffer_bytearray� � ��z;SelectorDatagramTransportTests.test_sendto_buffer_bytearrayc C r� r� ) r# ri r� rz rq rF ru r} r r{ r4 r� r� r
r
r � test_sendto_buffer_memoryview� r� z<SelectorDatagramTransportTests.test_sendto_buffer_memoryviewc C sJ d}t | jj_| �� }|�|d� | j�d|j� | �dgt |j
�� d S )Nr� rx rR rw ) r] ru rq r_ ri r- r, �
_sendto_readyr r{ r� r r
r
r �test_sendto_tryagain� s
�z3SelectorDatagramTransportTests.test_sendto_tryagainr5 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 )Z123r7 )rA ru rq r_ ri r r r� rD r} rW r� �_addressr8 )r r9 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 ) �ConnectionRefusedErrorru rq r_ ri r r r� r r� rF 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 )Nr� �rk r rf )r� ru re r_ ri r r r� rq rF r} rD r� rn r r
r
r �$test_sendto_error_received_connected� s
zCSelectorDatagramTransportTests.test_sendto_error_received_connectedc C s | � � }| �t|jdd� d S )Nr; r
)ri rO r< rq r� r
r
r �test_sendto_str� s z.SelectorDatagramTransportTests.test_sendto_strc C s" | j dd�}| �t|jdd� d S )Nr� rf s str)rk r+ )ri rO �
ValueErrorrq r� r
r
r �test_sendto_connected_addr� r� z9SelectorDatagramTransportTests.test_sendto_connected_addrc C s@ | j dd�}|�� | �|jd� |�dd� | �|jd� d S )N)r rf r r� r+ )ri r$ r r� rq 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� rx rR r )r ru rq r, ri r� rz r- r: r� rD r} r r~ rF 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
rR )r ru re r, ri r@ r� rz r- r: r� rq rW rF 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� ) ri r- r: r� rF ru rq 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)r' r
)r( r
rR )
r] ru rq r_ ri r� r� r- r: r� r, r r{ r� r
r
r �test_sendto_ready_tryagain* s
�z9SelectorDatagramTransportTests.test_sendto_ready_tryagainc C rI )N�r� r
r� ) rA ru rq r_ ri r r r� r� rz r� rW rJ r
r
r �test_sendto_ready_exception7 s
�z:SelectorDatagramTransportTests.test_sendto_ready_exceptionc C sB t | jj_| �� }t�� |_|j� d� |�
� | � |jj � d S )Nr� )
r� ru rq r_ ri r r r� r� rz r� rF r} r� r
r
r � test_sendto_ready_error_receivedC s
z?SelectorDatagramTransportTests.test_sendto_ready_error_receivedc C sV t | jj_| jdd�}t�� |_|j� d� |�
� | � |jj � | �
| jjj � d S )Nr� rf r� )r� ru re r_ ri r r r� r� rz r� rF r} rD r� rn r� r
r
r �+test_sendto_ready_error_received_connectionM s
zJSelectorDatagramTransportTests.test_sendto_ready_error_received_connectionz asyncio.base_events.logger.errorc C s8 | j dd�}t� }|�|� | �| jjj� |�� d S )Nr� rf )ri r� r� rF r� rn r} r� )r r� r% r r
r
r �test_fatal_error_connectedX s
z9SelectorDatagramTransportTests.test_fatal_error_connectedc C sZ G dd� dt �}| jdd�}|� }|�|� | �| jjj� |jt� d�|t
t
fd� d S )Nc @ r� )z[SelectorDatagramTransportTests.test_fatal_error_connected_custom_error.<locals>.MyExceptionNr� r
r
r
r � MyExceptionb r� r� r� rf r� r� ) r� ri r� rF r� rn r} rW r> r� r� )r r� r� r% r r
r
r �'test_fatal_error_connected_custom_error` s
�
�zFSelectorDatagramTransportTests.test_fatal_error_connected_custom_errorr )#r r r r) ri r r r ro rr rt rv 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 re C sB
re �__main__)r
) �__doc__r� rn r� r r8 � ImportErrorr Zasyncio.selector_eventsr r r r Ztest.test_asyncior r> �ANYr� r r
r" r& ZTestCaser' r� r� rW re r �mainr
r
r
r �<module> sJ �
\|
4 - �