Your IP : 216.73.216.213
3
� f�� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddl mZ e jdkr�e jd��ddlZddlmZ ddlmZ ddlmZ ejZd d
� Ze jed �G d d
� d
ej��Ze jeed�d�G dd� dej��Z G dd� dej�Z G dd� dej�Z G dd� de j�Z G dd� de j�Z ej!dd)�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&e'd(k�r�e j(� dS )*zTests for unix_events.py.� N)�mockZwin32z UNIX only)�log)�
test_utils)� unix_eventsc C s" | j d krd S | j j� d | _ d S )N)Z_pipe�close)� transport� r �G/opt/alt/python36/lib64/python3.6/test/test_asyncio/test_unix_events.py�close_pipe_transport s
r
zSignals are not supportedc sF e Zd Z� fdd�Zdd� Zdd� Zdd� Zejd �d
d � �Z ejd �d d
� �Z
ejd �dd� �Z ejd �dd� �Z ejd �ejd�dd� ��Z
ejd �ejd�dd� ��Zejd �dd� �Zejd �dd� �Zejd �ejd�dd � ��Zejd �d d � �Zejd �d d � �Zejd �d!d"� �Zejd#�ejd �d$d%� ��Z� ZS )&� SelectorEventLoopSignalTestsc s$ t � j� tj� | _| j| j� d S )N)�super�setUp�asyncio�SelectorEventLoop�loop�set_event_loop)�self)� __class__r r r
+ s
z"SelectorEventLoopSignalTests.setUpc C s. | j t| jjd� | j t| jjtjd � d S )N�1� )� assertRaises� TypeErrorr Z
_check_signal�
ValueError�signal�NSIG)r r r r �test_check_signal0 s z.SelectorEventLoopSignalTests.test_check_signalc C s | j jtjd � d S )Nr )r �_handle_signalr r )r r r r � test_handle_signal_no_handler6 s z:SelectorEventLoopSignalTests.test_handle_signal_no_handlerc C sh t jtj� f tj� d�}|j� || jjtjd < tj� | j_ | jj
tjd � | jj j tjd � d S )N)r r ) r �Handler �MockZcancelr �_signal_handlersr r �remove_signal_handlerr �assert_called_with)r �hr r r �$test_handle_signal_cancelled_handler9 s zASelectorEventLoopSignalTests.test_handle_signal_cancelled_handlerzasyncio.unix_events.signalc C s. t j|_t|j_| jt| jjt j dd� � d S )Nc S s dS )NTr r r r r �<lambda>J s zRSelectorEventLoopSignalTests.test_add_signal_handler_setup_error.<locals>.<lambda>)
r r r �
set_wakeup_fd� side_effectr � RuntimeErrorr �add_signal_handler�SIGINT)r �m_signalr r r �#test_add_signal_handler_setup_errorB s z@SelectorEventLoopSignalTests.test_add_signal_handler_setup_errorc C sZ t j|_tjdd� �}|}|� }| j|j� x(||fD ] }| jtd| jj t j
|� q6W d S )Nc s s g E d H d S )Nr r r r r �simple_coroutineP s z^SelectorEventLoopSignalTests.test_add_signal_handler_coroutine_error.<locals>.simple_coroutinez1coroutines cannot be used with add_signal_handler) r r r � coroutine�
addCleanupr �assertRaisesRegexr r r) r* )r r+ r- Z coro_funcZcoro_obj�funcr r r �'test_add_signal_handler_coroutine_errorL s zDSelectorEventLoopSignalTests.test_add_signal_handler_coroutine_errorc C sP t j|_dd� }| jjt j|� | jjjt j�}| j|tj � | j
|j |� d S )Nc S s dS )NTr r r r r r% b s zFSelectorEventLoopSignalTests.test_add_signal_handler.<locals>.<lambda>) r r r r) �SIGHUPr �get�assertIsInstancer r � assertEqualZ _callback)r r+ �cbr# r r r �test_add_signal_handler^ s z4SelectorEventLoopSignalTests.test_add_signal_handlerc C sL t j|_dd� }||_G dd� dt�}||j _| j|| jjt jdd� � d S )Nc S s | dkrt � �d S )Nr ���)r )�fdr r r r& l 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 �Errq s r? c S s dS )NTr r r r r r% x s zTSelectorEventLoopSignalTests.test_add_signal_handler_install_error.<locals>.<lambda>) r r r& �OSErrorr' r r r) r* )r r+ r& r? r r r �%test_add_signal_handler_install_errorh s zBSelectorEventLoopSignalTests.test_add_signal_handler_install_errorzasyncio.base_events.loggerc C sn t j|_G dd� dt�}||j _dd� | jjt j< | jt| jj t j
dd� � | j |j j
� | jd|jj� d S )Nc @ s e Zd ZejZdS )zPSelectorEventLoopSignalTests.test_add_signal_handler_install_error2.<locals>.ErrN)r; r< r= r> �EINVALr r r r r? s r? c S s dS )NTr r r r r r% � s zUSelectorEventLoopSignalTests.test_add_signal_handler_install_error2.<locals>.<lambda>c S s dS )NTr r r r r r% � s r )r r r@ r' r r r3 r r( r) r* � assertFalse�info�calledr6 r& �
call_count)r � m_loggingr+ r? r r r �&test_add_signal_handler_install_error2z s zCSelectorEventLoopSignalTests.test_add_signal_handler_install_error2c C s\ G dd� dt �}||j_tj|_| jt| jjtjdd� � | j |j
j � | j d|j
j� d S )Nc @ s e Zd ZejZdS )zPSelectorEventLoopSignalTests.test_add_signal_handler_install_error3.<locals>.ErrN)r; r< r= r> rB r r r r r? � s r? c S s dS )NTr r r r r r% � s zUSelectorEventLoopSignalTests.test_add_signal_handler_install_error3.<locals>.<lambda>� )r@ r r' r r r( r r) r* rC rD rE r6 r& rF )r rG r+ r? r r r �&test_add_signal_handler_install_error3� s zCSelectorEventLoopSignalTests.test_add_signal_handler_install_error3c C sl t j|_| jjt jdd� � | j| jjt j�� | j|jj� | j|j j� | j t j|j
f|j j d � d S )Nc S s dS )NTr r r r r r% � s zISelectorEventLoopSignalTests.test_remove_signal_handler.<locals>.<lambda>r ) r r r r) r3 �
assertTruer! r& rE r6 �SIG_DFL� call_args)r r+ r r r �test_remove_signal_handler� s z7SelectorEventLoopSignalTests.test_remove_signal_handlerc C s� t j|_t j|_| jjt jdd� � t� | jjt j< |jj � | j
| jj t j�� | j |jj
� | j
|j j
� | jt j|jf|j jd � d S )Nc S s dS )NTr r r r r r% � s zKSelectorEventLoopSignalTests.test_remove_signal_handler_2.<locals>.<lambda>r )r r r* r r) �objectr r3 r& �
reset_mockrK r! rC rE r6 �default_int_handlerrM )r r+ r r r � test_remove_signal_handler_2� s
z9SelectorEventLoopSignalTests.test_remove_signal_handler_2c C sB t j|_| jjt jdd� � t|j_| jjt j� | j |j
� d S )Nc S s dS )NTr r r r r r% � s zWSelectorEventLoopSignalTests.test_remove_signal_handler_cleanup_error.<locals>.<lambda>) r r r r) r3 r r& r' r! rK rD )r rG r+ r r r �(test_remove_signal_handler_cleanup_error� s
zESelectorEventLoopSignalTests.test_remove_signal_handler_cleanup_errorc C s< t j|_| jjt jdd� � t|j _| jt| jjt j� d S )Nc S s dS )NTr r r r r r% � s zOSelectorEventLoopSignalTests.test_remove_signal_handler_error.<locals>.<lambda>) r r r r) r3 r@ r' r r! )r r+ r r r � test_remove_signal_handler_error� s
z=SelectorEventLoopSignalTests.test_remove_signal_handler_errorc C sL t j|_| jjt jdd� � G dd� dt�}||j _| jt| jj t j� d S )Nc S s dS )NTr r r r r r% � s zPSelectorEventLoopSignalTests.test_remove_signal_handler_error2.<locals>.<lambda>c @ s e Zd ZejZdS )zKSelectorEventLoopSignalTests.test_remove_signal_handler_error2.<locals>.ErrN)r; r< r= r> rB r r r r r? � s r? )
r r r r) r3 r@ r' r r( r! )r r+ r? r r r �!test_remove_signal_handler_error2� s z>SelectorEventLoopSignalTests.test_remove_signal_handler_error2c C s| t j|_| jjt jdd� � | jjt jdd� � | jt| jj�d� |j j
� | jj � | jt| jj�d� |j j d� d S )Nc S s dS )NTr r r r r r% � s z9SelectorEventLoopSignalTests.test_close.<locals>.<lambda>c S s dS )NTr r r r r r% � s rI r r r9 )
r r r r) r3 �SIGCHLDr6 �lenr r& rP r �assert_called_once_with)r r+ r r r �
test_close� s
z'SelectorEventLoopSignalTests.test_closezasyncio.unix_events.sysc
C s� t j|_| jjt jdd� � | jt| jj�d� d|j_ |j j
� | j t d�� | jj
� W d Q R X | jt| jj�d� | j|j j� d S )Nc S s dS )NTr r r r r r% � s zGSelectorEventLoopSignalTests.test_close_on_finalizing.<locals>.<lambda>r Tz skipping signal handlers removalr )r r r r) r3 r6 rW r �
is_finalizing� return_valuerP �assertWarnsRegex�ResourceWarningr rC rE )r r+ Zm_sysr r r �test_close_on_finalizing� s
z5SelectorEventLoopSignalTests.test_close_on_finalizing)r; r< r= r
r r r$ r �patchr, r2 r8 rA rH rJ rN rR rS rT rU rY r^ �
__classcell__r r )r r r ( s*
r �AF_UNIXz UNIX Sockets are not supportedc s� e Zd Z� fdd�Zdd� Zejeed�d�dd� �Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zejeed�d�dd� �Zdd� Zejd�dd� �Zd d � Zd d � Zd d!� Zd"d#� Z� ZS )$� SelectorEventLoopUnixSocketTestsc s$ t � j� tj� | _| j| j� d S )N)r r
r r r r )r )r r r r
� s
z&SelectorEventLoopUnixSocketTests.setUpc
C sv t j� �d}tjtj�}|j|� |jd� |j� | jjdd� |�}| jj |�}|j� | jj |j
� � W d Q R X d S )Nr c S s d S )Nr r r r r r% s z]SelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_sock.<locals>.<lambda>) r �unix_socket_path�socketra �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_sock�fspathz no os.fspathc
C sX t j� �F}tj|�}| jjdd� |�}| jj|�}|j� | jj|j� � W d Q R X d S )Nc S s d S )Nr r r r r r% s zRSelectorEventLoopUnixSocketTests.test_create_unix_server_pathlib.<locals>.<lambda>) r rc �pathlibZPathr rf rg r rh )r ri Zsrv_cororl r r r � test_create_unix_server_pathlib s
z@SelectorEventLoopUnixSocketTests.test_create_unix_server_pathlibc C sP t j� �>}| jjdd� |j�}| jtd�� | jj|� W d Q R X W d Q R X d S )Nc S s d S )Nr r r r r r% s z`SelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_nonsock.<locals>.<lambda>zAddress.*is already in use)�tempfile�NamedTemporaryFiler rf �namer0 r@ rg )r �filerk r r r �-test_create_unix_server_existing_path_nonsock s
zNSelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_nonsockc C s> | j jdd� ddd�}| jtd�� | j j|� W d Q R X d S )Nc S s d S )Nr r r r r r% ! s zSSelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_bool.<locals>.<lambda>�spamT)ri �sslz"ssl argument must be an SSLContext)r rf r0 r rg )r rk r r r � test_create_unix_server_ssl_bool s
zASelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_boolc C s< | j jdd� d d�}| jtd�� | j j|� W d Q R X d S )Nc S s d S )Nr r r r r r% ( s zXSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosock.<locals>.<lambda>)ri z#path was not specified, and no sock)r rf r0 r rg )r rk r r r �%test_create_unix_server_nopath_nosock' s zFSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosockc C sV t j � }|�@ | jjdd� d |d�}| jtd�� | jj|� W d Q R X W d Q R X d S )Nc S s d S )Nr r r r r r% 0 s zXSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsock.<locals>.<lambda>)ri rj z"A UNIX Domain Stream.*was expected)rd r rf r0 r rg )r rj rk r r r �%test_create_unix_server_path_inetsock- s zFSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsockc C s^ t j t jt j�}|�@ | jjdd� d |d�}| jtd�� | jj|� W d Q R X W d Q R X d S )Nc S s d S )Nr r r r r r% 9 s zUSelectorEventLoopUnixSocketTests.test_create_unix_server_path_dgram.<locals>.<lambda>)ri rj z"A UNIX Domain Stream.*was expected)rd ra Z
SOCK_DGRAMr rf r0 r rg )r rj rk r r r �"test_create_unix_server_path_dgram6 s zCSelectorEventLoopUnixSocketTests.test_create_unix_server_path_dgram�
SOCK_NONBLOCKz$no socket.SOCK_NONBLOCK (linux only)c C s� t j t jt jt jB �}tj� � }|j}W d Q R X zX|�J |j|� | jj dd� d |d�}| jj
|�}|j � | jj
|j � � W d Q R X W d t
j|� X d S )Nc S s d S )Nr r r r r r% I s z^SelectorEventLoopUnixSocketTests.test_create_unix_server_path_stream_bittype.<locals>.<lambda>)ri rj )rd ra Z SOCK_STREAMr| rq rr rs re r rf rg r rh �os�unlink)r rj rt �fnrk rl r r r �+test_create_unix_server_path_stream_bittype? s
zLSelectorEventLoopUnixSocketTests.test_create_unix_server_path_stream_bittypec C sV t j � }|�@ | jjdd� d |d�}| jtd�� | jj|� W d Q R X W d Q R X d S )Nc S s d S )Nr r r r r r% T s z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_path_inetsock.<locals>.<lambda>)ri rj z"A UNIX Domain Stream.*was expected)rd r �create_unix_connectionr0 r rg )r rj rk r r r �)test_create_unix_connection_path_inetsockQ s zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_inetsockzasyncio.unix_events.socketc C s� t j� }||j_t|j_| jjdd� dd�}| j t�� | jj
|� W d Q R X | j |j j
� t|j_| jjdd� dd�}| j t�� | jj
|� W d Q R X | j |j j
� d S )Nc S s d S )Nr r r r r r% a s zUSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_error.<locals>.<lambda>z/test)ri c S s d S )Nr r r r r r% g s )r r rd r[ r@ re r' r rf r rg rK r rE � MemoryError)r Zm_socketrj rk r r r �"test_create_unix_server_bind_errorZ s zCSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_errorc C sB | j jdd� tjt� d�}| jtd�� | j j|� W d Q R X d S )Nc S s d S )Nr r r r r r% n s zXSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sock.<locals>.<lambda>)rj zpath and sock can not be)r r� r} �devnullrO r0 r rg )r rk r r r �%test_create_unix_connection_path_sockl s zFSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sockc C s: | j jdd� d �}| jtd�� | j j|� W d Q R X d S )Nc S s d S )Nr r r r r r% t s z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosock.<locals>.<lambda>z no path and sock were specified)r r� r0 r rg )r rk r r r �)test_create_unix_connection_nopath_nosockr s
zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosockc C s@ | j jdd� tjdd�}| jtd�� | j j|� W d Q R X d S )Nc S s d S )Nr r r r r r% { s z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhost.<locals>.<lambda>rv )Zserver_hostnamez"server_hostname is only meaningful)r r� r} r� r0 r rg )r rk r r r �,test_create_unix_connection_nossl_serverhosty s
zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhostc C s@ | j jdd� tjdd�}| jtd�� | j j|� W d Q R X d S )Nc S s d S )Nr r r r r r% � s z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhost.<locals>.<lambda>T)rw z/you have to pass server_hostname when using ssl)r r� r} r� r0 r rg )r rk r r r �,test_create_unix_connection_ssl_noserverhost� s
zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhost)r; r< r= r
rm �unittest�
skipUnless�hasattrr} rp ru rx ry rz r{ rd r� r� r r_ r� r� r� r� r� r` r r )r r rb � s rb c s� e Zd Z� fdd�Zd dd�Zdd� Zejd�d d
� �Zejd�d d � �Z ejd�d
d� �Z
ejd�ejd�dd� ��Z ejd�dd� �Z ejd�dd� �Z
ejd�dd� �Zejd�dd� �Zejd�dd� �Zd d � Zd d � Z� ZS )!�UnixReadPipeTransportTestsc s� t � j� | j� | _tjtj�| _t j
t j d�| _
d| j
j_t jd�}|j� | j|j� t jd�}|j� }t j
� }tj|_||_| j|j� d S )N)�spec_set� z$asyncio.unix_events._set_nonblockingzos.fstat)r r
�
new_test_loopr r �make_test_protocolr ZProtocol�protocolr r �io� RawIOBase�pipe�filenor[ r_ �startr/ �stop�stat�S_IFIFO�st_mode)r �blocking_patcher�
fstat_patcher�m_fstat�st)r r r r
� s
z UnixReadPipeTransportTests.setUpNc C s( t j| j| j| j|d�}| jt|� |S )N)�waiter)r Z_UnixReadPipeTransportr r� r� r/ r
)r r� r r r r �read_pipe_transport� s
z.UnixReadPipeTransportTests.read_pipe_transportc C sV t j| jd�}| j|d�}| jj|� | jjj|� | jjd|j � | j
|j � � d S )N)r )r� r� ) r �Futurer r� rg r� �connection_mader" �
assert_reader� _read_ready� assertIsNone�result)r r� �trr r r � test_ctor� s z$UnixReadPipeTransportTests.test_ctorzos.readc C s6 | j � }d|_|j� |jd|j� | jjjd� d S )Ns datar� )r� r[ r� r" �max_sizer� �
data_received)r �m_readr� r r r �test__read_ready� s
z+UnixReadPipeTransportTests.test__read_readyc C s\ | j � }d|_|j� |jd|j� | j| jj� tj | j� | j
j j� | j
j jd � d S )N� r� )
r� r[ r� r" r� rC r �readersr � run_brieflyr� Z eof_received�connection_lost)r r� r� r r r �test__read_ready_eof� s z/UnixReadPipeTransportTests.test__read_ready_eofc C sD | j � }t|_|j� |jd|j� tj| j� | j | j
j j � d S )Nr� )
r� �BlockingIOErrorr' r� r" r� r r� r rC r� r� rE )r r� r� r r r �test__read_ready_blocked� s z3UnixReadPipeTransportTests.test__read_ready_blockedzasyncio.log.logger.errorc C s^ | j � }t� }||_tj� |_|j� |jd|j� |jj|� |jt j
d�tt t fd� d S )Nr� z;Fatal read error on pipe transport
protocol:.*
transport:.*)�exc_info) r� r@ r' r r �_closer� r" r� r � MockPattern�MOCK_ANY)r r� �m_logexcr� �errr r r �test__read_ready_error� s
z1UnixReadPipeTransportTests.test__read_ready_errorc C s8 | j � }tj� }| jjd|� |j� | j| jj� d S )Nr� )r� r r r �
add_readerZ
pause_readingrC r� )r r� r� �mr r r �test_pause_reading� s
z-UnixReadPipeTransportTests.test_pause_readingc C s$ | j � }|j� | jjd|j� d S )Nr� )r� Zresume_readingr r� r� )r r� r� r r r �test_resume_reading� s z.UnixReadPipeTransportTests.test_resume_readingc C s* | j � }tj� |_|j� |jjd � d S )N)r� r r r� r r" )r r� r� r r r rY � s
z%UnixReadPipeTransportTests.test_closec C s2 | j � }d|_tj� |_|j� | j|jj� d S )NT)r� �_closingr r r� r rC rE )r r� r� r r r �test_close_already_closing� s
z5UnixReadPipeTransportTests.test_close_already_closingc C sR | j � }t� }|j|� | j|j� � | j| jj� tj | j� | j
j j |� d S )N)
r� rO r� rK �
is_closingrC r r� r r� r� r� r" )r r� r� r� r r r � test__close� s
z&UnixReadPipeTransportTests.test__closec C sd | j � }| j|j� | j|j� d }|j|� | jjj|� | jj j� | j
|j� | j
|j� d S )N) r� �assertIsNotNone� _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|j� | j|j� t� }|j|� | jjj|� | j j
j� | j |j� | j |j� d S )N) r� r� r� r� r@ r� r� 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_err)N)r; r< r= r
r� r� r r_ r� r� r� r� r� r� rY r� r� r� r� r` r r )r r r� � s
r� c s~ e Zd Z� fdd�Zd7dd�Zdd� Zdd � Zejd
�d d � �Z ejd
�d
d� �Z
ejd
�dd� �Z ejd
�dd� �Z ejd
�dd� �Z
ejd�ejd
�dd� ��Zejd
�dd� �Zdd� Zejd
�d d � �Zejd
�d d � �Zejd
�d d!� �Zejd
�d"d#� �Zejd$�ejd
�d%d&� ��Zejd
�d'd(� �Zejd
�d)d*� �Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z � Z S )8�UnixWritePipeTransportTestsc s� t � j� | j� | _tjtj�| _t j
t j d�| _
d| j
j_t jd�}|j� | j|j� t jd�}|j� }t j
� }tj|_||_| j|j� d S )N)r� r� z$asyncio.unix_events._set_nonblockingzos.fstat)r r
r� r r r� r Z BaseProtocolr� r r r� r� r� r� r[ r_ r� r/ r� r� �S_IFSOCKr� )r r� r� r� r� )r r r r
" s
z!UnixWritePipeTransportTests.setUpNc C s( t j| j| j| j|d�}| jt|� |S )N)r� )r Z_UnixWritePipeTransportr r� r� r/ r
)r r� r r r r �write_pipe_transport4 s
z0UnixWritePipeTransportTests.write_pipe_transportc C sX t j| jd�}| j|d�}| jj|� | jjj|� | jjd|j � | j
d |j � � d S )N)r )r� r� ) r r� r r� rg r� r� r" r� r� r6 r� )r r� r� r r r r� ; s z%UnixWritePipeTransportTests.test_ctorc C s | j � }| j|j� � d S )N)r� rK Z
can_write_eof)r r� r r r �test_can_write_eofD s z.UnixWritePipeTransportTests.test_can_write_eofzos.writec C sF | j � }d|_|jd� |jdd� | j| jj� | jt� |j � d S )N� s datar� )
r� r[ �writer" rC r �writersr6 � bytearray�_buffer)r �m_writer� r r r �
test_writeH s
z&UnixWritePipeTransportTests.test_writec C sB | j � }|jd� | j|j� | j| jj� | jtd�|j� d S )Nr� ) r� r� rC rE r r� r6 r� r� )r r� r� r r r �test_write_no_dataQ s
z.UnixWritePipeTransportTests.test_write_no_datac C s> | j � }d|_|jd� | jjd|j� | jtd�|j� d S )NrI s datar� s ta) r� r[ r� r �
assert_writer� _write_readyr6 r� r� )r r� r� r r r �test_write_partialY s
z.UnixWritePipeTransportTests.test_write_partialc C s^ | j � }| jjd|j� td�|_|jd� | j|j� | jj d|j� | j
td�|j� d S )Nr� s previouss datas previousdata) r� r �
add_writerr� r� r� r� rC rE r� r6 )r r� r� r r r �test_write_buffera s
z-UnixWritePipeTransportTests.test_write_bufferc C sP | j � }t� |_|jd� |jdtd�� | jjd|j� | j td�|j
� d S )Ns datar� ) r� r� r' r� r" r� r r� r� r6 r� )r r� r� r r r �test_write_againk s
z,UnixWritePipeTransportTests.test_write_againzasyncio.unix_events.loggerc C s� | j � }t� }||_tj� |_|jd� |jdd� | j| j j
� | j t � |j
� |jj|d� | j d|j� |jd� | j d|j� |jd� |jd� |jd� |jd� |jjd� |j� d S )Ns datar� z#Fatal write error on pipe transportr rI z=pipe closed by peer or os.write(pipe, data) raised exception.)r� r@ r' r r Z _fatal_errorr� r" rC r r� r6 r� r� �
_conn_lost�warningr )r r� Zm_logr� r� r r r �test_write_errt s*
z*UnixWritePipeTransportTests.test_write_errc C sD | j � }|j� |jd� | j|jd� |jd� | j|jd� d S )Ns datar rI )r� r� r� r6 r� )r r� r� r r r �test_write_close� s
z,UnixWritePipeTransportTests.test_write_closec C sX | j � }|j� | j| jj� | j| jj� | j|j� � tj | j� | j
j j d � d S )N)
r� r� rC r r� r� rK r� r r� r� r� r" )r r� r r r r� � s z,UnixWritePipeTransportTests.test__read_readyc C sR | j � }| jjd|j� td�|_d|_|j� | j| jj� | j t� |j� d S )Nr� s datar� )
r� r r� r� r� r� r[ rC r� r6 )r r� r� r r r �test__write_ready� s
z-UnixWritePipeTransportTests.test__write_readyc C sV | j � }| jjd|j� td�|_d|_|j� | jjd|j� | jtd�|j� d S )Nr� s data� � a) r� r r� r� r� r� r[ r� r6 )r r� r� r r r �test__write_ready_partial� s
z5UnixWritePipeTransportTests.test__write_ready_partialc C sh | j � }| jjd|j� td�|_t� |_|j� |jdtd�� | jj d|j� | j
td�|j� d S )Nr� s data) r� r r� r� r� r� r� r' r" r� r6 )r r� r� r r r �test__write_ready_again� s
z3UnixWritePipeTransportTests.test__write_ready_againc C sf | j � }| jjd|j� td�|_d|_|j� |jdtd�� | jjd|j� | j td�|j� d S )Nr� s datar )
r� r r� r� r� r� r[ r" r� r6 )r r� r� r r r �test__write_ready_empty� s
z3UnixWritePipeTransportTests.test__write_ready_emptyzasyncio.log.logger.errorc C s� | j � }| jjd|j� td�|_t� |_}|j� | j| jj � | j| jj
� | j t� |j� | j |j
� � |jtjd�tttfd� | j d|j� tj| j� | jjj|� d S )Nr� s dataz<Fatal write error on pipe transport
protocol:.*
transport:.*)r� r )r� r r� r� r� r� r@ r' rC r� r� r6 rK r� r" r r� r� 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 � }| jjd|j� d|_td�|_d|_|j� | j| jj � | j| jj
� | j t� |j� | j j
jd � | jjj� d S )Nr� Ts datar� )r� r r� r� r� r� r� r[ rC r� r� r6 r� r� r" r� r )r r� r� r r r �test__write_ready_closing� s
z5UnixWritePipeTransportTests.test__write_ready_closingc C s� | j � }| jjd|j� | jjd|j� ddg|_|j� | j|j � | j| jj
� | j| jj � | j g |j� | j
|j� � tj| j� | jjjd � d S )Nr� s das ta)r� r r� r� r� r� r� �abortrC rE r� r� r6 rK r� r r� r� r� r" )r r� r� r r r �
test_abort� s
z&UnixWritePipeTransportTests.test_abortc C sd | j � }| j|j� | j|j� d }|j|� | jjj|� | jj j� | j
|j� | j
|j� d S )N) r� r� r� r� r� r� r� r" r� r r� )r r� r� r r r r� � s
z6UnixWritePipeTransportTests.test__call_connection_lostc C sf | j � }| j|j� | j|j� t� }|j|� | jjj|� | j j
j� | j |j� | j |j� d S )N) r� r� r� r� r@ r� r� r� r" r� r r� )r r� r� r r r r� s
z?UnixWritePipeTransportTests.test__call_connection_lost_with_errc C s0 | j � }tj� |_|j� |jj� |j� d S )N)r� r r � write_eofr r" )r r� r r r rY s
z&UnixWritePipeTransportTests.test_closec C s2 | j � }tj� |_d|_|j� | j|jj� d S )NT)r� r r r� r� r rC rE )r r� r r r �test_close_closing! s
z.UnixWritePipeTransportTests.test_close_closingc C sJ | j � }|j� | j|j� � | j| jj� tj| j� | j j
j d � d S )N) r� r� rK r� rC r r� r r� r� r� r" )r r� r r r �test_write_eof( s z*UnixWritePipeTransportTests.test_write_eofc C s: | j � }dg|_|j� | j|j� � | j| jjj� d S )Ns data) r� r� r� rK r� rC r� r� rE )r r� r r r �test_write_eof_pending0 s
z2UnixWritePipeTransportTests.test_write_eof_pending)N) r; r< r= r
r� r� r� r r_ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rY r� r� r� r` r r )r r r� s6
r� c @ s e Zd Zdd� ZdS )�AbstractChildWatcherTestsc C sv t j� }tj� }| jt|j||� | jt|j|� | jt|j|� | jt|j � | jt|j
� | jt|j |||� d S )N) r r r ZAbstractChildWatcherr �NotImplementedError�add_child_handler�remove_child_handler� attach_loopr � __enter__�__exit__)r �f�watcherr r r �test_not_implemented: s
z.AbstractChildWatcherTests.test_not_implementedN)r; r< r= r� r r r r r� 8 s r� c @ s e Zd Zdd� ZdS )�BaseChildWatcherTestsc C s$ t j� }tj� }| jt|j|� d S )N)r r r ZBaseChildWatcherr r� Z _do_waitpid)r r� r� r r r r� M s z*BaseChildWatcherTests.test_not_implementedN)r; r< r= r� r r r r r� K s r� � WaitPidMocks�waitpid� WIFEXITED� WIFSIGNALED� WEXITSTATUS�WTERMSIGc s e Zd Zejjejd�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 ed0d1� �Z � Z!S )2�ChildWatcherTestsMixinr� c sZ t � j� | j� | _d| _i | _tjj| jd�� | _ | j
� | _ | j j | j� W d Q R X d S )NFr) )
r r
r� r �running�zombiesr r_ rO �m_add_signal_handler�create_watcherr� r� )r )r r r r
a s
z ChildWatcherTestsMixin.setUpc C sr t | jtj�s|dkr"| j|d� y&|dk r6| jj� S || jj|�fS W n tk
r\ Y nX | j rhdS t
� �d S )Nr r r9 )r r ) �
isinstancer� r �SafeChildWatcherZ
assertGreaterr �popitem�pop�KeyErrorr �ChildProcessError)r �pid�flagsr r r r� l s
z ChildWatcherTestsMixin.waitpidc C s |d | j |< d S )Ni � )r )r r
�
returncoder r r �
add_zombie{ s z!ChildWatcherTestsMixin.add_zombiec C s |dkS )Ni � r )r �statusr r r r� ~ s z ChildWatcherTestsMixin.WIFEXITEDc C s d| k odk S S )Ni� i � r )r r r r r r� � s z"ChildWatcherTestsMixin.WIFSIGNALEDc C s | j | j|�� |d S )Ni � )rK r� )r r r r r r� � s z"ChildWatcherTestsMixin.WEXITSTATUSc C s | j | j|�� d| S )Ni � )rK r� )r r r r r r� � s z ChildWatcherTestsMixin.WTERMSIGc C s | j jtj| jj� d S )N)r rX r rV r� � _sig_chld)r 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 d Q R X W d Q R X W d Q R X W d Q R X W d Q R X d S )Nc S s t j| |t jd�S )N)�wrapsZ new_callable)r r_ r )�target�wrapperr r r r_ � s zIChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_func.<locals>.patchz os.WTERMSIGzos.WEXITSTATUSzos.WIFSIGNALEDz os.WIFEXITEDz
os.waitpid)r� r� r� r� r� r� )r r_ Z
m_WTERMSIGZ
m_WEXITSTATUSZ
m_WIFSIGNALEDZ m_WIFEXITEDZ m_waitpid)r1 r r � wrapped_func� s z:ChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_funcr )r1 r r )r1 r �
waitpid_mocks� s z$ChildWatcherTestsMixin.waitpid_mocksc C s t j� }| j� d| _| jjd|ddd� W d Q R X | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | jj � | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d| _| j dd� | jj � | j
|jj� | j
|j j� | j|j
j� |jddddd� |jj� |jj� |j j� |j� | j dd� | j� | jj � W d Q R X | j|j� | j|j
j� |jj� |jj� |j j� | jj� | jj � | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d S ) NT�* � �
� F� �
)r r r� r r� rC rE r� r� r� r� r r
rK rX rP �ignore_warningsr �clear)r r� �callbackr r r � test_sigchld� sR
z#ChildWatcherTestsMixin.test_sigchldc C s� t j� }t j� }| j� d| _| jjd|dd� W d Q R X | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | j� | jjd|dd� W d Q R X | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | jj � | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | j dd
� | jj � |j
dddd� | j|j� | j|jj� | j|j j� | j|j
j� |jj� |jj� |j
j� |j� | jj � | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | j dd � d
| _| jj � |j
dd dd� | j|j� | j|jj� | j|j j� | j|j
j� |jj� |jj� |j j� |j� | j dd � | j dd � | j� | jj � W d Q R X | j|j� | j|j� | j|j
j� |jj� |jj� |j j� | jj� | jj � | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d S )NT�+ � � �, � � r� �l Fr � ���r( )r r r� r r� rC rE r� r� r� r� r r
rX rK rP r r r )r r� � callback1� callback2r r r �test_sigchld_two_children� s�
z0ChildWatcherTestsMixin.test_sigchld_two_childrenc C sn t j� }t j� }| j� d| _| jjd|dd� W d Q R X | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | j� | jjd|dd� W d Q R X | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | jj � | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | j dd� | j dd
� d
| _| jj � |j
dddd� |j
dddd� | j|jj� | j|jj� | j|j j� | j|j
j� |jj� |jj� |j
j� |j j� |j� |j� | j dd � | j dd � | j� | jj � W d Q R X | j|j� | j|j� | j|j
j� d S )NT�- � r" �. i{ r% �N r� Fr r' ���r0 )r r r� r r� rC rE r� r� r� r� r r
rX rK rP r )r r� r) r* r r r �.test_sigchld_two_children_terminating_togetherF sb
zEChildWatcherTestsMixin.test_sigchld_two_children_terminating_togetherc C s� t j� }| j�. | jdd� | jj� | jjd|dd� W d Q R X |jdddd� |j� | jdd� | j� | jj� W d Q R X | j |j
� d S )N�2 r� r r r9 ) r r r� r
r r� rX rP r rC rE )r r� r r r r �test_sigchld_race_condition� s
z2ChildWatcherTestsMixin.test_sigchld_race_conditionc C s� t j� }t j� }| j� d| _| jjd|d� W d Q R X | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | j� | jjd|d� W d Q R X | j|j� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d| _| j dd� | jj � |j
dd d� | j|j� | j|jj� | j|j j� | j|j
j� |jj� |jj� |j
j� |j� | j dd� | j� | jj � W d Q R X | j|j� | j|j� | j|j
j� d S )
NT�3 � � Fr" r i����i����)r r r� r r� rC rE r� r� r� r� r
r rX rK rP r )r r� r) r* r r r � test_sigchld_replace_handler� sJ
z3ChildWatcherTestsMixin.test_sigchld_replace_handlerc C s t j� }| j� d| _| jjd|d� W d Q R X | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� | jj d� | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d| _| j dd� | j
� | jj� W d Q R X | j|j� d S )NT�4 i� F�c )r r r� r r� rC rE r� r� r� r� r� r
r r )r r� r r r r �test_sigchld_remove_handler� s( z2ChildWatcherTestsMixin.test_sigchld_remove_handlerc C s( t j� }| j� d| _| jjd|d� W d Q R X | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d| j d< d| _| jj � |j
ddd� | j|jj� | j|jj� | j|j j� | j|j
j� |j� |jj� |jj� | jdd� | j� | jj � W d Q R X | j|j� d S ) NT�5 r5 i� F�e i��i��)r r r� r r� rC rE r� r� r� r� r r rX rK rP r
r )r r� r r r r �test_sigchld_unknown_status� s0
z2ChildWatcherTestsMixin.test_sigchld_unknown_statusc C s t j� }t j� }t j� }| j�< d| _| jjd|d� | jjd|d� | jjd|d� W d Q R X | j| jjd�� | j| jjd�� | j| jjd�� | j| jjd�� | jdd� | jdd� | jdd� d | _| j � | jj
� W d Q R X | j|j � | j|j � |j ddd� d S )
NT�6 r �7 rI �8 r� r F)
r r r� r r� rK r� rC r
r r rE rX )r r� r) r* � callback3r r r �test_remove_child_handler s* z0ChildWatcherTestsMixin.test_remove_child_handlerc C st t j� }| j� d| _| jjd|� W d Q R X t|j_t jj t
j d��$}| j | jj
� d � | j|j� W d Q R X d S )NT�9 �error)r r r� r r� r r� r' r_ rO r �loggerr6 r rK rE )r r� r Zm_errorr r r � test_sigchld_unhandled_exception< s
z7ChildWatcherTestsMixin.test_sigchld_unhandled_exceptionc C s� t j� }| j� d| _| jjd|� W d Q R X | j|j� | j|jj� | j|jj� | j|j j� | j|j
j� d| _| j dd� t j
dt j� |j
j� | j� | jj� W d Q R X t| jtj�r�| j|j� n |jdd� d S )NT�: Fr� � )r r r� r r� rC rE r� r� r� r� r
r} r� �WNOHANGrP r r r r �FastChildWatcherrX )r r� r r r r �#test_sigchld_child_reaped_elsewhereN s$
z:ChildWatcherTestsMixin.test_sigchld_child_reaped_elsewherec C s� t j� }t j� }| j�\ | j�J d| _| jdd� | jdd� | jj� | jjd|� | jjd|� W d Q R X W d Q R X |jdd� | j |j
� d S )NTiO r! iQ r- iP ) r r r r� r r
r r� rX rC rE )r r� r) r* r r r �,test_sigchld_unknown_pid_during_registrationq s
" zCChildWatcherTestsMixin.test_sigchld_unknown_pid_during_registrationc C s� t j� }| j� d| _| jjd|� W d Q R X | j}| j� | _t jj}||d��J}|| jd��2}| jj | j� |j
t j � |j
t j | jj
� W d Q R X W d Q R X d| _| jdd� | jj
� |j
dd� d S )NT�< r! r) Fr )r r r� r r� r r� r_ rO r� rX r rV r r
)r r� r �old_loopr_ Z m_old_removeZ m_new_addr r r �
test_set_loop� s$
"
z$ChildWatcherTestsMixin.test_set_loopc C s� t j� }t j� }t j� }| j�6 d| _| jjd|� | jjd|� | jjd|� W d Q R X | j}d | _t jj|d��6}| jt d�� | jj
d � W d Q R X |j t j
� W d Q R X | jdd� | jdd � | j|j� | j|j� | j|j� | j� | _t jj| jd ��J}| jj
| j� |j t j
| jj� |j dd� |j dd
� | j|j� W d Q R X |j� |j� d
| _| jdd � | jj� | j|j� | j|j� |j dd � d S )NT�= �> in r! zA loop is being detached� r� r) Fr5 r0 r0 )r r r� r r� r r_ rO r\ �RuntimeWarningr� rX r rV r
rC rE r� r rP )r r� r) r* rA rN �m_remove_signal_handlerr r r r � test_set_loop_race_condition� sN
z3ChildWatcherTestsMixin.test_set_loop_race_conditionc C s� t j� }| j�� d| _| jdd� | jdd� | jj� | jjd|� | jjd|� | jt| jj �d� t
| jt j �r�| jt| jj
�d� t jj| jd��F}| jj� |jtj� | j| jj � t
| jt j �r�| j| jj
� W d Q R X W d Q R X d S ) NT�? r �A r% �@ r r! )r r r� r r
r r� r6 rW Z
_callbacksr r rJ Z_zombiesr_ rO r r rX r rV rC )r r� r) rT r r r rY � s(
z!ChildWatcherTestsMixin.test_closec C sD t j� }| j� �*}| jtd�� |jd|� W d Q R X W d Q R X d S )Nz/the child watcher does not have a loop attached�d )r r r r0 r( r� )r r� r r� r r r �,test_add_child_handler_with_no_loop_attached s
zCChildWatcherTestsMixin.test_add_child_handler_with_no_loop_attached)"r; r< r= r r_ rO r rE r r
r� r
r� r� r� r� r r r r+ r1 r3 r7 r: r= rB rF rK rL rO rU rY rZ r` r r )r r r� ] s2 = h E 5 % ! # > r� c @ s e Zd Zdd� ZdS )�SafeChildWatcherTestsc C s t j� S )N)r r )r r r r r s z$SafeChildWatcherTests.create_watcherN)r; r< r= r r r r r r[
s r[ c @ s e Zd Zdd� ZdS )�FastChildWatcherTestsc C s t j� S )N)r rJ )r r r r r s z$FastChildWatcherTests.create_watcherN)r; r< r= r r r r r r\ s r\ c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� PolicyTestsc C s t j� S )N)r ZDefaultEventLoopPolicy)r r r r �
create_policy s zPolicyTests.create_policyc C sX | j � }| j|j� |j� }| j|tj� | j|j|� | j||j� � | j|j� d S )N) r^ r� �_watcher�get_child_watcherr5 r r �assertIsr� )r �policyr� r r r �test_get_child_watcher s z"PolicyTests.test_get_child_watcherc C s< | j � }tj� }|j|� | j|j|� | j||j� � d S )N)r^ r rJ Zset_child_watcherra r_ r` )r rb r� r r r � test_get_child_watcher_after_set( s
z,PolicyTests.test_get_child_watcher_after_setc C sL | j � }|j� }| j|j� |j� }| j|tj� | j|j |� |j
� d S )N) r^ �get_event_loopr� r_ r` r5 r r ra r� r )r rb r r� r r r �-test_get_child_watcher_with_mainloop_existing0 s z9PolicyTests.test_get_child_watcher_with_mainloop_existingc s6 � �fdd�}�j � � tj|d�}|j� |j� d S )Nc sR � j � j� � �j� j� tj� � j� } �j| tj� �j| j � � j� j
� d S )N) r �new_event_loopr5 re r ZAbstractEventLoopr` r r� r� r )r� )rb r r r r� > s
z4PolicyTests.test_get_child_watcher_thread.<locals>.f)r )r^ � threadingZThreadr� �join)r r� Zthr )rb r r � test_get_child_watcher_thread< s
z)PolicyTests.test_get_child_watcher_threadc C sr | j � }|j� }|j� }| j|j|� |j� }|j|� | j|j|� |jd � | j|jd � |j� |j� d S )N)r^ re r` ra r� rg r r )r rb r r� Znew_loopr r r �,test_child_watcher_replace_mainloop_existingP s
z8PolicyTests.test_child_watcher_replace_mainloop_existingN) r; r< r= r^ rc rd rf rj rk r r r r r] s r] �__main__)r� r� r� r� r� ))�__doc__� collectionsr> r� r} ro r rd r� �sysrq rh r� �warningsr �platformZSkipTestr r r r �ANYr� r
r� ZTestCaser r� rb r� r� r� r� �
namedtupler� r� r[ r\ r] r; �mainr r r r �<module> sj
S
5N