Your IP : 216.73.216.213
U
i�f� � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddl
mZ ddlmZ e
jdkr�e
�d��ddlZddlmZ ddlmZ dd lmZ ejZd
d � Zd d
� Z e
� ed�G dd� dej ��Z e
� e ed�d�G dd� dej ��Z!e
� e ed�d�G dd� dej ��Z"G dd� dej �Z#G dd � d ej �Z$G d d � d e
j �Z%G d d � d e
j �Z&e�'d!d"�Z(G d#d$� d$�Z)G d%d&� d&e)ej �Z*G d'd(� d(e)ej �Z+G d)d*� d*e
j �Z,G d+d,� d,e
j �Z-e.d-k�r e
�/� dS ).zTests for unix_events.py.� N)�mock)�supportZwin32z UNIX only)�log)� unix_events)�utilsc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �G/opt/alt/python38/lib64/python3.8/test/test_asyncio/test_unix_events.py�tearDownModule s r c C s" | j d krd S | j �� d | _ d S r )Z_pipe�close)� transportr r r
�close_pipe_transport$ s
r zSignals are not supportedc sF e Zd Z� fdd�Zdd� Zdd� Zdd� Ze�d �d
d � �Z e�d �d d
� �Z
e�d �dd� �Z e�d �dd� �Z e�d �e�d�dd� ��Z
e�d �e�d�dd� ��Ze�d �dd� �Ze�d �dd� �Ze�d �e�d�dd � ��Ze�d �d d � �Ze�d �d d � �Ze�d �d!d"� �Ze�d#�e�d �d$d%� ��Z� ZS )&� SelectorEventLoopSignalTestsc s$ t � �� t�� | _| �| j� d S r ��super�setUpr ZSelectorEventLoop�loop�set_event_loop��self�� __class__r r
r 0 s
z"SelectorEventLoopSignalTests.setUpc C s. | � t| jjd� | � t| jjtjd � d S )N�1� )� assertRaises� TypeErrorr Z
_check_signal�
ValueError�signal�NSIGr r r r
�test_check_signal5 s � �z.SelectorEventLoopSignalTests.test_check_signalc C s | j �tjd � d S )Nr )r �_handle_signalr r r r r r
� test_handle_signal_no_handler; s z:SelectorEventLoopSignalTests.test_handle_signal_no_handlerc C sh t jt�� dt�� d�}|�� || jjtjd < t�� | j_ | j�
tjd � | jj � tjd � d S )Nr )r r ) r �Handler �Mock�cancelr �_signal_handlersr r �remove_signal_handlerr! �assert_called_with)r �hr r r
�$test_handle_signal_cancelled_handler> s � zASelectorEventLoopSignalTests.test_handle_signal_cancelled_handlerzasyncio.unix_events.signalc C s6 t j|_t j|_t|j_| �t| jj t j
dd� � d S )Nc S s dS �NTr r r r r
�<lambda>P � zRSelectorEventLoopSignalTests.test_add_signal_handler_setup_error.<locals>.<lambda>) r r �
valid_signalsr �
set_wakeup_fd� side_effectr � RuntimeErrorr �add_signal_handler�SIGINT�r �m_signalr r r
�#test_add_signal_handler_setup_errorG s �z@SelectorEventLoopSignalTests.test_add_signal_handler_setup_errorc C sP t j|_dd� }|}|� }| �|j� ||fD ] }| �td| jjt j|� q.d S )Nc � s d S r r r r r r
�simple_coroutineV s z^SelectorEventLoopSignalTests.test_add_signal_handler_coroutine_error.<locals>.simple_coroutinez1coroutines cannot be used with add_signal_handler) r r �
addCleanupr �assertRaisesRegexr r r2 r3 )r r5 r7 Z coro_funcZcoro_obj�funcr r r
�'test_add_signal_handler_coroutine_errorR s �zDSelectorEventLoopSignalTests.test_add_signal_handler_coroutine_errorc C sX t j|_t j|_dd� }| j�t j|� | jj�t j�}| �|t j
� | � |j |� d S )Nc S s dS r+ r r r r r
r, h r- zFSelectorEventLoopSignalTests.test_add_signal_handler.<locals>.<lambda>)
r r r. r r2 �SIGHUPr& �get�assertIsInstancer r# � assertEqualZ _callback)r r5 �cbr) r r r
�test_add_signal_handlerc s z4SelectorEventLoopSignalTests.test_add_signal_handlerc C sT t j|_t j|_dd� }||_G dd� dt�}||j _| �|| jjt j dd� � d S )Nc S s | dkrt � �d S )N���)r )�fdr r r
r/ s s zYSelectorEventLoopSignalTests.test_add_signal_handler_install_error.<locals>.set_wakeup_fdc @ s e Zd ZejZdS )zOSelectorEventLoopSignalTests.test_add_signal_handler_install_error.<locals>.ErrN)�__name__�
__module__� __qualname__�errnoZEFAULTr r r r
�Errx s rH c S s dS r+ r r r r r
r, r- zTSelectorEventLoopSignalTests.test_add_signal_handler_install_error.<locals>.<lambda>)
r r r. r/ �OSErrorr0 r r r2 r3 )r r5 r/ rH r r r
�%test_add_signal_handler_install_errorn s �zBSelectorEventLoopSignalTests.test_add_signal_handler_install_errorzasyncio.base_events.loggerc C sv t j|_t j|_G dd� dt�}||j _dd� | jjt j< | �t | jj
t j dd� � | � |j
j� | �d|jj� d S )Nc @ s e Zd ZejZdS )zPSelectorEventLoopSignalTests.test_add_signal_handler_install_error2.<locals>.ErrN�rD rE rF rG ZEINVALr r r r
rH � s rH c S s dS r+ r r r r r
r, � r- zUSelectorEventLoopSignalTests.test_add_signal_handler_install_error2.<locals>.<lambda>c S s dS r+ r r r r r
r, � r- r )r r r. rI r0 r r& r< r r1 r2 r3 � assertFalse�info�calledr? r/ �
call_count�r � m_loggingr5 rH r r r
�&test_add_signal_handler_install_error2� s �zCSelectorEventLoopSignalTests.test_add_signal_handler_install_error2c C sd G dd� dt �}||j_tj|_tj|_| �t| jjtj dd� � | �
|j j � | �
d|jj� d S )Nc @ s e Zd ZejZdS )zPSelectorEventLoopSignalTests.test_add_signal_handler_install_error3.<locals>.ErrNrK r r r r
rH � s rH c S s dS r+ r r r r r
r, � r- zUSelectorEventLoopSignalTests.test_add_signal_handler_install_error3.<locals>.<lambda>� )rI r r0 r r. r r1 r r2 r3 rL rM rN r? r/ rO rP r r r
�&test_add_signal_handler_install_error3� s �zCSelectorEventLoopSignalTests.test_add_signal_handler_install_error3c C st t j|_t j|_| j�t jdd� � | �| j�t j�� | �|jj � | �|j j � | �
t j|j f|j j d � d S )Nc S s dS r+ r r r r r
r, � r- zISelectorEventLoopSignalTests.test_remove_signal_handler.<locals>.<lambda>r )
r r r. r r2 r< �
assertTruer' r/ rN r? �SIG_DFL� call_argsr4 r r r
�test_remove_signal_handler� s �
�z7SelectorEventLoopSignalTests.test_remove_signal_handlerc C s� t j|_t j|_t j|_| j�t jdd� � t� | jjt j< |j �
� | � | j� t j�� | �
|j j� | � |j j� | �t j|jf|j jd � d S )Nc S s dS r+ r r r r r
r, � r- zKSelectorEventLoopSignalTests.test_remove_signal_handler_2.<locals>.<lambda>r )r r r3 r. r r2 �objectr& r< r/ �
reset_mockrU r' rL rN r? �default_int_handlerrW r4 r r r
� test_remove_signal_handler_2� s
�
�z9SelectorEventLoopSignalTests.test_remove_signal_handler_2c C sJ t j|_t j|_| j�t jdd� � t|j_| j� t j� | �
|j � d S )Nc S s dS r+ r r r r r
r, � r- zWSelectorEventLoopSignalTests.test_remove_signal_handler_cleanup_error.<locals>.<lambda>) r r r. r r2 r< r r/ r0 r' rU rM )r rQ r5 r r r
�(test_remove_signal_handler_cleanup_error� s zESelectorEventLoopSignalTests.test_remove_signal_handler_cleanup_errorc C sD t j|_t j|_| j�t jdd� � t|j _| �t| jj t j� d S )Nc S s dS r+ r r r r r
r, � r- zOSelectorEventLoopSignalTests.test_remove_signal_handler_error.<locals>.<lambda>)
r r r. r r2 r< rI r0 r r' r4 r r r
� test_remove_signal_handler_error� s �z=SelectorEventLoopSignalTests.test_remove_signal_handler_errorc C sT t j|_t j|_| j�t jdd� � G dd� dt�}||j _| �t | jj
t j� d S )Nc S s dS r+ r r r r r
r, � r- zPSelectorEventLoopSignalTests.test_remove_signal_handler_error2.<locals>.<lambda>c @ s e Zd ZejZdS )zKSelectorEventLoopSignalTests.test_remove_signal_handler_error2.<locals>.ErrNrK r r r r
rH � s rH ) r r r. r r2 r< rI r0 r r1 r' )r r5 rH r r r
�!test_remove_signal_handler_error2� s �z>SelectorEventLoopSignalTests.test_remove_signal_handler_error2c C s� t j|_t j|_| j�t jdd� � | j�t jdd� � | �t| jj �d� |j
� � | j� � | �t| jj �d� |j
�
d� d S )Nc S s dS r+ r r r r r
r, � r- z9SelectorEventLoopSignalTests.test_close.<locals>.<lambda>c S s dS r+ r r r r r
r, � r- rS r rB )r r r. r r2 r< �SIGCHLDr? �lenr& r/ rZ r �assert_called_once_withr4 r r r
�
test_close� s
z'SelectorEventLoopSignalTests.test_closezasyncio.unix_events.sysc C s� t j|_t j|_| j�t jdd� � | �t| jj�d� d|j _
|j � � | � t
d�� | j�� W 5 Q R X | �t| jj�d� | �|j j� d S )Nc S s dS r+ r r r r r
r, � r- zGSelectorEventLoopSignalTests.test_close_on_finalizing.<locals>.<lambda>r Tz skipping signal handlers removalr )r r r. r r2 r< r? ra r& �
is_finalizing� return_valuerZ �assertWarnsRegex�ResourceWarningr rL rN )r r5 Zm_sysr r r
�test_close_on_finalizing� s
�z5SelectorEventLoopSignalTests.test_close_on_finalizing)rD rE rF r r r" r* r �patchr6 r; rA rJ rR rT rX r\ r] r^ r_ rc rh �
__classcell__r r r r
r - sD
r �AF_UNIXz UNIX Sockets are not supportedc s� e Zd Z� fdd�Zejdd� �Zejdd� �Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
e�eed�d�ejdd� ��Zdd� Zdd� Ze�d�d d � �Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Z � Z S )(� SelectorEventLoopUnixSocketTestsc s$ t � �� t�� | _| �| j� d S r r r r r r
r s
z&SelectorEventLoopUnixSocketTests.setUpc C sv t �� �d}t�tj�}|�|� |�d� |�� | j�dd� |�}| j� |�}|�� | j� |�
� � W 5 Q R X d S )Nr c S s d S r r r r r r
r, r- z]SelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_sock.<locals>.<lambda>) �
test_utils�unix_socket_path�socketrk �bindZlistenr r �create_unix_server�run_until_complete� wait_closed)r �path�sock�coro�srvr r r
�*test_create_unix_server_existing_path_sock s
zKSelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_sockc C sX t �� �F}t�|�}| j�dd� |�}| j�|�}|�� | j�|�� � W 5 Q R X d S )Nc S s d S r r r r r r
r, % r- zRSelectorEventLoopUnixSocketTests.test_create_unix_server_pathlib.<locals>.<lambda>) rm rn �pathlib�Pathr rq rr r rs )r rt Zsrv_cororw r r r
� test_create_unix_server_pathlib! s
z@SelectorEventLoopUnixSocketTests.test_create_unix_server_pathlibc
C sV t �� �D}t�|�}| j�dd� |�}| �t�� | j�|� W 5 Q R X W 5 Q R X d S )Nc S s d S r r r r r r
r, - r- zVSelectorEventLoopUnixSocketTests.test_create_unix_connection_pathlib.<locals>.<lambda>) rm rn ry rz r �create_unix_connectionr �FileNotFoundErrorrr )r rt rv r r r
�#test_create_unix_connection_pathlib* s
zDSelectorEventLoopUnixSocketTests.test_create_unix_connection_pathlibc
C sP t �� �>}| j�dd� |j�}| �td�� | j�|� W 5 Q R X W 5 Q R X d S )Nc S s d S r r r r r r
r, 5 r- z`SelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_nonsock.<locals>.<lambda>zAddress.*is already in use)�tempfile�NamedTemporaryFiler rq �namer9 rI rr )r �filerv r r r
�-test_create_unix_server_existing_path_nonsock3 s
�zNSelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_nonsockc C s> | j jdd� ddd�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, ; r- zSSelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_bool.<locals>.<lambda>�spamT)rt �sslz"ssl argument must be an SSLContext)r rq r9 r rr �r rv r r r
� test_create_unix_server_ssl_bool: s ��zASelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_boolc C s< | j jdd� d d�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, B r- zXSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosock.<locals>.<lambda>�rt z#path was not specified, and no sock�r rq r9 r rr r� r r r
�%test_create_unix_server_nopath_nosockA s
�zFSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosockc
C sV t � � }|�@ | jjdd� d |d�}| �td�� | j�|� W 5 Q R X W 5 Q R X d S )Nc S s d S r r r r r r
r, J r- zXSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsock.<locals>.<lambda>�rt ru �"A UNIX Domain Stream.*was expected)ro r rq r9 r rr �r ru rv r r r
�%test_create_unix_server_path_inetsockG s ��zFSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsockc
C s^ t � t jt j�}|�@ | jjdd� d |d�}| �td�� | j�|� W 5 Q R X W 5 Q R X d S )Nc S s d S r r r r r r
r, S r- zUSelectorEventLoopUnixSocketTests.test_create_unix_server_path_dgram.<locals>.<lambda>r� r� )ro rk Z
SOCK_DGRAMr rq r9 r rr r� r r r
�"test_create_unix_server_path_dgramP s ��zCSelectorEventLoopUnixSocketTests.test_create_unix_server_path_dgram�
SOCK_NONBLOCKz$no socket.SOCK_NONBLOCK (linux only)c C s� t � t jt jt jB �}t�� � }|j}W 5 Q R X zX|�J |� |� | j
j dd� d |d�}| j
� |�}|�
� | j
� |�� � W 5 Q R X W 5 t�|� X d S )Nc S s d S r r r r r r
r, d r- z^SelectorEventLoopUnixSocketTests.test_create_unix_server_path_stream_bittype.<locals>.<lambda>r� )ro rk � SOCK_STREAMr� r r� r� �os�unlinkrp r rq rr r rs )r ru r� �fnrv rw r r r
�+test_create_unix_server_path_stream_bittypeY s
�
� zLSelectorEventLoopUnixSocketTests.test_create_unix_server_path_stream_bittypec C s> | j jdd� ddd�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, m r- zfSelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_timeout_with_plain_sock.<locals>.<lambda>r� r �rt Zssl_handshake_timeout�1ssl_handshake_timeout is only meaningful with sslr� r� r r r
�3test_create_unix_server_ssl_timeout_with_plain_sockl s ��zTSelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_timeout_with_plain_sockc
C sT t � � }|�> | jjdd� |d�}| �td�� | j�|� W 5 Q R X W 5 Q R X d S )Nc S s d S r r r r r r
r, w r- z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_path_inetsock.<locals>.<lambda>�ru r� )ro r r| r9 r rr r� r r r
�)test_create_unix_connection_path_inetsockt s ��zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_inetsockzasyncio.unix_events.socketc C s� t �� }||j_t|j_| jjdd� dd�}| � t�� | j�
|� W 5 Q R X | � |j j
� t|j_| jjdd� dd�}| � t�� | j�
|� W 5 Q R X | � |j j
� d S )Nc S s d S r r r r r r
r, � r- zUSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_error.<locals>.<lambda>z/testr� c S s d S r r r r r r
r, � r- )r r$ ro re rI rp r0 r rq r rr rU r rN � MemoryError)r Zm_socketru rv r r r
�"test_create_unix_server_bind_error} s zCSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_errorc C sB | j jdd� tjt� d�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, � r- zXSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sock.<locals>.<lambda>r� zpath and sock can not be)r r| r� �devnullrY r9 r rr r� r r r
�%test_create_unix_connection_path_sock� s �zFSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sockc C s: | j �dd� d �}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, � r- z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosock.<locals>.<lambda>z no path and sock were specified�r r| r9 r rr r� r r r
�)test_create_unix_connection_nopath_nosock� s ��zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosockc C s@ | j jdd� tjdd�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, � r- z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhost.<locals>.<lambda>r� )Zserver_hostnamez"server_hostname is only meaningful�r r| r� r� r9 r rr r� r r r
�,test_create_unix_connection_nossl_serverhost� s ��zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhostc C s@ | j jdd� tjdd�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, � r- z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhost.<locals>.<lambda>T)r� z/you have to pass server_hostname when using sslr� r� r r r
�,test_create_unix_connection_ssl_noserverhost� s � �zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhostc C s> | j jdd� ddd�}| �td�� | j �|� W 5 Q R X d S )Nc S s d S r r r r r r
r, � r- zjSelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_timeout_with_plain_sock.<locals>.<lambda>r� r r� r� r� r� r r r
�7test_create_unix_connection_ssl_timeout_with_plain_sock� s ��zXSelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_timeout_with_plain_sock) rD rE rF r r Z skip_unless_bind_unix_socketrx r{ r~ r� r� r� r� r� �unittest�
skipUnless�hasattrro r� r� r� r ri r� r� r� r� r� r� rj r r r r
rl s2
�
rl �sendfilezsendfile is not supportedc s� e Zd Zdd ZG dd� dej�Ze� fdd��Ze� fdd��Z � fd d
�Z
d&d d
�Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Z� ZS )'�&SelectorEventLoopUnixSockSendfileTestss� 12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde12345abcde� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) z.SelectorEventLoopUnixSockSendfileTests.MyProtoc C s2 d| _ d| _t� | _|�� | _d | _|�� | _d S )NF)�started�closed� bytearray�data�
create_future�futr
�_ready)r r r r r
�__init__� s
z7SelectorEventLoopUnixSockSendfileTests.MyProto.__init__c C s d| _ || _| j�d � d S r+ )r� r
r� �
set_result)r r
r r r
�connection_made� s z>SelectorEventLoopUnixSockSendfileTests.MyProto.connection_madec C s | j �|� d S r )r� �extend)r r� r r r
�
data_received� s z<SelectorEventLoopUnixSockSendfileTests.MyProto.data_receivedc C s d| _ | j�d � d S r+ )r� r� r� )r �excr r r
�connection_lost� s z>SelectorEventLoopUnixSockSendfileTests.MyProto.connection_lostc � s | j I d H d S r )r� r r r r
rs � s z:SelectorEventLoopUnixSockSendfileTests.MyProto.wait_closedN)rD rE rF r� r� r� r� rs r r r r
�MyProto� s
r� c s2 t tjd��}|�| j� W 5 Q R X t� �� d S )N�wb)�openr �TESTFN�write�DATAr �
setUpClass)�cls�fpr r r
r� � s z1SelectorEventLoopUnixSockSendfileTests.setUpClassc s t �t j� t� �� d S r )r r� r� r �
tearDownClass)r� r r r
r� � s z4SelectorEventLoopUnixSockSendfileTests.tearDownClassc s@ t �� | _| �| j� ttjd�| _| �| jj � t
� � � d S )N�rb) r �new_event_loopr r r� r r� r� r8 r r r r r r r
r � s
z,SelectorEventLoopUnixSockSendfileTests.setUpTc C sR t � t jt j�}|�d� |�t jt jd� |�t jt jd� |rN| �|j � |S )NFr� )
ro ZAF_INETr� Z setblockingZ
setsockoptZ
SOL_SOCKETZ SO_SNDBUFZ SO_RCVBUFr8 r )r �cleanupru r r r
� make_socket� s
z2SelectorEventLoopUnixSockSendfileTests.make_socketc C s | j �|�S r )r rr r� r r r
�run_loop� s z/SelectorEventLoopUnixSockSendfileTests.run_loopc s� �� � }���j�� t�� }�j dd�}|�tj|f� ���jj� fdd�|d������j� |tj|f�� ��� j
� � ��fdd�}�� |� |� fS )NF)r� c s � S r r r )�protor r
r, � r- z@SelectorEventLoopUnixSockSendfileTests.prepare.<locals>.<lambda>r� c s2 � j �� ��� �� � ��� ����� � d S r )r
r r� rs r �r� r Zserverr r
r� � s
z?SelectorEventLoopUnixSockSendfileTests.prepare.<locals>.cleanup) r� r� r r Zfind_unused_portrp ZHOSTr� Z
create_serverZ sock_connectr� r8 )r ru ZportZsrv_sockr� r r� r
�prepare� s
�
z.SelectorEventLoopUnixSockSendfileTests.preparec
C sp | � � \}}tjdg d��: | �tjd�� | �| j�|| j dd �� W 5 Q R X W 5 Q R X | �
| j � � d� d S )Nzasyncio.unix_events.os)�specz$os[.]sendfile[(][)] is not availabler ) r� r ri r9 r �SendfileNotAvailableErrorr� r �_sock_sendfile_nativer� r? �tell)r ru r� r r r
� test_sock_sendfile_not_available s � �zGSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_availablec C sZ | � � \}}t� }| �tjd�� | �| j�||dd �� W 5 Q R X | �| j �
� d� d S �N�not a regular filer ) r� rY r9 r r� r� r r� r? r� r� �r ru r� �fr r r
� test_sock_sendfile_not_a_file s � �zDSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_a_filec C s\ | � � \}}t�� }| �tjd�� | �| j�||dd �� W 5 Q R X | � | j
� � d� d S r� ) r� �io�BytesIOr9 r r� r� r r� r? r� r� r� r r r
�test_sock_sendfile_iobuffer s � �zBSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_iobufferc C sd | � � \}}t�� }d|j_| �tjd�� | �| j �
||dd �� W 5 Q R X | � | j �
� d� d S )NrB r� r )r� r r$ �filenore r9 r r� r� r r� r? r� r� r� r r r
�#test_sock_sendfile_not_regular_file" s � �zJSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_not_regular_filec
C s� | � � \}}| j�� }| j�� }| j�|d ||dd t| j�d� |�� t �
t j �� | �
|� W 5 Q R X | �t�� | jj�|� W 5 Q R X d S �Nr )r� r r� r� r� �_sock_sendfile_native_implra r� r% �
contextlib�suppressr ZCancelledErrorr� r �KeyError� _selector�get_key�r ru r� r� r� r r r
�test_sock_sendfile_cancel1, s
� zASelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_cancel1c
C s� | � � \}}| j�� }| j�� }| j�|d ||dd t| j�d� |�� | j�||�� ||dd t| j�d� | � t
�� | jj � |� W 5 Q R X d S r� )
r� r r� r� r� r� ra r� r% r r� r� r� r� r r r
�test_sock_sendfile_cancel29 s$
� � zASelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_cancel2c C s� | � � \}}| j�� }t�� }d|j_tjdt� d��& | j �
|d ||dd t | j �d� W 5 Q R X | j j
�|�}| �|� |j�tj� d S )NF� os.sendfile�r0 r )r� r� r� r r$ Z cancelledre ri �BlockingIOErrorr r� ra r� r� r� �assertIsNotNoneZadd_done_callbackrb �ANY)r ru r� r� r� �keyr r r
�!test_sock_sendfile_blocking_errorF s
�
zHSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_blocking_errorc C s� | � � \}}| j�� }| j�� }tjdt� d��& | j�|d ||dd t | j
�d� W 5 Q R X | � t �� | jj
�|� W 5 Q R X |�� }| �|tj� | �d| j�� � d S )Nr� r� r )r� r� r� r r� r ri rI r� ra r� r r� r� r� � exceptionr> r r� r? r� )r ru r� r� r� r� r r r
�&test_sock_sendfile_os_error_first_callS s
� zMSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_os_error_first_callc C s� | � � \}}| j�� }| j�� }t� }tjd|d��* | j�||�� ||dd t | j
�d� W 5 Q R X | � t �� | jj
�|� W 5 Q R X |�� }| �||� | �d| j�� � d S �Nr� r� i� )r� r� r� r r� rI r ri r� ra r� r r� r� r� r� �assertIsr? r� �r ru r� r� r� �errr� r r r
�%test_sock_sendfile_os_error_next_calla s$
� zLSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_os_error_next_callc C s� | � � \}}| j�� }| j�� }t�� }tjd|d��* | j� ||�� ||dd t
| j �d� W 5 Q R X | � t
�� | jj�|� W 5 Q R X |�� }| �||� | �d| j�� � d S r� )r� r� r� r r� r r� r ri r� ra r� r r� r� r� r� r� r? r� r� r r r
� test_sock_sendfile_exceptionr s$
� zCSelectorEventLoopUnixSockSendfileTests.test_sock_sendfile_exception)T)rD rE rF r� r �Protocolr� � classmethodr� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� rj r r r r
r� � s(
r� c s
e Zd Z� fdd�Zd(dd�Zdd� Ze�d�d d
� �Ze�d�d d � �Z e�d�d
d� �Z
e�d�e�d�dd� ��Z e�d�dd� �Z e�d�dd� �Z
e�d�dd� �Ze�d�dd� �Ze�d�dd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Z� ZS ))�UnixReadPipeTransportTestsc s� t � �� | �� | _t�tj�| _t j
t j d�| _
d| j
j_t �d�}|�� | �|j� t �d�}|�� }t �
� }tj|_||_| �|j� d S �N)Zspec_set� zos.set_blockingzos.fstat)r r �
new_test_loopr rm �make_test_protocolr r� �protocolr r$ r� � RawIOBase�piper� re ri �startr8 �stop�stat�S_IFIFO�st_mode�r Zblocking_patcherZ
fstat_patcherZm_fstat�str r r
r � s
z UnixReadPipeTransportTests.setUpNc C s( t j| j| j| j|d�}| �t|� |S �N��waiter)r Z_UnixReadPipeTransportr r� r� r8 r �r r r
r r r
�read_pipe_transport� s � z.UnixReadPipeTransportTests.read_pipe_transportc C sR | j �� }| j|d�}| j �|� | jj�|� | j �d|j� | � |�
� � d S �Nr r� ) r r� r rr r� r� r( �
assert_reader� _read_ready� assertIsNone�result�r r �trr r r
� test_ctor� s
z$UnixReadPipeTransportTests.test_ctorzos.readc C s6 | � � }d|_|�� |�d|j� | jj�d� d S �N� datar� )r re r r( �max_sizer� r� �r �m_readr r r r
�test__read_ready� s
z+UnixReadPipeTransportTests.test__read_readyc C s\ | � � }d|_|�� |�d|j� | �| jj� t� | j� | j
j �� | j
j �d � d S )Nr- r� )
r re r r( r rL r �readersrm � run_brieflyr� Z eof_receivedr� r r r r
�test__read_ready_eof� s z/UnixReadPipeTransportTests.test__read_ready_eofc C sD | � � }t|_|�� |�d|j� t�| j� | � | j
j j � d S �Nr� )
r r� r0 r r( r rm r r rL r� r� rN r r r r
�test__read_ready_blocked� s z3UnixReadPipeTransportTests.test__read_ready_blocked�asyncio.log.logger.errorc C s^ | � � }t� }||_t�� |_|�� |�d|j� |j�|� |jt �
d�tt t fd� d S )Nr� z;Fatal read error on pipe transport
protocol:.*
transport:.*)�exc_info) r rI r0 r r$ �_closer r( r rm Z MockPattern�MOCK_ANY)r r �m_logexcr r� r r r
�test__read_ready_error� s
��z1UnixReadPipeTransportTests.test__read_ready_errorc C s8 | � � }t�� }| j�d|� |�� | �| jj� d S r )r r r$ r �
add_reader�
pause_readingrL r )r r r �mr r r
�test_pause_reading� s
z-UnixReadPipeTransportTests.test_pause_readingc C s, | � � }|�� |�� | j�d|j� d S r )r r# �resume_readingr r
r r r r r
�test_resume_reading� s z.UnixReadPipeTransportTests.test_resume_readingc C s* | � � }t�� |_|�� |j�d � d S r )r r r$ r r r( r r r r
rc � s
z%UnixReadPipeTransportTests.test_closec C s2 | � � }d|_t�� |_|�� | �|jj� d S r+ )r �_closingr r$ r r rL rN r r r r
�test_close_already_closing� s
z5UnixReadPipeTransportTests.test_close_already_closingc C sR | � � }t� }|�|� | �|�� � | �| jj� t� | j� | j
j � |� d S r )
r rY r rU �
is_closingrL r r rm r r� r� r( )r r r r� r r r
� test__close� s
z&UnixReadPipeTransportTests.test__closec C sd | � � }| �|j� | �|j� d }|�|� | jj�|� | jj �� | �
|j� | �
|j� d S r ) r r� � _protocol�_loop�_call_connection_lostr� r� r( r� r r �r r r� r r r
�test__call_connection_lost s
z5UnixReadPipeTransportTests.test__call_connection_lostc C sf | � � }| �|j� | �|j� t� }|�|� | jj�|� | j j
�� | � |j� | � |j� d S r ) r r� r, r- rI r. r� r� r( r� r r r/ r r r
�#test__call_connection_lost_with_err
s
z>UnixReadPipeTransportTests.test__call_connection_lost_with_errc C s4 | � � }|�� t�| j� | �|j� |�� d S r )r r rm r r r r- r# �r r r r r
�!test_pause_reading_on_closed_pipe s
z<UnixReadPipeTransportTests.test_pause_reading_on_closed_pipec C s | � � }|�� |�� d S r )r r# r2 r r r
�!test_pause_reading_on_paused_pipe! s z<UnixReadPipeTransportTests.test_pause_reading_on_paused_pipec C s4 | � � }|�� t�| j� | �|j� |�� d S r )r r rm r r r r- r& r2 r r r
�"test_resume_reading_on_closed_pipe' s
z=UnixReadPipeTransportTests.test_resume_reading_on_closed_pipec C s | � � }|�� d S r )r r& r2 r r r
�"test_resume_reading_on_paused_pipe. s z=UnixReadPipeTransportTests.test_resume_reading_on_paused_pipe)N)rD rE rF r r r r ri r r r r! r% r' rc r) r+ r0 r1 r3 r4 r5 r6 rj r r r r
r� � s8
r� c s~ e Zd Z� fdd�Zd7dd�Zdd� Zdd � Ze�d
�d d � �Z e�d
�d
d� �Z
e�d
�dd� �Z e�d
�dd� �Z e�d
�dd� �Z
e�d�e�d
�dd� ��Ze�d
�dd� �Zdd� Ze�d
�d d � �Ze�d
�d d � �Ze�d
�d d!� �Ze�d
�d"d#� �Ze�d$�e�d
�d%d&� ��Ze�d
�d'd(� �Ze�d
�d)d*� �Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z � Z S )8�UnixWritePipeTransportTestsc s� t � �� | �� | _t�tj�| _t j
t j d�| _
d| j
j_t �d�}|�� | �|j� t �d�}|�� }t �
� }tj|_||_| �|j� d S r� )r r r� r rm r� r Z BaseProtocolr� r r$ r� r� r� r� re ri r� r8 r� r� �S_IFSOCKr r r r r
r 7 s
z!UnixWritePipeTransportTests.setUpNc C s( t j| j| j| j|d�}| �t|� |S r )r Z_UnixWritePipeTransportr r� r� r8 r r r r r
�write_pipe_transportI s � z0UnixWritePipeTransportTests.write_pipe_transportc C sT | j �� }| j|d�}| j �|� | jj�|� | j �d|j� | � d |�
� � d S r ) r r� r9 rr r� r� r( r
r r? r
r r r r
r P s
z%UnixWritePipeTransportTests.test_ctorc C s | � � }| �|�� � d S r )r9 rU Z
can_write_eofr2 r r r
�test_can_write_eofY s z.UnixWritePipeTransportTests.test_can_write_eofzos.writec C sF | � � }d|_|�d� |�dd� | �| jj� | �t� |j � d S )N� r r� )
r9 re r� r( rL r �writersr? r� �_buffer�r �m_writer r r r
�
test_write] s
z&UnixWritePipeTransportTests.test_writec C sB | � � }|�d� | �|j� | �| jj� | �td�|j� d S )Nr- ) r9 r� rL rN r r< r? r� r= r> r r r
�test_write_no_dataf s
z.UnixWritePipeTransportTests.test_write_no_datac C s> | � � }d|_|�d� | j�d|j� | �td�|j� d S )NrS r r� � ta) r9 re r� r �
assert_writer� _write_readyr? r� r= r> r r r
�test_write_partialn s
z.UnixWritePipeTransportTests.test_write_partialc C s^ | � � }| j�d|j� td�|_|�d� | �|j� | j� d|j� | �
td�|j� d S )Nr� s previousr s previousdata) r9 r �
add_writerrD r� r= r� rL rN rC r? r> r r r
�test_write_bufferv s
z-UnixWritePipeTransportTests.test_write_bufferc C sP | � � }t� |_|�d� |�dtd�� | j�d|j� | � td�|j
� d S r ) r9 r� r0 r� r( r� r rC rD r? r= r> r r r
�test_write_again� s
z,UnixWritePipeTransportTests.test_write_againzasyncio.unix_events.loggerc C s� | � � }t� }||_t�� |_|�d� |�dd� | �| j j
� | � t � |j
� |j�|d� | � d|j� |�d� | � d|j� |�d� |�d� |�d� |�d� |j�d� |�� d S )Nr r� z#Fatal write error on pipe transportr rS z=pipe closed by peer or os.write(pipe, data) raised exception.)r9 rI r0 r r$ Z _fatal_errorr� r( rL r r< r? r� r= �
_conn_lost�warningr )r r? Zm_logr r� r r r
�test_write_err� s.
�
�z*UnixWritePipeTransportTests.test_write_errc C sD | � � }|�� |�d� | �|jd� |�d� | �|jd� d S )Nr r rS )r9 r r� r? rI r> r r r
�test_write_close� s
z,UnixWritePipeTransportTests.test_write_closec C sX | � � }|�� | �| jj� | �| jj� | �|�� � t� | j� | j
j � d � d S r )
r9 r rL r r r< rU r* rm r r� r� r( r2 r r r
r � s z,UnixWritePipeTransportTests.test__read_readyc C sR | � � }| j�d|j� td�|_d|_|�� | �| jj� | � t� |j� d S )Nr� r r; )
r9 r rF rD r� r= re rL r< r? r> r r r
�test__write_ready� s
z-UnixWritePipeTransportTests.test__write_readyc C sV | � � }| j�d|j� td�|_d|_|�� | j�d|j� | �td�|j� d S )Nr� r � � a) r9 r rF rD r� r= re rC r? r> r r r
�test__write_ready_partial� s
z5UnixWritePipeTransportTests.test__write_ready_partialc C sh | � � }| j�d|j� td�|_t� |_|�� |�dtd�� | j� d|j� | �
td�|j� d S )Nr� r ) r9 r rF rD r� r= r� r0 r( rC r? r> r r r
�test__write_ready_again� s
z3UnixWritePipeTransportTests.test__write_ready_againc C sf | � � }| j�d|j� td�|_d|_|�� |�dtd�� | j�d|j� | � td�|j� d S )Nr� r r )
r9 r rF rD r� r= re r( rC r? r> r r r
�test__write_ready_empty� s
z3UnixWritePipeTransportTests.test__write_ready_emptyr c C s� | � � }| j�d|j� td�|_t� |_}|�� | �| jj � | �| jj
� | � t� |j� | � |�
� � |�� | � d|j� t�| j� | jj�|� d S )Nr� r r )r9 r rF rD r� r= rI r0 rL r< r r? rU r* Zassert_not_calledrI rm r r� r� r( )r r? r r r� r r r
�test__write_ready_err� s
z1UnixWritePipeTransportTests.test__write_ready_errc C s� | � � }| j�d|j� d|_td�|_d|_|�� | �| jj � | �| jj
� | � t� |j� | j j
�d � | jj�� d S )Nr� Tr r; )r9 r rF rD r( r� r= re rL r< r r? r� r� r( r� r r> r r r
�test__write_ready_closing� s
z5UnixWritePipeTransportTests.test__write_ready_closingc C s� | � � }| j�d|j� | j�d|j� ddg|_|�� | �|j � | �| jj
� | �| jj � | � g |j� | �
|�� � t�| j� | jj�d � d S )Nr� s darB )r9 r rF rD r" r r= �abortrL rN r r< r? rU r* rm r r� r� r( r> r r r
�
test_abort s
z&UnixWritePipeTransportTests.test_abortc C sd | � � }| �|j� | �|j� d }|�|� | jj�|� | jj �� | �
|j� | �
|j� d S r ) r9 r� r, r- r. r� r� r( r� r r r/ r r r
r0 s
z6UnixWritePipeTransportTests.test__call_connection_lostc C sf | � � }| �|j� | �|j� t� }|�|� | jj�|� | j j
�� | � |j� | � |j� d S r ) r9 r� r, r- rI r. r� r� r( r� r r r/ r r r
r1 s
z?UnixWritePipeTransportTests.test__call_connection_lost_with_errc C s0 | � � }t�� |_|�� |j�� |�� d S r )r9 r r$ � write_eofr r( r2 r r r
rc ) s
z&UnixWritePipeTransportTests.test_closec C s2 | � � }t�� |_d|_|�� | �|jj� d S r+ )r9 r r$ rW r( r rL rN r2 r r r
�test_close_closing2 s
z.UnixWritePipeTransportTests.test_close_closingc C sJ | � � }|�� | �|�� � | �| jj� t�| j� | j j
� d � d S r ) r9 rW rU r* rL r r rm r r� r� r( r2 r r r
�test_write_eof9 s z*UnixWritePipeTransportTests.test_write_eofc C s: | � � }dg|_|�� | �|�� � | �| jjj� d S )Nr ) r9 r= rW rU r* rL r� r� rN r2 r r r
�test_write_eof_pendingA s
z2UnixWritePipeTransportTests.test_write_eof_pending)N) rD rE rF r r9 r r: r ri r@ rA rE rG rH rK rL r rM rP rQ rR rS rT rV r0 r1 rc rX rY rZ rj r r r r
r7 5 sR
r7 c @ s e Zd Zdd� ZdS )�AbstractChildWatcherTestsc C s� t �� }t�� }| �t|j||� | �t|j|� | �t|j|� | �t|j � | �t|j
� | �t|j � | �t|j |||� d S r )
r r$ r ZAbstractChildWatcherr �NotImplementedError�add_child_handler�remove_child_handler� attach_loopr Z is_active� __enter__�__exit__�r r� �watcherr r r
�test_not_implementedK sJ � � � � � � �z.AbstractChildWatcherTests.test_not_implementedN�rD rE rF rd r r r r
r[ I s r[ c @ s e Zd Zdd� ZdS )�BaseChildWatcherTestsc C s$ t �� }t�� }| �t|j|� d S r )r r$ r ZBaseChildWatcherr r\ Z _do_waitpidrb r r r
rd ` s �z*BaseChildWatcherTests.test_not_implementedNre r r r r
rf ^ s rf � WaitPidMocks)�waitpid� WIFEXITED� WIFSIGNALED� WEXITSTATUS�WTERMSIGc s e Zd Zej�ejd�Z� fdd�Z dd� Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zedd� �Zedd� �Zedd� �Zedd� �Zed d � �Zed d � �Zed d!� �Zed"d#� �Zed$d%� �Zed&d'� �Zed(d)� �Z ed*d+� �Z ed,d-� �Z ed.d/� �Z � Z S )0�ChildWatcherTestsMixinrJ c sZ t � �� | �� | _d| _i | _tj�| jd�� | _ | �
� | _ | j � | j� W 5 Q R X d S )NFr2 )
r r r� r �running�zombiesr ri rY �m_add_signal_handler�create_watcherrc r_ r r r r
r t s
�
z ChildWatcherTestsMixin.setUpc C sv t | jtj�s|dkr"| �|d� z*|dk r8| j�� W S || j�|�fW S W n tk
r` Y nX | j rldS t
� �d S )NrB r )r r ) �
isinstancerc r �SafeChildWatcherZ
assertGreaterro �popitem�popr� rn �ChildProcessError)r �pid�flagsr r r
rh s z ChildWatcherTestsMixin.waitpidc C s |d | j |< d S �N� � )ro )r rw �
returncoder r r
�
add_zombie� s z!ChildWatcherTestsMixin.add_zombiec C s |dkS ry r �r Zstatusr r r
ri � s z ChildWatcherTestsMixin.WIFEXITEDc C s d| k odk S S )Ni� rz r r} r r r
rj � s z"ChildWatcherTestsMixin.WIFSIGNALEDc C s | � | �|�� |d S ry )rU ri r} r r r
rk � s z"ChildWatcherTestsMixin.WEXITSTATUSc C s | � | �|�� d| S ry )rU rj r} r r r
rl � s z ChildWatcherTestsMixin.WTERMSIGc C s | j �tj| jj� d S r )rp rb r r` rc � _sig_chldr r r r
�test_create_watcher� s �z*ChildWatcherTestsMixin.test_create_watcherc s � fdd�}|S )Nc
s� dd� }|d| j ��|}|d| j��d}|d| j��L}|d| j��4}|d| j�� }� | t|||||�� W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S )Nc S s t j| |t jd�S )N)�wrapsZ new_callable)r ri r$ )�target�wrapperr r r
ri � s �zIChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_func.<locals>.patchz os.WTERMSIGzos.WEXITSTATUSzos.WIFSIGNALEDz os.WIFEXITEDz
os.waitpid)rl rk rj ri rh rg )r ri Z
m_WTERMSIGZ
m_WEXITSTATUSZ
m_WIFSIGNALEDZ m_WIFEXITEDZ m_waitpid�r: r r
� wrapped_func� s(
�
�
�
� �z:ChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_funcr )r: r� r r� r
�
waitpid_mocks� s z$ChildWatcherTestsMixin.waitpid_mocksc C s t �� }| j� d| _| j�d|ddd� W 5 Q R X | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� � | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d| _| � dd� | j� � | �
|jj� | �
|j j� | �|j
j� |�ddddd� |j�� |j�� |j �� |�� | � dd� | j� | j� � W 5 Q R X | �|j� | �|j
j� |j�� |j�� |j �� | j�� | j� � | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d S ) NT�* � �
� F� �
)r r$ rc rn r] rL rN ri rj rk rl r~ r| rU rb rZ �ignore_warningsro �clear�r r$ �callbackr r r
� test_sigchld� sR
z#ChildWatcherTestsMixin.test_sigchldc C s� t �� }t �� }| j� d| _| j�d|dd� W 5 Q R X | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� | j�d|dd� W 5 Q R X | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� � | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | � dd� | j� � |�
dddd� | �|j� | �|jj� | �|j j� | �|j
j� |j�� |j�� |j
�� |�� | j� � | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | � dd � d
| _| j� � |�
dd dd� | �|j� | �|jj� | �|j j� | �|j
j� |j�� |j�� |j �� |�� | � dd � | � dd � | j� | j� � W 5 Q R X | �|j� | �|j� | �|j
j� |j�� |j�� |j �� | j�� | j� � | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d S )
NT�+ � � �, � � ����l Fr� � )r r$ rc rn r] rL rN ri rj rk rl r~ r| rb rU rZ r� ro r� �r r$ � callback1� callback2r r r
�test_sigchld_two_children� s�
z0ChildWatcherTestsMixin.test_sigchld_two_childrenc C sn t �� }t �� }| j� d| _| j�d|dd� W 5 Q R X | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� | j�d|dd� W 5 Q R X | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� � | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | � dd� | � dd � d
| _| j� � |�
dddd� |�
dd dd� | �|jj� | �|jj� | �|j j� | �|j
j� |j�� |j�� |j
�� |j �� |�� |�� | � dd � | � dd � | j� | j� � W 5 Q R X | �|j� | �|j� | �|j
j� d S )
NT�- � r� �. i{ r� �N ���Fr� r� )r r$ rc rn r] rL rN ri rj rk rl r~ r| rb rU rZ r� r� r r r
�.test_sigchld_two_children_terminating_togetherY sb
zEChildWatcherTestsMixin.test_sigchld_two_children_terminating_togetherc C s� t �� }| j�. | �dd� | j�� | j�d|dd� W 5 Q R X |�dddd� |�� | �dd� | j� | j�� W 5 Q R X | � |j
� d S )N�2 r; r r� rB ) r r$ rc r| r~ r] rb rZ r� rL rN r� r r r
�test_sigchld_race_condition� s
z2ChildWatcherTestsMixin.test_sigchld_race_conditionc C s� t �� }t �� }| j� d| _| j�d|d� W 5 Q R X | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� | j�d|d� W 5 Q R X | �|j� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d| _| � dd� | j� � |�
ddd� | �|j� | �|jj� | �|j j� | �|j
j� |j�� |j�� |j
�� |�� | � dd� | j� | j� � W 5 Q R X | �|j� | �|j� | �|j
j� d S )NT�3 � � Fi����r� )r r$ rc rn r] rL rN ri rj rk rl r| r~ rb rU rZ r� r� r r r
� test_sigchld_replace_handler� sJ
z3ChildWatcherTestsMixin.test_sigchld_replace_handlerc C s t �� }| j� d| _| j�d|d� W 5 Q R X | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� | j� d� | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d| _| � dd� | j
� | j�� W 5 Q R X | �|j� d S )NT�4 i� F�c )r r$ rc rn r] rL rN ri rj rk rl r^ r| r� r~ r� r r r
�test_sigchld_remove_handler� s( z2ChildWatcherTestsMixin.test_sigchld_remove_handlerc C s( t �� }| j� d| _| j�d|d� W 5 Q R X | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d| j d< d| _| j� � |�
ddd� | �|jj� | �|jj� | �|j j� | �|j
j� |�� |j�� |j�� | �dd� | j� | j� � W 5 Q R X | �|j� d S )NT�5 i��i� F�e )r r$ rc rn r] rL rN ri rj rk rl ro r~ rb rU rZ r| r� r� r r r
�test_sigchld_unknown_status s0
z2ChildWatcherTestsMixin.test_sigchld_unknown_statusc C s t �� }t �� }t �� }| j�< d| _| j�d|d� | j�d|d� | j�d|d� W 5 Q R X | �| j�d�� | �| j�d�� | �| j�d�� | �| j�d�� | �dd� | �dd� | �dd� d | _| j � | j�
� W 5 Q R X | �|j � | �|j � |� ddd� d S )
NT�6 r �7 rS �8 rN r F)
r r$ rc rn r] rU r^ rL r| r� r~ rN rb )r r$ r� r� � callback3r r r
�test_remove_child_handler. s* z0ChildWatcherTestsMixin.test_remove_child_handlerc C st t �� }| j� d| _| j�d|� W 5 Q R X t|j_t j� t
j d��$}| � | j�
� d � | �|j� W 5 Q R X d S )NT�9 �error)r r$ rc rn r] r rh r0 ri rY r �loggerr? r~ rU rN )r r$ r� Zm_errorr r r
� test_sigchld_unhandled_exceptionO s
�z7ChildWatcherTestsMixin.test_sigchld_unhandled_exceptionc C s� t �� }| j� d| _| j�d|� W 5 Q R X | �|j� | �|jj� | �|jj� | �|j j� | �|j
j� d| _| � dd� t �
dt j� |j
�� | j� | j�� W 5 Q R X t| jtj�r�| �|j� n |�dd� d S )NT�: Fr; � )r r$ rc rn r] rL rN ri rj rk rl r| r� rh �WNOHANGrZ r� r~ rr r �FastChildWatcherrb r� r r r
�#test_sigchld_child_reaped_elsewherea s$
z:ChildWatcherTestsMixin.test_sigchld_child_reaped_elsewherec
C s� t �� }t �� }| j�\ | j�J d| _| �dd� | �dd� | j�� | j�d|� | j�d|� W 5 Q R X W 5 Q R X |�dd� | � |j
� d S )NTiO r� iQ r� iP ) r r$ r� rc rn r| r~ r] rb rL rN r� r r r
�,test_sigchld_unknown_pid_during_registration� s
" zCChildWatcherTestsMixin.test_sigchld_unknown_pid_during_registrationc
C s� t �� }| j� d| _| j�d|� W 5 Q R X | j}| �� | _t jj}||d��J}|| jd��2}| j� | j� |�
t j � |�
t j | jj
� W 5 Q R X W 5 Q R X d| _| �dd� | j�
� |�
dd� d S )NT�< r' r2 Fr� )r r$ rc rn r] r r� ri rY r_ rb r r` r~ r| )r r$ r� �old_loopri Z m_old_removeZ m_new_addr r r
�
test_set_loop� s.
�� �
z$ChildWatcherTestsMixin.test_set_loopc
C s� t �� }t �� }t �� }| j�6 d| _| j�d|� | j�d|� | j�d|� W 5 Q R X | j}d | _t j�|d��6}| �t d�� | j�
d � W 5 Q R X |� t j
� W 5 Q R X | �dd� | �dd� | �|j� | �|j� | �|j� | �� | _t j�| jd ��J}| j�
| j� |� t j
| jj� |� dd� |� dd� | �|j� W 5 Q R X |�� |�� d
| _| �dd � | j�� | �|j� | �|j� |� dd � d S ) NT�= �> in r' zA loop is being detached� r� r2 Fr� )r r$ rc rn r] r ri rY rf �RuntimeWarningr_ rb r r` r| rL rN r� r~ rZ )r r$ r� r� r� r� �m_remove_signal_handlerrp r r r
� test_set_loop_race_condition� sd � ��
� �
z3ChildWatcherTestsMixin.test_set_loop_race_conditionc
C s� t �� }| j�� d| _| �dd� | �dd� | j�� | j�d|� | j�d|� | �t| jj �d� t
| jt j �r�| �t| jj
�d� t j�| jd��F}| j�� |�tj� | �| jj � t
| jt j �r�| �| jj
� W 5 Q R X W 5 Q R X d S ) NT�? r� �A r� �@ r r' )r r$ rc rn r| r~ r] r? ra Z
_callbacksrr r r� Z_zombiesri rY r r rb r r` rL )r r$ r� r� r r r
rc � s.
�
�z!ChildWatcherTestsMixin.test_close)!rD rE rF r ri rY r r� r� r rh r| ri rj rk rl r r� r� r� r� r� r� r� r� r� r� r� r� r� r� rc rj r r r r
rm p sL
<
g
D
4
$
"
=rm c @ s e Zd Zdd� ZdS )�SafeChildWatcherTestsc C s t �� S r )r rs r r r r
rq s z$SafeChildWatcherTests.create_watcherN�rD rE rF rq r r r r
r� s r� c @ s e Zd Zdd� ZdS )�FastChildWatcherTestsc C s t �� S r )r r� r r r r
rq s z$FastChildWatcherTests.create_watcherNr� r r r r
r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) � PolicyTestsc C s t �� S r )r ZDefaultEventLoopPolicyr r r r
�
create_policy# s zPolicyTests.create_policyc C sL | � � }| �|j� |�� }| �|tj� | �|j|� | �||�� � d S r )r� r �_watcher�get_child_watcherr> r ZThreadedChildWatcherr� �r �policyrc r r r
� test_get_default_child_watcher& s z*PolicyTests.test_get_default_child_watcherc C s< | � � }t�� }|�|� | �|j|� | �||�� � d S r )r� r r� �set_child_watcherr� r� r� r� r r r
� test_get_child_watcher_after_set1 s
z,PolicyTests.test_get_child_watcher_after_setc sD � �fdd�}�� � � � �t�� � tj|d�}|�� |�� d S )Nc sR � � � �� � ��� �� tj� � �� } ��| tj� ��| j � � �� �
� d S r ) r r� r> �get_event_loopr ZAbstractEventLoopr� rs r r- r )rc �r� r r r
r� ; s
� z4PolicyTests.test_get_child_watcher_thread.<locals>.f)r� )r� r� r rs � threadingZThreadr� �join)r r� Zthr r� r
� test_get_child_watcher_thread9 s z)PolicyTests.test_get_child_watcher_threadc C s� | � � }|�� }t�� }|�|� |�|� | �|j|� |�� }|� |� | �|j|� |� d � | �|jd � |�
� |�
� d S r ) r� r� r rs r� r_ r� r- r� r r )r r� r rc Znew_loopr r r
�,test_child_watcher_replace_mainloop_existingN s
z8PolicyTests.test_child_watcher_replace_mainloop_existingN)rD rE rF r� r� r� r� r� r r r r
r� ! s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestFunctionalc C s t �� | _t �| j� d S r )r r� r r r r r r
r i s
zTestFunctional.setUpc C s | j �� t�d � d S r )r r r r r r r r
�tearDownm s
zTestFunctional.tearDownc s� � fdd�}dd� }|� � � j �t� |� W 5 Q R X |� � � j �t� |� W 5 Q R X |� � � j �t� � W 5 Q R X |� � � j �t� � W 5 Q R X d S )Nc s � � td�S )NzInvalid file object)r9 r r r r r
�
assert_raisesr s zFTestFunctional.test_add_reader_invalid_argument.<locals>.assert_raisesc S s d S r r r r r r
r, u r- zATestFunctional.test_add_reader_invalid_argument.<locals>.<lambda>)r r" rY rF �
remove_reader�
remove_writer)r r� r@ r r r
� test_add_reader_invalid_argumentq s z/TestFunctional.test_add_reader_invalid_argumentc sR �fdd�� � ��fdd�}t �� \�}z�j�|� � W 5 ��� |�� X d S )Nc s � � td�S )Nz'File descriptor .* is used by transport)r9 r1 r r r r
r� � s �zLTestFunctional.test_add_reader_or_writer_transport_fd.<locals>.assert_raisesc � sD �j jdd� �d�I d H \} }�zdd� }� � � �j ��|� W 5 Q R X � � � �j ���� |� W 5 Q R X � � � �j ��� W 5 Q R X � � � �j ���� � W 5 Q R X � � � �j ��|� W 5 Q R X � � � �j ���� |� W 5 Q R X � � � �j ��� W 5 Q R X � � � �j ���� � W 5 Q R X W 5 | �� X d S )Nc S s t �� S r )r r� r r r r
r, � r- zWTestFunctional.test_add_reader_or_writer_transport_fd.<locals>.runner.<locals>.<lambda>r� c S s d S r r r r r r
r, � r- )r Zcreate_connectionr r" r� r� rF r� )r Zprr@ �r� Zrsockr r r
�runner� s. � zETestFunctional.test_add_reader_or_writer_transport_fd.<locals>.runner)ro Z
socketpairr r rr )r r� Zwsockr r� r
�&test_add_reader_or_writer_transport_fd� s z5TestFunctional.test_add_reader_or_writer_transport_fdN)rD rE rF r r� r� r� r r r r
r� g s r� �__main__)0�__doc__� collectionsr� rG r� r� ry r ro r� �sysr r� r� r Ztestr �platformZSkipTestr r r Ztest.test_asyncior rm r� r r r r� ZTestCaser r� rl r� r� r7 r[ rf �
namedtuplerg rm r� r� r� r� rD �mainr r r r
�<module> sv
^ � ) � N 2 � ,FF