Your IP : 216.73.216.213
3
� f
� � @ sh d Z ddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z y ddl
Z
W n ek
rx dZ
Y nX ddlZddlZddlZddlZddlZddlZddlmZ ddlZej dkr�ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd
l m Z m Z m!Z! ydd l"m#Z# W n" ek
�rZ dd lm$Z# Y nX d
d� Z%dd� Z&e j!d�Z'e j!d�Z(dTdUdVd]d d d dfdhd$d%�
Z)G d&d'� d'ej*�Z+G d(d)� d)e+�Z,G d*d+� d+ej-�Z.G d,d-� d-ej*�Z/G d.d/� d/ej0�Z1G d0d1� d1ej2�Z3G d2d3� d3�Z4G d4d5� d5�Z5ej dk�rXG d6d7� d7e4e j6�Z7G d8d9� d9e4e5e j6�Z8n�dd:lm9Z9 G d;d<� d<e4�Z:e;e9d=��r�G d>d?� d?e:e5e j6�Z<e;e9d@��r�G dAdB� dBe:e5e j6�Z=e;e9dC��r�G dDdE� dEe:e5e j6�Z>G dFd7� d7e:e5e j6�Z7dGdH� Z?G dIdJ� dJe j6�Z@G dKdL� dLej6�ZAG dMdN� dNej6�ZBG dOdP� dPej6�ZCG dQdR� dRej6�ZDeEdSk�rdejF� dS )izTests for events.py.� N)�mock�win32)�
coroutines)�proactor_events)�selector_events)�sslproto)�
test_utils)�ONLYKEY�ONLYCERT� data_file)�support)� test_supportc C s6 t jdkrdS tj� d } ttt| jd���} | dk S )z4Return True if the platform is Mac OS 10.4 or older.�darwinFr �.�
� )r r )�sys�platformZmac_ver�tuple�map�int�split)�version� r �B/opt/alt/python36/lib64/python3.6/test/test_asyncio/test_events.py� osx_tiger* s
r c C s& dd� } t j� }t j|� |j| � �S )Nc � s dS )N�hellor r r r r �doit4 s z8_test_get_event_loop_new_process__sub_proc.<locals>.doit)�asyncio�new_event_loop�set_event_loop�run_until_complete)r �loopr r r �*_test_get_event_loop_new_process__sub_proc3 s
r# z keycert3.pemz pycacert.pem�)http://testca.pythontest.net/testca/ocsp/�0http://testca.pythontest.net/testca/pycacert.cer�2http://testca.pythontest.net/testca/revocation.crl� countryName�XY�organizationName� Python Software Foundation CA�
commonName�
our-ca-serverzJul 7 14:23:16 2028 GMTzAug 29 14:23:16 2018 GMTZCB2D80995A69525C� localityName�Castle Anthrax�Python Software Foundation� localhost�DNS� )
ZOCSPZ caIssuersZcrlDistributionPointsZissuerZnotAfterZ notBeforeZ serialNumberZsubjectZsubjectAltNamer c @ s> e Zd ZdZdZd dd�Zdd� Zdd� Zdd � Zd
d � Z dS )
� MyBaseProtoNc C s: d | _ d| _d| _|d k r6tj|d�| _tj|d�| _d S )N�INITIALr )r" )� transport�state�nbytesr �Future� connected�done)�selfr" r r r �__init__U s zMyBaseProto.__init__c C s6 || _ | jdkst| j��d| _| jr2| jjd � d S )Nr4 � CONNECTED)r5 r6 �AssertionErrorr9 �
set_result)r; r5 r r r �connection_made] s
zMyBaseProto.connection_madec C s* | j dkst| j ��| jt|�7 _d S )Nr= )r6 r> r7 �len)r; �datar r r �
data_receivedd s zMyBaseProto.data_receivedc C s | j dkst| j ��d| _ d S )Nr= �EOF)r6 r> )r; r r r � eof_receivedh s zMyBaseProto.eof_receivedc C s0 | j dkst| j ��d| _ | jr,| jjd � d S )Nr= rD �CLOSED)r= rD )r6 r> r: r? )r; �excr r r �connection_lostl s zMyBaseProto.connection_lost)N)
�__name__�
__module__� __qualname__r9 r: r< r@ rC rE rH r r r r r3 Q s
r3 c s e Zd Z� fdd�Z� ZS )�MyProtoc s t � j|� |jd� d S )Ns% GET / HTTP/1.0
Host: example.com
)�superr@ �write)r; r5 )� __class__r r r@ t s zMyProto.connection_made)rI rJ rK r@ �
__classcell__r r )rO r rL s s rL c @ s: e Zd ZdZd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
�MyDatagramProtoNc C s& d| _ d| _|d k r"tj|d�| _d S )Nr4 r )r" )r6 r7 r r8 r: )r; r" r r r r< | s zMyDatagramProto.__init__c C s$ || _ | jdkst| j��d| _d S )Nr4 � INITIALIZED)r5 r6 r> )r; r5 r r r r@ � s z MyDatagramProto.connection_madec C s* | j dkst| j ��| jt|�7 _d S )NrR )r6 r> r7 rA )r; rB �addrr r r �datagram_received� s z!MyDatagramProto.datagram_receivedc C s | j dkst| j ��d S )NrR )r6 r> )r; rG r r r �error_received� s z MyDatagramProto.error_receivedc C s0 | j dkst| j ��d| _ | jr,| jjd � d S )NrR rF )r6 r> r: r? )r; rG r r r rH � s z MyDatagramProto.connection_lost)N) rI rJ rK r: r< r@ rT rU rH r r r r rQ y s
rQ c @ s: e Zd ZdZd dd�Zdd� Zdd� Zdd � Zd
d � ZdS )
�MyReadPipeProtoNc C s. dg| _ d| _d | _|d k r*tj|d�| _d S )Nr4 r )r" )r6 r7 r5 r r8 r: )r; r" r r r r< � s
zMyReadPipeProto.__init__c C s, || _ | jdgks t| j��| jjd� d S )Nr4 r= )r5 r6 r> �append)r; r5 r r r r@ � s z MyReadPipeProto.connection_madec C s. | j ddgkst| j ��| jt|�7 _d S )Nr4 r= )r6 r> r7 rA )r; rB r r r rC � s z MyReadPipeProto.data_receivedc C s( | j ddgkst| j ��| j jd� d S )Nr4 r= rD )r6 r> rW )r; r r r rE � s z MyReadPipeProto.eof_receivedc C sR d| j kr| j jd� | j dddgks0t| j ��| j jd� | jrN| jjd � d S )NrD r4 r= rF )r6 rW r> r: r? )r; rG r r r rH � s
z MyReadPipeProto.connection_lost)N) rI rJ rK r: r< r@ rC rE rH r r r r rV � s
rV c @ s* e Zd ZdZddd�Zdd� Zdd� ZdS ) �MyWritePipeProtoNc C s& d| _ d | _|d k r"tj|d�| _d S )Nr4 )r" )r6 r5 r r8 r: )r; r" r r r r< � s zMyWritePipeProto.__init__c C s$ || _ | jdkst| j��d| _d S )Nr4 r= )r5 r6 r> )r; r5 r r r r@ � s z MyWritePipeProto.connection_madec C s0 | j dkst| j ��d| _ | jr,| jjd � d S )Nr= rF )r6 r> r: r? )r; rG r r r rH � s z MyWritePipeProto.connection_lost)N)rI rJ rK r: r< r@ rH r r r r rX � s
rX c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�MySubprocessProtocolc sr d| _ d | _tj� d�| _tj� d�| _� fdd�td�D �| _ddd�| _d | _ tj
� d�tj
� d�d�| _ d S )Nr4 )r" c s i | ]}t j� d �|�qS ))r" )r r8 )�.0�fd)r" r r �
<dictcomp>� s z1MySubprocessProtocol.__init__.<locals>.<dictcomp>r2 � )� � ) r6 r5 r r8 r9 � completed�range� disconnectsrB �
returncodeZEvent�got_data)r; r" r )r" r r< � s
z MySubprocessProtocol.__init__c C s0 || _ | jdkst| j��d| _| jjd � d S )Nr4 r= )r5 r6 r> r9 r? )r; r5 r r r r@ � s z$MySubprocessProtocol.connection_madec C s* | j dkst| j ��d| _ | jjd � d S )Nr= rF )r6 r> r` r? )r; rG r r r rH � s z$MySubprocessProtocol.connection_lostc C s8 | j dkst| j ��| j| |7 < | j| j� d S )Nr= )r6 r> rB rd �set)r; r[ rB r r r �pipe_data_received� s z'MySubprocessProtocol.pipe_data_receivedc C s> | j dkst| j ��|r*| j| j|� n| j| j|� d S )Nr= )r6 r> rb Z
set_exceptionr? )r; r[ rG r r r �pipe_connection_lost� s z)MySubprocessProtocol.pipe_connection_lostc C s$ | j dkst| j ��| jj� | _d S )Nr= )r6 r> r5 Zget_returncoderc )r; r r r �process_exited� s z#MySubprocessProtocol.process_exitedN) rI rJ rK r< r@ rH rf rg rh r r r r rY � s rY c s� e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zejeed �d �d!d"� �Zd#d$� Zd%d&� Zejeed'�d(�d)d*� �Zejeed+�d,�d-d.� �Zejeed+�d,�d/d0� �Z d�d2d3�Z d4d5� Z ejeed �d �d6d7� �Z d8d9� Z d1d:i fd;d<�Z!d�d=d>�Z"d�d?d@�Z#ej$e%d:kdA�dBdC� �Z&dDdE� Z'ej$e%d:kdA�ejeed �d �dFdG� ��Z(dHdI� Z)dJdK� Z*dLdM� Z+d�dNdO�Z,ej$e%d:kdA�dPdQ� �Z-e.j/dR�dSdT� �Z0dUdV� Z1dWdX� Z2dYdZ� Z3ejeed[�d\�d]d^� �Z4d_d`� Z5ejeed �d �dadb� �Z6ejeed �d �dcdd� �Z7d�dedf�Z8d�dgdh�Z9d�didj�Z:ej$e%d:kdA�dkdl� �Z;dmdn� Z<ej$e%d:kdA�ejeed �d �dodp� ��Z=dqdr� Z>ej$e%d:kdA�dsdt� �Z?dudv� Z@ej$e%d:kdA�ejeed �d �dwdx� ��ZAdydz� ZBej$e%d:kdA�d{d|� �ZCd}d~� ZDej$e%d:kdA�ejeed �d �dd�� ��ZEd�d�� ZFej$e%d:kdA�d�d�� �ZGd�d�� ZHd�d�� ZId�d�� ZJejeKjLd��d�d�� �ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQejeRjSd�kd��d�d�� �ZTejeRjSd�kd��d�d�� �ZUejeRjSd�kd��eKjVd�d��eKjWd��d�d�� ���ZXejeRjSd�kd��d�d�� �ZYejeRjSd�kd��d�d�� �ZZejeRjSd�kd��eKjVd�d��d�d�� ��Z[ejeRjSd�kd��eKjVd�d��d�d�� ��Z\d�d�� Z]d�d�� Z^d�d�� Z_d�d�� Z`d�d�� Zad�d�� Zb� ZcS )��EventLoopTestsMixinc s$ t � j� | j� | _| j| j� d S )N)rM �setUp�create_event_loopr" r )r; )rO r r rj � s
zEventLoopTestsMixin.setUpc s4 | j j� stj| j � | j� tj� t� j� d S )N) r" � is_closedr � run_brieflyZ
doCleanupsr Z
gc_collectrM �tearDown)r; )rO r r rn � s
z EventLoopTestsMixin.tearDownc s: t jdd� �� t j� �fdd��}�jt�jj|� � d S )Nc s s
d V d S )Nr r r r r �coro1 s zBEventLoopTestsMixin.test_run_until_complete_nesting.<locals>.coro1c s" �j �jj� � �jj� � � d S )N)�
assertTruer" �
is_runningr! r )ro r; r r �coro2 s zBEventLoopTestsMixin.test_run_until_complete_nesting.<locals>.coro2)r � coroutine� assertRaises� RuntimeErrorr" r! )r; rr r )ro r; r � test_run_until_complete_nesting s z3EventLoopTestsMixin.test_run_until_complete_nestingc C sX | j j� }| j jtjd| j d�� | j j� }| jd|| koFdkn || � d S )Ng�������?)r" g{�G�z�?g�������?)r" �timer! r �sleeprp )r; �t0�t1r r r �test_run_until_complete s
z+EventLoopTestsMixin.test_run_until_completec s. t j� fdd��}|� }� jt� jj|� d S )Nc 3 s$ � j j� tjd� j d�E d H d S )Ng�������?)r" )r" �stopr rx r )r; r r �cb s
z?EventLoopTestsMixin.test_run_until_complete_stopped.<locals>.cb)r rs rt ru r" r! )r; r} Ztaskr )r; r � test_run_until_complete_stopped s z3EventLoopTestsMixin.test_run_until_complete_stoppedc sv g � � �fdd�}�j jd|d� tj� }�j j� tj� }�j� dg� �jd|| koddkn || � d S )Nc s � j | � �jj� d S )N)rW r" r| )�arg)�resultsr; r r �callback% s
z5EventLoopTestsMixin.test_call_later.<locals>.callbackg�������?z hello worldg{�G�z�?g�������?)r" �
call_laterrw � monotonic� run_forever� assertEqualrp )r; r� ry rz r )r� r; r �test_call_later" s
z#EventLoopTestsMixin.test_call_laterc s> g � � �fdd�}�j j|dd� �j j� �j� dg� d S )Nc s � j | |f� �jj� d S )N)rW r" r| )Zarg1Zarg2)r� r; r r r� 3 s z4EventLoopTestsMixin.test_call_soon.<locals>.callbackr �world)r r� )r" � call_soonr� r� )r; r� r )r� r; r �test_call_soon0 s
z"EventLoopTestsMixin.test_call_soonc s� g �t j� ���fdd�� � ��fdd�}�j� t j|d�}|j� �� �jj� d� �jj� W d Q R X |j� �j �ddg� d S )Nc s$ � j | � t� �dkr �jj� d S )Nr_ )rW rA r" r| )r )r� r; r r r� ? s
z?EventLoopTestsMixin.test_call_soon_threadsafe.<locals>.callbackc s �j j� d� �j� d S )Nr )r" �call_soon_threadsafe�releaser )r� �lockr; r r �
run_in_threadD s zDEventLoopTestsMixin.test_call_soon_threadsafe.<locals>.run_in_thread)�targetr� r )
� threadingZLock�acquire�Thread�startr" r� r� �joinr� )r; r� �tr )r� r� r� r; r �test_call_soon_threadsafe; s z-EventLoopTestsMixin.test_call_soon_threadsafec sL g � � �fdd�}�j j|d� �j j|d� �j j� �j� ddg� d S )Nc s$ � j | � t� �dkr �jj� d S )Nr_ )rW rA r" r| )r )r� r; r r r� U s
zKEventLoopTestsMixin.test_call_soon_threadsafe_same_thread.<locals>.callbackr r� )r" r� r� r� r� )r; r� r )r� r; r �%test_call_soon_threadsafe_same_threadR s
z9EventLoopTestsMixin.test_call_soon_threadsafe_same_threadc C sH dd� }| j jd |d�}| j j|�\}}| j|d� | j|tj� � d S )Nc S s | t j� fS )N)r� � get_ident)r r r r �run` s z5EventLoopTestsMixin.test_run_in_executor.<locals>.runZyo)r" �run_in_executorr! r� ZassertNotEqualr� r� )r; r� �f2�resZ thread_idr r r �test_run_in_executor_ s
z(EventLoopTestsMixin.test_run_in_executorc s` d� � fdd�}dd� }| j jd |�}|j� | j j� || j _|| j _tjd� | j� � d S )NFc s d� d S )NTr )�args)�calledr r �patched_call_soonj s zJEventLoopTestsMixin.test_run_in_executor_cancel.<locals>.patched_call_soonc S s t jd� d S )Ng�������?)rw rx r r r r r� n s z<EventLoopTestsMixin.test_run_in_executor_cancel.<locals>.rung�������?) r" r� �cancel�closer� r� rw rx � assertFalse)r; r� r� r� r )r� r �test_run_in_executor_cancelg s
z/EventLoopTestsMixin.test_run_in_executor_cancelc s� t j� \�}�jd� t� � � ��fdd�}�jj�j� |� �jj|jd� t j �j� fdd�� �jj|jd� t j �j� fdd�� �jj|j
� �jj�jj � �jj � �j
� d � d S )
NFc sV y�j d�} W n tk
r" d S X | r4� j| � n �j�jj�j� �� �j� d S )Ni )�recv�BlockingIOError�extendrp r" �
remove_reader�filenor� )rB )�
bytes_read�rr; r r �reader~ s z8EventLoopTestsMixin.test_reader_callback.<locals>.readers abcc s t � �dkS )Nr2 )rA r )r� r r �<lambda>� s z:EventLoopTestsMixin.test_reader_callback.<locals>.<lambda>s defc s t � �dkS )N� )rA r )r� r r r� � s s abcdef)r �
socketpair� setblocking� bytearrayr" �
add_readerr� r� �send� run_untilr� r| r� r� )r; �wr� r )r� r� r; r �test_reader_callbacky s
z(EventLoopTestsMixin.test_reader_callbackc s� t j� \}��jd� � �fdd�}dd }� jj�j� ||� � jj� � j� jj�j� �� � j � jj�j� �� �j
� |j t |�d �}|j
� � j
||� d S )NFc s �j | � � jj� d S )N)r� r" r| )rB )r; r� r r �writer� s
z8EventLoopTestsMixin.test_writer_callback.<locals>.writer� xi r_ )r r� r� r" �
add_writerr� r� rp �
remove_writerr� r� r� rA r� )r; r� r� rB �readr )r; r� r �test_writer_callback� s
z(EventLoopTestsMixin.test_writer_callbackc C sT t | jtj�s�| jjd� |jd� | jt�� | jj| jj ||j
�� W d Q R X | jt�� | jj| jj |d�� W d Q R X | jt�� | jj| jj |d�� W d Q R X | jt�� | jj| jj
|�� W d Q R X |jd� | jj| jj ||j
�� | jj| jj |d�� | jj| jj |d��}| jj| jj |d�� |j� | j|jd�� d S )NTs GET / HTTP/1.0
i Fs HTTP/1.0 200 OK)�
isinstancer" r �BaseProactorEventLoop� set_debugr� rt �
ValueErrorr! � sock_connect�address� sock_sendall� sock_recv� sock_acceptr� rp �
startswith)r; �httpd�sockrB r r r �_basetest_sock_client_ops� s4
z-EventLoopTestsMixin._basetest_sock_client_opsc
C s, t j� �}tj� }| j||� W d Q R X d S )N)r �run_test_server�socketr� )r; r� r� r r r �test_sock_client_ops� s
z(EventLoopTestsMixin.test_sock_client_ops�AF_UNIXzNo UNIX Socketsc
C s0 t j� � }tjtj�}| j||� W d Q R X d S )N)r �run_test_unix_serverr� r� r� )r; r� r� r r r �test_unix_sock_client_ops� s
z-EventLoopTestsMixin.test_unix_sock_client_opsc C sx d }z t j � }|jd� |j� }W d |j� X t j � }|jd� | jt�� | jj| jj ||�� W d Q R X |j� d S )N� 127.0.0.1r F)r� r )
r� �bind� getsocknamer� r� rt �ConnectionRefusedErrorr" r! r� )r; r� �sr� r r r �test_sock_client_fail� s
z)EventLoopTestsMixin.test_sock_client_failc C s� t j � }|jd� |jd� |jd� t j � }|j|j� � | jj|�}| jj|�\}}| j |j
� d� | j ||j� � | j |j � |j� � |j � |j � |j � d S )NF� 127.0.0.1r r^ )r� r )
r� r� r� �listen�connectr� r" r� r! r� Z
gettimeoutZ getpeernamer� )r; Zlistener�client�f�connrS r r r �test_sock_accept� s
z$EventLoopTestsMixin.test_sock_accept�SIGKILLz
No SIGKILLc sX d� � fdd�}| j t| jjd|� | j t| jjd� | j t| jjtjd |� | j t| jjtjd � | j t| jjd|� | j t| jjd� | j t| jjd|� | j t| jjd � | j t| jjtj |� | j
| jjtj �� | jjtj |� t j
t j� tj � tj| j� fdd�� | j| jjtj �� | jtjtj �tj� | j
| jjtj �� d S )
Nr c s � d7 � d S )Nr^ r r )�caughtr r �
my_handler� s z?EventLoopTestsMixin.test_add_signal_handler.<locals>.my_handlerZboomr^ c s � S )Nr r )r� r r r� s z=EventLoopTestsMixin.test_add_signal_handler.<locals>.<lambda>���r� )rt � TypeErrorr" �add_signal_handler�remove_signal_handlerr� �signal�NSIGru r� r� �SIGINT�os�kill�getpidr r� rp r� � getsignal�default_int_handler)r; r� r )r� r �test_add_signal_handler� s< z+EventLoopTestsMixin.test_add_signal_handler�SIGALRMz
No SIGALRMc s^ d� � �fdd�}�j jtj|� tjtjdd� �j jd�j j� �j j� �j � d� d S )Nr c s � d7 � �j j� d S )Nr^ )r" r| r )r� r; r r r� ( s zLEventLoopTestsMixin.test_signal_handling_while_selecting.<locals>.my_handlerg{�G�z�?�< r^ )
r" r� r� r� � setitimer� ITIMER_REALr� r| r� r� )r; r� r )r� r; r �$test_signal_handling_while_selecting# s
z8EventLoopTestsMixin.test_signal_handling_while_selectingc sj d�d� � ��fdd�}�j jtj|f��� tjtjdd� �j jd�j j� �j j� �j � d� d S ) N�* r c s" � d7 � �j | �� �jj� d S )Nr^ )r� r" r| )r� )r� r; � some_argsr r r� 9 s zAEventLoopTestsMixin.test_signal_handling_args.<locals>.my_handlerg�������?r� r^ )r� )
r" r� r� r� r� r� r� r| r� r� )r; r� r )r� r; r� r �test_signal_handling_args4 s
z-EventLoopTestsMixin.test_signal_handling_argsTc C sv | j j|�\}}| j|tj� | j|tj� | j|j|� |rN| j|j d�� | j j|j
� | j |j d� |j
� d S )N�socknamer )r" r! �assertIsInstancer � Transport�Protocol�assertIsr5 �assertIsNotNone�get_extra_infor: �
assertGreaterr7 r� )r; �connection_fut�check_sockname�tr�prr r r �_basetest_create_connectionF s z/EventLoopTestsMixin._basetest_create_connectionc s> t j� �,}� jj� fdd�f|j�� }� j|� W d Q R X d S )Nc s t � jd�S )N)r" )rL r" r )r; r r r� T s z<EventLoopTestsMixin.test_create_connection.<locals>.<lambda>)r r� r" �create_connectionr� r )r; r� �conn_futr )r; r �test_create_connectionQ s
z*EventLoopTestsMixin.test_create_connectionc sD t � }tj� �*}� jj� fdd�|j�}� j||� W d Q R X d S )Nc s t � jd�S )N)r" )rL r" r )r; r r r� _ s zAEventLoopTestsMixin.test_create_unix_connection.<locals>.<lambda>)r r r� r" �create_unix_connectionr� r )r; r� r� r r )r; r �test_create_unix_connectionW s
z/EventLoopTestsMixin.test_create_unix_connectionc s
t j� ��}d }� jj� jj|jdtji��}xf|D ]R\}}}}}y4tj|||d�}|jd� � jj� jj ||�� W n Y q2X P q2W ds�t
d��� jj � fdd�|d�} � jj| �\}
} � j |
t
j� � j | t
j� � jj| j� � j| jd� |
j� W d Q R X d S ) N�type)�familyr �protoFzCan not create socket.c s t � jd�S )N)r" )rL r" r )r; r r r� v s zAEventLoopTestsMixin.test_create_connection_sock.<locals>.<lambda>)r� r )r r� r" r! � getaddrinfor� r� � SOCK_STREAMr� r� r> r r� r r� r� r: r� r7 r� ) r; r� r� �infosr r r �cnamer� r� r� r� r )r; r �test_create_connection_sockb s.
z/EventLoopTestsMixin.test_create_connection_sockNc C s |r| j |jd�� |r,| j||jd�� n| j |jd�� | j||jd�� |jd�}| j|t� | jt|�d|� | j|d t� | j|d t� | j|d t� |jd �}| j |� | j|j� |jd
�� | j|j � |jd�� | j|j
� |jd�� | j|j� |jd
�� d S ) Nr� �peername�peercert�cipherr2 r r^ r_ Z
ssl_object� compression) r� r� r� r� r rA �strr r r Z getpeercert)r; r� r� r r r Zsslobjr r r �check_ssl_extra_info~ s0
z(EventLoopTestsMixin.check_ssl_extra_infoc C sx | j j|�\}}| j|tj� | j|tj� | jd|jjj � k� | j
|||� | j j|j � | j |j
d� |j� d S )N�sslr )r" r! r� r r� r� rp rO rI �lowerr r: r� r7 r� )r; r� r� r r� r� r r r � _basetest_create_ssl_connection� s z3EventLoopTestsMixin._basetest_create_ssl_connectionc s� |t j� d�}� j|||� ttd�r�tjjfd d d d�� fdd� }tjd|d��,}|dd�}� j|||� � j |j
d � W d Q R X � j tj ��2}|dd�}t j
� � � j|||� W d Q R X W d Q R X � j |jjd
� d S ) N)r �Purpose)�cafile�capath�cadatac s � j | tjj� tj� S )z�
A ssl.create_default_context() replacement that doesn't enable
cert validation.
)r� r r � SERVER_AUTHr �dummy_ssl_context)Zpurposer r r )r; r r �_dummy_ssl_create_context� s zREventLoopTestsMixin._test_create_ssl_connection.<locals>._dummy_ssl_create_contextzssl.create_default_context)� side_effectTr^ ZCERTIFICATE_VERIFY_FAILED)r r r �hasattrr r r r �patchr� Z
call_countrt �SSLError�disable_logger� exception�reason) r; r� r r� r r r �m�cmr )r; r �_test_create_ssl_connection� s&
z/EventLoopTestsMixin._test_create_ssl_connectionz
No ssl modulec sN t jdd��8}tj� jj� fdd�f|j�� }� j|||jd� W d Q R X d S )NT)�use_sslc s t � jd�S )N)r" )rL r" r )r; r r r� � s z@EventLoopTestsMixin.test_create_ssl_connection.<locals>.<lambda>)r )r r� � functools�partialr" r r� r' )r; r� r r )r; r �test_create_ssl_connection� s
z.EventLoopTestsMixin.test_create_ssl_connectionc C s t j� � | j� W d Q R X d S )N)r �force_legacy_ssl_supportr+ )r; r r r �!test_legacy_create_ssl_connection� s
z5EventLoopTestsMixin.test_legacy_create_ssl_connectionc
sX t � }tjdd��:}tj� jj� fdd�|jdd�}� j||||jd� W d Q R X d S )NT)r( c s t � jd�S )N)r" )rL r" r )r; r r r� � s zEEventLoopTestsMixin.test_create_ssl_unix_connection.<locals>.<lambda>z 127.0.0.1)�server_hostname)r ) r r r� r) r* r" r r� r' )r; r� r� r r )r; r � test_create_ssl_unix_connection� s z3EventLoopTestsMixin.test_create_ssl_unix_connectionc C s t j� � | j� W d Q R X d S )N)r r, r/ )r; r r r �&test_legacy_create_ssl_unix_connection� s
z:EventLoopTestsMixin.test_legacy_create_ssl_unix_connectionc s� t j� �n}tj� }� jj� fdd�f|j�d|jd |fi�}� jj|�\}}|jj d�d }� j
||� |j � W d Q R X d S )Nc s t � jd�S )N)r" )rL r" r )r; r r r� � s zGEventLoopTestsMixin.test_create_connection_local_addr.<locals>.<lambda>�
local_addrr r� r^ ) r r� r �find_unused_portr" r r� r! r5 r� r� r� )r; r� �portr� r� r� Zexpectedr )r; r �!test_create_connection_local_addr� s
z5EventLoopTestsMixin.test_create_connection_local_addrc s� t j� �t}� jj� fdd�f|j�d|ji�}� jt��}� jj|� W d Q R X � j|j j
t
j � � j t
|j�|j j� W d Q R X d S )Nc s t � jd�S )N)r" )rL r" r )r; r r r� s zNEventLoopTestsMixin.test_create_connection_local_addr_in_use.<locals>.<lambda>r1 )r r� r" r r� rt �OSErrorr! r� r# �errno�
EADDRINUSE�assertInr �strerror)r; r� r� r& r )r; r �(test_create_connection_local_addr_in_use� s
z<EventLoopTestsMixin.test_create_connection_local_addr_in_usec s | j �G ��fdd�dt�}tj� }|jd� |jd� |j� � d�d �d�� ���fdd �}tj|d
d �}|j� |j � \}}|�d ����_ �j
�j �fd
d�||d�� �j � �j
j� |j� |jd� | j|j� � | j�jd� | j�jt��� | j��� d S )Nc s, e Zd Z� �fdd�Z� �fdd�Z� ZS )zAEventLoopTestsMixin.test_connect_accepted_socket.<locals>.MyProtoc s t � j|� �j�j� d S )N)rM rH r� r| )r; rG )rO r" r r rH s zQEventLoopTestsMixin.test_connect_accepted_socket.<locals>.MyProto.connection_lostc s t � j|� | jj�� d S )N)rM rC r5 rN )r; rB )rO �expected_responser r rC s zOEventLoopTestsMixin.test_connect_accepted_socket.<locals>.MyProto.data_received)rI rJ rK rH rC rP r )r; r" )rO r rL
s rL � 127.0.0.1r r^ s test datas rogerc sv yDt j � } �d k r �j| �} | j� � | j�� | jd��| j� W n, tk
rp } ztd|� W Y d d }~X nX d S )N�c z8Failure in client thread in test_connect_accepted_socket)r� Z wrap_socketr� �sendallr� r� � Exception�print)ZcsockrG )rS �
client_ssl�message�responser r r� s
z@EventLoopTestsMixin.test_connect_accepted_socket.<locals>.clientT)r� Zdaemon)r" c s � S )Nr r )r r r r� 4 s zBEventLoopTestsMixin.test_connect_accepted_socket.<locals>.<lambda>)r rF )r< r )r" r3 r� r� r� r� r� r� r� Zacceptr! Zconnect_accepted_socketr� r5 r� r� r� Zis_aliver� r6 r7 rA ) r; Z
server_sslrA rL Zlsockr� Zthreadr� �_r )rS rA r; r" rB r rC r � test_connect_accepted_socket s4
z0EventLoopTestsMixin.test_connect_accepted_socketc C s� t jdkr,t jdk r,t| jtj�r,tjd��t j
t j �}|j t
t� t|d�rTd|_t j|_t j
t j �}t|d�rxd|_t j|_| j||� d S )Nr r2 r z=SSL not supported with proactor event loops before Python 3.5�check_hostnameF)r2 r )r r � version_infor� r" r r� �unittest�SkipTestr �
SSLContext�PROTOCOL_SSLv23�load_cert_chainr
r r rF Z CERT_NONE� verify_moderE )r; Zserver_contextZclient_contextr r r � test_ssl_connect_accepted_socket? s
z4EventLoopTestsMixin.test_ssl_connect_accepted_socketzasyncio.base_events.socketc s� t j� fdd�����fdd�}t|�}� tjkrJdd� |D �|j� j_ndd� |D �|j� j_|�j_t j
� �j_ t j
� �j_ �jj
�fdd �|d
�}�jj|�}�j|j� d d � |jD �}�j||� d S )
Nc s> � t jkr � t jdd| |ffgS � t jdd| |ddffgS d S )Nr� � r )r� �AF_INETr
)�hostr3 r� �kw)r r r r X s
zEEventLoopTestsMixin.create_server_multiple_hosts.<locals>.getaddrinfoc s t j� | |��jd�S )N)r" )r �Taskr" )r� �kwds)r r; r r �getaddrinfo_task_ s zJEventLoopTestsMixin.create_server_multiple_hosts.<locals>.getaddrinfo_taskc S s g | ] }|d f�qS )�P r )rZ rQ r r r �
<listcomp>f s zDEventLoopTestsMixin.create_server_multiple_hosts.<locals>.<listcomp>c S s g | ]}|d ddf�qS )rV r r )rZ rQ r r r rW i s c s
t � j�S )N)rL r" r )r; r r r� m s zBEventLoopTestsMixin.create_server_multiple_hosts.<locals>.<lambda>rV c S s h | ]}|j � d �qS )r )�
getsockbyname)rZ r� r r r � <setcomp>p s zCEventLoopTestsMixin.create_server_multiple_hosts.<locals>.<setcomp>)r rs re r� rP rX r r" r r �MockZ_start_serving�
_stop_serving�
create_serverr! �
addCleanupr� �socketsr� ) r; r ZhostsZ mock_sockrU Z unique_hostsr� �serverZ server_hostsr )r r r; r � create_server_multiple_hostsV s
z0EventLoopTestsMixin.create_server_multiple_hostsc C s | j tjdddg� d S )Nz1.2.3.4z5.6.7.8)r` r� rP )r; r r r �&test_create_server_multiple_hosts_ipv4s s z:EventLoopTestsMixin.test_create_server_multiple_hosts_ipv4c C s | j tjdddg� d S )Nz::1z::2)r` r� �AF_INET6)r; r r r �&test_create_server_multiple_hosts_ipv6w s z:EventLoopTestsMixin.test_create_server_multiple_hosts_ipv6c s& t | j�� | jj� fdd�dd�}| jj|�}| jt|j�d� |jd }|j� \}}| j|d� tj� }|j d|f� |j
d� | jj� j � | jd� j � t
j| j� fd d�� | jd
� j� | j� jjd �� | jd� jjd �d � � jj� | jj� j� | jd
� j � |j� |j� d S )Nc s � S )Nr r )r r r r� } s z8EventLoopTestsMixin.test_create_server.<locals>.<lambda>z0.0.0.0r r^ z 127.0.0.1s xxxr= c s
� j dkS )Nr )r7 r )r r r r� � s r2 r� r rF )rL r" r\ r! r� rA r^ r� r� r� r> r9 r6 r r� r7 r� r5 r� r� r: )r; r� r_ r� rQ r3 r� r )r r �test_create_server{ s,
z&EventLoopTestsMixin.test_create_server� SO_REUSEPORTzNo SO_REUSEPORTc s� t | j�� | jj� fdd�dd�}| jj|�}| jt|j�d� |jd }| j|jt j
t j �� |j � t
j| j� t | j�� | jj� fdd�dddd�}| jj|�}| jt|j�d� |jd }| j|jt j
t j �� |j � d S ) Nc s � S )Nr r )r r r r� � s zCEventLoopTestsMixin.test_create_server_reuse_port.<locals>.<lambda>z0.0.0.0r r^ c s � S )Nr r )r r r r� � s T)Z
reuse_port)rL r" r\ r! r� rA r^ r� Z
getsockoptr� �
SOL_SOCKETre r� r rm rp )r; r� r_ r� r )r r � test_create_server_reuse_port� s*
z1EventLoopTestsMixin.test_create_server_reuse_portc s@ t j� � | j� fdd�� | jj|� f|�}| jj|�}|� fS )Nc s t jj� �ot j� �S )N)r� �path�exists�unlinkr )rh r r r� � s z7EventLoopTestsMixin._make_unix_server.<locals>.<lambda>)r Zgen_unix_socket_pathr] r" �create_unix_serverr! )r; �factory�kwargsr� r_ r )rh r �_make_unix_server� s
z%EventLoopTestsMixin._make_unix_serverc s� t | jd�� | j� fdd��\}}| jt|j�d� tjtj�}|j|� |j d� | jj
� j � | jd� j � t
j| j� fdd�� | jd� j� � jj� | jj
� j� | jd � j � |j� |j� d S )
N)r" c s � S )Nr r )r r r r� � s z=EventLoopTestsMixin.test_create_unix_server.<locals>.<lambda>r^ s xxxr= c s
� j dkS )Nr )r7 r )r r r r� � s r2 rF )rL r" rn r� rA r^ r� r� r� r> r! r9 r6 r r� r7 r5 r� r: )r; r_ rh r� r )r r �test_create_unix_server� s
z+EventLoopTestsMixin.test_create_unix_serverc sf t | jd�� tj� }|�D | jj� fdd�d|d�}| jtd�� | jj|� W d Q R X W d Q R X d S )N)r" c s � S )Nr r )r r r r� � s zOEventLoopTestsMixin.test_create_unix_server_path_socket_error.<locals>.<lambda>z/test)r� z3path and sock can not be specified at the same time)rL r" r� rk �assertRaisesRegexr� r! )r; r� r� r )r r �)test_create_unix_server_path_socket_error� s z=EventLoopTestsMixin.test_create_unix_server_path_socket_errorc C s, t jt j�}| jt jO _|j||� |S )N)r rJ rK �options� OP_NO_SSLv2rL )r; �certfile�keyfile�
sslcontextr r r �_create_ssl_context� s z'EventLoopTestsMixin._create_ssl_contextc
C sX | j ||�}| jj|dd|d�}| jj|�}|jd }|j� \}} | j|d� ||| fS )Nz 127.0.0.1r )r )rw r" r\ r! r^ r� r� )
r; rl rt ru rv r� r_ r� rQ r3 r r r �_make_ssl_server� s
z$EventLoopTestsMixin._make_ssl_serverc C s | j ||�}| j||d�S )N)r )rw rn )r; rl rt ru rv r r r �_make_ssl_unix_server� s z)EventLoopTestsMixin._make_ssl_unix_serverc s� t | jd�� | j� fdd�tt�\}}}| jjt||tj� d�}| jj |�\}}|j
d� | jj � j � | j d� j
� tj| j� fdd�� | j d� j� | j|||fd � � jj� | jj � j� | j d
� j
� |j� |j� d S ) N)r" c s � S )Nr r )r r r r� s z<EventLoopTestsMixin.test_create_server_ssl.<locals>.<lambda>)r s xxxr= c s
� j dkS )Nr )r7 r )r r r r� s r2 )r rF )rL r" rx r
r r r3 r r r! rN r9 r� r6 r� r7 r r5 r� r: )r; r_ rQ r3 �f_cr� r� r )r r �test_create_server_ssl� s"
z*EventLoopTestsMixin.test_create_server_sslc C s t j� � | j� W d Q R X d S )N)r r, r{ )r; r r r � test_legacy_create_server_ssl s
z1EventLoopTestsMixin.test_legacy_create_server_sslc s� t | jd�� | j� fdd�tt�\}}| jjt|tj� dd�}| jj |�\}}|j
d� | jj � j � | j d� j
� tj| j� fdd�� | j d � j� � jj� | jj � j� | j d
� j
� |j� |j� d S ) N)r" c s � S )Nr r )r r r r� ' s zAEventLoopTestsMixin.test_create_unix_server_ssl.<locals>.<lambda>rO )r r. s xxxr= c s
� j dkS )Nr )r7 r )r r r r� 2 s r2 rF )rL r" ry r
r r r3 r r r! rN r9 r� r6 r� r7 r5 r� r: )r; r_ rh rz r� r� r )r r �test_create_unix_server_ssl" s"
z/EventLoopTestsMixin.test_create_unix_server_sslc C s t j� � | j� W d Q R X d S )N)r r, r} )r; r r r �"test_legacy_create_unix_server_sslA s
z6EventLoopTestsMixin.test_legacy_create_unix_server_sslc s� t | jd�� | j� fdd�t�\}}}tjtj�}| jtjO _tj |_
t |d�rZd|_ | jj
t |||d�}tjj| jd��L tj� �, | jtjd�� | jj|� W d Q R X W d Q R X tj| j� W d Q R X | j� j� |j� d S ) N)r" c s � S )Nr r )r r r r� I s zJEventLoopTestsMixin.test_create_server_ssl_verify_failed.<locals>.<lambda>rF T)r �call_exception_handlerz (?i)certificate.verify.failed)rL r" rx �SIGNED_CERTFILEr rJ rK rr rs �
CERT_REQUIREDrM r rF r r r �objectr r" rp r! r! rm � assertIsNoner5 r� )r; r_ rQ r3 �sslcontext_clientrz r )r r �$test_create_server_ssl_verify_failedE s$
z8EventLoopTestsMixin.test_create_server_ssl_verify_failedc C s t j� � | j� W d Q R X d S )N)r r, r� )r; r r r �+test_legacy_create_server_ssl_verify_failedb s
z?EventLoopTestsMixin.test_legacy_create_server_ssl_verify_failedc s� t | jd�� | j� fdd�t�\}}tjtj�}| jtjO _tj |_
t |d�rXd|_ | jj
t ||dd�}tjj| jd��L tj� �, | jtjd �� | jj|� W d Q R X W d Q R X tj| j� W d Q R X | j� j� |j� d S )
N)r" c s � S )Nr r )r r r r� k s zOEventLoopTestsMixin.test_create_unix_server_ssl_verify_failed.<locals>.<lambda>rF TZinvalid)r r. r z (?i)certificate.verify.failed)rL r" ry r� r rJ rK rr rs r� rM r rF r r r r� r r" rp r! r! rm r� r5 r� )r; r_ rh r� rz r )r r �)test_create_unix_server_ssl_verify_failedf s&
z=EventLoopTestsMixin.test_create_unix_server_ssl_verify_failedc C s t j� � | j� W d Q R X d S )N)r r, r� )r; r r r �0test_legacy_create_unix_server_ssl_verify_failed� s
zDEventLoopTestsMixin.test_legacy_create_unix_server_ssl_verify_failedc s� t | jd�� | j� fdd�t�\}}}tjtj�}| jtjO _tj |_
|j t d� t
|d�rfd|_| jjt |||d�}tjj| jd��@ tj� �, | jtjd �� | jj|� W d Q R X W d Q R X W d Q R X � jd k r� jj� |j� d S )
N)r" c s � S )Nr r )r r r r� � s zIEventLoopTestsMixin.test_create_server_ssl_match_failed.<locals>.<lambda>)r rF T)r r z.hostname '127.0.0.1' doesn't match 'localhost')rL r" rx r� r rJ rK rr rs r� rM �load_verify_locations�
SIGNING_CAr rF r r r r� r r" rp ZCertificateErrorr! r5 r� )r; r_ rQ r3 r� rz r )r r �#test_create_server_ssl_match_failed� s*
*
z7EventLoopTestsMixin.test_create_server_ssl_match_failedc C s t j� � | j� W d Q R X d S )N)r r, r� )r; r r r �*test_legacy_create_server_ssl_match_failed� s
z>EventLoopTestsMixin.test_legacy_create_server_ssl_match_failedc s� t | jd�� | j� fdd�t�\}}tjtj�}| jtjO _tj |_
|j t d� t
|d�rdd|_| jjt ||dd�}| jj|�\}}� jj� |j� |j� | jj� j� d S ) N)r" c s � S )Nr r )r r r r� � s zJEventLoopTestsMixin.test_create_unix_server_ssl_verified.<locals>.<lambda>)r rF Tr0 )r r. )rL r" ry r� r rJ rK rr rs r� rM r� r� r rF r r! r5 r� r: )r; r_ rh r� rz r� r� r )r r �$test_create_unix_server_ssl_verified� s"
z8EventLoopTestsMixin.test_create_unix_server_ssl_verifiedc C s t j� � | j� W d Q R X d S )N)r r, r� )r; r r r �+test_legacy_create_unix_server_ssl_verified� s
z?EventLoopTestsMixin.test_legacy_create_unix_server_ssl_verifiedc s� t | jd�� | j� fdd�t�\}}}tjtj�}| jtjO _tj |_
|j t d� t
|d�rfd|_| jjt |||dd�}| jj|�\}}| j|||ftd � � jj� |j� |j� | jj� j� d S )
N)r" c s � S )Nr r )r r r r� � s zEEventLoopTestsMixin.test_create_server_ssl_verified.<locals>.<lambda>)r rF Tr0 )r r. )r r )rL r" rx r� r rJ rK rr rs r� rM r� r� r rF r r! r �PEERCERTr5 r� r: )r; r_ rQ r3 r� rz r� r� r )r r � test_create_server_ssl_verified� s&
z3EventLoopTestsMixin.test_create_server_ssl_verifiedc C s t j� � | j� W d Q R X d S )N)r r, r� )r; r r r �&test_legacy_create_server_ssl_verified� s
z:EventLoopTestsMixin.test_legacy_create_server_ssl_verifiedc s� t j| jd�� G � fdd�dt�}tjtjd�}|jtjtjd� |j d � | jj
||d�}| jj |�}|j d }| j
||� |j� \}}| j|d� tj� }|jd |f� |jd
� |j� |j� d S ) N)r" c s e Zd Z� �fdd�Z� ZS )z@EventLoopTestsMixin.test_create_server_sock.<locals>.TestMyProtoc s t � j|� �j| � d S )N)rM r@ r? )r; r5 )rO r r r r@ � s zPEventLoopTestsMixin.test_create_server_sock.<locals>.TestMyProto.connection_made)rI rJ rK r@ rP r )r )rO r � TestMyProto� s r� )r r^ �0.0.0.0r )r� z 127.0.0.1s xxx)r� r )r r8 r" rL r� r
�
setsockoptrf � SO_REUSEADDRr� r\ r! r^ r� r� r� r� r� r� ) r; r� �sock_obr� r_ r� rQ r3 r� r )r r �test_create_server_sock� s
z+EventLoopTestsMixin.test_create_server_sockc
C s� t j t jd�}|jt jt jd� |jd� | jjt|d�}| jj |�}|j
d }|j � \}}| jjt||d�}| j t
��}| jj |� W d Q R X | j|jjtj� |j� d S )N)r r^ �0.0.0.0r )r� )rQ r3 )r� r )r� r
r� rf r� r� r" r\ rL r! r^ r� rt r5 r� r# r6 r7 r� )r; r� r� r_ r� rQ r3 r& r r r � test_create_server_addr_in_use s
z2EventLoopTestsMixin.test_create_server_addr_in_usez IPv6 not supported or enabledc s< t j| jd�� G � fdd�dt�}d}xzy*tj� }| jj|d |d�}| jj|�}W nH tk
r� } z,|j t j
kr�|d7 }| j d|� w(n� W Y d d }~X q(X P q(W t j � }|j
d|f� |jd � | jj� �}|jj� |j� t j| jd�� t j t j�}|j
d
|f� |jd � | jj� �}|jj� |j� |j� d S ) N)r" c s e Zd Z� �fdd�Z� ZS )zFEventLoopTestsMixin.test_create_server_dual_stack.<locals>.TestMyProtoc s t � j|� �j| � d S )N)rM r@ r? )r; r5 )rO �f_protor r r@ s zVEventLoopTestsMixin.test_create_server_dual_stack.<locals>.TestMyProto.connection_made)rI rJ rK r@ rP r )r� )rO r r� s r� r )rQ r3 r^ r z 127.0.0.1s xxxz::1)r r8 r" rL r r2 r\ r! r5 r6 r7 ZassertGreaterEqualr� r� r� r5 r� rb ) r; r� Z try_countr3 r� r_ Zexr� r r )r� r � test_create_server_dual_stack s:
z1EventLoopTestsMixin.test_create_server_dual_stackc C s� | j jtdd�}| j j|�}|jd }|j� \}}tj� }|jd|f� |jd� |j � |j � tj� }| j
t |jd|f� |j � d S )Nz0.0.0.0r z 127.0.0.1s xxx) r" r\ rL r! r^ r� r� r� r� r� rt r� )r; r� r_ r� rQ r3 r� r r r �test_server_close? s
z%EventLoopTestsMixin.test_server_closec sn G �fdd�dt �}�jj|dd�}�jj|�\}�|jd�\}}�j|tj� �j�|� �jd�j � �j
�j |� �jj�fdd �||fd
�}�jj|�\}� �j|tj� �j� t � �jd� j � �j
� j |� |j d � t
j�j�fd d �� �jd
�j� t
j�j� fdd �� �jd� j� �j|jd�� |j� �jj� j� �jd� j � �j j� d S )Nc s* e Zd Z� �fdd�Z� fdd�Z� ZS )zNEventLoopTestsMixin.test_create_datagram_endpoint.<locals>.TestMyDatagramProtoc s t � j�jd� d S )N)r" )rM r< r" )Z
inner_self)rO r; r r r< S s zWEventLoopTestsMixin.test_create_datagram_endpoint.<locals>.TestMyDatagramProto.__init__c s$ t � j||� | jjd| |� d S )Ns resp:)rM rT r5 �sendto)r; rB rS )rO r r rT V s z`EventLoopTestsMixin.test_create_datagram_endpoint.<locals>.TestMyDatagramProto.datagram_received)rI rJ rK r< rT rP r )r; )rO r �TestMyDatagramProtoR s r� � 127.0.0.1r )r1 r� rR c s t � jd�S )N)r" )rQ r" r )r; r r r� e s zCEventLoopTestsMixin.test_create_datagram_endpoint.<locals>.<lambda>)Z remote_addrs xxxc s � j S )N)r7 r )r_ r r r� o s r2 c s � j S )N)r7 r )r� r r r� q s � rF )r� r )rQ r" �create_datagram_endpointr! r� r� r r� r� r6 r� r5 r� r r� r7 r� r� r: )r; r� �coroZ s_transportrQ r3 r5 r )r� r; r_ r � test_create_datagram_endpointQ s6
z1EventLoopTestsMixin.test_create_datagram_endpointc s� t jdkr"t� jtj�r"tjd��d }d }� jj� jj |dt
j i��}xZ|D ]F\}}}}}y(t
j
|||d�}|j d� |j
|� W n Y qLX P qLW ds�td��� jj� fd d
�|d �} � jj| �\}
} � j|
tj� � j| t� |
j� � jj| j� d S )
Nr z.UDP is not supported with proactor event loops� 127.0.0.1r r )r r r FzCan not create socket.c s t � jd�S )N)r" )rQ r" r )r; r r r� � s zHEventLoopTestsMixin.test_create_datagram_endpoint_sock.<locals>.<lambda>)r� )r� r )r r r� r" r r� rH rI r! r r� Z
SOCK_DGRAMr� r� r> r� r� r r� rQ r� r: ) r; r� Z
local_addressr r r r r r� r� r� r� r )r; r �"test_create_datagram_endpoint_sock s2
z6EventLoopTestsMixin.test_create_datagram_endpoint_sockc C sf | j � }t|tj�s&|j� | jd� | jd|j� |j� | jd|j� | j|j � | j|j
� d S )Nz#loop is not a BaseSelectorEventLoopr^ r ) rk r� r ZBaseSelectorEventLoopr� ZskipTestr� Z
_internal_fdsr� Z_csockZ_ssock)r; r" r r r �test_internal_fds� s
z%EventLoopTestsMixin.test_internal_fdsr z Don't support pipes for Windowsc s t �jd��tj� \}}tj|dd�� tj� ��fdd��}�jj|� � tj |d� t
j �j�fdd�� �j d �j
� tj |d
� t
j �j�fd d�� �j d d
g�j� �j d�j
� tj|� �jj�j� �j d d
ddg�j� �j�jjd�� d S )N)r" �rbi c 3 s^ �j j�fdd�� �E d H \} }�j|�� �j| �j� �jddg�j� �jd�j� d S )Nc s � S )Nr r )r r r r� � s zEEventLoopTestsMixin.test_read_pipe.<locals>.connect.<locals>.<lambda>r4 r= r )r" �connect_read_piper� r5 r� r6 r7 )r� �p)�pipeobjr r; r r r� � s z3EventLoopTestsMixin.test_read_pipe.<locals>.connect� 1c s
� j dkS )Nr^ )r7 r )r r r r� � s z4EventLoopTestsMixin.test_read_pipe.<locals>.<lambda>r^ s 2345c s
� j dkS )Nr )r7 r )r r r r� � s r4 r= r rD rF �pipe)rV r" r� r� �io�openr rs r! rN r r� r� r7 r6 r� r: r� r5 r� )r; �rpipe�wpiper� r )r� r r; r �test_read_pipe� s"
z"EventLoopTestsMixin.test_read_pipec s� | j � � t� d��t� d��tj� \}}tj|dd��tj|dd��tj� ����fdd��}� j |� �\}}� j
� | j dt |�� | j dt |�� �j
� �j
� d |_
d |_
d S )N)r" r� i r� c 3 sD � j �fdd���E d H \} }� j�fdd���E d H \}}| |fS )Nc s � S )Nr r )�
read_protor r r� � s zSEventLoopTestsMixin.test_unclosed_pipe_transport.<locals>.connect.<locals>.<lambda>c s � S )Nr r )� write_protor r r� � s )r� �connect_write_pipe)�read_transportrD �write_transport)r" r� �rpipeobj�wpipeobjr� r r r� � s
zAEventLoopTestsMixin.test_unclosed_pipe_transport.<locals>.connectr� )rk rV rX r� r� r� r� r rs r! r� r8 �reprZ_pipe)r; r� r� r� r� r� r )r" r� r� r� r� r � test_unclosed_pipe_transport� s
z0EventLoopTestsMixin.test_unclosed_pipe_transportr r� r� c s
t �jd��tj� \}}tj|dd�� tj� ��fdd��}�jj|� � tj |d� t
j �j�fdd�� �j d �j
� tj |d
� t
j �j�fd d�� �j d d
g�j� �j d�j
� tj|� �jj� �jj�j� �j d d
ddg�j� �j�jjd�� d S )N)r" r� r c 3 s^ �j j�fdd�� �E d H \} }�j|�� �j| �j� �jddg�j� �jd�j� d S )Nc s � S )Nr r )r r r r� � s zKEventLoopTestsMixin.test_read_pty_output.<locals>.connect.<locals>.<lambda>r4 r= r )r" r� r� r5 r� r6 r7 )r� r� )�master_read_objr r; r r r� � s z9EventLoopTestsMixin.test_read_pty_output.<locals>.connectr� c s � j S )N)r7 r )r r r r� s z:EventLoopTestsMixin.test_read_pty_output.<locals>.<lambda>r^ s 2345c s
� j dkS )Nr )r7 r )r r r r�
s r4 r= r rD rF r� )rV r" r� �openptyr� r� r rs r! rN r r� r� r7 r6 r� r5 r: r� r� )r; �master�slaver� r )r� r r; r �test_read_pty_output� s$
z(EventLoopTestsMixin.test_read_pty_outputc s6 t j� \�}tj|dd�}t| jd��| jj�fdd�|�}| jj|�\}}| j|�� | j|�j � | j
d�j � |j d� t
� � �fdd ��tj| j� �fd
d�� | j
d� � |j d � tj| j� �fd d�� | j
d
� � | j
d�j � t j�� | j�j jd�� �j j� | jj�j� | j
d�j � d S )N�wbi )r" c s � S )Nr r )r r r r� s z5EventLoopTestsMixin.test_write_pipe.<locals>.<lambda>r= r� c s t j� d�}| |7 } t| �S )Ni )r� r� rA )rB �chunk)r� r r r� ) s z3EventLoopTestsMixin.test_write_pipe.<locals>.readerc s �� �dkS )Nr^ r r )rB r� r r r� . s s 2345c s �� �dkS )Nr r r )rB r� r r r� 2 s s 12345r� rF )r� r� r� r� rX r" r� r! r� r5 r� r6 rN r� r r� r� r� r� r: )r; r� r� r� r5 r� r )rB r r� r� r �test_write_pipe s,
z#EventLoopTestsMixin.test_write_pipec s� t j� \}}|jd� tj|j� dd�}t| jd�� | jj� fdd�|�}| jj |�\}}| j
|� � | j
|� j � | j d� j
� |jd� | jj | jj|d��}| j d|� |j� | jj � j� | j d � j
� d S )
NFr� i )r" c s � S )Nr r )r r r r� H s zIEventLoopTestsMixin.test_write_pipe_disconnect_on_close.<locals>.<lambda>r= r� rF )r r� r� r� r� �detachrX r" r� r! r� r5 r� r6 rN r� r� r: )r; ZrsockZwsockr� r� r5 r� rB r )r r �#test_write_pipe_disconnect_on_close@ s
z7EventLoopTestsMixin.test_write_pipe_disconnect_on_closec s> t j� \�}tj|dd�}t| jd��| jj�fdd�|�}| jj|�\}}| j|�� | j|�j � | j
d�j � |j d� t
� � �fdd ��tj| j� �fd
d�d d � | j
d� � |j d
� tj| j� �fdd�d d � | j
d� � | j
d�j � t j�� | j�j jd�� �j j� | jj�j� | j
d�j � d S )Nr� r )r" c s � S )Nr r )r r r r� a s z4EventLoopTestsMixin.test_write_pty.<locals>.<lambda>r= r� c s t j� d�}| |7 } t| �S )Ni )r� r� rA )rB r� )r� r r r� j s z2EventLoopTestsMixin.test_write_pty.<locals>.readerc s �� �dkS )Nr^ r r )rB r� r r r� o s r )�timeouts 2345c s �� �dkS )Nr r r )rB r� r r r� t s s 12345r� rF )r� r� r� r� rX r" r� r! r� r5 r� r6 rN r� r r� r� r� r� r: )r; r� �slave_write_objr� r5 r� r )rB r� r r� r �test_write_ptyW s0
z"EventLoopTestsMixin.test_write_ptyc
s� t j� \�}t j|�}tj|� tj|dd�}t| jd��| jj �fdd�|�}| jj
|�\}}| j |�� | j |�j � | j
ddg�j� | j
d�j� tj|dd�}t| jd��| jj�fd d�|�}| jj
|�\} }| j |�� | j | �j � | j
d�j� t� � �fd
d ��| jd � tj| j� �fd
d�dd� | j
d � � | j
ddg�j� | j
d�j� t j�d� tj| j�fdd�dd� | j
ddg�j� | j
d�j� | j
d�j� | jd� tj| j� �fdd�dd� | j
d� � | j
ddg�j� | j
d�j� t j�d� tj| j�fdd�dd� | j
ddg�j� | j
d�j� | j
d�j� t j�� |j� | jj
�j� | j
ddddg�j� | j� | jj
�j� | j
d�j� d S )Nr� r )r" c s � S )Nr r )r� r r r� � s z<EventLoopTestsMixin.test_bidirectional_pty.<locals>.<lambda>r4 r= r� c s � S )Nr r )r� r r r� � s c s t j� d�}| |7 } t| �S )Ni )r� r� rA )rB r� )r� r r r� � s z:EventLoopTestsMixin.test_bidirectional_pty.<locals>.readerr� c s �� �dkS )Nr^ r r )rB r� r r r� � s r )r� � ac s
� j dkS )Nr^ )r7 r )r� r r r� � s r^ s 2345c s �� �dkS )Nr r r )rB r� r r r� � s s 12345s bcdec s
� j dkS )Nr )r7 r )r� r r r� � s r rD rF )r� r� �dup�ttyZsetrawr� r� rV r" r� r! r� r5 r� r6 r7 rX r� r� rN r r� r� r: )
r; Z
read_slaveZ write_slaveZslave_read_objZ read_connectr� r� r� Z
write_connectr� r )rB r� r� r� r� r �test_bidirectional_pty� sh
z*EventLoopTestsMixin.test_bidirectional_ptyc s� t j� \}}|jd� �jj|d�� t� dd �}|d k rD�j|j� tj � �fdd��}t
j � }tj |� �jd�}�jj
� t
j � | }�j|d� �j|j� d� �jtj� j� |d k rȈj|j� �jj|� |j� |j� d S ) NFr^ �ovc 3 sR z@y �j j� j� � E d H W n tjk
r8 d} Y nX d } W d �j j� X | S )N� cancelled)r" r� r� r �CancelledErrorr| )r� )r� r; r r �main� s
z:EventLoopTestsMixin.test_prompt_cancellation.<locals>.main)r" g�������?r� )r r� r� r" r� �getattrrp �pendingr rs rw r� rS r� Z
assertLessr� �resultrt r� r� r[ r� )r; r� r� r� r� r� r� �elapsedr )r� r; r �test_prompt_cancellation� s&
z,EventLoopTestsMixin.test_prompt_cancellationc sn � �fdd�}�j j� d�j _|�j _tj�fdd��}�j j|� � �j�j jd�j j�j jj j
d�� d S )Nc s �j jd7 _� � d S )Nr^ )r" �_run_once_counterr )�
orig_run_oncer; r r � _run_once� s z<EventLoopTestsMixin.test_timeout_rounding.<locals>._run_oncer c 3 sn � j } tjd| d�E d H tjd| d�E d H tjd| d�E d H tjd| d�E d H tjd| d�E d H d S )Ng{�G�z�?)r" g-C ��6?g���ư>g:�0�yE>g�����|�=)r" r rx )r" )r; r r �wait� s z7EventLoopTestsMixin.test_timeout_rounding.<locals>.wait� )Zclock_resolutionZselector) r" r� r� r rs r! ZassertLessEqualZ_clock_resolutionZ _selectorrO rI )r; r� r� r )r� r; r �test_timeout_rounding� s z)EventLoopTestsMixin.test_timeout_roundingc C sx | j � }dd� }tj� \}}| j|j� | j|j� |j||� |j||� |j� | j|j|�� | j|j |�� d S )Nc S s d S )Nr r r r r r�
s zCEventLoopTestsMixin.test_remove_fds_after_closing.<locals>.<lambda>)
rk r r� r] r� r� r� r� r� r� )r; r" r� r� r� r r r � test_remove_fds_after_closing s z1EventLoopTestsMixin.test_remove_fds_after_closingc C s� | j � }dd� }tj� \}}| j|j� | j|j� |j� | jt�� |j||� W d Q R X | jt�� |j||� W d Q R X d S )Nc S s d S )Nr r r r r r� s z@EventLoopTestsMixin.test_add_fds_after_closing.<locals>.<lambda>) rk r r� r] r� rt ru r� r� )r; r" r� r� r� r r r �test_add_fds_after_closing s z.EventLoopTestsMixin.test_add_fds_after_closingc sB t j� fdd��}|� j�}� jt�� � jj|� W d Q R X d S )Nc s � j j� d S )N)r" r� )r" )r; r r �
close_loop$ s zEEventLoopTestsMixin.test_close_running_event_loop.<locals>.close_loop)r rs r" rt ru r! )r; r� r� r )r; r � test_close_running_event_loop# s
z1EventLoopTestsMixin.test_close_running_event_loopc B C s� | j j� tjdd� �}dd� }|� }| j|j� | jt�� | j j� W d Q R X | jt�� tj| j d�}| j j |� W d Q R X | jt�� | j j
|� W d Q R X | jt�� | j j |� W d Q R X | jt�� | j j d|� W d Q R X | jt�� | j j
| j j� d |� W d Q R X | jt�� | j jd |� W d Q R X | jt�� | j j|� W d Q R X | jt�� | j jtj|� W d Q R X d S )Nc S s d S )Nr r r r r �test/ s z,EventLoopTestsMixin.test_close.<locals>.testc S s dS )NFr r r r r r� 3 s z0EventLoopTestsMixin.test_close.<locals>.<lambda>)r" g �?g )r" r� r rs r] rt ru r� r8 r! r� r� r� �call_atrw r� � create_taskr� r� �SIGTERM)r; r� �funcr� Zfutr r r �
test_close, s0
" z EventLoopTestsMixin.test_close)T)TN)TN)NN)N)N)N)drI rJ rK rj rn rv r{ r~ r� r� r� r� r� r� r� r� r� r� rH �
skipUnlessr r� r� r� r� r� r� r� r� r r r r
r r r' �skipIfr r+ r- r/ r0 r4 r: rE rN r r r` ra rc rd rg rn ro rq rw rx ry r{ r| r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r Z IPV6_ENABLEDr� r� r� r� r� r r r� r� �requires_mac_verZrequires_freebsd_versionr� r� r� r� r� r� r� r� r� r� r� rP r r )rO r ri � s�
!,
$
8 $
). " "
# & ( E$ ri c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e j
dd�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�SubprocessTestsMixinc C s, t jdkr| j|t� n| jtj |� d S )Nr )r r r� r r� r� r� )r; rc r r r �check_terminatedO s
z%SubprocessTestsMixin.check_terminatedc C s, t jdkr| j|t� n| jtj |� d S )Nr )r r r� r r� r� r� )r; rc r r r � check_killedV s
z!SubprocessTestsMixin.check_killedc C s� t jjt jjt�d�}| jjtjt | j�t
j |�}| jj |�\}}| j
|t � | jj |j� | jd|j� |jd�}|jd� | jj |jd j� � tj� � |j� W d Q R X | jj |j� | j|j� | jd|jd � d S )Nzecho.pyr= r s Python The Winnerr^ ) r� rh r� �dirname�__file__r" �subprocess_execr) r* rY r �
executabler! r� r9 r� r6 �get_pipe_transportrN rd r� r r" r� r` r� rc rB )r; �progr� �transpr �stdinr r r �test_subprocess_exec] s
z)SubprocessTestsMixin.test_subprocess_execc C s t jjt jjt�d�}| jjtjt | j�t
j |�}| jj |�\}}| j
|t � | jj |j� | jd|j� |jd�}|jd� | jj |jd j� � |jd j� | jd|jd � |jd� | jj |jd j� � | jd|jd � tj� � |j� W d Q R X | jj |j� | j|j � d S )Nzecho.pyr= r s Python r^ s
The Winners Python The Winner) r� rh r� r� r� r" r� r) r* rY r r� r! r� r9 r� r6 r� rN rd r� �clearrB r r" r� r` r� rc )r; r� r� r� r r� r r r �test_subprocess_interactiveq s(
z0SubprocessTestsMixin.test_subprocess_interactivec C s� | j jtjt| j �d�}| j j|�\}}| j|t� | j j|j� |jd�j � | j j|j
� | j d|j � | j
tdd� |jj� D ��� | j |jd jd�d� | j |jd d � |j � d S )
Nz echo Pythonr c s s | ]}|j � V qd S )N)r: )rZ r� r r r � <genexpr>� s z=SubprocessTestsMixin.test_subprocess_shell.<locals>.<genexpr>r^ s
s Pythonr_ r] )r" �subprocess_shellr) r* rY r! r� r9 r� r� r` r� rc rp �allrb �valuesrB �rstrip)r; r� r� r r r r �test_subprocess_shell� s z*SubprocessTestsMixin.test_subprocess_shellc C sd | j jtjt| j �dd d d d�}| j j|�\}}| j|t� | j j|j� | jd|j � |j
� d S )Nzexit 7)r� �stdout�stderr� ) r" r� r) r* rY r! r� r` r� rc r� )r; r� r� r r r r �test_subprocess_exitcode� s z-SubprocessTestsMixin.test_subprocess_exitcodec C s� | j jtjt| j �dd d d d�}| j j|�\}}| j|t� | j|jd�� | j|jd�� | j|jd�� | j j|j � | j
d|j � | j|j � � d S )Nzexit 7)r� r� r� r r^ r_ r� )
r" r� r) r* rY r! r� r� r� r` r� rc r� )r; r� r� r r r r �"test_subprocess_close_after_finish� s z7SubprocessTestsMixin.test_subprocess_close_after_finishc C s� t jjt jjt�d�}| jjtjt | j�t
j |�}| jj |�\}}| j
|t � | jj |j� |j� | jj |j� | j|j� |j� d S )Nzecho.py)r� rh r� r� r� r" r� r) r* rY r r� r! r� r9 r� r` r� rc r� )r; r� r� r� r r r r �test_subprocess_kill� s
z)SubprocessTestsMixin.test_subprocess_killc C s� t jjt jjt�d�}| jjtjt | j�t
j |�}| jj |�\}}| j
|t � | jj |j� |j� | jj |j� | j|j� |j� d S )Nzecho.py)r� rh r� r� r� r" r� r) r* rY r r� r! r� r9 Z terminater` r� rc r� )r; r� r� r� r r r r �test_subprocess_terminate� s
z.SubprocessTestsMixin.test_subprocess_terminateTz"Temporarily skipped (rhbz#1088233)c C s� t j t jt j�}z�tjjtjjt�d�}| jj t
j t | j�t
j|�}| jj|�\}}| j|t � | jj|j� |jt j� | jj|j� | jt j |j� |j� W d t j t j|� X d S )Nzecho.py)r� �SIGHUP�SIG_DFLr� rh r� r� r� r" r� r) r* rY r r� r! r� r9 Z send_signalr` r� rc r� )r; Z old_handlerr� r� r� r r r r �test_subprocess_send_signal� s
z0SubprocessTestsMixin.test_subprocess_send_signalc C s� t jjt jjt�d�}| jjtjt | j�t
j |�}| jj |�\}}| j
|t � | jj |j� |jd�}|jd� | jj |j� |j� | jd|jd � | j|jd jd�|jd � | jd|j� d S )Nzecho2.pyr s tests OUT:testr^ r_ s ERR:test)r� rh r� r� r� r" r� r) r* rY r r� r! r� r9 r� rN r` r� r� rB rp r� rc )r; r� r� r� r r� r r r �test_subprocess_stderr� s
z+SubprocessTestsMixin.test_subprocess_stderrc C s� t jjt jjt�d�}| jjtjt | j�t
j |t j
d�}| jj|�\}}| j|t � | jj|j� |jd�}| j|jd�� | j|jd�� |jd� | jj|j� | j|jd jd�|jd � | jd|jd � |j� | jd|j� d S ) Nzecho2.py)r� r r^ r_ s tests OUT:testERR:testr] ) r� rh r� r� r� r" r� r) r* rY r r� �
subprocessZSTDOUTr! r� r9 r� r� r� rN r` rp rB r� r� r� rc )r; r� r� r� r r� r r r �)test_subprocess_stderr_redirect_to_stdout� s"
z>SubprocessTestsMixin.test_subprocess_stderr_redirect_to_stdoutc C sF t jjt jjt�d�}| jjtjt | j�t
j |�}| jj |�\}}| j
|t � | jj |j� |jd�}|jd�}|jd� | jj |jd j� � | jd|jd � |j� | jj |jd � |jd� | jj |jd j� � t
jdkr�| jd |jd � n| jd
|jd � tj� � |j� W d Q R X | jj |j� | j|j � d S ) Nzecho3.pyr r^ s tests OUT:tests xxxr_ r s ERR:BrokenPipeErrors ERR:OSError) r� rh r� r� r� r" r� r) r* rY r r� r! r� r9 r� rN rd r� r� rB r� rb r r r" r` r� rc )r; r� r� r� r r� r� r r r �#test_subprocess_close_client_stream s.
z8SubprocessTestsMixin.test_subprocess_close_client_streamc c s` | j jtjt| j �dd d d dd�}| j j|�V \}}| j|t� | j j|j� | jd|j � d S )Nzexit 7T)r� r� r� Zstart_new_sessionr� )
r" r� r) r* rY r! r� r` r� rc )r; r� rD r r r r �"test_subprocess_wait_no_same_group3 s z7SubprocessTestsMixin.test_subprocess_wait_no_same_groupc s� t j� fdd��}� jt�� � jj|dd�� W d Q R X � jt�� � jj|dd�� W d Q R X � jt�� � jj|dd�� W d Q R X d S )Nc ; s � j jtjdf| �E d H d S )N�pwd)r" r� r �SubprocessProtocol)rT )r; r r r� ? s zGSubprocessTestsMixin.test_subprocess_exec_invalid_args.<locals>.connectT)�universal_newlinesi )�bufsize)�shell)r rs rt r� r" r! )r; r� r )r; r �!test_subprocess_exec_invalid_args> s z6SubprocessTestsMixin.test_subprocess_exec_invalid_argsc s� t jd � fdd� �}� jt�� � jj|ddg�� W d Q R X � jt�� � jj|dd�� W d Q R X � jt�� � jj|dd�� W d Q R X � jt�� � jj|d d
�� W d Q R X d S ) Nc ; s& | sd} � j jtj| f|�E d H d S )Nr )r" r� r r )�cmdrT )r; r r r� M s
zHSubprocessTestsMixin.test_subprocess_shell_invalid_args.<locals>.connectZlsz-lT)r
i )r F)r )N)r rs rt r� r" r! )r; r� r )r; r �"test_subprocess_shell_invalid_argsL s z7SubprocessTestsMixin.test_subprocess_shell_invalid_argsN)rI rJ rK r� r� r� r� r� r� r r r rH r� r r r r r
r r r r r r r� M s
! r� c @ s e Zd Zdd� ZdS )�SelectEventLoopTestsc C s t j� S )N)r �SelectorEventLoop)r; r r r rk c s z&SelectEventLoopTests.create_event_loopN)rI rJ rK rk r r r r r a s r c @ s� e Zd Zdd� Zej� s@dd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�ProactorEventLoopTestsc C s t j� S )N)r ZProactorEventLoop)r; r r r rk j s z(ProactorEventLoopTests.create_event_loopc C s t jd��d S )Nz need python 3.5 (ssl.MemoryBIO))rH rI )r; r r r r+ n s z1ProactorEventLoopTests.test_create_ssl_connectionc C s t jd��d S )Nz need python 3.5 (ssl.MemoryBIO))rH rI )r; r r r r{ q s z-ProactorEventLoopTests.test_create_server_sslc C s t jd��d S )Nz need python 3.5 (ssl.MemoryBIO))rH rI )r; r r r r� t s z;ProactorEventLoopTests.test_create_server_ssl_verify_failedc C s t jd��d S )Nz need python 3.5 (ssl.MemoryBIO))rH rI )r; r r r r� w s z:ProactorEventLoopTests.test_create_server_ssl_match_failedc C s t jd��d S )Nz need python 3.5 (ssl.MemoryBIO))rH rI )r; r r r r� z s z6ProactorEventLoopTests.test_create_server_ssl_verifiedc C s t jd��d S )Nz*IocpEventLoop incompatible with legacy SSL)rH rI )r; r r r r- } s z8ProactorEventLoopTests.test_legacy_create_ssl_connectionc C s t jd��d S )Nz*IocpEventLoop incompatible with legacy SSL)rH rI )r; r r r r| � s z4ProactorEventLoopTests.test_legacy_create_server_sslc C s t jd��d S )Nz*IocpEventLoop incompatible with legacy SSL)rH rI )r; r r r r� � s zBProactorEventLoopTests.test_legacy_create_server_ssl_verify_failedc C s t jd��d S )Nz*IocpEventLoop incompatible with legacy SSL)rH rI )r; r r r r� � s zAProactorEventLoopTests.test_legacy_create_server_ssl_match_failedc C s t jd��d S )Nz*IocpEventLoop incompatible with legacy SSL)rH rI )r; r r r r� � s z=ProactorEventLoopTests.test_legacy_create_server_ssl_verifiedc C s t jd��d S )Nz(IocpEventLoop does not have add_reader())rH rI )r; r r r r� � s z+ProactorEventLoopTests.test_reader_callbackc C s t jd��d S )Nz(IocpEventLoop does not have add_reader())rH rI )r; r r r �test_reader_callback_cancel� s z2ProactorEventLoopTests.test_reader_callback_cancelc C s t jd��d S )Nz(IocpEventLoop does not have add_writer())rH rI )r; r r r r� � s z+ProactorEventLoopTests.test_writer_callbackc C s t jd��d S )Nz(IocpEventLoop does not have add_writer())rH rI )r; r r r �test_writer_callback_cancel� s z2ProactorEventLoopTests.test_writer_callback_cancelc C s t jd��d S )Nz6IocpEventLoop does not have create_datagram_endpoint())rH rI )r; r r r r� � s z4ProactorEventLoopTests.test_create_datagram_endpointc C s t jd��d S )Nz(IocpEventLoop does not have add_reader())rH rI )r; r r r r� � s z4ProactorEventLoopTests.test_remove_fds_after_closingN)rI rJ rK rk r Z_is_sslproto_availabler+ r{ r� r� r� r- r| r� r� r� r� r r� r r� r� r r r r r f s$ r )� selectorsc s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )�UnixEventLoopTestsMixinc s, t � j� tj� }|j| j� tj|� d S )N)rM rj r ZSafeChildWatcherZ attach_loopr" �set_child_watcher)r; Zwatcher)rO r r rj � s
z UnixEventLoopTestsMixin.setUpc s t jd � t� j� d S )N)r r rM rn )r; )rO r r rn � s
z UnixEventLoopTestsMixin.tearDownc s. � fdd�}� j � � j� jj|� �d� d S )Nc � s* t jj� } � jj| t�I d H }| j� |S )N)�
concurrentZfuturesZProcessPoolExecutorr" r� r# Zshutdown)Zpoolr� )r; r r r� � s
zEUnixEventLoopTestsMixin.test_get_event_loop_new_process.<locals>.mainr )Zunpatch_get_running_loopr� r" r! )r; r� r )r; r � test_get_event_loop_new_process� s
z7UnixEventLoopTestsMixin.test_get_event_loop_new_process)rI rJ rK rj rn r rP r r )rO r r � s r �KqueueSelectorc s\ e Zd Zdd� Zejdd�ejej j
d�d�� fdd���Z ejdd�� fd d
��Z � Z
S ) �KqueueEventLoopTestsc C s t jtj� �S )N)r r r r )r; r r r rk � s z&KqueueEventLoopTests.create_event_loopr � Zopenbsdztest hangs on OpenBSDc s t � j� d S )N)rM r� )r; )rO r r r� � s z)KqueueEventLoopTests.test_read_pty_outputc s t � j� d S )N)rM r� )r; )rO r r r� � s z#KqueueEventLoopTests.test_write_pty)rI rJ rK rk r r� rH r� r r r� r� r� rP r r )rO r r � s
r �
EpollSelectorc @ s e Zd Zdd� ZdS )�EPollEventLoopTestsc C s t jtj� �S )N)r r r r )r; r r r rk � s z%EPollEventLoopTests.create_event_loopN)rI rJ rK rk r r r r r! � s r! � PollSelectorc @ s e Zd Zdd� ZdS )�PollEventLoopTestsc C s t jtj� �S )N)r r r r" )r; r r r rk � s z$PollEventLoopTests.create_event_loopN)rI rJ rK rk r r r r r# � s r# c @ s e Zd Zdd� ZdS )r c C s t jtj� �S )N)r r r ZSelectSelector)r; r r r rk � s z&SelectEventLoopTests.create_event_loopN)rI rJ rK rk r r r r r � s c O s d S )Nr )r� rm r r r �noop� s r$ c sh e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
j e e
jd�d�dd� �Z� ZS )� HandleTestsc s" t � j� tj� | _d| jj_d S )NT)rM rj r rZ r" � get_debug� return_value)r; )rO r r rj � s
zHandleTests.setUpc C s\ dd� }f }t j||| j�}| j|j|� | j|j|� | j|j� |j� | j |j� d S )Nc W s | S )Nr )r� r r r r� � s z)HandleTests.test_handle.<locals>.callback)
r �Handler" r� � _callback�_argsr� �
_cancelledr� rp )r; r� r� �hr r r � test_handle� s zHandleTests.test_handlec C s\ dd� }t j� | _t j� | j_tj|f | j�}|j� | jjjtj d�t j
||j d�� d S )Nc S s
t � �d S )N)r� r r r r r� s z:HandleTests.test_callback_with_exception.<locals>.callbackzException in callback.*)rB r# ZhandleZsource_traceback) r rZ r" r r r( Z_run�assert_called_withr Z MockPattern�ANY�_source_traceback)r; r� r, r r r � test_callback_with_exception s
z(HandleTests.test_callback_with_exceptionc C s( t j� }tjdd� f | j�}||d< d S )Nc S s d S )Nr r r r r r� s z1HandleTests.test_handle_weakref.<locals>.<lambda>r, )�weakref�WeakValueDictionaryr r( r" )r; Zwdr, r r r �test_handle_weakref s z HandleTests.test_handle_weakrefc C sx d| j j_tjtd| j �}tjt�\}}| jt |�d||f � |j
� | jt |�d� tj t�}tj|f | j �}| jt |�d||f � t j
tdd�}tj|d| j �}dtj|�|f }| jt |�|� t j
tdd �}tj|d| j �}d
tj|�|f }| jt |�|� tjdk�rttj}t j|�}tj|�\}}tj|f | j �}d }d
| }d|tj|�|f }| jt |�|� d S )NFr^ r_ z <Handle noop(1, 2) at %s:%s>z<Handle cancelled>z<Handle noop() at %s:%s>r2 z%^<Handle noop\(1, 2\)\(3\) at %s:%s>$)�xz'^<Handle noop\(x=1\)\(2, 3\) at %s:%s>$� z*<function HandleTests.test_handle_repr .*>z%functools.partialmethod\(%s, , \)\(\)z^<Handle %s at %s:%s>$)r^ r_ )r2 )r_ r2 )r2 r6 )r" r& r' r r( r$ r �get_function_sourcer� r� r� rs r) r* �re�escapeZ assertRegexr rG r% �test_handle_repr�
partialmethod)r; r, �filename�linenor} Zregex�methodZcb_regexr r r r: sD
z HandleTests.test_handle_reprc C s� d| j j_t}tj� jd }tjt d| j �}t
j t �\}}| j t
|�d||||f � |j� | j t
|�d||||f � |j� | j t
|�d||||f � d S )NTr^ r_ z-<Handle noop(1, 2) at %s:%s created at %s:%s>z7<Handle cancelled noop(1, 2) at %s:%s created at %s:%s>)r^ r_ )r" r& r' r� r � _getframe�f_linenor r( r$ r r7 r� r� r� )r; �create_filename�
create_linenor, r< r= r r r �test_handle_repr_debugK s$
z"HandleTests.test_handle_repr_debugc s| t j� j� }|jd� � j|� � fdd�}|jt�}||� |jt�}||� |jdt�}||� |jdt�}||� d S )NTc sB t jd�jd }� j| jt� � j| jd d d� t|df� d S )Nr^ r2 � test_handle_source_tracebackr� )r r? r@ r� r0 �listr� r� )r, r= )r; r r �check_source_tracebackj s zHHandleTests.test_handle_source_traceback.<locals>.check_source_tracebackr ) r �get_event_loop_policyr r� r r� r$ r� r� )r; r" rF r, r )r; r rD e s
z(HandleTests.test_handle_source_traceback� Coroutinez No collections.abc.Coroutinec C s8 G dd� d�}|� }d|_ | jtj|�� | jtj|�d� d|_| jtj|�d� d|_| jtj|�d� |� }| jtj|�d � |� }d|_d |_ | jtj|�d� |� }d|_d |_ d |_
| jtj|�d� |� }d |_d |_ d |_
| jtj|�t |�� d
�� |� }d |_ d |_
| jtj|�t |�� d
�� d S ) Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
zEHandleTests.test_coroutine_like_object_debug_formatting.<locals>.Coroc S s d S )Nr )r; �vr r r r� � s zJHandleTests.test_coroutine_like_object_debug_formatting.<locals>.Coro.sendc W s d S )Nr )r; rG r r r �throw� s zKHandleTests.test_coroutine_like_object_debug_formatting.<locals>.Coro.throwc S s d S )Nr )r; r r r r� � s zKHandleTests.test_coroutine_like_object_debug_formatting.<locals>.Coro.closec S s d S )Nr )r; r r r � __await__� s zOHandleTests.test_coroutine_like_object_debug_formatting.<locals>.Coro.__await__N)rI rJ rK r� rJ r� rK r r r r �Coro� s rL ZAAAzAAA()ZBBBzBBB()Tz
BBB() runningzCoro()z()) rI rp r Z iscoroutiner� r Z_format_coroutinerK �
cr_running�cr_code�cr_framer� )r; rL r� r r r �+test_coroutine_like_object_debug_formatting� s:
z7HandleTests.test_coroutine_like_object_debug_formatting)rI rJ rK rj r- r1 r4 r: rC rD rH r� r � collections�abcrP rP r r )rO r r% � s
1 r% c sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�
TimerTestsc s t � j� tj� | _d S )N)rM rj r rZ r" )r; )rO r r rj � s
zTimerTests.setUpc C s8 t j� }tj|dd� f tj� �}| jt|�t|�� d S )Nc S s dS )NFr r r r r r� � s z&TimerTests.test_hash.<locals>.<lambda>)rw r� r � TimerHandler rZ r� �hash)r; �whenr, r r r � test_hash� s
zTimerTests.test_hashc C s� dd� }d}t j� }tj|||tj� �}| j|j|� | j|j|� | j |j
� |j � | j |j
� | j
|j� | j
|j� | jttjd ||| j� d S )Nc W s | S )Nr )r� r r r r� � s z'TimerTests.test_timer.<locals>.callbackr^ r_ r2 )r^ r_ r2 )rw r� r rT r rZ r� r) r* r� r+ r� rp r� rt r> r" )r; r� r� rV r, r r r �
test_timer� s
zTimerTests.test_timerc C sV d| j j_tjdtf | j �}tjt�}| jt |�d| � |j
� | jt |�d� d S )NF�{ z&<TimerHandle when=123 noop() at %s:%s>z <TimerHandle cancelled when=123>) r" r& r' r rT r$ r r7 r� r� r� )r; r, �srcr r r �test_timer_repr� s
zTimerTests.test_timer_reprc C s� d| j j_t}tj� jd }tjdt f | j �}t
j t �\}}| j t
|�d||||f � |j� | j t
|�d||||f � d S )NTr^ rY z7<TimerHandle when=123 noop() at %s:%s created at %s:%s>zA<TimerHandle cancelled when=123 noop() at %s:%s created at %s:%s>)r" r& r' r� r r? r@ r rT r$ r r7 r� r� r� )r; rA rB r, r< r= r r r �test_timer_repr_debug� s
z TimerTests.test_timer_repr_debugc C s� dd� }t j� }tj||f | j�}tj||f | j�}| j||k � | j||k � | j||k� | j||k� | j||k� | j||k� | j||k� | j||k� | j||k� | j||k� |j� | j||k� tj||f | j�}tj|d |f | j�}| j||k � | j||k � | j||k� | j||k� | j||k� | j||k� | j||k� | j||k� | j||k� | j||k� tj|f | j�}| j t
|j |�� | j t
|j |�� d S )Nc W s | S )Nr )r� r r r r�
s z2TimerTests.test_timer_comparison.<locals>.callbackg $@)
rw r� r rT r" r� rp r� r( r� �NotImplemented�__eq__�__ne__)r; r� rV Zh1Zh2Zh3r r r �test_timer_comparison
s> z TimerTests.test_timer_comparison)
rI rJ rK rj rW rX r[ r\ r` rP r r )rO r rS � s rS c @ s e Zd Zdd� ZdS )�AbstractEventLoopTestsc C s� t j� }tj� }| jt|j� | jt|jd � | jt|j� | jt|j � | jt|j
� | jt|j � | jt|j d � | jt|j
d d � | jt|j||� | jt|jd � | jt|j� | jt|jd � | jt|j||� | jt|j|� | jt|jdd� | jt|jd� | jt|j|� | jt|j|� | jt|j|� | jt|jd|� | jt|jd� | jt|jd|� | jt|j d� | jt|j |d� | jt|j |d� | jt|j ||� | jt|j |� | jt|j!d|� | jt|j"d� | jt|j"d� | jt|j#|t j$j%� | jt|j&|t j$j%� | jt|j'|t j$� | jt|j(|� | jt|j)|� | jt|j*|� | jt|j+|� | jt|j,� | jt|j-|� d S )Nr0 � r^ r )r0 rb ).r rZ r �AbstractEventLooprt �NotImplementedErrorr� r! r| rq rl r� r� r� r� r� rw r� r� Zset_default_executorr Z getnameinfor r\ r� r� r� r� r� r� r� r� r� r� r� r� �sentinelr� r� r� r� Zset_exception_handlerZdefault_exception_handlerr r& r� )r; r� r" r r r �test_not_implemented-
s�
z+AbstractEventLoopTests.test_not_implementedN)rI rJ rK rf r r r r ra +
s ra c @ s e Zd Zdd� ZdS )�ProtocolsAbsTestsc C s� t j� }tj� }| j|j|�� | j|j|�� | j|j|�� | j|j� � tj � }| j|j|�� | j|j|�� | j|j
|�� | j|j ||�� tj � }| j|j|�� | j|j|�� | j|j
d|�� | j|jd|�� | j|j� � d S )Nr^ )r rZ r r� r� r@ rH rC rE �DatagramProtocolrU rT r rf rg rh )r; r� r� ZdpZspr r r �
test_empty�
s" z ProtocolsAbsTests.test_emptyN)rI rJ rK ri r r r r rg �
s rg c @ sf e Zd Zdd� Zdd� Zdd� Zdd� Zejd �d
d � �Z d d
� Z
dd� Z dd� Z dd� Z
dd� ZdS )� PolicyTestsc C sZ t j� }| jt|j� | jt|jt� � | jt|j� | jt|j� | jt|j t� � d S )N)
r �AbstractEventLoopPolicyrt rd �get_event_loopr r� r Zget_child_watcherr )r; �policyr r r �test_event_loop_policy�
s
z"PolicyTests.test_event_loop_policyc C sX t j� }| j|jj� |j� }| j|t j� | j|jj|� | j||j� � |j � d S )N)
r �DefaultEventLoopPolicyr� �_localZ_looprl r� rc r� r� )r; rm r" r r r �test_get_event_loop�
s z PolicyTests.test_get_event_loopc C sF t j� }tjj|d|jd��}|j� }|j|� W d Q R X |j� d S )Nr )�wraps) r ro r r r� r rl r. r� )r; rm Zm_set_event_loopr" r r r �(test_get_event_loop_calls_set_event_loop�
s z4PolicyTests.test_get_event_loop_calls_set_event_loopc C s$ t j� }|jd � | jt|j� d S )N)r ro r rt ru rl )r; rm r r r �"test_get_event_loop_after_set_none�
s
z.PolicyTests.test_get_event_loop_after_set_nonez'asyncio.events.threading.current_threadc s, � fdd�}t j|d�}|j� |j� d S )Nc s t j� } � jt| j� d S )N)r ro rt ru rl )rm )r; r r r� �
s z1PolicyTests.test_get_event_loop_thread.<locals>.f)r� )r� r� r� r� )r; Zm_current_threadr� Zthr )r; r �test_get_event_loop_thread�
s z&PolicyTests.test_get_event_loop_threadc C s* t j� }|j� }| j|t j� |j� d S )N)r ro r r� rc r� )r; rm r" r r r �test_new_event_loop�
s z PolicyTests.test_new_event_loopc C sh t j� }|j� }| jt|jt� � |j� }|j|� | j||j� � | j ||j� � |j
� |j
� d S )N) r ro rl rt r> r r� r r� � assertIsNotr� )r; rm Zold_loopr" r r r �test_set_event_loop�
s
z PolicyTests.test_set_event_loopc C s* t j� }| j|t j� | j|t j� � d S )N)r rG r� rk r� )r; rm r r r �test_get_event_loop_policy�
s z&PolicyTests.test_get_event_loop_policyc C sL | j ttjt� � tj� }tj� }tj|� | j|tj� � | j||� d S )N) rt r> r �set_event_loop_policyr� rG ro r� rw )r; �
old_policyrm r r r �test_set_event_loop_policy�
s
z&PolicyTests.test_set_event_loop_policyc s� G dd� dt j�}d � t j� }zBt j|� � t j� � �jt j� d � � �fdd�}� j|� � W d t j|� � d k r|� j� X �jt j� d � d S )Nc @ s e Zd Zdd� ZdS )zDPolicyTests.test_get_event_loop_returns_running_loop.<locals>.Policyc S s t �d S )N)rd )r; r r r rl �
s zSPolicyTests.test_get_event_loop_returns_running_loop.<locals>.Policy.get_event_loopN)rI rJ rK rl r r r r �Policy�
s r} c � s$ �j tj� � � �j tj� � � d S )N)r� r rl �_get_running_loopr )r" r; r r r� s zBPolicyTests.test_get_event_loop_returns_running_loop.<locals>.func) r ro rG rz r r� r~ r! r� )r; r} r{ r� r )r" r; r �(test_get_event_loop_returns_running_loop�
s
z4PolicyTests.test_get_event_loop_returns_running_loopN)rI rJ rK rn rq rs rt r r ru rv rx ry r| r r r r r rj �
s
rj �__main__)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 �r� )r� r� r� r� �r1 r0 )r� )G�__doc__�collections.abcrQ Zconcurrent.futuresr r) �gcr� r� r r8 r� r� r � ImportErrorr r r� rw r6 rH r r2 r� r r r r r r Zasyncio.test_utilsr r
r r� r r
r r# r� r� r� r� r3 rL rh rQ rV Z BaseProtocolrX r rY ri r� ZTestCaser r r r r r r! r# r$ r% rS ra rg rj rI r� r r r r �<module> s�
" ) f
7 NnXt