Your IP : 216.73.216.213
�
f fA� � @ sq d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l m Z e j
d k r� e j d � � n d d l Z d d l m Z d d l m Z d d l m Z e j Z d d
� Z e j e d � Gd d
� d
e j � � Z e j e e d � d � Gd d � d e j � � Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z e j d d d d d d f � Z Gd d! � d! � Z! Gd" d# � d# e! e j � Z" Gd$ d% � d% e! e j � Z# Gd& d' � d' e j � Z$ e% d( k rme j& � n d S))zTests for unix_events.py.� N)�mock�win32z UNIX only)�log)�
test_utils)� unix_eventsc C s- | j d k r d S| j j � d | _ d S)N)Z_pipe�close)� transport� r �G/opt/alt/python34/lib64/python3.4/test/test_asyncio/test_unix_events.py�close_pipe_transport s
r zSignals are not supportedc @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z e j d � d
d � � Z e j d � d d
� � Z
e j d � d d � � Z e j d � d d � � Z e j d � e j d � d d � � � Z
e j d � e j d � d d � � � Z e j d � d d � � Z e j d � d d � � Z e j d � e j d � d d � � � Z e j d � d d � � Z e j d � d d � � Z e j d � d! d" � � Z d# S)$� SelectorEventLoopSignalTestsc C s# t j � | _ | j | j � d S)N)�asyncio�SelectorEventLoop�loop�set_event_loop)�selfr r r
�setUp) s z"SelectorEventLoopSignalTests.setUpc C s= | j t | j j d � | j t | j j t j d � d S)N�1� )� assertRaises� TypeErrorr Z
_check_signal�
ValueError�signal�NSIG)r r r r
�test_check_signal- s z.SelectorEventLoopSignalTests.test_check_signalc C s | j j t j d � d S)Nr )r �_handle_signalr r )r r r r
� test_handle_signal_no_handler3 s z:SelectorEventLoopSignalTests.test_handle_signal_no_handlerc C s� t j t j � f d t j � �} | j � | | j j t j d <t j � | j _ | j j
t j d � | j j j t j d � d S)Nr r ) r
�Handler �MockZcancelr �_signal_handlersr r �remove_signal_handlerr �assert_called_with)r �hr r r
�$test_handle_signal_cancelled_handler6 s
zASelectorEventLoopSignalTests.test_handle_signal_cancelled_handlerzasyncio.unix_events.signalc C sA t j | _ t | j _ | j t | j j t j d d � � d S)Nc S s d S)NTr r r r r
�<lambda>G 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_error? s z@SelectorEventLoopSignalTests.test_add_signal_handler_setup_errorc C s} t j | _ t j d d � � } | } | � } | j | j � x6 | | f D]( } | j t d | j j t j
| � qM Wd S)Nc s s
g Dd Hd S)Nr r r r r
�simple_coroutineM s z^SelectorEventLoopSignalTests.test_add_signal_handler_coroutine_error.<locals>.simple_coroutinez1coroutines cannot be used with add_signal_handler) r r r
Z coroutine�
addCleanupr �assertRaisesRegexr r r( r) )r r* r, Z coro_funcZcoro_obj�funcr r r
�'test_add_signal_handler_coroutine_errorI s zDSelectorEventLoopSignalTests.test_add_signal_handler_coroutine_errorc C sp t j | _ d d � } | j j t j | � | j j j t j � } | j | t j � | j
| j | � d S)Nc S s d S)NTr r r r r
r$ _ 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 | _ d d � } | | _ Gd d � d t � } | | j _ | j | | j j t j d d � � d S)Nc S s | d k r t � � n d S)Nr ���)r )�fdr r r
r% i s zYSelectorEventLoopSignalTests.test_add_signal_handler_install_error.<locals>.set_wakeup_fdc @ s e Z d Z e j Z d S)zOSelectorEventLoopSignalTests.test_add_signal_handler_install_error.<locals>.ErrN)�__name__�
__module__� __qualname__�errnoZEFAULTr r r r
�Errn s r= c S s d S)NTr r r r r
r$ u 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_errore s zBSelectorEventLoopSignalTests.test_add_signal_handler_install_errorzasyncio.base_events.loggerc C s� t j | _ Gd d � d t � } | | j _ d d � | j j t j <| j t | j j t j
d d � � | j | j j
� | j d | j j � d S)Nc @ s e Z d Z e j Z d S)zPSelectorEventLoopSignalTests.test_add_signal_handler_install_error2.<locals>.ErrN)r9 r: r; r< �EINVALr r r r
r= | s r= c S s d S)NTr r r r r
r$ � s zUSelectorEventLoopSignalTests.test_add_signal_handler_install_error2.<locals>.<lambda>c S s d S)NTr r r r r
r$ � s r )r r r> r& r r r1 r r' r( r) � assertFalse�info�calledr4 r% �
call_count)r � m_loggingr* r= r r r
�&test_add_signal_handler_install_error2w s zCSelectorEventLoopSignalTests.test_add_signal_handler_install_error2c C s� Gd d � d t � } | | j _ t j | _ | j t | j j t j d d � � | j | j
j � | j d | j
j � d S)Nc @ s e Z d Z e j Z d S)zPSelectorEventLoopSignalTests.test_add_signal_handler_install_error3.<locals>.ErrN)r9 r: r; r< r@ r r r r
r= � s r= c S s d S)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) rA rB rC r4 r% rD )r rE r* r= r r r
�&test_add_signal_handler_install_error3� s zCSelectorEventLoopSignalTests.test_add_signal_handler_install_error3c C s� t j | _ | j j t j d d � � | j | j j t j � � | j | j j � | j | j j � | j t j | j
f | j j d � d S)Nc S s d S)NTr r r r r
r$ � s zISelectorEventLoopSignalTests.test_remove_signal_handler.<locals>.<lambda>r ) r r r r( r1 �
assertTruer r% rC r4 �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 | _ | j j t j d d � � t � | j j t j <| j j � | j
| j j t j � � | j | j j
� | j
| j j
� | j t j | j f | j j d � d S)Nc S s d S)NTr r r r r
r$ � s zKSelectorEventLoopSignalTests.test_remove_signal_handler_2.<locals>.<lambda>r )r r r) r r( �objectr r1 r% �
reset_mockrI r rA rC r4 �default_int_handlerrK )r r* r r r
� test_remove_signal_handler_2� s
z9SelectorEventLoopSignalTests.test_remove_signal_handler_2c C s[ t j | _ | j j t j d d � � t | j _ | j j t j � | j | j
� d S)Nc S s d S)NTr r r r r
r$ � s zWSelectorEventLoopSignalTests.test_remove_signal_handler_cleanup_error.<locals>.<lambda>) r r r r( r1 r r% r& r rI rB )r rE r* r r r
�(test_remove_signal_handler_cleanup_error� s
zESelectorEventLoopSignalTests.test_remove_signal_handler_cleanup_errorc C sT t j | _ | j j t j d d � � t | j _ | j t | j j t j � d S)Nc S s d S)NTr r r r r
r$ � s zOSelectorEventLoopSignalTests.test_remove_signal_handler_error.<locals>.<lambda>) r r r r( r1 r> r& r r )r r* r r r
� test_remove_signal_handler_error� s
z=SelectorEventLoopSignalTests.test_remove_signal_handler_errorc C sj t j | _ | j j t j d d � � Gd d � d t � } | | j _ | j t | j j t j � d S)Nc S s d S)NTr r r r r
r$ � s zPSelectorEventLoopSignalTests.test_remove_signal_handler_error2.<locals>.<lambda>c @ s e Z d Z e j Z d S)zKSelectorEventLoopSignalTests.test_remove_signal_handler_error2.<locals>.ErrN)r9 r: r; r< r@ r r r r
r= � s r= )
r r r r( r1 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 | _ | j j t j d d � � | j j t j d d � � | j t | j j � d � | j j
� | j j � | j t | j j � d � | j j d � d S)Nc S s d S)NTr r r r r
r$ � s z9SelectorEventLoopSignalTests.test_close.<locals>.<lambda>c S s d S)NTr r r r r
r$ � s rG r r r7 )
r r r r( r1 �SIGCHLDr4 �lenr r% rN r �assert_called_once_with)r r* r r r
�
test_close� s
z'SelectorEventLoopSignalTests.test_closeN)r9 r: r; r r r r# r �patchr+ r0 r6 r? rF rH rL rP rQ rR rS rW r r r r
r & s&
r �AF_UNIXz UNIX Sockets are not supportedc @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z e j
d
� d d � � Z d d � Z d d � Z
d d � Z d d � Z d S)� SelectorEventLoopUnixSocketTestsc C s# t j � | _ | j | j � d S)N)r
r r r )r r r r
r � s z&SelectorEventLoopUnixSocketTests.setUpc C s� t j � �w } t j t j � } | j | � | �I | j j d d � | � } | j t d � � | j j | � Wd QXWd QXWd QXd S)Nc S s d S)Nr r r r r
r$ � s z]SelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_sock.<locals>.<lambda>zAddress.*is already in use)
r Zunix_socket_path�socketrY �bindr �create_unix_serverr. r> �run_until_complete)r �path�sock�coror r r
�*test_create_unix_server_existing_path_sock� s
zKSelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_sockc C s` t j � �N } | j j d d � | j � } | j t d � � | j j | � Wd QXWd QXd 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)�tempfileZNamedTemporaryFiler r] �namer. r> r^ )r �filera r r r
�-test_create_unix_server_existing_path_nonsock� s
zNSelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_nonsockc C sQ | j j d d � d d d d �} | j t d � � | j j | � Wd QXd S)Nc S s d S)Nr r r r r
r$ s zSSelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_bool.<locals>.<lambda>r_ �spam�sslTz"ssl argument must be an SSLContext)r r] r. r r^ )r ra r r r
� test_create_unix_server_ssl_bool s
zASelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_boolc C sK | j j d d � d d �} | j t d � � | j j | � Wd QXd S)Nc S s d S)Nr r r r r
r$ s zXSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosock.<locals>.<lambda>r_ z#path was not specified, and no sock)r r] r. r r^ )r ra r r r
�%test_create_unix_server_nopath_nosock s
zFSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosockc C sj t j � } | �R | j j d d � d d d | �} | j t d � � | j j | � Wd QXWd QXd S)Nc S s d S)Nr r r r r
r$ s zXSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsock.<locals>.<lambda>r_ r` z!A UNIX Domain Socket was expected)r[ r r] r. r r^ )r r` ra r r r
�%test_create_unix_server_path_inetsock
s
zFSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsockzasyncio.unix_events.socketc C s� t j � } | | j _ t | j _ | j j d d � d d �} | j t � � | j j
| � Wd QX| j | j j
� t | j _ | j j d d � d d �} | j t � � | j j
| � Wd QX| j | j j
� d S)Nc S s d S)Nr r r r r
r$ s zUSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_error.<locals>.<lambda>r_ z/testc S s d S)Nr r r r r
r$ # s )r r r[ � return_valuer> r\ r& r r] r r^ rI r rC � MemoryError)r Zm_socketr` ra r r r
�"test_create_unix_server_bind_error s zCSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_errorc C sT | j j d d � t j d t � �} | j t d � � | j j | � Wd QXd S)Nc S s d S)Nr r r r r
r$ * s zXSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sock.<locals>.<lambda>r` zpath and sock can not be)r �create_unix_connection�os�devnullrM r. r r^ )r ra r r r
�%test_create_unix_connection_path_sock( s zFSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sockc C sH | j j d d � d � } | j t d � � | j j | � Wd QXd S)Nc S s d S)Nr r r r r
r$ 0 s z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosock.<locals>.<lambda>z no path and sock were specified)r ro r. r r^ )r ra r r r
�)test_create_unix_connection_nopath_nosock. s
zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosockc C sQ | j j d d � t j d d �} | j t d � � | j j | � Wd QXd S)Nc S s d S)Nr r r r r
r$ 7 s z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhost.<locals>.<lambda>Zserver_hostnamerg z"server_hostname is only meaningful)r ro rp rq r. r r^ )r ra r r r
�,test_create_unix_connection_nossl_serverhost5 s
zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhostc C sQ | j j d d � t j d d �} | j t d � � | j j | � Wd QXd S)Nc S s d S)Nr r r r r
r$ > s z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhost.<locals>.<lambda>rh Tz/you have to pass server_hostname when using ssl)r ro rp rq r. r r^ )r ra r r r
�,test_create_unix_connection_ssl_noserverhost< s
zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhostN)r9 r: r; r rb rf ri rj rk r rX rn rr rs rt ru r r r r
rZ � s
rZ c @ sQ e Z d Z d d � Z d d d � Z d d � Z e j d � d d
� � Z e j d � d d � � Z e j d � d
d � � Z
e j d � e j d � d d � � � Z e j d � d d � � Z e j d � d d � � Z
e j d � d d � � Z e j d � d d � � Z e j d � d d � � Z d d � Z d d � Z d S) �UnixReadPipeTransportTestsc C s� | j � | _ t j t j � | _ t j d t j
� | _ d | j j _
t j d � } | j � | j | j � t j d � } | j � } t j � } t j | _ | | _
| j | j � d S)N�spec_set� z$asyncio.unix_events._set_nonblockingzos.fstat)�
new_test_loopr r �make_test_protocolr
ZProtocol�protocolr r �io� RawIOBase�pipe�filenorl rX �startr- �stop�stat�S_IFIFO�st_mode)r �blocking_patcher�
fstat_patcher�m_fstat�str r r
r H s
z UnixReadPipeTransportTests.setUpNc C s8 t j | j | j | j d | �} | j t | � | S)N�waiter)r Z_UnixReadPipeTransportr r~ r{ r- r )r r� r r r r
�read_pipe_transportY s
z.UnixReadPipeTransportTests.read_pipe_transportc C sw t j d | j � } | j d | � } | j j | � | j j j | � | j j d | j � | j
| j � � d S)Nr r� rx ) r
�Futurer r� r^ r{ �connection_mader! �
assert_reader� _read_ready� assertIsNone�result)r r� �trr r r
� test_ctor` s z$UnixReadPipeTransportTests.test_ctorzos.readc C sI | j � } d | _ | j � | j d | j � | j j j d � d S)Ns datarx )r� rl r� r! �max_sizer{ �
data_received)r �m_readr� r r r
�test__read_readyi s
z+UnixReadPipeTransportTests.test__read_readyc C s| | j � } d | _ | j � | j d | j � | j | j j � t j | j � | j
j j � | j
j j d � d S)N� rx )
r� rl r� r! r� rA r �readersr � run_brieflyr{ Z eof_received�connection_lost)r r� r� r r r
�test__read_ready_eofr s
z/UnixReadPipeTransportTests.test__read_ready_eofc C s\ | j � } t | _ | j � | j d | j � t j | j � | j | j
j j � d S)Nrx )
r� �BlockingIOErrorr& r� r! r� r r� r rA r{ r� rC )r r� r� r r r
�test__read_ready_blocked~ s
z3UnixReadPipeTransportTests.test__read_ready_blockedzasyncio.log.logger.errorc C s� | j � } t � } | | _ t j � | _ | j � | j d | j � | j j | � | j t j
d � d t t t f �d S)Nrx 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 sL | j � } t j � } | j j d | � | j � | j | j j � d S)Nrx )r� r r r �
add_readerZ
pause_readingrA r� )r r� r� �mr r r
�test_pause_reading� s
z-UnixReadPipeTransportTests.test_pause_readingc C s0 | j � } | j � | j j d | j � d S)Nrx )r� Zresume_readingr r� r� )r r� r� r r r
�test_resume_reading� s
z.UnixReadPipeTransportTests.test_resume_readingc C s9 | j � } t j � | _ | j � | j j d � d S)N)r� r r r� r r! )r r� r� r r r
rW � s
z%UnixReadPipeTransportTests.test_closec C sE | j � } d | _ t j � | _ | j � | j | j j � d S)NT)r� �_closingr r r� r rA rC )r r� r� r r r
�test_close_already_closing� s
z5UnixReadPipeTransportTests.test_close_already_closingc C so | j � } t � } | j | � | j | j � � | j | j j � t j | j � | j
j j | � d S)N)
r� rM r� rI �
is_closingrA r r� r r� r{ r� r! )r r� r� r� r r r
� test__close� s
z&UnixReadPipeTransportTests.test__closec C s� | j � } | j | j � | j | j � d } | j | � | j j j | � | j j 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 s� | j � } | j | j � | j | j � t � } | j | � | j j j | � | 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)r9 r: r; r r� r� r rX r� r� r� r� r� r� rW r� r� r� r� r r r r
rv F s
rv c @ s/ e Z d Z d d � Z d d d � Z d d � Z d d � Z e j d
� d d � � Z e j d
� d
d � � Z
e j d
� d d � � Z e j d
� d d � � Z e j d
� d d � � Z
e j d � e j d
� d d � � � Z e j d
� d d � � Z d d � Z e j d
� d d � � Z e j d
� d d � � Z e j d
� d d! � � Z e j d
� d" d# � � Z e j d$ � e j d
� d% d& � � � Z e j d
� d' d( � � Z e j d
� d) d* � � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d S)7�UnixWritePipeTransportTestsc C s� | j � | _ t j t j � | _ t j d t j
� | _ d | j j _
t j d � } | j � | j | j � t j d � } | j � } t j � } t j | _ | | _
| j | j � d S)Nrw rx z$asyncio.unix_events._set_nonblockingzos.fstat)ry r r rz r
Z BaseProtocolr{ r r r| r} r~ r rl rX r� r- r� r� �S_IFSOCKr� )r r� r� r� r� r r r
r � s
z!UnixWritePipeTransportTests.setUpNc C s8 t j | j | j | j d | �} | j t | � | S)Nr� )r Z_UnixWritePipeTransportr r~ r{ r- r )r r� r r r r
�write_pipe_transport� s
z0UnixWritePipeTransportTests.write_pipe_transportc C sz t j d | j � } | j d | � } | j j | � | j j j | � | j j d | j � | j
d | j � � d S)Nr r� rx ) r
r� r r� r^ r{ r� r! r� r� r4 r� )r r� r� r r r
r� � s z%UnixWritePipeTransportTests.test_ctorc C s# | j � } | j | j � � d S)N)r� rI Z
can_write_eof)r r� r r r
�test_can_write_eof� s z.UnixWritePipeTransportTests.test_can_write_eofzos.writec C s\ | j � } d | _ | j d � | j d d � | j | j j � | j g | j � d S)N� s datarx ) r� rl �writer! rA r �writersr4 �_buffer)r �m_writer� r r r
�
test_write s
z&UnixWritePipeTransportTests.test_writec C sS | j � } | j d � | j | j � | j | j j � | j g | j � d S)Nr� )r� r� rA rC r r� r4 r� )r r� r� r r r
�test_write_no_data s
z.UnixWritePipeTransportTests.test_write_no_datac C sb | j � } d | _ | j d � | j d d � | j j d | j � | j d g | j � d S)NrG s datarx s ta) r� rl r� r! r �
assert_writer� _write_readyr4 r� )r r� r� r r r
�test_write_partial s
z.UnixWritePipeTransportTests.test_write_partialc C s~ | j � } | j j d | j � d g | _ | j d � | j | j � | j j d | j � | j d d g | j � d S)Nrx s previouss data)
r� r �
add_writerr� r� r� rA rC r� r4 )r r� r� r r r
�test_write_buffer s
z-UnixWritePipeTransportTests.test_write_bufferc C se | j � } t � | _ | j d � | j d d � | j j d | j � | j d g | j � d S)Ns datarx )
r� r� r& r� r! r r� r� r4 r� )r r� r� r r r
�test_write_again& s
z,UnixWritePipeTransportTests.test_write_againzasyncio.unix_events.loggerc C s | j � } t � } | | _ t j � | _ | j d � | j d d � | j | j j
� | j g | j � | j j | d � | j d | j
� | j d � | j d | j
� | j d � | j d � | j d � | j d � | j j d � | j � d S)Ns datarx z#Fatal write error on pipe transportr rG z=pipe closed by peer or os.write(pipe, data) raised exception.)r� r> r& r r Z _fatal_errorr� r! rA r r� r4 r� �
_conn_lost�warningr )r r� Zm_logr� r� r r r
�test_write_err/ s*
z*UnixWritePipeTransportTests.test_write_errc C sZ | j � } | j � | j d � | j | j d � | j d � | j | j d � d S)Ns datar rG )r� r� r� r4 r� )r r� r� r r r
�test_write_closeJ s
z,UnixWritePipeTransportTests.test_write_closec C sv | j � } | j � | j | j j � | j | j j � | j | j � � t j | j � | j
j j d � d S)N)
r� r� rA r r� r� rI r� r r� r{ r� r! )r r� r r r
r� T s
z,UnixWritePipeTransportTests.test__read_readyc C s~ | j � } | j j d | j � d d g | _ d | _ | j � | j d d � | j | j j � | j g | j � d S)Nrx s das tar� s data)
r� r r� r� r� rl r! rA r� r4 )r r� r� r r r
�test__write_ready] s
z-UnixWritePipeTransportTests.test__write_readyc C s� | j � } | j j d | j � d d g | _ d | _ | j � | j d d � | j j d | j � | j d g | j � d S)Nrx s das ta� s data� a) r� r r� r� r� rl r! r� r4 )r r� r� r r r
�test__write_ready_partialh s
z5UnixWritePipeTransportTests.test__write_ready_partialc C s� | j � } | j j d | j � d d g | _ t � | _ | j � | j d d � | j j d | j � | j d g | j � d S)Nrx s das tas data)
r� r r� r� r� r� r& r! r� r4 )r r� r� r r r
�test__write_ready_agains s
z3UnixWritePipeTransportTests.test__write_ready_againc C s� | j � } | j j d | j � d d g | _ d | _ | j � | j d d � | j j d | j � | j d g | j � d S)Nrx s das tar s data) r� r r� r� r� rl r! r� r4 )r r� r� r r r
�test__write_ready_empty~ s
z3UnixWritePipeTransportTests.test__write_ready_emptyzasyncio.log.logger.errorc C s | j � } | j j d | j � d d g | _ t � | _ } | j � | j d d � | j | j j � | j | j j
� | j g | j � | j | j
� � | j t j d � d t t t f �| j d | j � t j | j � | j j j | � d S)Nrx s das tas dataz<Fatal write error on pipe transport
protocol:.*
transport:.*r� r )r� r r� r� r� r> r& r! rA r� r� r4 rI 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 � } | j j d | j � d | _ d d g | _ d | _ | j � | j d d � | j | j j � | j | j j
� | j g | j � | j j
j d � | j j j � d S)Nrx Ts das tar� s data)r� r r� r� r� r� rl r! rA r� r� r4 r{ r� r~ r )r r� r� r r r
�test__write_ready_closing� s
z5UnixWritePipeTransportTests.test__write_ready_closingc C s� | j � } | j j d | j � | j j d | j � d d g | _ | j � | j | j � | j | j j
� | j | j j � | j g | j � | j
| j � � t j | j � | j j j d � d S)Nrx s das ta)r� r r� r� r� r� r� �abortrA rC r� r� r4 rI r� r r� r{ r� r! )r r� r� r r r
�
test_abort� s
z&UnixWritePipeTransportTests.test_abortc C s� | j � } | j | j � | j | j � d } | j | � | j j j | � | 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� � s
z6UnixWritePipeTransportTests.test__call_connection_lostc C s� | j � } | j | j � | j | j � t � } | j | � | j j j | � | 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 s@ | j � } t j � | _ | j � | j j � | j � d S)N)r� r r � write_eofr r! )r r� r r r
rW � s
z&UnixWritePipeTransportTests.test_closec C sE | j � } t j � | _ d | _ | j � | j | j j � d S)NT)r� r r r� r� r rA rC )r r� r r r
�test_close_closing� s
z.UnixWritePipeTransportTests.test_close_closingc C sc | j � } | j � | j | j � � | j | j j � t j | j � | j j
j d � d S)N) r� r� rI r� rA r r� r r� r{ r� r! )r r� r r r
�test_write_eof� s
z*UnixWritePipeTransportTests.test_write_eofc C sO | j � } d g | _ | j � | j | j � � | j | j j j � d S)Ns data) r� r� r� rI r� rA r{ r� rC )r r� r r r
�test_write_eof_pending� s
z2UnixWritePipeTransportTests.test_write_eof_pending) r9 r: r; r r� r� r� r rX r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rW r� r� r� r r r r
r� � s6
r� c @ s e Z d Z d d � Z d S)�AbstractChildWatcherTestsc C s� t j � } t j � } | j t | j | | � | j t | j | � | j t | j | � | j t | j � | j t | j
� | j t | 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)r9 r: r; r� r r r r
r� � s r� c @ s e Z d Z d d � Z d S)�BaseChildWatcherTestsc C s2 t j � } t j � } | j t | j | � d S)N)r r r ZBaseChildWatcherr r� Z _do_waitpid)r r� r� r r r
r� s z*BaseChildWatcherTests.test_not_implementedN)r9 r: r; r� r r r r
r�
s r� � WaitPidMocks�waitpid� WIFEXITED� WIFSIGNALED� WEXITSTATUS�WTERMSIGc @ s� e Z d Z e j j e j d � Z d d � Z d d � Z
d d � Z d d � Z d
d � Z
d d
� Z d d � Z d d � Z d d � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d � � Z e d d! � � Z e d" d# � � Z e d$ d% � � Z e d& d' � � Z e d( d) � � Z e d* d+ � � Z e d, d- � � Z e d. d/ � � Z d0 S)1�ChildWatcherTestsMixinr� c C sk | j � | _ d | _ i | _ t j j | j d � �, | _ | j � | _ | j j
| j � Wd QXd S)NFr( ) ry r �running�zombiesr rX rM �m_add_signal_handler�create_watcherr� r� )r r r r
r s z ChildWatcherTestsMixin.setUpc C s� t | j t j � s! | d k r4 | j | d � n y3 | d k rP | j j � S| | j j | � f SWn t k
r{ Yn X| j r� d St
� � d S)Nr r r7 )r r ) �
isinstancer� r
�SafeChildWatcherZ
assertGreaterr� �popitem�pop�KeyErrorr� �ChildProcessError)r �pid�flagsr r r
r� * s !
z ChildWatcherTestsMixin.waitpidc C s | d | j | <d S)Ni � )r� )r r� �
returncoder r r
�
add_zombie9 s z!ChildWatcherTestsMixin.add_zombiec C s
| d k S)Ni � r )r �statusr r r
r� < s z ChildWatcherTestsMixin.WIFEXITEDc C s d | k o d k SS)Ni� i � r )r r� r r r
r� ? s z"ChildWatcherTestsMixin.WIFSIGNALEDc C s | j | j | � � | d S)Ni � )rI r� )r r� r r r
r� B s z"ChildWatcherTestsMixin.WEXITSTATUSc C s | j | j | � � d | S)Ni � )rI r� )r r� r r r
r� F s z ChildWatcherTestsMixin.WTERMSIGc C s | j j t j | j j � d S)N)r� rV r rT r� � _sig_chld)r r r r
�test_create_watcherJ s z*ChildWatcherTestsMixin.test_create_watcherc s � f d d � } | S)Nc + s� d d � } | d | j � �� } | d | j � �w } | d | j � �\ } | d | j � �A } | d | j � �& } � | t | | | | | � � Wd QXWd QXWd QXWd QXWd QXd S)Nc S s t j | d | d t j �S)N�wrapsZ new_callable)r rX r )�target�wrapperr r r
rX P s zIChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_func.<locals>.patchz os.WTERMSIGzos.WEXITSTATUSzos.WIFSIGNALEDz os.WIFEXITEDz
os.waitpid)r� r� r� r� r� r� )r rX Z
m_WTERMSIGZ
m_WEXITSTATUSZ
m_WIFSIGNALEDZ m_WIFEXITEDZ m_waitpid)r/ r r
� wrapped_funcO s z:ChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_funcr )r/ r� r )r/ r
�
waitpid_mocksN s z$ChildWatcherTestsMixin.waitpid_mocksc C s� t j � } | j �* d | _ | j j d | d d d � Wd QX| j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d | _ | j d d � | j j � | j
| j j � | j
| j j � | j | j
j � | j d d d d d � | j j � | j j � | j j � | j � | j d d � | j � | j j � Wd QX| j | j � | j | j
j � | j j � | j j � | j j � | j j � | j j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d S) NT�* � �
� F� �
)r r r� r� r� rA rC r� r� r� r� r� r� rI rV rN �ignore_warningsr� �clear)r r� �callbackr r r
� test_sigchld_ sR
"
z#ChildWatcherTestsMixin.test_sigchldc C s� t j � } t j � } | j �' d | _ | j j d | d d � Wd QX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j d | d d � Wd QX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j j � | j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j d d
� | j j � | j
d d d d � | j | j � | j | j j � | j | j j � | j | j
j � | j j � | j j � | j
j � | j � | j j � | j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j d d � d
| _ | j j � | j
d d d d � | j | j � | j | j j � | j | j j � | j | j
j � | j j � | j j � | j j � | j � | j d d � | j d d � | j � | j j � Wd QX| j | j � | j | j � | j | j
j � | j j � | j j � | j j � | j j � | j j � | j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d S)NT�+ � � �, � � r� �l Fr � ���r )r r r� r� r� rA rC r� r� r� r� r� r� rV rI rN r r� r )r r� � callback1� callback2r r r
�test_sigchld_two_children� s�
z0ChildWatcherTestsMixin.test_sigchld_two_childrenc C s. t j � } t j � } | j �' d | _ | j j d | d d � Wd QX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j d | d d � Wd QX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j j � | j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j d d � | j d d
� d
| _ | j j � | j
d d d d � | j
d d d d � | j | j j � | j | j j � | j | j j � | j | j
j � | j j � | j j � | j
j � | j j � | j � | j � | j d d � | j d d � | j � | j j � Wd QX| j | j � | j | j � | j | j
j � d S)NT�- � r
�. i{ r
�N rx Fr r ���r )r r r� r� r� rA rC r� r� r� r� r� r� rV rI rN r )r r� r r r r r
�.test_sigchld_two_children_terminating_together sb
zEChildWatcherTestsMixin.test_sigchld_two_children_terminating_togetherc C s� t j � } | j �; | j d d � | j j � | j j d | d d � Wd QX| j d d d d � | j � | j d d � | j � | j j � Wd QX| j | j
� d S)N�2 r� r r r7 ) r r r� r� r� r� rV rN r rA rC )r r� r r r r
�test_sigchld_race_conditionI s
z2ChildWatcherTestsMixin.test_sigchld_race_conditionc C s\ t j � } t j � } | j �$ d | _ | j j d | d � Wd QX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j d | d � Wd QX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d | _ | j d d � | j j � | j
d d d � | j | j � | j | j j � | j | j j � | j | j
j � | j j � | j j � | j
j � | j � | j d d � | j � | j j � Wd QX| j | j � | j | j � | j | j
j � d S)
NT�3 � � Fr
r i����i����)r r r� r� r� rA rC r� r� r� r� r� r� rV rI rN r )r r� r r r r r
� test_sigchld_replace_handler_ sJ
z3ChildWatcherTestsMixin.test_sigchld_replace_handlerc C sM t j � } | j �$ d | _ | j j d | d � Wd QX| j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j j d � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d | _ | j d d � | j
� | j j � Wd QX| j | j � d S)NT�4 i� F�c )r r r� r� r� rA rC 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 | _ | j j d | d � Wd QX| j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d | j d <d | _ | j j � | j
d d d � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j � | j j � | j d d � | j � | j j � Wd QX| j | j � d S) NT�5 r i� F�e i��i��)r r r� r� r� rA rC r� r� r� r� r� r� rV rI rN r� r )r r� r r r r
�test_sigchld_unknown_status� s0
z2ChildWatcherTestsMixin.test_sigchld_unknown_statusc C sp t j � } t j � } t j � } | j �P d | _ | j j d | d � | j j d | d � | j j d | d � Wd QX| j | j j d � � | j | j j d � � | j | j j d � � | j | j j d � � | j d d � | j d d � | j d d � d | _ | j � | j j
� Wd QX| j | j � | j | j � | j d d d � d S)
NT�6 r �7 rG �8 r� r F)
r r r� r� r� rI r� rA r� r r� rC rV )r r� r r � callback3r r r
�test_remove_child_handler� s*
z0ChildWatcherTestsMixin.test_remove_child_handlerc C s� t j � } | j �! d | _ | j j d | � Wd QXt | j _ t j j t
j d � �0 } | j | j j
� d � | j | j � Wd QXd S)NT�9 �error)r r r� r� r� r r� r& rX rM r �loggerr4 r� rI rC )r r� r Zm_errorr r r
� test_sigchld_unhandled_exception� s
z7ChildWatcherTestsMixin.test_sigchld_unhandled_exceptionc C s& t j � } | j �! d | _ | j j d | � Wd QX| j | j � | j | j j � | j | j j � | j | j j � | j | j
j � d | _ | j d d � t j
d t j � | j
j � | j � | j j � Wd QXt | j t j � r| j | j � n | j d d � d S)NT�: Fr� � )r r r� r� r� rA rC r� r� r� r� r� rp r� �WNOHANGrN r r� r� r
�FastChildWatcherrV )r r� r r r r
�#test_sigchld_child_reaped_elsewhere s$
z:ChildWatcherTestsMixin.test_sigchld_child_reaped_elsewherec C s� t j � } t j � } | j �q | j �a d | _ | j d d � | j d d � | j j � | j j d | � | j j d | � Wd QXWd QX| j d d � | j | j
� d S)NTiO r iQ r iP ) r r r r� r� r� r� r� rV rA rC )r r� r r r r r
�,test_sigchld_unknown_pid_during_registration/ s
zCChildWatcherTestsMixin.test_sigchld_unknown_pid_during_registrationc C s t j � } | j �! d | _ | j j d | � Wd QX| j } | j � | _ t j j } | | d � �^ } | | j d � �C } | j j | j � | j
t j � | j
t j | j j
� Wd QXWd QXd | _ | j d d � | j j
� | j
d d � d S)NT�< r r( Fr� )r r r� r� r� r ry rX rM r� rV r rT r� r� )r r� r �old_looprX Z m_old_removeZ m_new_addr r r
�
test_set_loopD s$
z$ChildWatcherTestsMixin.test_set_loopc C s t j � } t j � } t j � } | j �G d | _ | j j d | � | j j d | � | j j d | � Wd QX| j } d | _ t j j | d � �' } | j j d � | j t
j � Wd QX| j d d � | j d d � | j
| j � | j
| j � | j
| j � | j � | _ t j j | j d � �c } | j j | j � | j t
j | j j � | j d d � | j d d � | j
| j � Wd QX| j � | j � d | _ | j d d
� | j j � | j
| j � | j
| j � | j d d
� d S)
NT�= �> in r � rx r( Fr r r )r r r� r� r� r rX rM r� rV r rT r� rA rC ry r� rN )r r� r r r) r6 �m_remove_signal_handlerr� r r r
� test_set_loop_race_conditionc sJ
z3ChildWatcherTestsMixin.test_set_loop_race_conditionc C sH t j � } | j �-d | _ | j d d � | j d d � | j j � | j j d | � | j j d | � | j t | j j � d � t
| j t j � r� | j t | j j
� d � n t j j | j d � �b } | j j � | j t j � | j | j j � t
| j t j � r8| j | j j
� n Wd QXWd QXd S) NT�? r� �A r
�@ r r )r r r� r� r� r� r� r4 rU Z
_callbacksr� r
r2 Z_zombiesrX rM r r rV r rT rA )r r� r r; r r r
rW � s(
z!ChildWatcherTestsMixin.test_closeN) r9 r: r; r rX rM r r- r r r� r� r� r� r� r� r� r� r r r r r r" r% r* r. r3 r4 r7 r< rW r r r r
r� s0
=hE5 %!# <r� c @ s e Z d Z d d � Z d S)�SafeChildWatcherTestsc C s
t j � S)N)r
r� )r r r r
r� � s z$SafeChildWatcherTests.create_watcherN)r9 r: r; r� r r r r
r@ � s r@ c @ s e Z d Z d d � Z d S)�FastChildWatcherTestsc C s
t j � S)N)r
r2 )r r r r
r� � s z$FastChildWatcherTests.create_watcherN)r9 r: r; r� r r r r
rA � s rA c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
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 | t j � | j | j | � | j | | j � � | j | j � d S)N) rC r� �_watcher�get_child_watcherr3 r
r� �assertIsr� )r �policyr� r r r
�test_get_child_watcher� s z"PolicyTests.test_get_child_watcherc C sR | j � } t j � } | j | � | j | j | � | j | | j � � d S)N)rC r
r2 Zset_child_watcherrF rD rE )r rG r� r r r
� test_get_child_watcher_after_set� s
z,PolicyTests.test_get_child_watcher_after_setc C sh | j � } | j � } | j | j � | j � } | j | t j � | j | j | � | j
� d S)N) rC �get_event_loopr� rD rE r3 r
r� rF r� r )r rG r r� r r r
�-test_get_child_watcher_with_mainloop_existing� s z9PolicyTests.test_get_child_watcher_with_mainloop_existingc sK � � f d d � } � j � � t j d | � } | j � | j � d S)Nc so � j � j � � � j � j � t j � � j � } � j | t j � � j | j � � j � j
� d S)N) r �new_event_loopr3 rJ r
ZAbstractEventLooprE r� r� r� r )r� )rG r r r
r� � s
z4PolicyTests.test_get_child_watcher_thread.<locals>.fr� )rC � threadingZThreadr� �join)r r� Zthr )rG r r
� test_get_child_watcher_thread� s
z)PolicyTests.test_get_child_watcher_threadc C s� | j � } | j � } | j � } | j | j | � | j � } | j | � | j | j | � | j d � | j | j d � | j � | j � d S)N)rC rJ rE rF r� rL r r )r rG r r� Znew_loopr r r
�,test_child_watcher_replace_mainloop_existing s
z8PolicyTests.test_child_watcher_replace_mainloop_existingN) r9 r: r; rC rH rI rK rO rP r r r r
rB � s rB �__main__)'�__doc__� collectionsr< r| rp r r[ r� �sysrc rM Zunittestr �platformZSkipTestr
r r r �ANYr� r Z
skipUnlessZTestCaser �hasattrrZ rv r� r� r� �
namedtupler� r� r@ rA rB r9 �mainr r r r
�<module> sX �]�� � � � N