File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_proactor_events.cpython-310.pyc
o
�i � � @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd l
mZ dd
l
mZ dd lmZ d d
� Zdd� ZG dd� dej�ZG dd� dej�ZG dd� dej�Ze�ejdkd�G dd� dej��Ze dkr�e� � dS dS )z Tests for proactor_events.py� N)�mock)�BaseProactorEventLoop)�_ProactorSocketTransport)�_ProactorWritePipeTransport)� _ProactorDuplexPipeTransport)�_ProactorDatagramTransport)� os_helper)�
socket_helper)�utilsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r
r
�C/usr/local/lib/python3.10/test/test_asyncio/test_proactor_events.py�tearDownModule 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 sz e Zd Z� fdd�ZdXdd�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.� Zd/d0� Ze�d1�d2d3� �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+dRdS� Z,dTdU� Z-dVdW� Z.� Z/S )Y� ProactorSocketTransportTestsc s\ t � �� | �� | _| �| jj� t�� | _| j| j_ t
� t j
�| _t�tj�| _d| _d S )Ni )�super�setUp�
new_test_loop�loop�
addCleanupr r �Mock�proactor� _proactor�
test_utils�make_test_protocolr �Protocol�protocol�socket�sock� buffer_size��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 sX | j �� }| j|d�}t�| j � | �|�� � | j�|� | j j
� | j t
| j�� d S r( )r �
create_futurer* r � run_briefly� assertIsNone�resultr �connection_mader � recv_into�assert_called_withr" � bytearrayr# �r% �fut�trr
r
r � test_ctor3 s
z&ProactorSocketTransportTests.test_ctorc C sN | � � }|�� | jjj�| jt| j�� | � | j
j j � | � | j
j
j � d S r )r* �
_loop_readingr r r0 r1 r" r2 r# � assertFalser �
data_received�called� eof_received�r% r5 r
r
r �test_loop_reading; s
z.ProactorSocketTransportTests.test_loop_readingc C s� d}| j �� }|�t|�� | �� }||_||jd t|��< |�|� t| j �}||d t|��< | j j
j � | j
|� | jj� t|�� d S �N� data)r r+ �
set_result�lenr* � _read_fut�_datar7 r2 r# r r0 r1 r" r r9 )r% �buf�resr5 Z
called_bufr
r
r �test_loop_reading_dataB s
z3ProactorSocketTransportTests.test_loop_reading_datac C sz | j �� }|�d� | �� }| �t|j|� t�� |_ ||_
|�|� | � | j j j
j� | �| jjj� | �|j j� d S �Nr )r r+ r@ r* � assertRaises�AssertionErrorr7 r r r rB r8 r r0 r: �
assertTruer r; �r% rE r5 r
r
r �test_loop_reading_no_dataP s
z6ProactorSocketTransportTests.test_loop_reading_no_datac C �> t � }| jjj_| �� }t�� |_|� � |j�
|d� d S �Nz"Fatal read error on pipe transport) �ConnectionAbortedErrorr r r0 � side_effectr* r r � _fatal_errorr7 r1 �r% �errr5 r
r
r �test_loop_reading_aborted^ �
�z6ProactorSocketTransportTests.test_loop_reading_abortedc C �@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S �NT)
rO r r r0 rP r* �_closingr r rQ r7 r8 r: r< r
r
r �!test_loop_reading_aborted_closingh s
z>ProactorSocketTransportTests.test_loop_reading_aborted_closingc C rV �NF)
rO r r r0 rP r* rX r r rQ r7 rJ r: r< r
r
r �"test_loop_reading_aborted_is_fatalq s
z?ProactorSocketTransportTests.test_loop_reading_aborted_is_fatalc C sZ t � }| jjj_| �� }d|_t�� |_ t�� |_
|� � | � |j j
� |j
�|� d S rZ )�ConnectionResetErrorr r r0 rP r* rX r r rQ � _force_closer7 r8 r: r1 rR r
r
r �!test_loop_reading_conn_reset_losty s
z>ProactorSocketTransportTests.test_loop_reading_conn_reset_lostc C rM rN ) �OSErrorr r r0 rP r* r r rQ r7 r1 rR r
r
r �test_loop_reading_exception� rU z8ProactorSocketTransportTests.test_loop_reading_exceptionc C s< | � � }t�� |_|�d� | �|jd � |jjdd� d S )Nr? )�data)r* r r �
_loop_writing�write� assertEqual�_bufferr1 r< r
r
r �
test_write� s
z'ProactorSocketTransportTests.test_writec C s" | � � }|�d� | �|j� d S )N� )r* rc r8 re r< r
r
r �test_write_no_data� s
z/ProactorSocketTransportTests.test_write_no_datac C sF | � � }t�� |_t�� |_|�d� | �|jd� | �|jj � d S r> )
r* r r �
_write_futrb rc rd re r8 r: 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* r2 re rb r r �sendr1 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.)r_ r r rk rP r* r r rQ re rb r1 �
_conn_lostrc rd �warning)r% �m_logrS r5 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* ri rb r- r3 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 �Nrp )
r r+ r@ r* ri r rb r- r r, r �connection_lostr1 r3 r
r
r �test_loop_writing_closing� s
z6ProactorSocketTransportTests.test_loop_writing_closingc C s* | � � }t�� |_|�� |j�d � d S r )r* r r r] �abortr1 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 rv )r* r r r, r r rw r1 rJ �
is_closingrd rq �
reset_mockr8 r: 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 ri r r r, r r8 r rw r: r< r
r
r �test_close_write_fut� s
z1ProactorSocketTransportTests.test_close_write_futc C s8 | � � }dg|_|�� t�| j� | �| jjj � d S r> )
r* re r r r, r r8 r rw r: r< r
r
r �test_close_buffer� s
z.ProactorSocketTransportTests.test_close_bufferc C sH | � � }d| jj_|�� t�| j� | jj �
d � | � | jj j
� d S )N���)r* r" �filenorl r r r, r r rw r1 r8 �shutdownr: r< r
r
r �test_close_invalid_sockobj� s
z7ProactorSocketTransportTests.test_close_invalid_sockobj�asyncio.base_events.loggerc C s< | � � }t�� |_|�d � | �|jj� | �|jj� d S r )r* r r r] rQ rJ r: �error)r% Z m_loggingr5 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? rp )r* re r r rB ri r] �cancelr1 r r, r r rw rd rq )r% r5 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 )Nrp r? )r r r Zset_exception_handlerr+ r@ r rk rl r* rc r] r r, �assert_not_called)r% Z exc_handlerr4 r5 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 rW )
r* rX r] r r, r rJ r rw r: r< r
r
r �test_force_close_idempotent s
z8ProactorSocketTransportTests.test_force_close_idempotentc C sT | � � }| �| jjj� d|_|�d � |�d � t�| j � | �
d| jjj � d S )NTrp ) r* r8 r rw r: rX r] r r, r rd �
call_countr< r
r
r �.test_force_close_protocol_connection_lost_once+ s
zKProactorSocketTransportTests.test_force_close_protocol_connection_lost_oncec C sJ | � � }| �| jjj� |�� |�� t�| j� | � d| jjj
� d S rv ) r* r8 r rw r: r r r, r rd r� r< r
r
r �(test_close_protocol_connection_lost_once6 s zEProactorSocketTransportTests.test_close_protocol_connection_lost_oncec C sF | � � }dg|_|�d � t�| j� | jj�d � | � d |j� d S r> )
r* re r] r r, r r rw r1 rd 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_lostrJ r rw r: r" r r< r
r
r �test_call_connection_lostI s
z6ProactorSocketTransportTests.test_call_connection_lostc C sT | � � }| �|�� � |�� | jj�tj� |�� | � | jjj
d� |� � d S rv ) r* rJ �
can_write_eof� write_eofr" r� r1 r! �SHUT_WRrd r� r r< r
r
r �test_write_eofO 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 rk rl rc r� rJ Z _eof_writtenr8 r" r� r: r1 r@ � _run_oncer! r� r �r% r5 �fr
r
r �test_write_eof_bufferX s
z2ProactorSocketTransportTests.test_write_eof_bufferc C s\ t | j| j| j�}| �|�� � |�� | �|�� � | j�� | �| jj j
� |� � d S r ) r r r" r rJ r� r� r{ r� r r: r< r
r
r �test_write_eof_write_pipef 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 rk rl rc r� rJ r{ r8 r� r: r1 r@ r� r r� r
r
r � test_write_eof_buffer_write_pipep s
z=ProactorSocketTransportTests.test_write_eof_buffer_write_pipec C s^ t | j| j| j�}| �|�� � | �t�� |�� W d � n1 s$w Y t |� d S r )
r r r" r r8 r� rH �NotImplementedErrorr� r r< r
r
r �test_write_eof_duplex_pipe s �
� z7ProactorSocketTransportTests.test_write_eof_duplex_pipec s� �� � }d}g d�}tt|��� � �fdd�}|�jjj_�j�� ��|j � ��
|� � � |d d� D ]}�j�� �j j
�t|�� q5|�� |�� ��
|j � ��|� � � td�D ]}�j�� q_�j j
�t|d �� |�� |�� ��|j � ��
|� � � |dd� D ]}�j�� �j j
�t|�� q�|�� |�� t�� �j_�j�� �jj�� �j j
�t|d �� |�� ��|� � � d S ) Nr )� data1� data2s data3s data4s data5rg c s@ �j �� }��� �|j�� ��fdd�}||_|�t��� |S )Nc s �� d t ���< �� S r )rA r
�ra �msgr. r
r �monkey� s zYProactorSocketTransportTests.test_pause_resume_reading.<locals>.recv_into.<locals>.monkey)r r+ �popr. r@ rA )r" ra r� r� �Z
reversed_msgsr% r� r r0 � s
zIProactorSocketTransportTests.test_pause_resume_reading.<locals>.recv_into� �
rp r� )r* �list�reversedr r r0 rP r� r8 Z_pausedrJ Z
is_readingr r9 r1 r2 Z
pause_reading�rangeZresume_readingr r �call_exception_handlerr� r )r% r5 �indexZmsgsr0 r� �ir
r� r �test_pause_resume_reading� sD
z6ProactorSocketTransportTests.test_pause_resume_readingc C sH | � � }|j|d� | �|�� d� | �| jjj� | �| jjj� |S )N��highr ) r* Zset_write_buffer_limitsrd �get_write_buffer_sizer8 r �
pause_writingr: �resume_writing)r% r� r5 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 rk rl rc r� rJ r r� r: r@ rd r� r� �r% r5 r4 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 rk rl rc r� rd r� r8 r r� r: rJ )r% r5 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� � 1rp 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 rk rl rc r� rd r� r8 r r� r: r� r
r
r �test_dont_pause_writing s
z4ProactorSocketTransportTests.test_dont_pause_writingr )0�__name__�
__module__� __qualname__r r* r6 r= rF rL rT rY r[ r^ r` rf rh rj rn r �patchrt ru rx rz 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 ! sZ
8 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" r� rl r$ r& r
r r s
z$ProactorDatagramTransportTests.setUpNc C s8 |rd nt | jj_t| j| j| j|d�}| �t|� |S )N��address) r_ r" � getpeernamerP 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� �sendtorJ r r: r1 r" �r% ra r r
r
r � test_sendto! s
�z*ProactorDatagramTransportTests.test_sendtoc C �F t d�}| �� }|�|d� | �| jjj� | jjj| jddd� d S r� )r2 r� r� rJ r r: r1 r" r� r
r
r �test_sendto_bytearray) �
�z4ProactorDatagramTransportTests.test_sendto_bytearrayc C r� r� )�
memoryviewr� r� rJ r r: r1 r" r� r
r
r �test_sendto_memoryview1 r� z5ProactorDatagramTransportTests.test_sendto_memoryviewc C sH | � � }|j�d� |�dd� | �| jjj� | �dgt|j�� d S )N)r? �r� i90 rg r
) r� re �appendr� r8 r" r: rd r� �r% r r
r
r �test_sendto_no_data9 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� re r� �objectri r� r8 r r: rd r� r� r
r
r �test_sendto_bufferA s ��z1ProactorDatagramTransportTests.test_sendto_bufferc C �p t d�}| �� }|j�d� t� |_|�|d� | �| jjj � | �
ddgt |j�� | � |jd d t
� d S �Nr� r� r� r� rp r )r2 r� re r� r� ri r� r8 r r: rd r� �assertIsInstance�bytes�r% Zdata2r r
r
r � test_sendto_buffer_bytearrayL � ��z;ProactorDatagramTransportTests.test_sendto_buffer_bytearrayc C r� r� )r� r� re r� r� ri r� r8 r r: rd r� r� r� r� r
r
r � test_sendto_buffer_memoryviewY r� z<ProactorDatagramTransportTests.test_sendto_buffer_memoryviewro c C s� d}t � }| jj_| �� }t�� |_|�|d� | �|jj � |j�
|d� d|_ d|_ |�|� |�|� |�|� |�|� |�|� |j
�
d� d S )Nr? r
�'Fatal write error on datagram transportrp )Z123z!socket.sendto() raised exception.)� RuntimeErrorr r� rP r� r r rQ rJ r: r1 rq �_addressrr )r% rs ra rS r r
r
r �test_sendto_exceptionf s$
�
z4ProactorDatagramTransportTests.test_sendto_exceptionc C sL d}t | jj_| �� }t�� |_|�|d� | �|j d� | �
|jj � d S )Nr? r
r ) �ConnectionRefusedErrorr" r� rP r� r r rQ rd rq r8 r: r� r
r
r �test_sendto_error_received} s
z9ProactorDatagramTransportTests.test_sendto_error_receivedc C sP d}t | jj_| jdd�}t�� |_|�|� | � |jj
� | � | j j
j
� d S )Nr? �r� rp r� )r� r rk rP r� r r rQ r� r8 r: rJ r �error_receivedr� r
r
r �$test_sendto_error_received_connected� s
zCProactorDatagramTransportTests.test_sendto_error_received_connectedc C s | � � }| �t|jdd� d S )N�strr
)r� rH � TypeErrorr� r� r
r
r �test_sendto_str� s z.ProactorDatagramTransportTests.test_sendto_strc C s" | j dd�}| �t|jdd� d S )Nr� r� s str)r� r� )r� rH �
ValueErrorr� r� r
r
r �test_sendto_connected_addr� s
�z9ProactorDatagramTransportTests.test_sendto_connected_addrc C s@ | j dd�}|�� | �|jd� |�dd� | �|jd� d S )N)rp r� rp r? r� )r� r rd rq r� r� r
r
r �test_sendto_closing� s
z2ProactorDatagramTransportTests.test_sendto_closingc C sL | � � }d|_|�� | �|j� t�| j� | jj �
� | j j �
d � d S rW )
r� rX rb r- ri r r, r r" r r1 r rw r� r
r
r �test__loop_writing_closing� s z9ProactorDatagramTransportTests.test__loop_writing_closingc C sH t � }| jj_| �� }t�� |_|j� d� |�
� |j� |d� d S )N�r? r
r� ) r� r r� rP r� r r rQ re r� rb r1 )r% rS 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� rP r� r r rQ re r� rb r8 r: 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 rk rP r� r r rQ re r� rb r8 r: rJ 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� rQ r8 r r� r: r� )r% Zm_excr rS r
r
r �test_fatal_error_connected� s
z9ProactorDatagramTransportTests.test_fatal_error_connectedr )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 d � n1 s<w Y W d � n1 sKw Y | �
| j � d S )N�)asyncio.proactor_events.socket.socketpair)rl �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 s� t �� t �� f \}}|_t �d��
t| j�}W d � n1 s"w Y | �|j|� | �|j|� | � |j
d� |� � d S )Nr� rp ) r r rl r� r r ZassertIs�_ssock�_csockrd �
_internal_fdsr )r% �
socketpairr r r r
r
r r6 � 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_piperd r rJ r r r: 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 rJ r: r r- r r| r8 r$ r
r
r r} � 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 �recvr1 r rl rm 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
rJ r. r: r r r1 r rl rm )r% r4 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� r_ r r rP r
rJ r: 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 rk r1 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' � z.BaseProactorEventLoopTests.test_process_eventsr� 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 rG )r r r � call_soon�_start_servingr" rJ r: � call_argsr �acceptr1 r. rl r
r_ rP r r� )r% rs �pfr r r4 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 rG ) r r r r r r" r r+ r� rJ r r: )r% r r r r4 r
r
r �test_create_server_cancelF 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_servingrJ r r: r� r r1 r8 )r% Zsock1Zfuture1Zsock2Zfuture2r
r
r �test_stop_servingS 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� d s z-BaseProactorEventLoopTests.datagram_transportc C s. | � � }| �|t� | �|tj� t|� d S r )r� r� r r ZDatagramTransportr r< r
r
r � test_make_datagram_transporth 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� re �
appendleftrb r r r� r1 r" rl rm r r< r
r
r �test_datagram_loop_writingn s
z5BaseProactorEventLoopTests.test_datagram_loop_writingc C sP | � � }|�� | jjj�| jd� | �| jj j
� | �| jj j
� t |� d S )N� )
r� r7 r r �recvfromr1 r" r8 r �datagram_receivedr: r� r r< r
r
r �test_datagram_loop_readingx 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� rB r7 r r% r1 r" r r& r rK r
r
r � test_datagram_loop_reading_data� 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)rg r )r r+ r@ r� rH rI r7 r r r rB rJ r r% r: r8 r r� r rK r
r
r �"test_datagram_loop_reading_no_data� s
z=BaseProactorEventLoopTests.test_datagram_loop_reading_no_datac C sR t � }| jjj_| �� }t�� |_t�� |j _
|� � |j j
� |� t
|� d S r )rO r r r% rP r� r r rQ � _protocolr� r7 r1 r rR r
r
r �"test_datagram_loop_reading_aborted� 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 )rO r r r� rP r� r r rQ r* r� re r" rb r1 r rR r
r
r �"test_datagram_loop_writing_aborted� s
z=BaseProactorEventLoopTests.test_datagram_loop_writing_aborted)r� r� r� r r r� r6 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� � s0
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 rZ )�started�closedr2 ra r+ r4 r )r% r r
r
r �__init__� s
z7ProactorEventLoopUnixSockSendfileTests.MyProto.__init__c C s d| _ || _d S rW )r0 r r� r
r
r r/ � s
z>ProactorEventLoopUnixSockSendfileTests.MyProto.connection_madec C s | j �|� d S r )ra �extend)r% ra r
r
r r9 � r z<ProactorEventLoopUnixSockSendfileTests.MyProto.data_receivedc C s d| _ | j�d � d S rW )r1 r4 r@ )r% �excr
r
r rw � s z>ProactorEventLoopUnixSockSendfileTests.MyProto.connection_lostc � s �| j I d H d S r )r4 r$ r
r
r � wait_closed� s �z:ProactorEventLoopUnixSockSendfileTests.MyProto.wait_closedN)r� r� r� r2 r/ r9 rw r5 r
r
r
r �MyProto� s r6 c sF t tjd��}|�| j� W d � n1 sw Y t� �� d S )N�wb)�openr �TESTFNrc �DATAr �
setUpClass)�cls�fpr& r
r r; � s �z1ProactorEventLoopUnixSockSendfileTests.setUpClassc s t �t j� t� �� d S r )r �unlinkr9 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 r8 r r9 �filer r r$ r& r
r r � s
z,ProactorEventLoopUnixSockSendfileTests.setUpTc C sR t � t jt j�}|�d� |�t jt jd� |�t jt jd� |r'| �|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)rI r! c s � S r r
r
)�protor
r �<lambda>� s z@ProactorEventLoopUnixSockSendfileTests.prepare.<locals>.<lambda>)r" c s< � j d ur� j �� ��� �� � ��� ����� � d S r )r r rL r5 r
�rM r% Zserverr
r rI � s
z?ProactorEventLoopUnixSockSendfileTests.prepare.<locals>.cleanup) rJ r6 r r Zfind_unused_port�bindrL �
create_serverZ sock_connect� getsocknamer )r% r" �portZsrv_sockrI r
rO r �prepare� s
�
z.ProactorEventLoopUnixSockSendfileTests.preparec C sn | � � \}}t� }| �tjd�� | �| j�||dd �� W d � n1 s'w Y | �| j �
� d� d S �N�not a regular filer ) rT r� �assertRaisesRegexr �SendfileNotAvailableErrorrL r �_sock_sendfile_nativerd rA �tell�r% r" rM r� r
r
r � test_sock_sendfile_not_a_file s �� �zDProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_a_filec C sp | � � \}}t�� }| �tjd�� | �| j�||dd �� W d � n1 s(w Y | � | j
� � d� d S rU ) rT �io�BytesIOrW r rX rL r rY rd rA rZ r[ r
r
r �test_sock_sendfile_iobuffer s �� �zBProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_iobufferc C sx | � � \}}t�� }d|j_| �tjd�� | �| j �
||dd �� W d � n1 s,w Y | � | j �
� d� d S )Nr� rV r )rT r r r� rl rW r rX rL r rY rd rA rZ r[ r
r
r �#test_sock_sendfile_not_regular_file s �� �zJProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_regular_file)T)r� r� r� r: r r r6 � classmethodr; r? r rJ rL rT r\ r_ r` r� r
r
r& r r. � s
r. �__main__) �__doc__r] r! Zunittest�sysr r Zasyncio.proactor_eventsr r r r r Z test.supportr r Ztest.test_asyncior
r r r ZTestCaser r� r� ZskipIf�platformr. r� �mainr
r
r
r �<module> s@ p K Y�n �