File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/test_proactor_events.cpython-37.pyc
B
��g�� � @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddlm Z ddlm Z dd l
mZ dd
lmZ d d � ZG d
d� dej�Ze�d�G dd� dej��ZG dd� dej�Ze�ejdkd�G dd� dej��Zedk�r
e� � dS )z Tests for proactor_events.py� N)�mock)�events)�BaseProactorEventLoop)�_ProactorSocketTransport)�_ProactorWritePipeTransport)� _ProactorDuplexPipeTransport)�support)�utilsc C s" | j d krd S | j �� d | _ d S )N)Z_sock�close)� transport� r �B/usr/local/lib/python3.7/test/test_asyncio/test_proactor_events.py�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 )N)�super�setUp�
new_test_loop�loop�
addCleanupr
r �Mock�proactor� _proactor�
test_utils�make_test_protocol�asyncio�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 sV t j| jd�}| j|d�}t�| j� | �|�� � | j� |� | j
j � | j
d� d S )N)r )r! i � )r �Futurer r" r � run_briefly� assertIsNone�resultr �connection_mader �recv�assert_called_withr )r �fut�trr r r
� test_ctor- s z&ProactorSocketTransportTests.test_ctorc C sH | � � }|�� | jjj�| jd� | �| jj j
� | �| jj j
� d S )Ni � ) r" �
_loop_readingr r r( r) r � assertFalser �
data_received�called� eof_received)r r+ r r r
�test_loop_reading5 s
z.ProactorSocketTransportTests.test_loop_readingc C sV t j| jd�}|�d� | �� }||_|�|� | jjj� | j
d� | j j � d� d S )N)r s datai � )
r r# r �
set_resultr" � _read_futr- r r( r) r r r/ )r �resr+ r r r
�test_loop_reading_data< s
z3ProactorSocketTransportTests.test_loop_reading_datac C s~ t j| jd�}|�d� | �� }| �t|j|� t� � |_
||_ |�|� | � | jj
jj� | �| jjj� | �|j
j� d S )N)r � )r r# r r3 r" � assertRaises�AssertionErrorr- r r r
r4 r. r r( r0 �
assertTruer r1 )r r5 r+ r r r
�test_loop_reading_no_dataF 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_errorr- r) )r �errr+ r r r
�test_loop_reading_abortedT s
z6ProactorSocketTransportTests.test_loop_reading_abortedc C s@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S )NT)
r< r r r( r= r" �_closingr r r> r- r. r0 )r r+ r r r
�!test_loop_reading_aborted_closing^ s
z>ProactorSocketTransportTests.test_loop_reading_aborted_closingc C s@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S )NF)
r< r r r( r= r" rA r r r> r- r: r0 )r r+ r r r
�"test_loop_reading_aborted_is_fatalg s
z?ProactorSocketTransportTests.test_loop_reading_aborted_is_fatalc C sZ t � }| jjj_| �� }d|_t�� |_ t�� |_
|� � | � |j j
� |j
�|� d S )NF)�ConnectionResetErrorr r r( r= r" rA r r r> � _force_closer- r. r0 r) )r r? r+ r r r
�!test_loop_reading_conn_reset_losto s
z>ProactorSocketTransportTests.test_loop_reading_conn_reset_lostc C s> t � }| jjj_| �� }t�� |_|� � |j�
|d� d S )Nz"Fatal read error on pipe transport) �OSErrorr r r( r= r" r r r> r- r) )r r? r+ r r r
�test_loop_reading_exceptionz s
z8ProactorSocketTransportTests.test_loop_reading_exceptionc C s< | � � }t�� |_|�d� | �|jd � |jjdd� d S )Ns data)�data)r" r r �
_loop_writing�write� assertEqual�_bufferr) )r r+ r r r
�
test_write� s
z'ProactorSocketTransportTests.test_writec C s" | � � }|�d� | �|j� d S )Nr7 )r" rK r. rM )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 )Ns data)
r" r r �
_write_futrJ rK rL rM r. r0 )r r+ r r r
�test_write_more� s
z,ProactorSocketTransportTests.test_write_morec C sH | � � }td�|_|�� | jjj�| jd� | jjjj j
�|j� d S )Ns data) r" � bytearrayrM rJ r r �sendr) r � return_value�add_done_callback)r r+ r r r
�test_loop_writing� s
z.ProactorSocketTransportTests.test_loop_writingz 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� s dataz socket.send() raised exception.)rG r r rS r= r" r r r> rM rJ r) �
_conn_lostrK rL Zwarning)r �m_logr? r+ r r r
�test_loop_writing_err� s
z2ProactorSocketTransportTests.test_loop_writing_errc C s@ t j| jd�}|�d� | �� }||_|�|� | �|j� d S )N)r s data)r r# r r3 r" rP rJ r% )r r* r+ r r r
�test_loop_writing_stop� s
z3ProactorSocketTransportTests.test_loop_writing_stopc C sb t j| jd�}|�d� | �� }||_|�� |�|� | �|j� t �
| j� | j j �
d � d S )N)r rW )r r# r r3 r" rP r
rJ r% r r$ r �connection_lostr) )r r* r+ r r r
�test_loop_writing_closing� s
z6ProactorSocketTransportTests.test_loop_writing_closingc C s* | � � }t�� |_|�� |j�d � d S )N)r" r r rE �abortr) )r 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 )NrW )r" r
r r$ r r r\ r) r: �
is_closingrL rX �
reset_mockr. r0 )r r+ r r r
�
test_close� s z'ProactorSocketTransportTests.test_closec C s: | � � }t�� |_|�� t�| j� | �| j j
j � d S )N) r" r r rP r
r r$ r r. r r\ r0 )r r+ r r r
�test_close_write_fut� s
z1ProactorSocketTransportTests.test_close_write_futc C s8 | � � }dg|_|�� t�| j� | �| jjj � d S )Ns data)
r" rM r
r r$ r r. r r\ r0 )r r+ r r r
�test_close_buffer� s
z.ProactorSocketTransportTests.test_close_bufferzasyncio.base_events.loggerc C s< | � � }t�� |_|�d � | �|jj� | �|jj� d S )N)r" r r rE r> r: r0 �error)r Z m_loggingr+ 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 )Ns datarW )r" rM r r r4 rP rE �cancelr) r r$ r r r\ rL rX )r r+ Zread_futZ write_futr r r
�test_force_close� s
z-ProactorSocketTransportTests.test_force_closec C sj t �� }| j�|� tj| jd�}|�d� || jj_ | �
� }|� d� |� d � t
�| j� |�� d S )N)r rW s data)r r r Zset_exception_handlerr r# r3 r rS rT r" rK rE r r$ �assert_not_called)r Z exc_handlerr* r+ 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 )NT)
r" rA rE r r$ r r. r r\ r0 )r 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 )Ns data)
r" rM rE r r$ r r r\ r) rL )r r+ r r r
�test_fatal_error_2 s
z/ProactorSocketTransportTests.test_fatal_error_2c C s6 | � � }|�d � | �| jjj� | �| jjj� d S )N)r" Z_call_connection_lostr: r r\ r0 r r
)r r+ r r r
�test_call_connection_lost s
z6ProactorSocketTransportTests.test_call_connection_lostc C sT | � � }| �|�� � |�� | jj�tj� |�� | � | jjj
d� |� � d S )NrW ) r" r: �
can_write_eof� write_eofr �shutdownr) r �SHUT_WRrL Z
call_countr
)r r+ r r r
�test_write_eof# s z+ProactorSocketTransportTests.test_write_eofc C s� | � � }tj| jd�}||jjj_|�d� |� � | �
|j � | � | j
jj� |jjj�| j
d� |�d� | j�� | j
j�tj� |�� d S )N)r s data� )r" r r# r �_loopr rS rT rK ro r: Z _eof_writtenr. r rp r0 r) r3 � _run_oncer rq r
)r r+ �fr r r
�test_write_eof_buffer, s
z2ProactorSocketTransportTests.test_write_eof_bufferc C s\ t | j| j| j�}| �|�� � |�� | �|�� � | j�� | �| jj j
� |� � d S )N) r r r r r: rn ro r` ru r
r0 )r r+ r r r
�test_write_eof_write_pipe: s
z6ProactorSocketTransportTests.test_write_eof_write_pipec C s� t | j| j| j�}tj| jd�}||jjj_ |�
d� |� � | � |�
� � | �| jjj� |jjj�| jd� |�d� | j�� | j�� | � | jjj� |�� d S )N)r s datars )r r r r r r# rt r rS rT rK ro r: r` r. rp r0 r) r3 ru r
)r r+ rv r r r
� test_write_eof_buffer_write_pipeD s
z=ProactorSocketTransportTests.test_write_eof_buffer_write_pipec C sJ t | j| j| j�}| �|�� � | �t�� |�� W d Q R X t |� d S )N)
r r r r r. rn r8 �NotImplementedErrorro r )r r+ r r r
�test_write_eof_duplex_pipeS s z7ProactorSocketTransportTests.test_write_eof_duplex_pipec C s� | � � }g }x.dD ]&}tj| jd�}|�|� |�|� qW || jjj_| j� � | �
|j � | � |�
� � | j� � | jj�d� | j� � | jj�d� |�� |�� | � |j � | �
|�
� � xtd�D ]}| j� � q�W | jj�d� |�� |�� | �
|j � | � |�
� � | j� � | jj�d� | j� � | jj�d� |�� |�� t�� | j_| j� � | jj�� | jj�d� |�� | �
|�
� � d S ) N)s data1s data2s data3s data4s data5r7 )r s data1s data2�
s data3s data4s data5)r" r r# r r3 �appendr r( r= ru r. �_pausedr: �
is_readingr r/ r) �
pause_reading�range�resume_readingr r �call_exception_handlerri r
)r r+ �futures�msgrv �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_limitsrL �get_write_buffer_sizer. r �
pause_writingr0 �resume_writing)r r� r+ r r r
�pause_writing_transport� s z4ProactorSocketTransportTests.pause_writing_transportc C s� | j dd�}tj| jd�}|| jjj_|�d� | j�� | � | j
j j � |�
d � | j�� | �|�� d� | � | j
jj � d S )Nrs )r� )r s
large datar )r� r r# r r rS rT rK ru r: r r� r0 r3 rL r� r� )r r+ r* r r r
�test_pause_resume_writing� s
z6ProactorSocketTransportTests.test_pause_resume_writingc C s� | j dd�}tj| jd�}|| jjj_|�d� | j�� | � |�
� d� | � | j j
j� |�d� | j�� | � |�
� d� | �| j j
j� d S )Nrs )r� )r s 123� s abc� )r� r r# r r rS rT rK ru rL r� r. r r� r0 r: )r r+ Zfut1r r r
�test_pause_writing_2write� s
z6ProactorSocketTransportTests.test_pause_writing_2writec C s� | j dd�}tj| jd�}|| jjj_|�d� | j�� | � |�
� d� | � | j j
j� |�d� | j�� | � |�
� d� | � | j j
j� |�d� | j�� | � |�
� d � | �| j j
j� d S )
Nrs )r� )r � 1rW s 23r� s abcr� )r� r r# r r rS rT rK ru rL r� r. r r� r0 r: )r r+ r* r r r
�test_pause_writing_3write� s
z6ProactorSocketTransportTests.test_pause_writing_3writec C sh | j dd�}tj| jd�}|�d � || jjj_|�d� | j� � | �
|� � d� | � | j
jj� d S )Nrs )r� )r s very large datar )r� r r# r r3 r rS rT rK ru rL r� r. r r� r0 )r r+ r* r r r
�test_dont_pause_writing� s
z4ProactorSocketTransportTests.test_dont_pause_writing)N)-�__name__�
__module__� __qualname__r r" r, r2 r6 r; r@ rB rC rF rH rN rO rQ rV r �patchrZ r[ r] r_ rb rc rd rf rh rj rk rl rm rr rw rx ry r{ r� r� r� r� r� r� �
__classcell__r r )r r
r sN
- r zeFIXME: bpo-33694: these tests are too close to the implementation and should be refactored or removedc s� e Zd Z� fdd�Zd'dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z e
� d�dd� �Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z� ZS )(�)ProactorSocketTransportBufferedProtoTestsc sr t � �� � �� � _� �� jj� t�� � _� j� j_ t
� t j
�� _td�� _� fdd�� jj_t�tj�� _d S )NrW c s � j S )N)�buf)�hint)r r r
�<lambda>� r7 zAProactorSocketTransportBufferedProtoTests.setUp.<locals>.<lambda>)r r r r r r
r r r r r r r �BufferedProtocolr rR r� �
get_bufferr= r r )r )r )r r
r � s
z/ProactorSocketTransportBufferedProtoTests.setUpNc C s& t | j| j| j|d�}| �t|� |S )N)r! )r r r r r r )r r! r r r r
r" � s
z:ProactorSocketTransportBufferedProtoTests.socket_transportc C sX t j| jd�}| j|d�}t�| j� | �|�� � | j� |� | j
j � | j
| j� d S )N)r )r! )r r# r r" r r$ r% r& r r' r � recv_intor) r r� )r r* r+ r r r
r, � s z3ProactorSocketTransportBufferedProtoTests.test_ctorc C sZ | � � }|�� | jjj�| j| j� | �| j j
j � | � | j j
j � | � | j jj � d S )N)r" r- r r r� r) r r� r: r r� r0 r. �buffer_updatedr1 )r r+ r r r
r2 � s z;ProactorSocketTransportBufferedProtoTests.test_loop_readingc C sd | � � }t�� |_t�� | j_t� | jj_ |�
� | � |jj � | � | jjj � | �
| jjj � d S )N)r" r r r> r r� � LookupErrorr r� r= r- r: r0 r. r� )r r r r r
�test_get_buffer_error� s
z?ProactorSocketTransportBufferedProtoTests.test_get_buffer_errorc C sf | � � }t�� |_t�� | j_dd� | jj_|� � | �
|jj � | �
| jjj � | � | jj
j � d S )Nc S s t d�S )Nr )rR )r� r r r
r� r7 zUProactorSocketTransportBufferedProtoTests.test_get_buffer_zerosized.<locals>.<lambda>)r" r r r> r r� r r� r= r- r: r0 r. r� )r r r r r
�test_get_buffer_zerosized s
zCProactorSocketTransportBufferedProtoTests.test_get_buffer_zerosizedc s� t �tj�| _| �� }tj| jd�}|�d� | �� }||_ |�
|� | jj j �
| jd� | jj�
d� t �tj�}td�� � fdd�|j_|�|� t �| j� tj| jd�}|�d� ||_ |�
|� | jj j�
| j� � |j�
d� d S )N)r s datai � rs c s � S )Nr )r� )r� r r
r� ) r7 zRProactorSocketTransportBufferedProtoTests.test_proto_type_switch.<locals>.<lambda>)r r r r r r" r# r r3 r4 r- r r( r) r r/ r� rR r� r= � set_protocolr$ r� r� )r r+ r5 � buf_protor )r� r
�test_proto_type_switch s(
z@ProactorSocketTransportBufferedProtoTests.test_proto_type_switchzbFIXME: bpo-33694: this test is too close to the implementation and should be refactored or removedc s� | � � }t�| j� | jj�d� t�tj �}t
d�� � fdd�|j_ dd� |j j
_ |�|� | �|jj� t�| j� |j�d� d S )N���rs c s � S )Nr )r� )r� r r
r� @ r7 zQProactorSocketTransportBufferedProtoTests.test_proto_buf_switch.<locals>.<lambda>c S s dS )NFr r r r r
r� A r7 )r" r r$ r r r� r) r r r� rR r= r4 Zdoner� r. r0 )r r+ r� r )r� r
�test_proto_buf_switch5 s
z?ProactorSocketTransportBufferedProtoTests.test_proto_buf_switchc C s� | � � }t�� |_t�� | j_t� | jj_ t
j | jd�}|� d� ||_
|�|� | �|jj� | �| jjj� | �| jjj� d S )N)r r| )r" r r r> r r� r� r r� r= r r# r3 r4 r- r: r0 r. r� )r r r5 r r r
�test_buffer_updated_errorG s
zCProactorSocketTransportBufferedProtoTests.test_buffer_updated_errorc C s� t j| jd�}|�d� t� | jj_| �� }t �
� |_ t �
� |_ ||_
|�|� | �| jjjj� | �| jjj� | �|j j� d S )N)r r )r r# r r3 r� r r1 r= r" r r r> r
r4 r- r. r r� r0 r: )r r5 r+ r r r
� test_loop_eof_received_errorW s
zFProactorSocketTransportBufferedProtoTests.test_loop_eof_received_errorc C sX t j| jd�}|�d� | �� }||_|�|� | jjj� | j
| j � | j j
� d� d S )N)r rs )r r# r r3 r" r4 r- r r� r) r r� r r� )r r5 r+ r r r
r6 g s
z@ProactorSocketTransportBufferedProtoTests.test_loop_reading_datac C s~ t j| jd�}|�d� | �� }| �t|j|� t� � |_
||_ |�|� | � | jj
jj� | �| jjj� | �|j
j� d S )N)r r )r r# r r3 r" r8 r9 r- r r r
r4 r. r r� r0 r: r r1 )r r5 r+ r r r
r; q s
zCProactorSocketTransportBufferedProtoTests.test_loop_reading_no_datac C s> t � }| jjj_| �� }t�� |_|� � |j�
|d� d S )Nz"Fatal read error on pipe transport) r< r r r� r= r" r r r> r- r) )r r? r+ r r r
r@ s
zCProactorSocketTransportBufferedProtoTests.test_loop_reading_abortedc C s@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S )NT)
r< r r r( r= r" rA r r r> r- r. r0 )r r+ r r r
rB � s
zKProactorSocketTransportBufferedProtoTests.test_loop_reading_aborted_closingc C s@ t � | jjj_| �� }d|_t�� |_ |�
� | � |j j � d S )NF)
r< r r r� r= r" rA r r r> r- r: r0 )r r+ r r r
rC � s
zLProactorSocketTransportBufferedProtoTests.test_loop_reading_aborted_is_fatalc C sZ t � }| jjj_| �� }d|_t�� |_ t�� |_
|� � | � |j j
� |j
�|� d S )NF)rD r r r� r= r" rA r r r> rE r- r. r0 r) )r r? r+ r r r
rF � s
zKProactorSocketTransportBufferedProtoTests.test_loop_reading_conn_reset_lostc C s> t � }| jjj_| �� }t�� |_|� � |j�
|d� d S )Nz"Fatal read error on pipe transport) rG r r r� r= r" r r r> r- r) )r r? r+ r r r
rH � s
zEProactorSocketTransportBufferedProtoTests.test_loop_reading_exceptionc C sd | � � }g }x.dD ]&}tj| jd�}|�|� |�|� qW || jjj_| j� � | �
|j � | � |�
� � | j� � | jj�d� | j� � | jj�d� |�� |�� | � |j � | �
|�
� � xtd�D ]}| j� � q�W | jj�d� |�� |�� | �
|j � | � |�
� � | j� � | jj�d� | j� � | jj�d� |�� | �
|�
� � d S )N)r| � � �( r )r r| r� r� r� )r" r r# r r3 r} r r� r= ru r. r~ r: r r r� r) r� r� r� r
)r r+ r� r� rv r� r r r
r� � s>
zCProactorSocketTransportBufferedProtoTests.test_pause_resume_reading)N)r� r� r� r r" r, r2 r� r� r� �unittest�skipr� r� r� r6 r; r@ rB rC rF rH r� r� r r )r r
r� � s$
r� c s� e Zd Z� fdd�Zej�ed�e�d�dd� ��Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Ze�d�dd� �Zdd� Zd d � Z� ZS ) �BaseProactorEventLoopTestsc sr t � �� t�� | _t�� | _t�� t�� | _| _ tj
d| j| j fd�� t | j�| _ W d Q R X | �
| j � d S )Nz)asyncio.proactor_events.socket.socketpair)rT )r r r Zmock_nonblocking_socketr r r r �ssock�csockr� r r �set_event_loop)r )r r r
r � s
z BaseProactorEventLoopTests.setUp� call_soonz)asyncio.proactor_events.socket.socketpairc C sf t �� t �� f \}}|_t| j�}| �|j|� | �|j|� | �|j d� |�
|j � |� � d S )NrW )
r r rT r r ZassertIs�_ssock�_csockrL �
_internal_fdsr) �_loop_self_readingr
)r Z
socketpairr� r� r� r r r r
r, � 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_piperL r� r: r� r
r0 r� r% r� r� Z_closed)r 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 )N) r r r r� r
r: r0 r r% r ra r. )r r r r
rb � s
z%BaseProactorEventLoopTests.test_closec C s, | j �| jt�� �}| �|t� t|� d S )N)r �_make_socket_transportr r r ZassertIsInstancer 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 r� r r( r) r� rT rU )r r r r
�test_loop_self_reading
s
z1BaseProactorEventLoopTests.test_loop_self_readingc C sN t �� }| j�|� | �|jj� | jj� | j
d� | jjj j � | jj� d S )Ni )
r r r r� r: r& r0 r r( r) r� rT rU )r r* 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 )N) r r r r� rG r r( r= r� r: r0 )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� rS r) )r r r r
�test_write_to_self s
z-BaseProactorEventLoopTests.test_write_to_selfc C s | j �g � d S )N)r Z_process_events)r r r r
�test_process_events" s z.BaseProactorEventLoopTests.test_process_eventszasyncio.base_events.loggerc 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 r� �_start_servingr r: r0 � call_argsr Zacceptr) r& rT r� rG r= r
re )r rY �pfr� r r* Zmake_trr r r
�test_create_server% s"
z-BaseProactorEventLoopTests.test_create_serverc C sh t �� }t �� }| j_| j�|| j� |jd d }tj| jd�}|� � ||� | �
| jj j � d S )Nr )r )
r r r r� r� r r� r r# rg r: r
r0 )r r� r� r r* r r r
�test_create_server_cancelA 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 )N)
r r �filenor Z_accept_futuresZ
_stop_servingr: r
r0 rg r r) r. )r Zsock1Zfuture1Zsock2Zfuture2r r r
�test_stop_servingN s z,BaseProactorEventLoopTests.test_stop_serving)r� r� r� r r r� �objectr r, r� rb 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� 12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcdei c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) z.ProactorEventLoopUnixSockSendfileTests.MyProtoc C s( d| _ d| _t� | _|�� | _d | _d S )NF)�started�closedrR rI Z
create_futurer* r )r r r r r
�__init__g s
z7ProactorEventLoopUnixSockSendfileTests.MyProto.__init__c C s d| _ || _d S )NT)r� r )r r r r r
r' n s z>ProactorEventLoopUnixSockSendfileTests.MyProto.connection_madec C s | j �|� d S )N)rI �extend)r rI r r r
r/ r s z<ProactorEventLoopUnixSockSendfileTests.MyProto.data_receivedc C s d| _ | j�d � d S )NT)r� r* r3 )r �excr r r
r\ u s z>ProactorEventLoopUnixSockSendfileTests.MyProto.connection_lostc � s | j I d H d S )N)r* )r r r r
� wait_closedy s z:ProactorEventLoopUnixSockSendfileTests.MyProto.wait_closedN)r� r� r� r� r' r/ r\ r� r r r r
�MyProtoe s
r� c s2 t tjd��}|�| j� W d Q R X t� �� d S )N�wb)�openr �TESTFNrK �DATAr �
setUpClass)�cls�fp)r r r
r� | s z1ProactorEventLoopUnixSockSendfileTests.setUpClassc s t �t j� t� �� d S )N)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 )NFi )
r ZAF_INETZ SOCK_STREAMZ setblockingZ
setsockoptZ
SOL_SOCKETZ SO_SNDBUFZ SO_RCVBUFr r
)r �cleanupr r r r
� make_socket� s
z2ProactorEventLoopUnixSockSendfileTests.make_socketc C s | j �|�S )N)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)r� z 127.0.0.1c s � S )Nr r )�protor r
r� � r7 z@ProactorEventLoopUnixSockSendfileTests.prepare.<locals>.<lambda>)r c s< � j d k r"� j �� ��� �� � ��� ����� � d S )N)r r
r� r� r )r� r �serverr r
r� � s
z?ProactorEventLoopUnixSockSendfileTests.prepare.<locals>.cleanup) r� r� r r Zfind_unused_portZbindr� Z
create_serverZ sock_connectZ getsocknamer )r r ZportZsrv_sockr� r )r� r r� r
�prepare� s
z.ProactorEventLoopUnixSockSendfileTests.preparec C sZ | � � \}}t� }| �tjd�� | �| j�||dd �� W d Q R X | �| j �
� d� d S )Nznot a regular filer ) r� r� �assertRaisesRegexr �SendfileNotAvailableErrorr� r �_sock_sendfile_nativerL r� �tell)r r r� rv r r r
� test_sock_sendfile_not_a_file� s zDProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_a_filec C s\ | � � \}}t�� }| �tjd�� | �| j�||dd �� W d Q R X | � | j
� � d� d S )Nznot a regular filer ) r� �io�BytesIOr� r r� r� r r� rL r� r� )r r r� rv r r r
�test_sock_sendfile_iobuffer� s zBProactorEventLoopUnixSockSendfileTests.test_sock_sendfile_iobufferc C sd | � � \}}t�� }d|j_| �tjd�� | �| j �
||dd �� W d Q R X | � | j �
� d� d S )Nr� znot a regular filer )r� r r r� rT r� r r� r� r r� rL r� r� )r r r� rv 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 r� r� r� r� r� r� r� r r )r r
r� ` s
r� �__main__) �__doc__r� r r� �sysr r r Zasyncio.proactor_eventsr r r r Ztestr Ztest.test_asyncior r r ZTestCaser r� r� r� ZskipIf�platformr� r� �mainr r r r
�<module> s8 >
|
n