Your IP : 216.73.216.213
��Yfg� � @ s� 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 Z d d l
Z
d d l m Z e j d k r� e j d � � 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 re j( � 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/python35/lib64/python3.5/test/test_asyncio/test_unix_events.py�close_pipe_transport s
r zSignals are not supportedc s� e Z d Z � f 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 � S)#� SelectorEventLoopSignalTestsc s0 t � j � t j � | _ | 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 | 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_signal0 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_handler6 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_handler9 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>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 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 Ed Hd 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 | _ 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& 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 | _ d d � } | | _ Gd d � d t � } | | j _ | j | | j j t j d d � � d S)Nc S s | d k r t � � d S)Nr ���)r )�fdr r r
r' l 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
�Errq s r@ c S s d S)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 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)r< 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 rA r( r r! r4 r r) r* r+ � assertFalse�info�calledr7 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� 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)r< r= r> r? rC 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>� )rA r r( r r r) r r* r+ rD rE rF r7 r' rG )r rH 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* r4 �
assertTruer" r' rF r7 �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! r4 r' �
reset_mockrL r" rD rF r7 �default_int_handlerrN )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* r4 r r' r( r" rL rE )r rH 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* r4 rA 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)r< r= r> r? rC r r r r
r@ � s r@ )
r r r r* r4 rA 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 rJ r r r: )
r r r r* r4 �SIGCHLDr7 �lenr! r' rQ r �assert_called_once_with)r r, r r r
�
test_close� s
z'SelectorEventLoopSignalTests.test_close)r< r= r> r r r r% r �patchr- r3 r9 rB rI rK rO rS rT rU rV rZ r r )r r
r ( s&
r �AF_UNIXz UNIX Sockets are not supportedc s e Z d Z � f d d � Z d d � Z e j e e d � d � d d � � Z d d
� Z
d d � Z d
d � Z d d � Z
d d � Z e j e e d � d � d d � � Z d d � Z e j d � d d � � Z d d � Z d d � Z d d! � Z d" d# � Z � S)$� SelectorEventLoopUnixSocketTestsc s0 t � j � t j � | _ | j | j � d S)N)r
r r r r r )r )r r r
r � s
z&SelectorEventLoopUnixSocketTests.setUpc
C s� t j � �� } t j t j � } | j | � | j d � | j � | j j d d � | � } | j j | � } | j � | j j | j
� � Wd QRXd 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�socketr\ �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_sockZfspathz no os.fspathc
C sv t j � �c } t j | � } | j j d d � | � } | j j | � } | j � | j j | j � � Wd QRXd S)Nc S s d S)Nr r r r r
r& s zRSelectorEventLoopUnixSocketTests.test_create_unix_server_pathlib.<locals>.<lambda>) r r^ �pathlibZPathr ra rb r rc )r rd Zsrv_cororg r r r
� test_create_unix_server_pathlib� s
z@SelectorEventLoopUnixSocketTests.test_create_unix_server_pathlibc C sb t j � �O } | j j d d � | j � } | j t d � � | j j | � Wd QRXWd QRXd 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 ra �namer1 rA rb )r �filerf r r r
�-test_create_unix_server_existing_path_nonsock s
zNSelectorEventLoopUnixSocketTests.test_create_unix_server_existing_path_nonsockc C sR | j j d d � d d d d �} | j t d � � | j j | � Wd QRXd S)Nc S s d S)Nr r r r r
r& s zSSelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_bool.<locals>.<lambda>rd �spam�sslTz"ssl argument must be an SSLContext)r ra r1 r rb )r rf r r r
� test_create_unix_server_ssl_bool s
zASelectorEventLoopUnixSocketTests.test_create_unix_server_ssl_boolc C sL | j j d d � d d �} | j t d � � | j j | � Wd QRXd S)Nc S s d S)Nr r r r r
r& s zXSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosock.<locals>.<lambda>rd z#path was not specified, and no sock)r ra r1 r rb )r rf r r r
�%test_create_unix_server_nopath_nosock s
zFSelectorEventLoopUnixSocketTests.test_create_unix_server_nopath_nosockc C sl t j � } | �S | j j d d � d d d | �} | j t d � � | j j | � Wd QRXWd QRXd S)Nc S s d S)Nr r r r r
r& s zXSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsock.<locals>.<lambda>rd re z"A UNIX Domain Stream.*was expected)r_ r ra r1 r rb )r re rf r r r
�%test_create_unix_server_path_inetsock s
zFSelectorEventLoopUnixSocketTests.test_create_unix_server_path_inetsockc C sx t j t j t j � } | �S | j j d d � d d d | �} | j t d � � | j j | � Wd QRXWd QRXd S)Nc S s d S)Nr r r r r
r& ( s zUSelectorEventLoopUnixSocketTests.test_create_unix_server_path_dgram.<locals>.<lambda>rd re z"A UNIX Domain Stream.*was expected)r_ r\ Z
SOCK_DGRAMr ra r1 r rb )r re rf r r r
�"test_create_unix_server_path_dgram% s
zCSelectorEventLoopUnixSocketTests.test_create_unix_server_path_dgram�
SOCK_NONBLOCKz$no socket.SOCK_NONBLOCK (linux only)c C s� t j t j t j t j B� } t j � � } | j } Wd QRXzu | �h | j | � | j j d d � d d d | �} | j j
| � } | j � | j j
| j � � Wd QRXWd t
j | � Xd S)Nc S s d S)Nr r r r r
r& 8 s z^SelectorEventLoopUnixSocketTests.test_create_unix_server_path_stream_bittype.<locals>.<lambda>rd re )r_ r\ Z SOCK_STREAMrv rk rl rm r` r ra rb r rc �os�unlink)r re rn �fnrf rg r r r
�+test_create_unix_server_path_stream_bittype. s
!zLSelectorEventLoopUnixSocketTests.test_create_unix_server_path_stream_bittypec C sl t j � } | �S | j j d d � d d d | �} | j t d � � | j j | � Wd QRXWd QRXd S)Nc S s d S)Nr r r r r
r& C s z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_path_inetsock.<locals>.<lambda>rd re z"A UNIX Domain Stream.*was expected)r_ r �create_unix_connectionr1 r rb )r re rf r r r
�)test_create_unix_connection_path_inetsock@ s
zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_inetsockzasyncio.unix_events.socketc C s� t j � } | | j _ t | j _ | j j d d � d d �} | j t � � | j j
| � Wd QRX| j | j j
� t | j _ | j j d d � d d �} | j t � � | j j
| � Wd QRX| j | j j
� d S)Nc S s d S)Nr r r r r
r& P s zUSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_error.<locals>.<lambda>rd z/testc S s d S)Nr r r r r
r& V s )r r r_ � return_valuerA r` r( r ra r rb rL r rF � MemoryError)r Zm_socketre rf r r r
�"test_create_unix_server_bind_errorI s zCSelectorEventLoopUnixSocketTests.test_create_unix_server_bind_errorc C sU | j j d d � t j d t � �} | j t d � � | j j | � Wd QRXd S)Nc S s d S)Nr r r r r
r& ] s zXSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sock.<locals>.<lambda>re zpath and sock can not be)r r{ rw �devnullrP r1 r rb )r rf r r r
�%test_create_unix_connection_path_sock[ s zFSelectorEventLoopUnixSocketTests.test_create_unix_connection_path_sockc C sI | j j d d � d � } | j t d � � | j j | � Wd QRXd S)Nc S s d S)Nr r r r r
r& c s z\SelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosock.<locals>.<lambda>z no path and sock were specified)r r{ r1 r rb )r rf r r r
�)test_create_unix_connection_nopath_nosocka s
zJSelectorEventLoopUnixSocketTests.test_create_unix_connection_nopath_nosockc C sR | j j d d � t j d d �} | j t d � � | j j | � Wd QRXd S)Nc S s d S)Nr r r r r
r& j s z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhost.<locals>.<lambda>Zserver_hostnamerp z"server_hostname is only meaningful)r r{ rw r� r1 r rb )r rf r r r
�,test_create_unix_connection_nossl_serverhosth s
zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_nossl_serverhostc C sR | j j d d � t j d d �} | j t d � � | j j | � Wd QRXd S)Nc S s d S)Nr r r r r
r& q s z_SelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhost.<locals>.<lambda>rq Tz/you have to pass server_hostname when using ssl)r r{ rw r� r1 r rb )r rf r r r
�,test_create_unix_connection_ssl_noserverhosto s
zMSelectorEventLoopUnixSocketTests.test_create_unix_connection_ssl_noserverhost)r< r= r> r rh �unittest�
skipUnless�hasattrrw rj ro rr rs rt ru r_ rz r| r r[ r r� r� r� r� r r )r r
r] � s ' r] c sW e Z d Z � f 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 � S) �UnixReadPipeTransportTestsc s� t � j � | 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)r
r �
new_test_loopr r �make_test_protocolr ZProtocol�protocolr r �io� RawIOBase�pipe�filenor} r[ �startr0 �stop�stat�S_IFIFO�st_mode)r �blocking_patcher�
fstat_patcher�m_fstat�st)r r r
r { s
z UnixReadPipeTransportTests.setUpNc C s8 t j | j | j | j d | �} | j t | � | S)N�waiter)r Z_UnixReadPipeTransportr r� r� r0 r )r r� r r r r
�read_pipe_transport� 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� r� ) r �Futurer r� rb 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 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 � | j d | j � | j | j j � t j | j � | j
j j � | j
j j d � d S)N� r� )
r� r} r� r# r� rD 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 s\ | j � } t | _ | j � | j d | j � t j | j � | j | j
j j � d S)Nr� )
r� �BlockingIOErrorr( r� r# r� r r� r rD r� r� rF )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)Nr� z;Fatal read error on pipe transport
protocol:.*
transport:.*�exc_info) r� rA 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)Nr� )r� r r r �
add_readerZ
pause_readingrD 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)Nr� )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
rZ � s
z%UnixReadPipeTransportTests.test_closec C sE | j � } d | _ t j � | _ | j � | j | j j � d S)NT)r� �_closingr r r� r rD rF )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� rP r� rL �
is_closingrD 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� rA 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)r< r= r> r r� r� r r[ r� r� r� r� r� r� rZ r� r� r� r� r r )r r
r� y s
r� c s5 e Z d Z � f 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 � S)7�UnixWritePipeTransportTestsc s� t � j � | 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)Nr� 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� r0 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� r0 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� r� ) r r� r r� rb r� r� r# r� r� r7 r� )r r� r� r r r
r� * s z%UnixWritePipeTransportTests.test_ctorc C s# | j � } | j | j � � d S)N)r� rL Z
can_write_eof)r r� r r r
�test_can_write_eof3 s z.UnixWritePipeTransportTests.test_can_write_eofzos.writec C s_ | j � } d | _ | j d � | j d d � | j | j j � | j t � | j � d S)N� s datar� )
r� r} �writer# rD r �writersr7 � bytearray�_buffer)r �m_writer� r r r
�
test_write7 s
z&UnixWritePipeTransportTests.test_writec C sY | j � } | j d � | j | j � | j | j j � | j t d � | j � d S)Nr� ) r� r� rD rF r r� r7 r� r� )r r� r� r r r
�test_write_no_data@ s
z.UnixWritePipeTransportTests.test_write_no_datac C sU | j � } d | _ | j d � | j j d | j � | j t d � | j � d S)NrJ s datar� s ta) r� r} r� r �
assert_writer� _write_readyr7 r� r� )r r� r� r r r
�test_write_partialH s
z.UnixWritePipeTransportTests.test_write_partialc C s� | j � } | j j d | j � t d � | _ | j d � | j | j � | j j d | j � | j
t d � | j � d S)Nr� s previouss datas previousdata) r� r �
add_writerr� r� r� r� rD rF r� r7 )r r� r� r r r
�test_write_bufferP s
z-UnixWritePipeTransportTests.test_write_bufferc C sn | j � } t � | _ | j d � | j d t d � � | j j d | j � | j t d � | j
� d S)Ns datar� ) r� r� r( r� r# r� r r� r� r7 r� )r r� r� r r r
�test_write_againZ s
z,UnixWritePipeTransportTests.test_write_againzasyncio.unix_events.loggerc C s | j � } t � } | | _ t j � | _ | j d � | j d d � | j | j j
� | j t � | 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 datar� z#Fatal write error on pipe transportr rJ z=pipe closed by peer or os.write(pipe, data) raised exception.)r� rA r( r r Z _fatal_errorr� r# rD r r� r7 r� r� �
_conn_lost�warningr )r r� Zm_logr� r� r r r
�test_write_errc s*
z*UnixWritePipeTransportTests.test_write_errc C sZ | j � } | j � | j d � | j | j d � | j d � | j | j d � d S)Ns datar rJ )r� r� r� r7 r� )r r� r� r r r
�test_write_close~ 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� rD r r� r� rL r� r r� r� r� r# )r r� r r r
r� � s
z,UnixWritePipeTransportTests.test__read_readyc C sq | j � } | j j d | j � t d � | _ d | _ | j � | j | j j � | j t � | j � d S)Nr� s datar� )
r� r r� r� r� r� r} rD r� r7 )r r� r� r r r
�test__write_ready� s
z-UnixWritePipeTransportTests.test__write_readyc C sw | j � } | j j d | j � t d � | _ d | _ | j � | j j d | j � | j t d � | j � d S)Nr� s data� � a) r� r r� r� r� r� r} r� r7 )r r� r� r r r
�test__write_ready_partial� s
z5UnixWritePipeTransportTests.test__write_ready_partialc C s� | j � } | j j d | j � t d � | _ t � | _ | j � | j d t d � � | j j d | j � | j
t d � | j � d S)Nr� s data) r� r r� r� r� r� r� r( r# r� r7 )r r� r� r r r
�test__write_ready_again� s
z3UnixWritePipeTransportTests.test__write_ready_againc C s� | j � } | j j d | j � t d � | _ d | _ | j � | j d t d � � | j j d | j � | j t d � | j � d S)Nr� s datar )
r� r r� r� r� r� r} r# r� r7 )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 � t d � | _ t � | _ } | j � | j | j j � | j | j j
� | j t � | j � | j | j
� � | j t j d � d t t t f �| j d | j � t j | j � | j j j | � d S)Nr� s dataz<Fatal write error on pipe transport
protocol:.*
transport:.*r� r )r� r r� r� r� r� rA r( rD r� r� r7 rL 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 � } | j j d | j � d | _ t d � | _ d | _ | j � | j | j j � | j | j j
� | j t � | j � | j j
j d � | j j j � d S)Nr� Ts datar� )r� r r� r� r� r� r� r} rD r� r� r7 r� 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)Nr� s das ta)r� r r� r� r� r� r� �abortrD rF r� r� r7 rL 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� rA 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
rZ s
z&UnixWritePipeTransportTests.test_closec C sE | j � } t j � | _ d | _ | j � | j | j j � d S)NT)r� r r r� r� r rD rF )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� rL r� rD 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� rL r� rD r� r� rF )r r� r r r
�test_write_eof_pending s
z2UnixWritePipeTransportTests.test_write_eof_pending) 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� rZ 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)r< 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)r< 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 � f 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 e d0 d1 � � Z � S)2�ChildWatcherTestsMixinr� c sy t � j � | j � | _ d | _ i | _ t j j | j d � �, | _ | j
� | _ | j j | j � Wd QRXd S)NFr* )
r
r r� r �running�zombiesr r[ rP �m_add_signal_handler�create_watcherr� r� )r )r r r
r P s
z ChildWatcherTestsMixin.setUpc C s� t | j t j � s! | d k r1 | j | d � y3 | d k rM | j j � S| | j j | � f SWn t k
rx Yn X| j r� d St
� � d S)Nr r r: )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_zombiej s z!ChildWatcherTestsMixin.add_zombiec C s
| d k S)Ni � r )r �statusr r r
r� m s z ChildWatcherTestsMixin.WIFEXITEDc C s d | k o d k SS)Ni� i � r )r r r r r
r� p s z"ChildWatcherTestsMixin.WIFSIGNALEDc C s | j | j | � � | d S)Ni � )rL r� )r r r r r
r� s s z"ChildWatcherTestsMixin.WEXITSTATUSc C s | j | j | � � d | S)Ni � )rL r� )r r r r r
r� w s z ChildWatcherTestsMixin.WTERMSIGc C s | j j t j | j j � d S)N)r� rY r rW r� � _sig_chld)r r r r
�test_create_watcher{ s z*ChildWatcherTestsMixin.test_create_watcherc s � f d d � } | S)Nc + s� d d � } | d | j � �� } | d | j � �z } | d | j � �^ } | d | j � �B } | d | j � �& } � | t | | | | | � � Wd QRXWd QRXWd QRXWd QRXWd QRXd S)Nc S s t j | d | d t j �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)r2 r r
� wrapped_func� s z:ChildWatcherTestsMixin.waitpid_mocks.<locals>.wrapped_funcr )r2 r r )r2 r
�
waitpid_mocks s z$ChildWatcherTestsMixin.waitpid_mocksc C s� t j � } | j �* d | _ | j j d | d d d � Wd QRX| 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 QRX| 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� rD rF r� r� r� r� r
r rL rY rQ �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 QRX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j d | d d � Wd QRX| 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 QRX| 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� rD rF r� r� r� r� r
r rY rL rQ r r� r )r r� � callback1� callback2r r r
�test_sigchld_two_children� s�
z0ChildWatcherTestsMixin.test_sigchld_two_childrenc C s1 t j � } t j � } | j �' d | _ | j j d | d d � Wd QRX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j d | d d � Wd QRX| 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 QRX| j | j � | j | j � | j | j
j � d S)NT�- � r �. i{ r �N r� Fr r" ���r+ )r r r� r� r� rD rF r� r� r� r� r
r rY rL rQ r )r r� r$ r% r r r
�.test_sigchld_two_children_terminating_together5 sb
zEChildWatcherTestsMixin.test_sigchld_two_children_terminating_togetherc C s� t j � } | j �; | j d d � | j j � | j j d | d d � Wd QRX| j d d d d � | j � | j d d � | j � | j j � Wd QRX| j | j
� d S)N�2 r� r r r: ) r r r� r r
r� rY rQ r rD rF )r r� r r r r
�test_sigchld_race_conditionz s
z2ChildWatcherTestsMixin.test_sigchld_race_conditionc C s_ t j � } t j � } | j �$ d | _ | j j d | d � Wd QRX| j | j � | j | j � | j | j j � | j | j j � | j | j j � | j | j
j � | j � | j j d | d � Wd QRX| 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 QRX| j | j � | j | j � | j | j
j � d S)
NT�3 � � Fr r i����i����)r r r� r� r� rD rF r� r� r� r� r r
rY rL rQ r )r r� r$ r% r r r
� test_sigchld_replace_handler� sJ
z3ChildWatcherTestsMixin.test_sigchld_replace_handlerc C sO t j � } | j �$ d | _ | j j d | d � Wd QRX| 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 QRX| j | j � d S)NT�4 i� F�c )r r r� r� r� rD rF 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 QRX| 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 QRX| j | j � d S) NT�5 r0 i� F�e i��i��)r r r� r� r� rD rF r� r� r� r� r� r
rY rL rQ r r )r r� r r r r
�test_sigchld_unknown_status� s0
z2ChildWatcherTestsMixin.test_sigchld_unknown_statusc C sr t j � } t j � } t j � } | j �P d | _ | j j d | d � | j j d | d � | j j d | d � Wd QRX| 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 QRX| j | j � | j | j � | j d d d � d S)
NT�6 r �7 rJ �8 r� r F)
r r r� r� r� rL r� rD r r r
rF rY )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 QRXt | j _ t j j t
j d � �0 } | j | j j
� d � | j | j � Wd QRXd S)NT�9 �error)r r r� r� r� r r� r( r[ rP r �loggerr7 r
rL rF )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 QRX| 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 QRXt | j t j � r| j | j � n | j d d � d S)NT�: Fr� � )r r r� r� r� rD rF r� r� r� r� r rw r� �WNOHANGrQ r r
r� r �FastChildWatcherrY )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 �r | j �a d | _ | j d d � | j d d � | j j � | j j d | � | j j d | � Wd QRXWd QRX| j d d � | j | j
� d S)NTiO r iQ r( iP ) r r r r� r� r r
r� rY rD rF )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 QRX| j } | j � | _ t j j } | | d � �_ } | | j d � �C } | j j | j � | j
t j � | j
t j | j j
� Wd QRXWd QRXd | _ | j d d � | j j
� | j
d d � d S)NT�< r" r* Fr )r r r� r� r� r r� r[ rP r� rY r rW r
r )r r� r �old_loopr[ Z m_old_removeZ m_new_addr r r
�
test_set_loopu 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 QRX| j } d | _ t j j | d � �A } | j t d � � | j j
d � Wd QRX| j t j
� Wd QRX| 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 QRX| j � | j � d
| _ | j d d � | j j � | j | j � | j | j � | j d d � d S)NT�= �> in r" zA loop is being detached� r� r* Fr0 r+ r+ )r r r� r� r� r r[ rP ZassertWarnsRegex�RuntimeWarningr� rY r rW r rD rF r� r
rQ )r r� r$ r% r< rI �m_remove_signal_handlerr� r r r
� test_set_loop_race_condition� sN
z3ChildWatcherTestsMixin.test_set_loop_race_conditionc C sD 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 � t j j | j d � �_ } | j j � | j t j � | j | j j � t
| j t j � r2| j | j j
� Wd QRXWd QRXd S) NT�? r �A r �@ r r" )r r r� r� r r
r� r7 rX Z
_callbacksr� r rE Z_zombiesr[ rP r r rY r rW rD )r r� r$ rO r r r
rZ � s(
z!ChildWatcherTestsMixin.test_closec C sP t j � } | j � �1 } | j t d � � | j d | � Wd QRXWd QRXd S)Nz/the child watcher does not have a loop attached�d )r r r� r1 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[ rP r r@ r r r� r r� r� r� r� r r r r& r, r. r2 r5 r8 r= rA rF rG rJ rP rZ rU r r )r r
r� L s2 =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)r< r= r> r� r r r r
rV � s rV c @ s e Z d Z d d � Z d S)�FastChildWatcherTestsc C s
t j � S)N)r rE )r r r r
r� s z$FastChildWatcherTests.create_watcherN)r< r= r> r� r r r r
rW s rW 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) rY r� �_watcher�get_child_watcherr6 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)rY r rE Zset_child_watcherr\ rZ r[ )r r] 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) rY �get_event_loopr� rZ r[ r6 r r r\ r� r )r r] 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_loopr6 r` r ZAbstractEventLoopr[ r r� r� r )r� )r] r r r
r� - s
z4PolicyTests.test_get_child_watcher_thread.<locals>.fr
)rY � threadingZThreadr� �join)r r� Zthr )r] 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)rY r` r[ r\ r� rb r r )r r] r r� Znew_loopr r r
�,test_child_watcher_replace_mainloop_existing? s
z8PolicyTests.test_child_watcher_replace_mainloop_existingN) r< r= r> rY r^ r_ ra re rf r r r r
rX s rX �__main__))�__doc__� collectionsr? r� rw ri r r_ r� �sysrk rc r� �warningsr �platformZSkipTestr r r r �ANYr� r r� ZTestCaser r� r] r� r� r� r� �
namedtupler� r� rV rW rX r< �mainr r r r
�<module> s\ ���� � � � N