File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/test_proactor_events.cpython-38.pyc
U
/D�g'� � @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd l
mZ dd
lmZ d d � Zd
d� ZG dd� dej�ZG dd� dej�ZG dd� dej�Ze�ejdkd�G dd� dej��Zedk�re� � dS )z Tests for proactor_events.py� N)�mock)�BaseProactorEventLoop)�_ProactorSocketTransport)�_ProactorWritePipeTransport)� _ProactorDuplexPipeTransport)�_ProactorDatagramTransport)�support)�utilsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �E/opt/cppython/lib/python3.8/test/test_asyncio/test_proactor_events.py�tearDownModule s r c C s" | j d krd S | j �� d | _ d S r
)�_sock�close)� transportr r r
�close_transport s
r c sb e Zd Z� fdd�ZdRdd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Ze�d �d!d"� �Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Ze�d/�d0d1� �Zd2d3� Z 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)dNdO� Z*dPdQ� Z+� Z,S )S� ProactorSocketTransportTestsc sV t � �� | �� | _| �| jj� t�� | _| j| j_ t
� t j
�| _t�tj�| _d S r
)�super�setUp�
new_test_loop�loop�
addCleanupr r �Mock�proactor� _proactor�
test_utils�make_test_protocolr �Protocol�protocol�socket�sock��self�� __class__r r
r " s
z"ProactorSocketTransportTests.setUpNc C s& t | j| j| j|d�}| �t|� |S )N��waiter)r r r! r r r )r# r' r r r r
�socket_transport+ s
� z-ProactorSocketTransportTests.socket_transportc C sR | j �� }| j|d�}t�| j � | �|�� � | j�|� | j j
� | j d� d S )Nr& � � )
r �
create_futurer( r � run_briefly� assertIsNone�resultr �connection_mader �recv�assert_called_withr! �r# �fut�trr r r
� test_ctor1 s
z&ProactorSocketTransportTests.test_ctorc C sH | � � }|�� | jjj�| jd� | �| jj j
� | �| jj j
� d S )Nr) ) r( �
_loop_readingr r r/ r0 r! � assertFalser �
data_received�called� eof_received�r# r3 r r r
�test_loop_reading9 s
z.ProactorSocketTransportTests.test_loop_readingc C sR | j �� }|�d� | �� }||_|�|� | j jj�| j d� | j
j �d� d S )N� datar) ) r r* �
set_resultr( � _read_futr5 r r/ r0 r! r r7 �r# �resr3 r r r
�test_loop_reading_data@ s
z3ProactorSocketTransportTests.test_loop_reading_datac C sz | j �� }|�d� | �� }| �t|j|� t�� |_ ||_
|�|� | � | j j j
j� | �| jjj� | �|j j� d S �N� )r r* r= r( � assertRaises�AssertionErrorr5 r r r r> r6 r r/ r8 �
assertTruer r9 r? r r r
�test_loop_reading_no_dataJ s
z6ProactorSocketTransportTests.test_loop_reading_no_datac C s> t � }| jjj_| �� }t�� |_|� � |j�
|d� d S �Nz"Fatal read error on pipe transport) �ConnectionAbortedErrorr r r/ � side_effectr( r r � _fatal_errorr5 r0 �r# �errr3 r r r
�test_loop_reading_abortedX s
�z6ProactorSocketTransportTests.test_loop_reading_abortedc C s@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S �NT)
rI r r r/ rJ r( �_closingr r rK r5 r6 r8 r: r r r
�!test_loop_reading_aborted_closingb s
z>ProactorSocketTransportTests.test_loop_reading_aborted_closingc C s@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S �NF)
rI r r r/ rJ r( rP r r rK r5 rF r8 r: r r r
�"test_loop_reading_aborted_is_fatalk s
z?ProactorSocketTransportTests.test_loop_reading_aborted_is_fatalc C sZ t � }| jjj_| �� }d|_t�� |_ t�� |_
|� � | � |j j
� |j
�|� d S rR )�ConnectionResetErrorr r r/ rJ r( rP r r rK � _force_closer5 r6 r8 r0 rL r r r
�!test_loop_reading_conn_reset_losts s
z>ProactorSocketTransportTests.test_loop_reading_conn_reset_lostc C s> t � }| jjj_| �� }t�� |_|� � |j�
|d� d S rH ) �OSErrorr r r/ rJ r( r r rK r5 r0 rL r r r
�test_loop_reading_exception~ s
�z8ProactorSocketTransportTests.test_loop_reading_exceptionc C s< | � � }t�� |_|�d� | �|jd � |jjdd� d S )Nr< )�data)r( r r �
_loop_writing�write� assertEqual�_bufferr0 r: r r r
�
test_write� s
z'ProactorSocketTransportTests.test_writec C s" | � � }|�d� | �|j� d S rB )r( r[ r6 r] r: r r r
�test_write_no_data� s
z/ProactorSocketTransportTests.test_write_no_datac C sF | � � }t�� |_t�� |_|�d� | �|jd� | �|jj � d S �Nr< )
r( r r �
_write_futrZ r[ r\ r] r6 r8 r: r r r
�test_write_more� s
z,ProactorSocketTransportTests.test_write_morec C sH | � � }td�|_|�� | jjj�| jd� | jjjj j
�|j� d S r` ) r( � bytearrayr] rZ r r �sendr0 r! � return_value�add_done_callbackr: r r r
�test_loop_writing� s
�z.ProactorSocketTransportTests.test_loop_writing� asyncio.proactor_events.loggerc C s� t � }| jjj_| �� }t�� |_ddg|_ |�
� |j� |d� d|_ |�
d� |�
d� |�
d� |�
d� |�
d� | �|j d � |j� d� d S )Ns das taz#Fatal write error on pipe transport� r< z socket.send() raised exception.)rW r r rd rJ r( r r rK r] rZ r0 �
_conn_lostr[ r\ �warning)r# �m_logrM r3 r r r
�test_loop_writing_err� s"
�
z2ProactorSocketTransportTests.test_loop_writing_errc C s< | j �� }|�d� | �� }||_|�|� | �|j� d S r` )r r* r= r( ra rZ r, r1 r r r
�test_loop_writing_stop� s
z3ProactorSocketTransportTests.test_loop_writing_stopc C s^ | j �� }|�d� | �� }||_|�� |�|� | �|j� t� | j � | j
j � d � d S �Nri )
r r* r= r( ra r rZ r, r r+ r �connection_lostr0 r1 r r r
�test_loop_writing_closing� s
z6ProactorSocketTransportTests.test_loop_writing_closingc C s* | � � }t�� |_|�� |j�d � d S r
)r( r r rU �abortr0 r: r r r
�
test_abort� s
z'ProactorSocketTransportTests.test_abortc C sz | � � }|�� t�| j� | jj�d � | �|� � � | �
|j d� | jj� � |�� t�| j� | �
| jjj� d S ro )r( r r r+ r r rp r0 rF �
is_closingr\ rj �
reset_mockr6 r8 r: r r r
�
test_close� s z'ProactorSocketTransportTests.test_closec C s: | � � }t�� |_|�� t�| j� | �| j j
j � d S r
) r( r r ra r r r+ r r6 r rp r8 r: r r r
�test_close_write_fut� s
z1ProactorSocketTransportTests.test_close_write_futc C s8 | � � }dg|_|�� t�| j� | �| jjj � d S r` )
r( r] r r r+ r r6 r rp r8 r: r r r
�test_close_buffer� s
z.ProactorSocketTransportTests.test_close_buffer�asyncio.base_events.loggerc C s< | � � }t�� |_|�d � | �|jj� | �|jj� d S r
)r( r r rU rK rF r8 �error)r# Z m_loggingr3 r r r
�test_fatal_error� s
z-ProactorSocketTransportTests.test_fatal_errorc C s� | � � }dg|_t�� }|_t�� }|_|�d � |j�� |j�� t �
| j � | j j
�d � | �d |j� | �|jd� d S )Nr< ri )r( r] r r r> ra rU �cancelr0 r r+ r r rp r\ rj )r# r3 Zread_futZ write_futr r r
�test_force_close� s
z-ProactorSocketTransportTests.test_force_closec C sf t �� }| j�|� | j�� }|�d� || jj_| � � }|�
d� |� d � t �
| j� |�� d S )Nri r< )r r r Zset_exception_handlerr* r= r rd re r( r[ rU r r+ �assert_not_called)r# Z exc_handlerr2 r3 r r r
� test_loop_writing_force_close s
z:ProactorSocketTransportTests.test_loop_writing_force_closec C s8 | � � }d|_|�d � t�| j� | �| jjj � d S rO )
r( rP rU r r+ r r6 r rp r8 r: r r r
�test_force_close_idempotent s
z8ProactorSocketTransportTests.test_force_close_idempotentc C sF | � � }dg|_|�d � t�| j� | jj�d � | � d |j� d S r` )
r( r] rU r r+ r r rp r0 r\ r: r r r
�test_fatal_error_2 s
z/ProactorSocketTransportTests.test_fatal_error_2c C s6 | � � }|�d � | �| jjj� | �| jjj� d S r
)r( Z_call_connection_lostrF r rp r8 r! r r: r r r
�test_call_connection_lost! s
z6ProactorSocketTransportTests.test_call_connection_lostc C sT | � � }| �|�� � |�� | jj�tj� |�� | � | jjj
d� |� � d S ro ) r( rF �
can_write_eof� write_eofr! �shutdownr0 r �SHUT_WRr\ Z
call_countr r: r r r
�test_write_eof' s z+ProactorSocketTransportTests.test_write_eofc C s� | � � }| j�� }||jjj_|�d� |�� | � |j
� | � | j j
j� |jjj�| j d� |�d� | j�� | j j
�tj� |�� d S �Nr< � )r( r r* �_loopr rd re r[ r� rF Z _eof_writtenr6 r! r� r8 r0 r= � _run_oncer r� r �r# r3 �fr r r
�test_write_eof_buffer0 s
z2ProactorSocketTransportTests.test_write_eof_bufferc C s\ t | j| j| j�}| �|�� � |�� | �|�� � | j�� | �| jj j
� |� � d S r
) r r r! r rF r� r� rt r� r r8 r: r r r
�test_write_eof_write_pipe> s �
z6ProactorSocketTransportTests.test_write_eof_write_pipec C s� t | j| j| j�}| j�� }||jjj_|� d� |�
� | � |� � � | �
| jjj� |jjj�| jd� |�d� | j�� | j�� | � | jjj� |�� d S r� )r r r! r r* r� r rd re r[ r� rF rt r6 r� r8 r0 r= r� r r� r r r
� test_write_eof_buffer_write_pipeH s
z=ProactorSocketTransportTests.test_write_eof_buffer_write_pipec C sJ t | j| j| j�}| �|�� � | �t�� |�� W 5 Q R X t |� d S r
)
r r r! r r6 r� rD �NotImplementedErrorr� r r: r r r
�test_write_eof_duplex_pipeW s � z7ProactorSocketTransportTests.test_write_eof_duplex_pipec C s� | � � }g }dD ]"}| j�� }|�|� |�|� q|| jjj_| j�� | � |j
� | � |� � � | j�� | j
j�d� | j�� | j
j�d� |�� |�� | � |j
� | � |� � � td�D ]}| j�� q�| j
j�d� |�� |�� | � |j
� | � |� � � | j�� | j
j�d� | j�� | j
j�d� |�� |�� t�� | j_| j�� | jj�� | j
j�d� |�� | � |� � � d S )N)� data1� data2� data3� data4� data5rC r� r� �
r� r� r� )r( r r* r= �appendr r/ rJ r� r6 Z_pausedrF Z
is_readingr r7 r0 Z
pause_reading�rangeZresume_readingr r �call_exception_handlerr~ r )r# r3 �futures�msgr� �ir r r
�test_pause_resume_reading_ sJ
z6ProactorSocketTransportTests.test_pause_resume_readingc C sH | � � }|j|d� | �|�� d� | �| jjj� | �| jjj� |S )N��highr ) r( Zset_write_buffer_limitsr\ �get_write_buffer_sizer6 r �
pause_writingr8 �resume_writing)r# r� r3 r r r
�pause_writing_transport� s z4ProactorSocketTransportTests.pause_writing_transportc C s~ | j dd�}| j�� }|| jjj_|�d� | j�� | �| j j
j � |� d � | j�� | �
|�� d� | �| j jj � d S )Nr� r� s
large datar )r� r r* r rd re r[ r� rF r r� r8 r= r\ r� r� �r# r3 r2 r r r
�test_pause_resume_writing� s
z6ProactorSocketTransportTests.test_pause_resume_writingc C s� | j dd�}| j�� }|| jjj_|�d� | j�� | �|� � d� | �
| j j j
� |�d� | j�� | �|� � d� | �| j j j
� d S )Nr� r� s 123� � abc� �r� r r* r rd re r[ r� r\ r� r6 r r� r8 rF )r# r3 Zfut1r r r
�test_pause_writing_2write� s
z6ProactorSocketTransportTests.test_pause_writing_2writec C s� | j dd�}| j�� }|| jjj_|�d� | j�� | �|� � d� | �
| j j j
� |�d� | j�� | �|� � d� | �
| j j j
� |�d� | j�� | �|� � d� | �| j j j
� d S ) Nr� r� � 1ri s 23r� r� r� r� r� r r r
�test_pause_writing_3write� s
z6ProactorSocketTransportTests.test_pause_writing_3writec C sd | j dd�}| j�� }|�d � || jjj_|�d� | j�� | � |�
� d� | � | j j
j� d S )Nr� r� s very large datar )r� r r* r= r rd re r[ r� r\ r� r6 r r� r8 r� r r r
�test_dont_pause_writing� s
z4ProactorSocketTransportTests.test_dont_pause_writing)N)-�__name__�
__module__� __qualname__r r( r4 r; rA rG rN rQ rS rV rX r^ r_ rb rg r �patchrm rn rq rs rv rw rx r{ r} r r� r� r� r� r� r� r� r� r� r� r� r� r� r� �
__classcell__r r r$ r
r sR
- r c s� e Zd Z� fdd�Zd,dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z e �
d�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Ze �
d)�d*d+� �Z� ZS )-� ProactorDatagramTransportTestsc sT t � �� | �� | _t�� | _| j| j_t� t
j �| _ tjt
j
d�| _d| jj_d S )N)Zspec_set� )r r r r r r r r r r r �DatagramProtocolr r r! �filenore r" r$ r r
r � s
z$ProactorDatagramTransportTests.setUpNc C s8 |rd nt | jj_t| j| j| j|d�}| �t|� |S )N��address) rW r! � getpeernamerJ r r r r r )r# r� r r r r
�datagram_transport� s
� z1ProactorDatagramTransportTests.datagram_transportc C sB d}| � � }|�|d� | �| jjj� | jjj| j|dd� d S �Nr< )�0.0.0.0i� ��addr)r� �sendtorF r r8 r0 r! �r# rY r r r r
� test_sendto� s �z*ProactorDatagramTransportTests.test_sendtoc C sF t d�}| �� }|�|d� | �| jjj� | jjj| jddd� d S r� )rc r� r� rF r r8 r0 r! r� r r r
�test_sendto_bytearray� s �z4ProactorDatagramTransportTests.test_sendto_bytearrayc C sF t d�}| �� }|�|d� | �| jjj� | jjj| jddd� d S r� )�
memoryviewr� r� rF r r8 r0 r! r� r r r
�test_sendto_memoryview� s �z5ProactorDatagramTransportTests.test_sendto_memoryviewc C sH | � � }|j�d� |�dd� | �| jjj� | �dgt|j�� d S )N)r< �r� i90 rC r ) r� r] r� r� r6 r! r8 r\ �list�r# r r r r
�test_sendto_no_data s �z2ProactorDatagramTransportTests.test_sendto_no_datac C sR | � � }|j�d� t� |_|�dd� | �| jjj� | � ddgt
|j�� d S )N�r� r� r� r� �r� r� ) r� r] r� �objectra r� r6 r r8 r\ r� r� r r r
�test_sendto_buffer s ��z1ProactorDatagramTransportTests.test_sendto_bufferc C sp t d�}| �� }|j�d� t� |_|�|d� | �| jjj � | �
ddgt |j�� | � |jd d t
� d S �Nr� r� r� r� ri r )rc r� r] r� r� ra r� r6 r r8 r\ r� �assertIsInstance�bytes�r# Zdata2r r r r
� test_sendto_buffer_bytearray s ��z;ProactorDatagramTransportTests.test_sendto_buffer_bytearrayc C sp t d�}| �� }|j�d� t� |_|�|d� | �| jjj � | �
ddgt |j�� | � |jd d t
� d S r� )r� r� r] r� r� ra r� r6 r r8 r\ r� r� r� r� r r r
� test_sendto_buffer_memoryview& s ��z<ProactorDatagramTransportTests.test_sendto_buffer_memoryviewrh c C s� d}t � }| jj_| �� }t�� |_|�|d� | �|jj � |j�
|d� d|_ d|_ |�|� |�|� |�|� |�|� |�|� |j
�
d� d S )Nr< r �'Fatal write error on datagram transportri )Z123z!socket.sendto() raised exception.)� RuntimeErrorr r� rJ r� r r rK rF r8 r0 rj �_addressrk )r# rl rY rM r r r r
�test_sendto_exception3 s$
�
z4ProactorDatagramTransportTests.test_sendto_exceptionc C sL d}t | jj_| �� }t�� |_|�|d� | �|j d� | �
|jj � d S )Nr< r r ) �ConnectionRefusedErrorr! r� rJ r� r r rK r\ rj r6 r8 r� r r r
�test_sendto_error_receivedJ s
z9ProactorDatagramTransportTests.test_sendto_error_receivedc C sP d}t | jj_| jdd�}t�� |_|�|� | � |jj
� | � | j j
j
� d S )Nr< �r� ri r� )r� r rd rJ r� r r rK r� r6 r8 rF r �error_receivedr� r r r
�$test_sendto_error_received_connectedV s
zCProactorDatagramTransportTests.test_sendto_error_received_connectedc C s | � � }| �t|jdd� d S )N�strr )r� rD � TypeErrorr� r� r r r
�test_sendto_strb s z.ProactorDatagramTransportTests.test_sendto_strc C s" | j dd�}| �t|jdd� d S )Nr� r� s str)r� � )r� rD �
ValueErrorr� r� r r r
�test_sendto_connected_addrf s �z9ProactorDatagramTransportTests.test_sendto_connected_addrc C s@ | j dd�}|�� | �|jd� |�dd� | �|jd� d S )N)ri r� ri r< r� )r� r r\ rj r� r� r r r
�test_sendto_closingk s
z2ProactorDatagramTransportTests.test_sendto_closingc C sL | � � }d|_|�� | �|j� t�| j� | jj �
� | j j �
d � d S rO )
r� rP rZ r, ra r r+ r r! r r0 r rp r� r r r
�test__loop_writing_closingr s z9ProactorDatagramTransportTests.test__loop_writing_closingc C sH t � }| jj_| �� }t�� |_|j� d� |�
� |j� |d� d S )N�r< r r� ) r� r r� rJ r� r r rK r] r� rZ r0 )r# rM r r r r
� test__loop_writing_exception{ s
�z;ProactorDatagramTransportTests.test__loop_writing_exceptionc C sB t | jj_| �� }t�� |_|j� d� |�
� | � |jj � d S )Nr� )
r� r r� rJ r� r r rK r] r� rZ r6 r8 r� r r r
�!test__loop_writing_error_received� s
z@ProactorDatagramTransportTests.test__loop_writing_error_receivedc C sV t | jj_| jdd�}t�� |_|j� d� |�
� | � |jj � | �
| jjj � d S )Nr� r� r� )r� r rd rJ r� r r rK r] r� rZ r6 r8 rF r r� r� r r r
�,test__loop_writing_error_received_connection� s
zKProactorDatagramTransportTests.test__loop_writing_error_received_connectionz asyncio.base_events.logger.errorc C s8 | j dd�}t� }|�|� | �| jjj� |�� d S )Nr� r� )r� r� rK r6 r r� r8 r~ )r# Zm_excr rM r r r
�test_fatal_error_connected� s
z9ProactorDatagramTransportTests.test_fatal_error_connected)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� � s,
r� c s� e Zd Z� fdd�Ze�d�dd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Ze�d�dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Z� ZS )-�BaseProactorEventLoopTestsc
s� t � �� t�� | _t�� | _t�� t�� | _| _ tj
d| j| j fd��( t�
d�� t | j�| _ W 5 Q R X W 5 Q R X | �
| j � d S )N�)asyncio.proactor_events.socket.socketpair)re �signal.set_wakeup_fd)r r r Zmock_nonblocking_socketr! r r r �ssock�csockr� r r �set_event_loopr" r$ r r
r � s
� z BaseProactorEventLoopTests.setUpr� c C sp t �� t �� f \}}|_t �d�� t| j�}W 5 Q R X | �|j|� | �|j|� | � |j
d� |� � d S )Nr� ri ) r r re r� r r ZassertIs�_ssock�_csockr\ �
_internal_fdsr )r# �
socketpairr� r� r r r r
r4 � s � z$BaseProactorEventLoopTests.test_ctorc C sb | j �� | �| j jd� | �| jjj� | �| jjj� | � | j j
� | � | j j � d| j _ d S )Nr T)
r �_close_self_piper\ r� rF r� r r8 r� r, r� r� �_closedr" r r r
�test_close_self_pipe� s
z/BaseProactorEventLoopTests.test_close_self_pipec C sn t �� | j_| j�� | �| jjj� | �| jjj� | �| jj � | jj�
� | j�� | � | jjj� d S r
) r r r r� r rF r8 r r, r ru r6 r" r r r
rv � s
z%BaseProactorEventLoopTests.test_closec C s, | j �| jt�� �}| �|t� t|� d S r
)r �_make_socket_transportr! r r r� r r r: r r r
�test_make_socket_transport� s z5BaseProactorEventLoopTests.test_make_socket_transportc C s6 | j �� | jj�| jd� | jjjj�| j j� d S �Ni )r �_loop_self_readingr r/ r0 r� re rf r" r r r
�test_loop_self_reading� s
�z1BaseProactorEventLoopTests.test_loop_self_readingc C sV t �� }|| j_| j�|� | �|jj� | jj �
| j d� | jj j j
�
| jj� d S r� )r r r Z_self_reading_futurer� rF r- r8 r r/ r0 r� re rf )r# r2 r r r
�test_loop_self_reading_fut� s �z5BaseProactorEventLoopTests.test_loop_self_reading_futc C s6 t �� | j_t� | jj_| j�� | � | jjj
� d S r
) r r r r� rW r r/ rJ r� rF r8 r" r r r
� test_loop_self_reading_exception� s
z;BaseProactorEventLoopTests.test_loop_self_reading_exceptionc C s | j �� | jj�d� d S )N� )r Z_write_to_selfr� rd r0 r" r r r
�test_write_to_self� s
z-BaseProactorEventLoopTests.test_write_to_selfc C s | j �g � d S r
)r Z_process_eventsr" r r r
�test_process_events� s z.BaseProactorEventLoopTests.test_process_eventsry c C s� t �� }t �� }| j_| j�|| j� | �|j� |jd d }|� | j j
� | j� t �� }t �� t �� f|j _
t �� }| j_||� | �|j j� | �|j� t� |j _||� | �| jjj� | �|jj� d S �Nr )r r r � call_soon�_start_servingr! rF r8 � call_argsr �acceptr0 r- re r� rW rJ r rz )r# rl �pfr r r2 Zmake_trr r r
�test_create_server� s"
z-BaseProactorEventLoopTests.test_create_serverc C sd t �� }t �� }| j_| j�|| j� |jd d }| j�� }|�� ||� | � | jj
j � d S r ) r r r r r r! r r* r| rF r r8 )r# r r r r2 r r r
�test_create_server_cancel s
z4BaseProactorEventLoopTests.test_create_server_cancelc C s� t �� }t �� }t �� }t �� }|�� ||�� |i| j_| j�|� | �|jj� | �|j j� | j
j� |� | � |jj� | � |j j� d S r
)
r r r� r Z_accept_futuresZ
_stop_servingrF r r8 r| r r0 r6 )r# Zsock1Zfuture1Zsock2Zfuture2r r r
�test_stop_serving s � z,BaseProactorEventLoopTests.test_stop_servingc C s t �tj�| _| j�| j| j�S r
)r r r r� r r Z_make_datagram_transportr! r" r r r
r� 1 s z-BaseProactorEventLoopTests.datagram_transportc C s | � � }| �|t� t|� d S r
)r� r� r r r: r r r
� test_make_datagram_transport5 s z7BaseProactorEventLoopTests.test_make_datagram_transportc C sV | � � }|j�d� |�� | jjjj| jddd� | jjjj j
�|j� t |� d S )N�r< �� 127.0.0.1i$/ r< r
r� ) r� r] �
appendleftrZ r r r� r0 r! re rf r r: r r r
�test_datagram_loop_writing: s �z5BaseProactorEventLoopTests.test_datagram_loop_writingc C sP | � � }|�� | jjj�| jd� | �| jj j
� | �| jj j
� t |� d S )N� )
r� r5 r r �recvfromr0 r! r6 r �datagram_receivedr8 r� r r: r r r
�test_datagram_loop_readingD s z5BaseProactorEventLoopTests.test_datagram_loop_readingc C s\ | j �� }|�d� | �� }||_|�|� | j jj�| j d� | j
j �dd� t |� d S )Nr r r< r
)
r r* r= r� r> r5 r r r0 r! r r r r? r r r
� test_datagram_loop_reading_dataL s
z:BaseProactorEventLoopTests.test_datagram_loop_reading_datac C s� | j �� }|�d� | �� }| �t|j|� t�� |_ ||_
|�|� | � | j j j
j� | �| jjj� | �|j j� t|� d S )N)rC r
)r r* r= r� rD rE r5 r r r r> rF r r r8 r6 r r� r r? r r r
�"test_datagram_loop_reading_no_dataW s
z=BaseProactorEventLoopTests.test_datagram_loop_reading_no_datac C sR t � }| jjj_| �� }t�� |_t�� |j _
|� � |j j
� |� t
|� d S r
)rI r r r rJ r� r r rK � _protocolr� r5 r0 r rL r r r
�"test_datagram_loop_reading_abortedf s
z=BaseProactorEventLoopTests.test_datagram_loop_reading_abortedc C s^ t � }| jjj_| �� }t�� |_t�� |j _
|j � d� |�
� |j j
�|� t|� d S )N)s Hellor
)rI r r r� rJ r� r r rK r r� r] r rZ r0 r rL r r r
�"test_datagram_loop_writing_abortedp s
z=BaseProactorEventLoopTests.test_datagram_loop_writing_aborted)r� r� r� r r r� r4 r� rv r� r� r� r� r r r r r
r� r r r r r r r r� r r r$ r
r� � s.
r� �win32z%Proactor is supported on Windows onlyc s� e Zd Zdd ZG dd� dej�Ze� fdd��Ze� fdd��Z � fd d
�Z
dd d
�Z dd� Z dd� Z
dd� Zdd� Zdd� Z� ZS )�&ProactorEventLoopUnixSockSendfileTestss� 12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) z.ProactorEventLoopUnixSockSendfileTests.MyProtoc C s( d| _ d| _t� | _|�� | _d | _d S rR )�started�closedrc rY r* r2 r )r# r r r r
�__init__� s
z7ProactorEventLoopUnixSockSendfileTests.MyProto.__init__c C s d| _ || _d S rO )r r r� r r r
r. � s z>ProactorEventLoopUnixSockSendfileTests.MyProto.connection_madec C s | j �|� d S r
)rY �extend)r# rY r r r
r7 � s z<ProactorEventLoopUnixSockSendfileTests.MyProto.data_receivedc C s d| _ | j�d � d S rO )r r2 r= )r# �excr r r
rp � s z>ProactorEventLoopUnixSockSendfileTests.MyProto.connection_lostc � s | j I d H d S r
)r2 r" r r r
� wait_closed� s z:ProactorEventLoopUnixSockSendfileTests.MyProto.wait_closedN)r� r� r� r r. r7 rp r" r r r r
�MyProto� s
r# c s2 t tjd��}|�| j� W 5 Q R X t� �� d S )N�wb)�openr �TESTFNr[ �DATAr �
setUpClass)�cls�fpr$ r r
r( � s z1ProactorEventLoopUnixSockSendfileTests.setUpClassc s t �t j� t� �� d S r
)r �unlinkr& r �
tearDownClass)r) r$ r r
r, � s z4ProactorEventLoopUnixSockSendfileTests.tearDownClassc sN t �� | _| �| j� | �| jj� ttjd�| _ | �| j j� t
� � � d S )N�rb) r ZProactorEventLoopr r� r r r% r r& �filer r r" r$ r r
r � s
z,ProactorEventLoopUnixSockSendfileTests.setUpTc C sR t � t jt j�}|�d� |�t jt jd� |�t jt jd� |rN| �|j � |S )NFr )
r �AF_INET� SOCK_STREAM� setblocking�
setsockopt�
SOL_SOCKET� SO_SNDBUF� SO_RCVBUFr r )r# �cleanupr! r r r
� make_socket� s
z2ProactorEventLoopUnixSockSendfileTests.make_socketc C s | j �|�S r
)r Zrun_until_complete)r# �coror r r
�run_loop� s z/ProactorEventLoopUnixSockSendfileTests.run_loopc s� �� � }���j�� t�� }�j dd�}|�d|f� ���jj� fdd�|d������j�||� � �� � ��fdd�}��
|� |� fS ) NF)r6 r c s � S r
r r )�protor r
�<lambda>� rC z@ProactorEventLoopUnixSockSendfileTests.prepare.<locals>.<lambda>)r! c s< � j d k r"� j �� ��� �� � ��� ����� � d S r
)r r r9 r" r �r: r# Zserverr r
r6 � s
z?ProactorEventLoopUnixSockSendfileTests.prepare.<locals>.cleanup) r7 r# r r Zfind_unused_port�bindr9 �
create_serverZ sock_connect� getsocknamer )r# r! �portZsrv_sockr6 r r< r
�prepare� s
�
z.ProactorEventLoopUnixSockSendfileTests.preparec C sZ | � � \}}t� }| �tjd�� | �| j�||dd �� W 5 Q R X | �| j �
� d� d S �N�not a regular filer ) rA r� �assertRaisesRegexr �SendfileNotAvailableErrorr9 r �_sock_sendfile_nativer\ r. �tell�r# r! r: r� r r r
� test_sock_sendfile_not_a_file� s � �zDProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_a_filec C s\ | � � \}}t�� }| �tjd�� | �| j�||dd �� W 5 Q R X | � | j
� � d� d S rB ) rA �io�BytesIOrD r rE r9 r rF r\ r. rG rH r r r
�test_sock_sendfile_iobuffer� s � �zBProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_iobufferc C sd | � � \}}t�� }d|j_| �tjd�� | �| j �
||dd �� W 5 Q R X | � | j �
� d� d S )N���rC r )rA r r r� re rD r rE r9 r rF r\ r. rG rH r r r
�#test_sock_sendfile_not_regular_file� s � �zJProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_regular_file)T)r� r� r� r' r r r# � classmethodr( r, r r7 r9 rA rI rL rN r� r r r$ r
r | s
r �__main__) �__doc__rJ r Zunittest�sysr r Zasyncio.proactor_eventsr r r r r �testr Ztest.test_asyncior r r r ZTestCaser r� r� ZskipIf�platformr r� �mainr r r r
�<module> s: > K X �n