Your IP : 216.73.216.213
�
f f�m � @ 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 y d d l Z Wn e
k
r� d Z Yn Xd 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 d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z y d d l m Z Wn" e
k
ryd d l m Z Yn Xd
d � Z d d
� Z e d � Z e d � Z e d � Z e d � Z i d d 6d d 6d d f f d d f f d d f f d d f f f d 6d d f f d d f f d d f f f d! 6d" d# 6d$ d% 6Z! Gd& d' � d' e j" � Z# Gd( d) � d) e# � Z$ Gd* d+ � d+ e j% � Z& Gd, d- � d- e j" � Z' Gd. d/ � d/ e j( � Z) Gd0 d1 � d1 e j* � Z+ Gd2 d3 � d3 � Z, Gd4 d5 � d5 � Z- e j d6 k rLGd7 d8 � d8 e, e j. � Z/ Gd9 d: � d: e, e- e j. � Z0 n� d d; l m1 Z1 Gd< d= � d= e, � Z2 e3 e1 d> � r�Gd? d@ � d@ e2 e- e j. � Z4 n e3 e1 dA � r�GdB dC � dC e2 e- e j. � Z5 n e3 e1 dD � rGdE dF � dF e2 e- e j. � Z6 n GdG d8 � d8 e2 e- e j. � Z/ dH dI � Z7 GdJ dK � dK e j. � Z8 GdL dM � dM e j. � Z9 GdN dO � dO e j. � Z: GdP dQ � dQ e j. � Z; GdR dS � dS e j. � Z< e= dT k r�e j> � n d S)UzTests for events.py.� N)�mock)�proactor_events)�selector_events)�sslproto)�
test_utils)�support)� test_supportc C s� t t d � r@ t j j t j | � } t j j | � r@ | Sn t j j t j j t � | � } t j j | � rw | St | � � d S)N�
TEST_HOME_DIR)
�hasattrr �os�path�joinr �isfile�dirname�__file__�FileNotFoundError)�filename�fullname� r �B/opt/alt/python34/lib64/python3.4/test/test_asyncio/test_events.py� data_file$ s !r c C sK t j d k r d St j � d } t t t | j d � � � } | d k S)z4Return True if the platform is Mac OS 10.4 or older.�darwinFr �.�
� )r r )�sys�platformZmac_ver�tuple�map�int�split)�versionr r r � osx_tiger/ s
r" z ssl_cert.pemz ssl_key.pemz keycert3.pemz pycacert.pemZB09264B1F2DA21D1Z serialNumber� r! Z countryNameZXYZ localityNamezCastle AnthraxZorganizationNamezPython Software FoundationZ
commonName� localhostZsubjectz Python Software Foundation CAz
our-ca-serverZissuerzNov 13 19:47:07 2022 GMTZnotAfterzJan 4 19:47:07 2013 GMTZ notBeforec @ s[ e Z d Z d Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d S) � MyBaseProtoNc C sX d | _ d | _ d | _ | d k rT t j d | � | _ t j d | � | _ n d S)N�INITIALr �loop)� transport�state�nbytes�asyncio�Future� connected�done)�selfr' r r r �__init__M s zMyBaseProto.__init__c C s2 | | _ d | _ | j r. | j j d � n d S)N� CONNECTED)r( r) r- �
set_result)r/ r( r r r �connection_madeU s zMyBaseProto.connection_madec C s | j t | � 7_ d S)N)r* �len)r/ �datar r r �
data_received\ s zMyBaseProto.data_receivedc C s
d | _ d S)N�EOF)r) )r/ r r r � eof_received` s zMyBaseProto.eof_receivedc C s) d | _ | j r% | j j d � n d S)N�CLOSED)r) r. r2 )r/ �excr r r �connection_lostd s zMyBaseProto.connection_lost)
�__name__�
__module__� __qualname__r- r. r0 r3 r6 r8 r; r r r r r% I s r% c s" e Z d Z � f d d � Z � S)�MyProtoc s! t � j | � | j d � d S)Ns% GET / HTTP/1.0
Host: example.com
)�superr3 �write)r/ r( )� __class__r r r3 l s zMyProto.connection_made)r< r= r> r3 r r )rB r r? k s r? c @ sU e Z d Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d S) �MyDatagramProtoNc C s: d | _ d | _ | d k r6 t j d | � | _ n d S)Nr& r r' )r) r* r+ r, r. )r/ r' r r r r0 t s zMyDatagramProto.__init__c C s | | _ d | _ d S)N� INITIALIZED)r( r) )r/ r( r r r r3 z s z MyDatagramProto.connection_madec C s | j t | � 7_ d S)N)r* r4 )r/ r5 �addrr r r �datagram_received s z!MyDatagramProto.datagram_receivedc C s d S)Nr )r/ r: r r r �error_received� s z MyDatagramProto.error_receivedc C s) d | _ | j r% | j j d � n d S)Nr9 )r) r. r2 )r/ r: r r r r; � s z MyDatagramProto.connection_lost) r< r= r> r. r0 r3 rF rG r; r r r r rC q s rC c @ sU e Z d Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d S) �MyReadPipeProtoNc C sF d g | _ d | _ d | _ | d k rB t j d | � | _ n d S)Nr& r r' )r) r* r( r+ r, r. )r/ r' r r r r0 � s
zMyReadPipeProto.__init__c C s | | _ | j j d � d S)Nr1 )r( r) �append)r/ r( r r r r3 � s z MyReadPipeProto.connection_madec C s | j t | � 7_ d S)N)r* r4 )r/ r5 r r r r6 � s z MyReadPipeProto.data_receivedc C s | j j d � d S)Nr7 )r) rI )r/ r r r r8 � s z MyReadPipeProto.eof_receivedc C sR d | j k r" | j j d � n | j j d � | j rN | j j d � n d S)Nr7 r9 )r) rI r. r2 )r/ r: r r r r; � s
z MyReadPipeProto.connection_lost) r< r= r> r. r0 r3 r6 r8 r; r r r r rH � s rH c @ s= e Z d Z d Z d d d � Z d d � Z d d � Z d S)�MyWritePipeProtoNc C s: d | _ d | _ | d k r6 t j d | � | _ n d S)Nr& r' )r) r( r+ r, r. )r/ r' r r r r0 � s zMyWritePipeProto.__init__c C s | | _ d | _ d S)Nr1 )r( r) )r/ r( r r r r3 � s z MyWritePipeProto.connection_madec C s) d | _ | j r% | j j d � n d S)Nr9 )r) r. r2 )r/ r: r r r r; � s z MyWritePipeProto.connection_lost)r< r= r> r. r0 r3 r; r r r r rJ � s rJ 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)�MySubprocessProtocolc s� d | _ d | _ t j d � � | _ t j d � � | _ � f d d � t d � D� | _ i d d 6d d 6| _ d | _ i t j
d � � d 6t j
d � � d 6| _ d S) Nr& r' c s% i | ] } t j d � � | � q S)r' )r+ r, )�.0�fd)r' r r �
<dictcomp>� s z1MySubprocessProtocol.__init__.<locals>.<dictcomp>� � r# � ) r) r( r+ r, r- � completed�range� disconnectsr5 �
returncodeZEvent�got_data)r/ r' r )r' r r0 � s " z MySubprocessProtocol.__init__c C s& | | _ d | _ | j j d � d S)Nr1 )r( r) r- r2 )r/ r( r r r r3 � s z$MySubprocessProtocol.connection_madec C s d | _ | j j d � d S)Nr9 )r) rR r2 )r/ r: r r r r; � s z$MySubprocessProtocol.connection_lostc C s& | j | | 7<| j | j � d S)N)r5 rV �set)r/ rM r5 r r r �pipe_data_received� s z'MySubprocessProtocol.pipe_data_receivedc C s5 | r | j | j | � n | j | j | � d S)N)rT Z
set_exceptionr2 )r/ rM r: r r r �pipe_connection_lost� s z)MySubprocessProtocol.pipe_connection_lostc C s | j j � | _ d S)N)r( Zget_returncoderU )r/ r r r �process_exited� s z#MySubprocessProtocol.process_exitedN) r< r= r> r0 r3 r; rX rY rZ r r r r rK � s rK c s� e Z d Z � f d 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 d d � Z
d d � Z d d � Z d d � Z e j e e d � d � d d � � Z d! d" � Z d# d$ � Z e j e e d% � d& � d' d( � � Z e j e e d) � d* � d+ d, � � Z e j e e d) � d* � d- d. � � Z d/ d0 d1 � Z d2 d3 � Z e j e e d � d � d4 d5 � � Z d6 d7 � Z d/ d8 i d9 d: � Z d/ d8 d; d< � Z! d/ d8 d= d> � Z" e j# e$ d8 k d? � d@ dA � � Z% dB dC � Z& e j# e$ d8 k d? � e j e e d � d � dD dE � � � Z' dF dG � Z( dH dI � Z) dJ dK � Z* e+ j, dL � dM dN � � Z- dO dP � Z. dQ dR � Z/ dS dT � Z0 e j e e dU � dV � dW dX � � Z1 dY dZ � Z2 e j e e d � d � d[ d\ � � Z3 e j e e d � d � d] d^ � � Z4 d8 d_ d` � Z5 d8 da db � Z6 d8 dc dd � Z7 e j# e$ d8 k d? � de df � � Z8 dg dh � Z9 e j# e$ d8 k d? � e j e e d � d � di dj � � � Z: dk dl � Z; e j# e$ d8 k d? � dm dn � � Z< do dp � Z= e j# e$ d8 k d? � e j e e d � d � dq dr � � � Z> ds dt � Z? e j# e$ d8 k d? � du dv � � Z@ dw dx � ZA e j# e$ d8 k d? � e j e e d � d � dy dz � � � ZB d{ d| � ZC e j# e$ d8 k d? � d} d~ � � ZD d d� � ZE d� d� � ZF d� d� � ZG e j eH jI d� � d� d� � � ZJ d� d� � ZK d� d� � ZL d� d� � ZM d� d� � ZN e j eO jP d� k d� � d� d� � � ZQ e j eO jP d� k d� � eH jR d� d� � eH jS d� � d� d� � � � � ZT e j eO jP d� k d� � d� d� � � ZU e j eO jP d� k d� � d� d� � � ZV e j eO jP d� k d� � eH jR d� d� � d� d� � � � ZW d� d� � ZX d� d� � ZY d� d� � ZZ d� d� � Z[ d� d� � Z\ d� d� � Z] d� d� � Z^ � S)��EventLoopTestsMixinc s0 t � j � | j � | _ | j | j � d S)N)r@ �setUp�create_event_loopr' �set_event_loop)r/ )rB r r r\ � s
zEventLoopTestsMixin.setUpc sJ | j j � s" t j | j � n | j j � t j � t � j � d S)N) r' � is_closedr � run_briefly�close�gcZcollectr@ �tearDown)r/ )rB r r rc � s
z EventLoopTestsMixin.tearDownc sS t j d d � � � t j � � f d d � � } � j t � j j | � � d S)Nc s s d Vd S)Nr r r r r �coro1� s zBEventLoopTestsMixin.test_run_until_complete_nesting.<locals>.coro1c s- � j � j j � � � j j � � � d S)N)�
assertTruer' �
is_running�run_until_completer )rd r/ r r �coro2 s zBEventLoopTestsMixin.test_run_until_complete_nesting.<locals>.coro2)r+ � coroutine� assertRaises� RuntimeErrorr' rg )r/ rh r )rd r/ r � test_run_until_complete_nesting� s z3EventLoopTestsMixin.test_run_until_complete_nestingc C sr | j j � } | j j t j d d | j �� | j j � } | j d | | k oa d k n | | � d S)Ng�������?r' g{�G�z�?g�������?)r' �timerg r+ �sleepre )r/ �t0�t1r r r �test_run_until_complete s "z+EventLoopTestsMixin.test_run_until_completec sA t j � f d d � � } | � } � j t � j j | � d S)Nc 3 s, � j j � t j d d � j �Dd Hd S)Ng�������?r' )r' �stopr+ rn r )r/ r r �cb s
z?EventLoopTestsMixin.test_run_until_complete_stopped.<locals>.cb)r+ ri rj rk r' rg )r/ rs Ztaskr )r/ r � test_run_until_complete_stopped s z3EventLoopTestsMixin.test_run_until_complete_stoppedc s� g � � � f d d � } � j j d | d � t j � } � j j � t j � } � j � d g � � j d | | k o� d k n | | � d S)Nc s � j | � � j j � d S)N)rI r' rr )�arg)�resultsr/ r r �callback s
z5EventLoopTestsMixin.test_call_later.<locals>.callbackg�������?z hello worldg{�G�z�?g�������?)r' �
call_laterrm � monotonic� run_forever� assertEqualre )r/ rw ro rp r )rv r/ r �test_call_later s
z#EventLoopTestsMixin.test_call_laterc sU g � � � f d d � } � j j | d d � � j j � � j � d g � d S)Nc s$ � j | | f � � j j � d S)N)rI r' rr )Zarg1Zarg2)rv r/ r r rw + s z4EventLoopTestsMixin.test_call_soon.<locals>.callback�hello�world)r} r~ )r' � call_soonrz r{ )r/ rw r )rv r/ r �test_call_soon( s
z"EventLoopTestsMixin.test_call_soonc
s� g � t j � � � � f d d � � � � � f d d � } � j � t j d | � } | j � � �% � j j � d � � j j � Wd QX| j � � j � d d g � d S)Nc s3 � j | � t � � d k r/ � j j � n d S)NrQ )rI r4 r' rr )ru )rv r/ r r rw 7 s
z?EventLoopTestsMixin.test_call_soon_threadsafe.<locals>.callbackc s! � j j � d � � j � d S)Nr} )r' �call_soon_threadsafe�releaser )rw �lockr/ r r �
run_in_thread< s zDEventLoopTestsMixin.test_call_soon_threadsafe.<locals>.run_in_thread�targetr~ r} )
� threadingZLock�acquire�Thread�startr' r rz r
r{ )r/ r� �tr )rw r� rv r/ r �test_call_soon_threadsafe3 s
z-EventLoopTestsMixin.test_call_soon_threadsafec sh g � � � f d d � } � j j | d � � j j | d � � j j � � j � d d g � d S)Nc s3 � j | � t � � d k r/ � j j � n d S)NrQ )rI r4 r' rr )ru )rv r/ r r rw M s
zKEventLoopTestsMixin.test_call_soon_threadsafe_same_thread.<locals>.callbackr} r~ )r' r� r rz r{ )r/ rw r )rv r/ r �%test_call_soon_threadsafe_same_threadJ s
z9EventLoopTestsMixin.test_call_soon_threadsafe_same_threadc C sf d d � } | j j d | d � } | j j | � \ } } | j | d � | j | t j � � d S)Nc S s | t j � f S)N)r� � get_ident)ru r r r �runX s z5EventLoopTestsMixin.test_run_in_executor.<locals>.runZyo)r' �run_in_executorrg r{ ZassertNotEqualr� r� )r/ r� �f2�resZ thread_idr r r �test_run_in_executorW s
z(EventLoopTestsMixin.test_run_in_executorc s
t j � \ � } � j d � t � � � � � f d d � } � j j � j � | � � j j | j d � t j � j � f d d � � � j j | j d � t j � j � f d d � � � j j | j
� � j j � j j � � j j � � j
� d � d S)
NFc so y � j d � } Wn t k
r+ d SYn X| rB � j | � n) � j � j j � j � � � � j � d S)Ni )�recv�BlockingIOError�extendre r' �
remove_reader�filenora )r5 )�
bytes_read�rr/ r r �readerd s
z8EventLoopTestsMixin.test_reader_callback.<locals>.readers abcc s t � � d k S)NrO )r4 r )r� r r �<lambda>s s z:EventLoopTestsMixin.test_reader_callback.<locals>.<lambda>s defc s t � � d k S)N� )r4 r )r� r r r� u s s abcdef)r �
socketpair� setblocking� bytearrayr' �
add_readerr� r �send� run_untilra rr rz r{ )r/ �wr� r )r� r� r/ r �test_reader_callback_ s
z(EventLoopTestsMixin.test_reader_callbackc s� t j � \ } � � j d � � � f d d � } d d } � j j � j � | | � � j j � � j � j j � j � � � � j � j j � j � � � � j
� | j t | � d � } | j
� � j
| | � d S)NFc s � j | � � j j � d S)N)r� r' rr )r5 )r/ r� r r �writer s
z8EventLoopTestsMixin.test_writer_callback.<locals>.writer� xi rQ )r r� r� r' �
add_writerr� rz re �
remove_writer� assertFalsera r� r4 r{ )r/ r� r� r5 �readr )r/ r� r �test_writer_callback{ s
z(EventLoopTestsMixin.test_writer_callbackc C s� t | j t j � s | j j d � | j d � | j t � �' | j j | j j | | j
� � Wd QX| j t � �$ | j j | j j | d � � Wd QX| j t � �$ | j j | j j | d � � Wd QX| j t � �! | j j | j j
| � � Wd QXn | j d � | j j | j j | | j
� � | j j | j j | d � � | j j | j j | d � � } | j j | j j | d � � | j � | j | j d � � d S)NTs GET / HTTP/1.0
i Fs HTTP/1.0 200 OK)�
isinstancer' r ZBaseProactorEventLoop� set_debugr� rj �
ValueErrorrg � sock_connect�address� sock_sendall� sock_recv� sock_acceptra re �
startswith)r/ �httpd�sockr5 r r r �_basetest_sock_client_ops� s4
z-EventLoopTestsMixin._basetest_sock_client_opsc
C s5 t j � �# } t j � } | j | | � Wd QXd S)N)r �run_test_server�socketr� )r/ r� r� r r r �test_sock_client_ops� s z(EventLoopTestsMixin.test_sock_client_ops�AF_UNIXzNo UNIX Socketsc
C s; t j � �) } t j t j � } | j | | � Wd QXd S)N)r �run_test_unix_serverr� r� r� )r/ r� r� r r r �test_unix_sock_client_ops� s z-EventLoopTestsMixin.test_unix_sock_client_opsc C s� d } z) t j � } | j d � | j � } Wd | j � Xt j � } | j d � | j t � �$ | j j | j j | | � � Wd QX| j � d S)N� 127.0.0.1r F)r� r )
r� �bind� getsocknamera r� rj �ConnectionRefusedErrorr' rg r� )r/ r� �sr� r r r �test_sock_client_fail� s
z)EventLoopTestsMixin.test_sock_client_failc C s� t j � } | j d � | j d � | j d � t j � } | j | j � � | j j | � } | j j | � \ } } | j | j
� d � | j | | j � � | j | j � | j � � | j � | j � | j � d S)NF� 127.0.0.1r r# )r� r )
r� r� r� Zlisten�connectr� r' r� rg r{ Z
gettimeoutZ getpeernamera )r/ Zlistener�client�fZconnrE r r r �test_sock_accept� s
z$EventLoopTestsMixin.test_sock_accept�SIGKILLz
No SIGKILLc s� d � � f d d � } | j t | j j d | � | j t | j j d � | j t | j j t j d | � | j t | j j t j d � | j t | j j d | � | j t | j j d � | j t | j j d | � | j t | j j d � | j t | j j t j | � | j
| j j t j � � | j j t j | � t j
t j � t j � t j | j � f d d � � | j | j j t j � � | j t j t j � t j � | j
| j j t j � � d S)
Nr c s � d 7� d S)Nr# r r )�caughtr r �
my_handler� s z?EventLoopTestsMixin.test_add_signal_handler.<locals>.my_handlerZboomr# c s � S)Nr r )r� r r r� s z=EventLoopTestsMixin.test_add_signal_handler.<locals>.<lambda>���r� )rj � TypeErrorr' �add_signal_handler�remove_signal_handlerr� �signal�NSIGrk r� r� �SIGINTr �kill�getpidr r� re r{ � getsignal�default_int_handler)r/ r� r )r� r �test_add_signal_handler� s<
z+EventLoopTestsMixin.test_add_signal_handler�SIGALRMz
No SIGALRMc sh d � � � f d d � } � j j t j | � t j t j d d � � j j � � j � d � d S)Nr c s � d 7� � j j � d S)Nr# )r' rr r )r� r/ r r r� s
zLEventLoopTestsMixin.test_signal_handling_while_selecting.<locals>.my_handlerg{�G�z�?r# )r' r� r� r� � setitimer� ITIMER_REALrz r{ )r/ r� r )r� r/ r �$test_signal_handling_while_selecting s
z8EventLoopTestsMixin.test_signal_handling_while_selectingc s� d � d � � � � f d d � } � j j t j | � � t j t j d d � � j j d � j j � � j j � � j � d � d S) N�* r c s � d 7� � j | � � d S)Nr# )r{ )�args)r� r/ � some_argsr r r� s
zAEventLoopTestsMixin.test_signal_handling_args.<locals>.my_handlerg�������?g �?r# )r� )
r' r� r� r� r� r� rx rr rz r{ )r/ r� r )r� r/ r� r �test_signal_handling_args s
z-EventLoopTestsMixin.test_signal_handling_argsTc C s� | j j | � \ } } | j | t j � | j | t j � | j | j | � | rp | j | j d � � n | j j | j
� | j | j d � | j
� d S)N�socknamer )r' rg �assertIsInstancer+ � Transport�Protocol�assertIsr( �assertIsNotNone�get_extra_infor. �
assertGreaterr* ra )r/ �connection_fut�check_sockname�tr�prr r r �_basetest_create_connection* s z/EventLoopTestsMixin._basetest_create_connectionc sJ t j � �8 } � j j � f d d � | j � } � j | � Wd QXd S)Nc s t d � j � S)Nr' )r? r' r )r/ r r r� 8 s z<EventLoopTestsMixin.test_create_connection.<locals>.<lambda>)r r� r' �create_connectionr� r� )r/ r� �conn_futr )r/ r �test_create_connection5 s z*EventLoopTestsMixin.test_create_connectionc sW t � } t j � �; } � j j � f d d � | j � } � j | | � Wd QXd S)Nc s t d � j � S)Nr' )r? r' r )r/ r r r� C s zAEventLoopTestsMixin.test_create_unix_connection.<locals>.<lambda>)r" r r� r' �create_unix_connectionr� r� )r/ r� r� r� r )r/ r �test_create_unix_connection; s
z/EventLoopTestsMixin.test_create_unix_connectionc sU t j � �C} d } � j j � j j d t j | j � � } xw | D]o \ } } } } } yN t j d | d | d | � } | j d � � j j � j j | | � � Wn YqF XPqF W� j j
� f d d � d | �} � j j | � \ }
} � j |
t j
� � j | t j � � j j | j � � j | j d � |
j � Wd QXd S) N�type�family�protoFc s t d � j � S)Nr' )r? r' r )r/ r r r� Z s zAEventLoopTestsMixin.test_create_connection_sock.<locals>.<lambda>r� r )r r� r' rg � getaddrinfor� � SOCK_STREAMr� r� r� r� r� r+ r� r� r. r� r* ra ) r/ r� r� �infosr� r� r� �cnamer� r� r� r� r )r/ r �test_create_connection_sockF s,
z/EventLoopTestsMixin.test_create_connection_sockNc C s� | r | j | j d � � n | rA | j | | j d � � n | j | j d � � | j | | j d � � | j d � } | j | t � | j t | � d | � | j | d t � | j | d t � | j | d t � | j d � } | j | � | j | j � | j d
� � | j | j � | j d � � | j | j
� | j d � � | j | j � | j d
� � d S) Nr� �peername�peercert�cipherrO r r# rQ Z
ssl_object� compression) r� r� r{ r� r r4 �strr r� r� Z getpeercert)r/ r� r� r� r� r� Zsslobjr r r �check_ssl_extra_infob s0
z(EventLoopTestsMixin.check_ssl_extra_infoc C s� | j j | � \ } } | j | t j � | j | t j � | j d | j j j � k � | j
| | | � | j j | j � | j | j
d � | j � d S)N�sslr )r' rg r� r+ r� r� re rB r< �lowerr� r. r� r* ra )r/ r� r� r� r� r� r r r � _basetest_create_ssl_connection� s z3EventLoopTestsMixin._basetest_create_ssl_connectionc s$ | d t j � � } � j | | | � t t d � r� t j j d d d d d d � f d d �} t j d d | ��< } | d d
� } � j | | | � � j | j
d � Wd QXn � j t j � �< } | d d
� } t j
� � � j | | | � Wd QXWd QX� j | j j d � d S)
Nr �Purpose�cafile�capath�cadatac s � j | t j j � t j � S)z�
A ssl.create_default_context() replacement that doesn't enable
cert validation.
)r{ r r � SERVER_AUTHr �dummy_ssl_context)Zpurposer r r )r/ r r �_dummy_ssl_create_context� s zREventLoopTestsMixin._test_create_ssl_connection.<locals>._dummy_ssl_create_contextzssl.create_default_context� side_effectTr# ZCERTIFICATE_VERIFY_FAILED)r r r r
r r r r �patchr{ Z
call_countrj �SSLError�disable_logger� exception�reason) r/ r� r� r� r� r� r �m�cmr )r/ r �_test_create_ssl_connection� s&
z/EventLoopTestsMixin._test_create_ssl_connectionz
No ssl modulec sb t j d d � �J } t j � j j � f d d � | j � } � j | | d | j �Wd QXd S)N�use_sslTc s t d � j � S)Nr' )r? r' r )r/ r r r� � s z@EventLoopTestsMixin.test_create_ssl_connection.<locals>.<lambda>r� )r r� � functools�partialr' r� r� r )r/ r� r� r )r/ r �test_create_ssl_connection� s z.EventLoopTestsMixin.test_create_ssl_connectionc C s! t j � � | j � Wd QXd S)N)r �force_legacy_ssl_supportr )r/ r r r �!test_legacy_create_ssl_connection� s
z5EventLoopTestsMixin.test_legacy_create_ssl_connectionc
su t � } t j d d � �S } t j � j j � f d d � | j d d �} � j | | | d | j �Wd QXd S)Nr Tc s t d � j � S)Nr' )r? r' r )r/ r r r� � s zEEventLoopTestsMixin.test_create_ssl_unix_connection.<locals>.<lambda>�server_hostnamez 127.0.0.1r� ) r" r r� r r r' r� r� r )r/ r� r� r� r )r/ r � test_create_ssl_unix_connection� s
z3EventLoopTestsMixin.test_create_ssl_unix_connectionc C s! t j � � | j � Wd QXd S)N)r r r )r/ r r r �&test_legacy_create_ssl_unix_connection� s
z:EventLoopTestsMixin.test_legacy_create_ssl_unix_connectionc s� t j � �� } t j � } � j j � f d d � d | j d | f | j �} � j j | � \ } } | j j d � d } � j
| | � | j � Wd QXd S)Nc s t d � j � S)Nr' )r? r' r )r/ r r r� � s zGEventLoopTestsMixin.test_create_connection_local_addr.<locals>.<lambda>�
local_addrr r� r# ) r r� r �find_unused_portr' r� r� rg r( r� r{ ra )r/ r� �portr� r� r� Zexpectedr )r/ r �!test_create_connection_local_addr� s z5EventLoopTestsMixin.test_create_connection_local_addrc s� t j � �� } � j j � f d d � d | j | j �} � j t � � } � j j | � Wd QX� j | j j
t
j � � j t
| j � | j j � Wd QXd S)Nc s t d � j � S)Nr' )r? r' r )r/ r r r� � s zNEventLoopTestsMixin.test_create_connection_local_addr_in_use.<locals>.<lambda>r )r r� r' r� r� rj �OSErrorrg r{ r �errno�
EADDRINUSE�assertInr� �strerror)r/ r� r� r r )r/ r �(test_create_connection_local_addr_in_use� s z<EventLoopTestsMixin.test_create_connection_local_addr_in_usezasyncio.base_events.socketc s t j � f d d � � � � � f d d � } � t j k ra d d � | D� | j � j _ n d d � | D� | j � j _ | � j _ t j � � j _
t j � � j _ � j j � f d d � | d
� } � j j
| � } � j | j � d d � | j D� } � j | | � d S) Nc sW � t j k r. � t j d d | | f g g S� t j d d | | d d f g g Sd S)Nr� � r )r� �AF_INETr� )�hostr r� �kw)r� r r r� � s zEEventLoopTestsMixin.create_server_multiple_hosts.<locals>.getaddrinfoc s t j � | | � d � j �S)Nr' )r+ �Taskr' )r� �kwds)r� r/ r r �getaddrinfo_task� s zJEventLoopTestsMixin.create_server_multiple_hosts.<locals>.getaddrinfo_taskc S s g | ] } | d f � q S)�P r )rL r( r r r �
<listcomp>� s zDEventLoopTestsMixin.create_server_multiple_hosts.<locals>.<listcomp>c S s" g | ] } | d d d f � q S)r- r r )rL r( r r r r. � s c s
t � j � S)N)r? r' r )r/ r r r� s zBEventLoopTestsMixin.create_server_multiple_hosts.<locals>.<lambda>r- c S s g | ] } | j � d � q S)r )�
getsockbyname)rL r� r r r r. s )r+ ri r� r' r/ r
r' r� r �MockZ_start_serving�
_stop_serving�
create_serverrg �
addCleanupra �socketsr{ )r/ r� ZhostsZ mock_sockr, r� �serverZ server_hostsr )r� r� r/ r � create_server_multiple_hosts� s $z0EventLoopTestsMixin.create_server_multiple_hostsc C s | j t j d d g � d S)Nz1.2.3.4z5.6.7.8)r6 r� r' )r/ r r r �&test_create_server_multiple_hosts_ipv4 s z:EventLoopTestsMixin.test_create_server_multiple_hosts_ipv4c C s | j t j d d g � d S)Nz::1z::2)r6 r� �AF_INET6)r/ r r r �&test_create_server_multiple_hosts_ipv6
s z:EventLoopTestsMixin.test_create_server_multiple_hosts_ipv6c s� t | j � � | j j � f d d � d d � } | j j | � } | j t | j � d � | j d } | j � \ } } | j | d � t j � } | j d | f � | j
d � | j j � j � | j d � j � t
j | j � f d d � � | j d
� j � | j � j j d � � | j d � j j d � d � � j j � | j j � j � | j d
� j � | j � | j � d S)Nc s � S)Nr r )r� r r r� s z8EventLoopTestsMixin.test_create_server.<locals>.<lambda>z0.0.0.0r r# z 127.0.0.1s xxxr1 c s
� j d k S)Nr )r* r )r� r r r� s rO r� r� r9 )r? r' r2 rg r{ r4 r4 r� r� r� �sendallr- r) r r� r* r� r( r� ra r. )r/ r� r5 r� r( r r� r )r� r �test_create_server
s, $
z&EventLoopTestsMixin.test_create_server� SO_REUSEPORTzNo SO_REUSEPORTc sB t | j � � | j j � f d d � d d � } | j j | � } | j t | j � d � | j d } | j | j t j
t j � � | j � t
j | j � t | j � � | j j � f d d � d d d d �} | j j | � } | j t | j � d � | j d } | j | j t j
t j � � | j � d S) Nc s � S)Nr r )r� r r r� 5 s zCEventLoopTestsMixin.test_create_server_reuse_port.<locals>.<lambda>z0.0.0.0r r# c s � S)Nr r )r� r r r� B s Z
reuse_portT)r? r' r2 rg r{ r4 r4 r� Z
getsockoptr� �
SOL_SOCKETr< ra r r` re )r/ r� r5 r� r )r� r � test_create_server_reuse_port1 s*
!
z1EventLoopTestsMixin.test_create_server_reuse_portc sY t j � � | j � f d d � � | j j | � | � } | j j | � } | � f S)Nc s t j j � � o t j � � S)N)r r �exists�unlinkr )r r r r� M s z7EventLoopTestsMixin._make_unix_server.<locals>.<lambda>)r Zgen_unix_socket_pathr3 r' �create_unix_serverrg )r/ �factory�kwargsr� r5 r )r r �_make_unix_serverK s
z%EventLoopTestsMixin._make_unix_serverc s t d | j � � | j � f d d � � \ } } | j t | j � d � t j t j � } | j | � | j d � | j j
� j � | j d � j � t
j | j � f d d � � | j d � j � � j j � | j j
� j � | j d � j � | j � | j � d S)
Nr' c s � S)Nr r )r� r r r� W s z=EventLoopTestsMixin.test_create_unix_server.<locals>.<lambda>r# s xxxr1 c s
� j d k S)Nr )r* r )r� r r r� ` s rO r9 )r? r' rD r{ r4 r4 r� r� r� r: rg r- r) r r� r* r( ra r. )r/ r5 r r� r )r� r �test_create_unix_serverT s !
z+EventLoopTestsMixin.test_create_unix_serverc s t d | j � � t j � } | �U | j j � f d d � d d | �} | j t d � � | j j | � Wd QXWd QXd S)Nr' c s � S)Nr r )r� r r r� u s zOEventLoopTestsMixin.test_create_unix_server_path_socket_error.<locals>.<lambda>z/testr� z3path and sock can not be specified at the same time)r? r' r� rA �assertRaisesRegexr� rg )r/ r� r� r )r� r �)test_create_unix_server_path_socket_errorp s '
z=EventLoopTestsMixin.test_create_unix_server_path_socket_errorc C s8 t j t j � } | j t j O_ | j | | � | S)N)r �
SSLContext�PROTOCOL_SSLv23�options� OP_NO_SSLv2Zload_cert_chain)r/ �certfile�keyfile�
sslcontextr r r �_create_ssl_context{ s z'EventLoopTestsMixin._create_ssl_contextc
C s~ | j | | � } | j j | d d d | �} | j j | � } | j d } | j � \ } } | j | d � | | | f S)Nz 127.0.0.1r r )rO r' r2 rg r4 r� r{ )
r/ rB rL rM rN r� r5 r� r( r r r r �_make_ssl_server� s
z$EventLoopTestsMixin._make_ssl_serverc C s% | j | | � } | j | d | �S)Nr )rO rD )r/ rB rL rM rN r r r �_make_ssl_unix_server� s z)EventLoopTestsMixin._make_ssl_unix_serverc sA t d | j � � | j � f d d � t t � \ } } } | j j t | | d t j � �} | j j | � \ } } | j
d � | j j � j � | j d � j
� t j | j � f d d � � | j d � j � | j | d | | f �� j j � | j j � j � | j d
� j
� | j � | j � d S) Nr' c s � S)Nr r )r� r r r� � s z<EventLoopTestsMixin.test_create_server_ssl.<locals>.<lambda>r s xxxr1 c s
� j d k S)Nr )r* r )r� r r r� � s rO r� r9 )r? r' rP �ONLYCERT�ONLYKEYr� r% r r rg rA r- r{ r) r� r* r� r( ra r. )r/ r5 r( r �f_cr� r� r )r� r �test_create_server_ssl� s" $
z*EventLoopTestsMixin.test_create_server_sslc C s! t j � � | j � Wd QXd S)N)r r rU )r/ r r r � test_legacy_create_server_ssl� s
z1EventLoopTestsMixin.test_legacy_create_server_sslc s( t d | j � � | j � f d d � t t � \ } } | j j t | d t j � d d �} | j j | � \ } } | j
d � | j j � j � | j d � j
� t j | j � f d d � � | j d
� j � � j j � | j j � j � | j d � j
� | j � | j � d S) Nr' c s � S)Nr r )r� r r r� � s zAEventLoopTestsMixin.test_create_unix_server_ssl.<locals>.<lambda>r r r& s xxxr1 c s
� j d k S)Nr )r* r )r� r r r� � s rO r9 )r? r' rQ rR rS r� r% r r rg rA r- r{ r) r� r* r( ra r. )r/ r5 r rT r� r� r )r� r �test_create_unix_server_ssl� s" !
z/EventLoopTestsMixin.test_create_unix_server_sslc C s! t j � � | j � Wd QXd S)N)r r rW )r/ r r r �"test_legacy_create_unix_server_ssl� s
z6EventLoopTestsMixin.test_legacy_create_unix_server_sslc s. t d | j � � | j � f d d � t � \ } } } t j t j � } | j t j O_ t j | _
t | d � r� d | _ n | j j
t | | d | �} t j j | j d � �T t j � �1 | j t j d � � | j j | � Wd QXWd QXt j | j � Wd QX| j � j � | j � d S) Nr' c s � S)Nr r )r� r r r� � s zJEventLoopTestsMixin.test_create_server_ssl_verify_failed.<locals>.<lambda>�check_hostnameTr �call_exception_handlerz (?i)certificate.verify.failed )r? r' rP �SIGNED_CERTFILEr rH rI rJ rK �
CERT_REQUIRED� verify_moder
rY r� r r �objectr r
rF r rg r` � assertIsNoner( ra )r/ r5 r( r �sslcontext_clientrT r )r� r �$test_create_server_ssl_verify_failed� s$ !
z8EventLoopTestsMixin.test_create_server_ssl_verify_failedc C s! t j � � | j � Wd QXd S)N)r r ra )r/ r r r �+test_legacy_create_server_ssl_verify_failed� s
z?EventLoopTestsMixin.test_legacy_create_server_ssl_verify_failedc s. t d | j � � | j � f d d � t � \ } } t j t j � } | j t j O_ t j | _
t | d � r� d | _ n | j j
t | d | d d �} t j j | j d � �T t j � �1 | j t j d
� � | j j | � Wd QXWd QXt j | j � Wd QX| j � j � | j � d S) Nr' c s � S)Nr r )r� r r r� � s zOEventLoopTestsMixin.test_create_unix_server_ssl_verify_failed.<locals>.<lambda>rY Tr r ZinvalidrZ z (?i)certificate.verify.failed )r? r' rQ r[ r rH rI rJ rK r\ r] r
rY r� r r r^ r r
rF r rg r` r_ r( ra )r/ r5 r r` rT r )r� r �)test_create_unix_server_ssl_verify_failed� s&
z=EventLoopTestsMixin.test_create_unix_server_ssl_verify_failedc C s! t j � � | j � Wd QXd S)N)r r rc )r/ r r r �0test_legacy_create_unix_server_ssl_verify_failed s
zDEventLoopTestsMixin.test_legacy_create_unix_server_ssl_verify_failedc s+ t d | j � � | j � f d d � t � \ } } } t j t j � } | j t j O_ t j | _
| j d t � t
| d � r� d | _ n | j j t | | d | �} t j j | j d � �D t j � �1 | j t j d � � | j j | � Wd QXWd QXWd QX� j j � | j � d S)
Nr' c s � S)Nr r )r� r r r� s zIEventLoopTestsMixin.test_create_server_ssl_match_failed.<locals>.<lambda>r rY Tr rZ z.hostname '127.0.0.1' doesn't match 'localhost')r? r' rP r[ r rH rI rJ rK r\ r] �load_verify_locations�
SIGNING_CAr
rY r� r r r^ r r
rF ZCertificateErrorrg r( ra )r/ r5 r( r r` rT r )r� r �#test_create_server_ssl_match_failed s( !
"
z7EventLoopTestsMixin.test_create_server_ssl_match_failedc C s! t j � � | j � Wd QXd S)N)r r rg )r/ r r r �*test_legacy_create_server_ssl_match_failed7 s
z>EventLoopTestsMixin.test_legacy_create_server_ssl_match_failedc s t d | j � � | j � f d d � t � \ } } t j t j � } | j t j O_ t j | _
| j d t � t
| d � r� d | _ n | j j t | d | d d �} | j j | � \ } } � j j � | j � | j � | j j � j � d S)
Nr' c s � S)Nr r )r� r r r� @ s zJEventLoopTestsMixin.test_create_unix_server_ssl_verified.<locals>.<lambda>r rY Tr r r$ )r? r' rQ r[ r rH rI rJ rK r\ r] re rf r
rY r� rg r( ra r. )r/ r5 r r` rT r� r� r )r� r �$test_create_unix_server_ssl_verified; s"
z8EventLoopTestsMixin.test_create_unix_server_ssl_verifiedc C s! t j � � | j � Wd QXd S)N)r r ri )r/ r r r �+test_legacy_create_unix_server_ssl_verifiedU s
z?EventLoopTestsMixin.test_legacy_create_unix_server_ssl_verifiedc s' t d | j � � | j � f d d � t � \ } } } t j t j � } | j t j O_ t j | _
| j d t � t
| d � r� d | _ n | j j t | | d | d d �} | j j | � \ } } | j | d
| | f d t �� j j � | j � | j � | j j � j � d S) Nr' c s � S)Nr r )r� r r r� ] s zEEventLoopTestsMixin.test_create_server_ssl_verified.<locals>.<lambda>r rY Tr r r$ r� r� )r? r' rP r[ r rH rI rJ rK r\ r] re rf r
rY r� rg r� �PEERCERTr( ra r. )r/ r5 r( r r` rT r� r� r )r� r � test_create_server_ssl_verifiedY s& !
z3EventLoopTestsMixin.test_create_server_ssl_verifiedc C s! t j � � | j � Wd QXd S)N)r r rl )r/ r r r �&test_legacy_create_server_ssl_verifiedv s
z:EventLoopTestsMixin.test_legacy_create_server_ssl_verifiedc s t j d | j � � G� f d d � d t � } t j d t j � } | j t j t j d � | j d � | j j
| d | �} | j j | � } | j d } | j
| | � | j � \ } } | j | d � t j � } | j d | f � | j d
� | j � | j � d S) Nr' c s% e Z d Z � � f d d � Z � S)z@EventLoopTestsMixin.test_create_server_sock.<locals>.TestMyProtoc s! t � j | � � j | � d S)N)r@ r3 r2 )r/ r( )rB r� r r r3 ~ s zPEventLoopTestsMixin.test_create_server_sock.<locals>.TestMyProto.connection_made)r< r= r> r3 r )r� )rB r � TestMyProto} s rn r� r# �0.0.0.0r r� z 127.0.0.1s xxx)ro r )r+ r, r' r? r� r� �
setsockoptr= � SO_REUSEADDRr� r2 rg r4 r� r� r{ r� r� ra ) r/ rn �sock_obr� r5 r� r( r r� r )r� r �test_create_server_sockz s
z+EventLoopTestsMixin.test_create_server_sockc C s� t j d t j � } | j t j t j d � | j d � | j j t d | �} | j j | � } | j
d } | j � \ } } | j j t d | d | �} | j t
� � } | j j | � Wd QX| j | j j t j � | j � d S) Nr� r# �0.0.0.0r r� r( r )rt r )r� r� rp r= rq r� r' r2 r? rg r4 r� rj r r{ r r! r" ra )r/ rr r� r5 r� r( r r r r r � test_create_server_addr_in_use� s
z2EventLoopTestsMixin.test_create_server_addr_in_usez IPv6 not supported or enabledc s� t j d | j � � G� f d d � d t � } d } x� y@ t j � } | j j | d d d | �} | j j | � } WnZ t k
r� } z: | j t j
k r� | d 7} | j d | � w: n � WYd d } ~ Xq: XPq: Wt j � } | j
d | f � | j d
� | j j � � } | j j � | j � t j d | j � � t j t j � } | j
d | f � | j d
� | j j � � } | j j � | j � | j � d S) Nr' c s% e Z d Z � � f d d � Z � S)zFEventLoopTestsMixin.test_create_server_dual_stack.<locals>.TestMyProtoc s! t � j | � � j | � d S)N)r@ r3 r2 )r/ r( )rB �f_protor r r3 � s zVEventLoopTestsMixin.test_create_server_dual_stack.<locals>.TestMyProto.connection_made)r< r= r> r3 r )rv )rB r rn � s rn r r( r r# r z 127.0.0.1s xxxz::1)r+ r, r' r? r r r2 rg r r! r" ZassertGreaterEqualr� r� r� r( ra r8 ) r/ rn Z try_countr r� r5 Zexr� r� r )rv r � test_create_server_dual_stack� s:
z1EventLoopTestsMixin.test_create_server_dual_stackc C s� | j j t d d � } | j j | � } | j d } | j � \ } } t j � } | j d | f � | j d � | j � | j � t j � } | j
t | j d | f � | j � d S)Nz0.0.0.0r z 127.0.0.1s xxx) r' r2 r? rg r4 r� r� r� r� ra rj r� )r/ r� r5 r� r( r r� r r r �test_server_close� s
z%EventLoopTestsMixin.test_server_closec s� G� f d d � d t � } � j j | d d �} � j j | � \ } � | j d � \ } } � j | t j � � j � | � � j d � j � � j
� j | � � j j � f d d � d
| | f �} � j j | � \ } � � j | t j � � j � t � � j d � j � � j
� j | � | j d � t
j � j � f d d � � � j d
� j � t
j � j � f d d � � � j d � j � � j | j d � � | j � � j j � j � � j d � j � � j j � d S)Nc s7 e Z d Z � � f d d � Z � f d d � Z � S)zNEventLoopTestsMixin.test_create_datagram_endpoint.<locals>.TestMyDatagramProtoc s t � j d � j � d S)Nr' )r@ r0 r' )Z
inner_self)rB r/ r r r0 � s zWEventLoopTestsMixin.test_create_datagram_endpoint.<locals>.TestMyDatagramProto.__init__c s. t � j | | � | j j d | | � d S)Ns resp:)r@ rF r( �sendto)r/ r5 rE )rB r r rF � s z`EventLoopTestsMixin.test_create_datagram_endpoint.<locals>.TestMyDatagramProto.datagram_received)r< r= r> r0 rF r )r/ )rB r �TestMyDatagramProto� s rz r � 127.0.0.1r r� rD c s t d � j � S)Nr' )rC r' r )r/ r r r� � s zCEventLoopTestsMixin.test_create_datagram_endpoint.<locals>.<lambda>Z remote_addrs xxxc s � j S)N)r* r )r5 r r r� � s rO c s � j S)N)r* r )r� r r r� � s � r9 )r{ r )rC r' �create_datagram_endpointrg r� r� r+ r� r{ r) r� r( ry r r� r* r� ra r. )r/ rz �coroZ s_transportr( r r( r )r� r/ r5 r � test_create_datagram_endpoint� s6
z1EventLoopTestsMixin.test_create_datagram_endpointc s d } d
} � j j � j j d t j | � � } xe | D]] \ } } } } } y< t j d | d | d | � } | j d � | j | � Wn Yq: XPq: W� j j � f d d � d | �} � j j | � \ }
} � j |
t j
� � j | t � |
j � � j j | j
� d S) N� 127.0.0.1r r� r� r� Fc s t d � j � S)Nr' )rC r' r )r/ r r r� s zHEventLoopTestsMixin.test_create_datagram_endpoint_sock.<locals>.<lambda>r� )r� r )r' rg r� r� �
SOCK_DGRAMr� r� r� r� r+ r� rC ra r. ) r/ r� Z
local_addressr� r� r� r� r� r� r� r� r� r )r/ r �"test_create_datagram_endpoint_sock
s(
z6EventLoopTestsMixin.test_create_datagram_endpoint_sockc C s� | j � } t | t j � s8 | j � | j d � n | j d | j � | j � | j d | j � | j | j � | j | j
� d S)Nz#loop is not a BaseSelectorEventLoopr# r ) r] r� r ZBaseSelectorEventLoopra ZskipTestr{ Z
_internal_fdsr_ Z_csockZ_ssock)r/ r' r r r �test_internal_fds' s
z%EventLoopTestsMixin.test_internal_fds�win32z Don't support pipes for Windowsc sf t d � j � � t j � \ } } t j | d d � � t j � � � f d d � � } � j j | � � t j | d � t
j � j � f d d � � � j d � j
� t j | d
� t
j � j � f d d � � � j d d
g � j � � j d � j
� t j | � � j j � j � � j d d
d d g � j � � j � j j d � � d S)Nr' �rbi c 3 s � j j � f d d � � � Dd H\ } } � j | � � � j | � j � � j d d g � j � � j d � j � d S)Nc s � S)Nr r )r� r r r� > s zEEventLoopTestsMixin.test_read_pipe.<locals>.connect.<locals>.<lambda>r& r1 r )r' �connect_read_piper� r( r{ r) r* )r� �p)�pipeobjr� r/ r r r� ; s #z3EventLoopTestsMixin.test_read_pipe.<locals>.connect� 1c s
� j d k S)Nr# )r* r )r� r r r� G s z4EventLoopTestsMixin.test_read_pipe.<locals>.<lambda>r# s 2345c s
� j d k S)Nr )r* r )r� r r r� K s r& r1 r r7 r9 �pipe)rH r' r r� �io�openr+ ri rg rA r r� r{ r* r) ra r. r� r( r� )r/ �rpipe�wpiper� r )r� r� r/ r �test_read_pipe3 s" !
z"EventLoopTestsMixin.test_read_piper r� r| c sf t d � j � � t j � \ } } t j | d d � � t j � � � f d d � � } � j j | � � t j | d � t
j � j � f d d � � � j d � j
� t j | d
� t
j � j � f d d � � � j d d
g � j � � j d � j
� t j | � � j j � j � � j d d
d d g � j � � j � j j d � � d S)Nr' r� r c 3 s � j j � f d d � � � Dd H\ } } � j | � � � j | � j � � j d d g � j � � j d � j � d S)Nc s � S)Nr r )r� r r r� e s zKEventLoopTestsMixin.test_read_pty_output.<locals>.connect.<locals>.<lambda>r& r1 r )r' r� r� r( r{ r) r* )r� r� )�master_read_objr� r/ r r r� c s z9EventLoopTestsMixin.test_read_pty_output.<locals>.connectr� c s � j S)N)r* r )r� r r r� o s z:EventLoopTestsMixin.test_read_pty_output.<locals>.<lambda>r# s 2345c s
� j d k S)Nr )r* r )r� r r r� s s r& r1 r r7 r9 r� )rH r' r �openptyr� r� r+ ri rg rA r r� r{ r* r) ra r. r� r( r� )r/ �master�slaver� r )r� r� r/ r �test_read_pty_outputV s" !
z(EventLoopTestsMixin.test_read_pty_outputc s� t j � \ � } t j | d d � } t d | j � � | j j � f d d � | � } | j j | � \ } } | j | � � | j | � j � | j
d � j � | j d � t
� � � f d d � � t j | j � � f d
d � � | j
d � � | j d � t j | j � � f d d � � | j
d
� � | j
d � j � t j � � | j � j j d � � � j j � | j j � j � | j
d � j � d S)N�wbi r' c s � S)Nr r )r� r r r� � s z5EventLoopTestsMixin.test_write_pipe.<locals>.<lambda>r1 r� c s& t j � d � } | | 7} t | � S)Ni )r r� r4 )r5 �chunk)r� r r r� � s
z3EventLoopTestsMixin.test_write_pipe.<locals>.readerc s � � � d k S)Nr# r r )r5 r� r r r� � s s 2345c s � � � d k S)Nr r r )r5 r� r r r� � s s 12345r� r9 )r r� r� r� rJ r' �connect_write_piperg r� r( r{ r) rA r� r r� ra r� r� r. )r/ r� r� r� r( r� r )r5 r� r� r� r �test_write_pipe~ s, !
"
"
z#EventLoopTestsMixin.test_write_pipec s- t j � \ } } | j d � t j | j � d d � } t d | j � � | j j � f d d � | � } | j j | � \ } } | j
| � � | j
| � j � | j d � j
� | j d � | j j | j j | d � � } | j d | � | j � | j j � j � | j d � j
� d S)
NFr� i r' c s � S)Nr r )r� r r r� � s zIEventLoopTestsMixin.test_write_pipe_disconnect_on_close.<locals>.<lambda>r1 r� r9 )r r� r� r� r� �detachrJ r' r� rg r� r( r{ r) rA r� ra r. )r/ ZrsockZwsockr� r� r( r� r5 r )r� r �#test_write_pipe_disconnect_on_close� s
!
!
z7EventLoopTestsMixin.test_write_pipe_disconnect_on_closec s� t j � \ � } t j | d d � } t d | j � � | j j � f d d � | � } | j j | � \ } } | j | � � | j | � j � | j
d � j � | j d � t
� � � f d d � � t j | j � � f d
d � d d �| j
d � � | j d
� t j | j � � f d d � d d �| j
d � � | j
d � j � t j � � | j � j j d � � � j j � | j j � j � | j
d � j � d S)Nr� r r' c s � S)Nr r )r� r r r� � s z4EventLoopTestsMixin.test_write_pty.<locals>.<lambda>r1 r� c s& t j � d � } | | 7} t | � S)Ni )r r� r4 )r5 r� )r� r r r� � s
z2EventLoopTestsMixin.test_write_pty.<locals>.readerc s � � � d k S)Nr# r r )r5 r� r r r� � s Ztimeoutr s 2345c s � � � d k S)Nr r r )r5 r� r r r� � s s 12345r� r9 )r r� r� r� rJ r' r� rg r� r( r{ r) rA r� r r� ra r� r� r. )r/ r� Zslave_write_objr� r( r� r )r5 r� r� r� r �test_write_pty� s0 !
!
!
z"EventLoopTestsMixin.test_write_ptyc sJ t j � \ } } | j d � � j j | d � � t � d d � } | d k re � j | j � n t j � � f d d � � } t
j � } t j | � d � j �} � j j
� t
j � | } � j | d � � j | j � d � � j t j � j � | d k r"� j | j � n � j j | � | j � | j � d S) NFr# �ovc 3 s] zH y � j j � j � � Dd HWn t j k
r@ d } Yn Xd } Wd � j j � X| S)N� cancelled)r' r �cancelr+ �CancelledErrorrr )r� )r� r/ r r �main� s
z:EventLoopTestsMixin.test_prompt_cancellation.<locals>.mainr' g�������?r� )r r� r� r' r� �getattrre �pendingr+ ri rm ry r* rz Z
assertLessr{ �resultrj r� r� r1 ra )r/ r� r� r� r� r� r� �elapsedr )r� r/ r �test_prompt_cancellation� s&
z,EventLoopTestsMixin.test_prompt_cancellationc s� � � f d d � } � j j � d � j _ | � j _ t j � f d d � � } � j j | � � � j � j j d i � j j d 6� j j j j
d 6� d S) Nc s � j j d 7_ � � d S)Nr# )r' �_run_once_counterr )�
orig_run_oncer/ r r � _run_once
s z<EventLoopTestsMixin.test_timeout_rounding.<locals>._run_oncer c 3 s� � j } t j d d | �Dd Ht j d d | �Dd Ht j d d | �Dd Ht j d d | �Dd Ht j d d | �Dd Hd S)Ng{�G�z�?r' g-C ��6?g���ư>g:�0�yE>g�����|�=)r' r+ rn )r' )r/ r r �wait s z7EventLoopTestsMixin.test_timeout_rounding.<locals>.wait� Zclock_resolutionZselector) r' r� r� r+ ri rg ZassertLessEqualZ_clock_resolutionZ _selectorrB r< )r/ r� r� r )r� r/ r �test_timeout_rounding s z)EventLoopTestsMixin.test_timeout_roundingc C s� t j d f g } t j rC | j t j d f t j d f f � n x� | D]� \ } } x� t j t j f D]� } t j | | � } | �h | j d � | j j
| | � } | j t � � } | j j
| � Wd QX| j d t | j � � Wd QXqi WqJ Wd S) N�www.python.orgr- r Fzaddress must be resolved)r� r- )r� r- )r� r- r r )r� r' r � IPV6_ENABLEDr� r8 r� r� r� r' r� rj r� rg r# r� r )r/ Z addressesr� r� Z sock_typer� r� r r r r �test_sock_connect_address' s
z-EventLoopTestsMixin.test_sock_connect_addressc C s� | j � } d d � } t j � \ } } | j | j � | j | j � | j | | � | j | | � | j � | j | j | � � | j | j | � � d S)Nc S s d S)Nr r r r r r� < s zCEventLoopTestsMixin.test_remove_fds_after_closing.<locals>.<lambda>)
r] r r� r3 ra r� r� r� r� r� )r/ r' rw r� r� r r r � test_remove_fds_after_closing: s
z1EventLoopTestsMixin.test_remove_fds_after_closingc C s� | j � } d d � } t j � \ } } | j | j � | j | j � | j � | j t � � | j | | � Wd QX| j t � � | j | | � Wd QXd S)Nc S s d S)Nr r r r r r� H s z@EventLoopTestsMixin.test_add_fds_after_closing.<locals>.<lambda>) r] r r� r3 ra rj rk r� r� )r/ r' rw r� r� r r r �test_add_fds_after_closingF s
z.EventLoopTestsMixin.test_add_fds_after_closingc sT t j � f d d � � } | � j � } � j t � � � j j | � Wd QXd S)Nc s � j j � d S)N)r' ra )r' )r/ r r �
close_loopS s zEEventLoopTestsMixin.test_close_running_event_loop.<locals>.close_loop)r+ ri r' rj rk rg )r/ r� r~ r )r/ r � test_close_running_event_loopR s z1EventLoopTestsMixin.test_close_running_event_loopc : C s� | j j � t j d d � � } d d � } | � } | j | j � | j t � � | j j � Wd QX| j t � �* t j d | j � } | j j | � Wd QX| j t � � | j j
| � Wd QX| j t � � | j j | � Wd QX| j t � � | j j d | � Wd QX| j t � �% | j j
| j j � d | � Wd QX| j t � � | j j d | � Wd QX| j t � � | j j | � Wd QX| j t � � | j j t j | � Wd QXd S)Nc S s d S)Nr r r r r �test^ s z,EventLoopTestsMixin.test_close.<locals>.testc S s d S)NFr r r r r r� b s z0EventLoopTestsMixin.test_close.<locals>.<lambda>r' g �?g )r' ra r+ ri r3 rj rk rz r, rg r r� rx �call_atrm r� � create_taskr� r� �SIGTERM)r/ r� �funcr~ Zfutr r r �
test_close[ s0
&z EventLoopTestsMixin.test_close)_r< r= r> r\ rc rl rq rt r| r� r� r� r� r� r� r� r� �unittestZ
skipUnlessr
r� r� r� r� r� r� r� r� r� r� r� r� r� r r �skipIfr r r r r r r% r r r6 r7 r9 r; r> rD rE rG rO rP rQ rU rV rW rX ra rb rc rd rg rh ri rj rl rm rs ru r r� rw rx r r� r� r r r� �requires_mac_verZrequires_freebsd_versionr� r� r� r� r� r� r� r� r� r� r� r r )rB r r[ � s�
! ' ','' ' $$
*
$' ' ' $ * $ * $ * $ !) . "!"&!( $ r[ c @ s� e Z 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 j
d d � d d � � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# S)$�SubprocessTestsMixinc C s: t j d k r" | j | t � n | j t j | � d S)Nr� )r r r� r r{ r� r� )r/ rU r r r �check_terminated~ s z%SubprocessTestsMixin.check_terminatedc C s: t j d k r" | j | t � n | j t j | � d S)Nr� )r r r� r r{ r� r� )r/ rU r r r � check_killed� s z!SubprocessTestsMixin.check_killedc
C s- t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j d | j � | j d � } | j d � | j j | j d j � � t j � � | j � Wd QX| j j | j � | j | j � | j d | j d � d S)Nzecho.pyr1 r s Python The Winnerr# ) r r r
r r r' �subprocess_execr r rK r �
executablerg r� r- r{ r) �get_pipe_transportrA rV r� r r
ra rR r� rU r5 )r/ �progr� �transpr� �stdinr r r �test_subprocess_exec� s !
z)SubprocessTestsMixin.test_subprocess_execc C s t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j d | j � | j d � } | j d � | j j | j d j � � | j d j � | j d | j d � | j d � | j j | j d j � � | j d | j d � t j � � | j � Wd QX| j j | j � | j | j � d S)Nzecho.pyr1 r s Python r# s
The Winners Python The Winner) r r r
r r r' r� r r rK r r� rg r� r- r{ r) r� rA rV r� �clearr5 r r
ra rR r� rU )r/ r� r� r� r� r� r r r �test_subprocess_interactive� s( !
z0SubprocessTestsMixin.test_subprocess_interactivec C s | j j t j t | j � d � } | j j | � \ } } | j | t � | j j | j � | j d � j � | j j | j
� | j d | j � | j
t d d � | j j � D� � � | j | j d j d � d � | j | j d d � | j � d S)
Nz echo Pythonr c s s | ] } | j � Vq d S)N)r. )rL r� r r r � <genexpr>� s z=SubprocessTestsMixin.test_subprocess_shell.<locals>.<genexpr>r# s
s PythonrQ rP )r' �subprocess_shellr r rK rg r� r- r� ra rR r{ rU re �allrT �valuesr5 �rstrip)r/ r� r� r� r r r �test_subprocess_shell� s ) z*SubprocessTestsMixin.test_subprocess_shellc C s� | j j t j t | j � d d d d d d d �} | j j | � \ } } | j | t � | j j | j � | j d | j � | j
� d S)Nzexit 7r� �stdout�stderr� ) r' r� r r rK rg r� rR r{ rU ra )r/ r� r� r� r r r �test_subprocess_exitcode� s z-SubprocessTestsMixin.test_subprocess_exitcodec C s� | j j t j t | j � d d d d d d d �} | j j | � \ } } | j | t � | j | j d � � | j | j d � � | j | j d � � | j j | j � | j
d | j � | j | j � � d S) Nzexit 7r� r� r� r r# rQ r� )
r' r� r r rK rg r� r_ r� rR r{ rU ra )r/ r� r� r� r r r �"test_subprocess_close_after_finish� s z7SubprocessTestsMixin.test_subprocess_close_after_finishc C s� t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j � | j j | j � | j | j � | j � d S)Nzecho.py)r r r
r r r' r� r r rK r r� rg r� r- r� rR r� rU ra )r/ r� r� r� r� r r r �test_subprocess_kill� s !
z)SubprocessTestsMixin.test_subprocess_killc C s� t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j � | j j | j � | j | j � | j � d S)Nzecho.py)r r r
r r r' r� r r rK r r� rg r� r- Z terminaterR r� rU ra )r/ r� r� r� r� r r r �test_subprocess_terminate� s !
z.SubprocessTestsMixin.test_subprocess_terminateTz"Temporarily skipped (rhbz#1088233)c C s� t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j t j � | j j | j � | j t j | j � | j � d S)Nzecho.py)r r r
r r r' r� r r rK r r� rg r� r- Z send_signalr� �SIGHUPrR r{ rU ra )r/ r� r� r� r� r r r �test_subprocess_send_signal� s ! z0SubprocessTestsMixin.test_subprocess_send_signalc C s t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j d � } | j d � | j j | j � | j � | j d | j d � | j | j d j d � | j d � | j d | j � d S)Nzecho2.pyr s tests OUT:testr# rQ s ERR:test)r r r
r r r' r� r r rK r r� rg r� r- r� rA rR ra r{ r5 re r� rU )r/ r� r� r� r� r� r r r �test_subprocess_stderr s !
'z+SubprocessTestsMixin.test_subprocess_stderrc C sI t j j t j j t � d � } | j j t j t | j � t
j | d t j
�} | j j | � \ } } | j | t � | j j | j � | j d � } | j | j d � � | j | j d � � | j d � | j j | j � | j | j d j d � | j d � | j d | j d � | j � | j d | j � d S) Nzecho2.pyr� r r# rQ s tests OUT:testERR:testrP ) r r r
r r r' r� r r rK r r� �
subprocessZSTDOUTrg r� r- r� r� r_ rA rR re r5 r� r{ ra rU )r/ r� r� r� r� r� r r r �)test_subprocess_stderr_redirect_to_stdout# s" !
z>SubprocessTestsMixin.test_subprocess_stderr_redirect_to_stdoutc C s� t j j t j j t � d � } | j j t j t | j � t
j | � } | j j | � \ } } | j
| t � | j j | j � | j d � } | j d � } | j d � | j j | j d j � � | j d | j d � | j � | j j | j d � | j d � | j j | j d j � � t
j d k rY| j d | j d � n | j d
| j d � t j � � | j � Wd QX| j j | j � | j | j � d S) Nzecho3.pyr r# s tests OUT:tests xxxrQ r� s ERR:BrokenPipeErrors ERR:OSError) r r r
r r r' r� r r rK r r� rg r� r- r� rA rV r� r{ r5 ra rT r r r
rR r� rU )r/ r� r� r� r� r� r� r r r �#test_subprocess_close_client_stream: s. !
z8SubprocessTestsMixin.test_subprocess_close_client_streamc c s� | j j t j t | j � d d d d d d d d d �} | j j | � V\ } } | j | t � | j j | j � | j d | j � d S)Nzexit 7r� r� r� Zstart_new_sessionTr� )
r' r� r r rK rg r� rR r{ rU )r/ r� �_r� r r r �"test_subprocess_wait_no_same_group[ s z7SubprocessTestsMixin.test_subprocess_wait_no_same_groupc s� t j � f d d � � } � j t � � � j j | d d � � Wd QX� j t � � � j j | d d � � Wd QX� j t � � � j j | d d � � Wd QXd S)Nc ; s" � j j t j d | � Dd Hd S)N�pwd)r' r� r+ �SubprocessProtocol)r+ )r/ r r r� g s zGSubprocessTestsMixin.test_subprocess_exec_invalid_args.<locals>.connect�universal_newlinesT�bufsizei �shell)r+ ri rj r� r' rg )r/ r� r )r/ r �!test_subprocess_exec_invalid_argsf s z6SubprocessTestsMixin.test_subprocess_exec_invalid_argsc s� t j d � f d d � � } � j t � �! � j j | d d g � � Wd QX� j t � � � j j | d d � � Wd QX� j t � � � j j | d d � � Wd QX� j t � � � j j | d d
� � Wd QXd S) Nc ; s1 | s d } n � j j t j | | � Dd Hd S)Nr� )r' r� r+ r� )�cmdr+ )r/ r r r� u s
zHSubprocessTestsMixin.test_subprocess_shell_invalid_args.<locals>.connectZlsz-lr� Tr� i r� F)r+ ri rj r� r' rg )r/ r� r )r/ r �"test_subprocess_shell_invalid_argst s " z7SubprocessTestsMixin.test_subprocess_shell_invalid_argsN)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� | s
! r� r� c @ s e Z d Z d d � Z d S)�SelectEventLoopTestsc C s
t j � S)N)r+ �SelectorEventLoop)r/ r r r r] � s z&SelectEventLoopTests.create_event_loopN)r< r= r> r] r r r r r� � s r� c @ s� e Z d Z d d � Z e j � sc d d � Z d d � Z d d � Z d d
� Z d d � Z
n 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 d d � Z d! d" � Z d# S)$�ProactorEventLoopTestsc C s
t j � S)N)r+ ZProactorEventLoop)r/ r r r r] � s z(ProactorEventLoopTests.create_event_loopc C s t j d � � d S)Nz need python 3.5 (ssl.MemoryBIO))r� �SkipTest)r/ r r r r � s z1ProactorEventLoopTests.test_create_ssl_connectionc C s t j d � � d S)Nz need python 3.5 (ssl.MemoryBIO))r� r� )r/ r r r rU � s z-ProactorEventLoopTests.test_create_server_sslc C s t j d � � d S)Nz need python 3.5 (ssl.MemoryBIO))r� r� )r/ r r r ra � s z;ProactorEventLoopTests.test_create_server_ssl_verify_failedc C s t j d � � d S)Nz need python 3.5 (ssl.MemoryBIO))r� r� )r/ r r r rg � s z:ProactorEventLoopTests.test_create_server_ssl_match_failedc C s t j d � � d S)Nz need python 3.5 (ssl.MemoryBIO))r� r� )r/ r r r rl � s z6ProactorEventLoopTests.test_create_server_ssl_verifiedc C s t j d � � d S)Nz*IocpEventLoop incompatible with legacy SSL)r� r� )r/ r r r r � s z8ProactorEventLoopTests.test_legacy_create_ssl_connectionc C s t j d � � d S)Nz*IocpEventLoop incompatible with legacy SSL)r� r� )r/ r r r rV � s z4ProactorEventLoopTests.test_legacy_create_server_sslc C s t j d � � d S)Nz*IocpEventLoop incompatible with legacy SSL)r� r� )r/ r r r rb � s zBProactorEventLoopTests.test_legacy_create_server_ssl_verify_failedc C s t j d � � d S)Nz*IocpEventLoop incompatible with legacy SSL)r� r� )r/ r r r rh � s zAProactorEventLoopTests.test_legacy_create_server_ssl_match_failedc C s t j d � � d S)Nz*IocpEventLoop incompatible with legacy SSL)r� r� )r/ r r r rm � s z=ProactorEventLoopTests.test_legacy_create_server_ssl_verifiedc C s t j d � � d S)Nz(IocpEventLoop does not have add_reader())r� r� )r/ r r r r� � s z+ProactorEventLoopTests.test_reader_callbackc C s t j d � � d S)Nz(IocpEventLoop does not have add_reader())r� r� )r/ r r r �test_reader_callback_cancel� s z2ProactorEventLoopTests.test_reader_callback_cancelc C s t j d � � d S)Nz(IocpEventLoop does not have add_writer())r� r� )r/ r r r r� � s z+ProactorEventLoopTests.test_writer_callbackc C s t j d � � d S)Nz(IocpEventLoop does not have add_writer())r� r� )r/ r r r �test_writer_callback_cancel� s z2ProactorEventLoopTests.test_writer_callback_cancelc C s t j d � � d S)Nz6IocpEventLoop does not have create_datagram_endpoint())r� r� )r/ r r r r � s z4ProactorEventLoopTests.test_create_datagram_endpointc C s t j d � � d S)Nz(IocpEventLoop does not have add_reader())r� r� )r/ r r r r� � s z4ProactorEventLoopTests.test_remove_fds_after_closingN)r< r= r> r] r Z_is_sslproto_availabler rU ra rg rl r rV rb rh rm r� r� r� r� r r� r r r r r� � s$ r� )� selectorsc s4 e Z d Z � f d d � Z � f d d � Z � S)�UnixEventLoopTestsMixinc s: t � j � t j � } | j | j � t j | � d S)N)r@ r\ r+ ZSafeChildWatcherZ attach_loopr' �set_child_watcher)r/ Zwatcher)rB r r r\ � s
z UnixEventLoopTestsMixin.setUpc s t j d � t � j � d S)N)r+ r� r@ rc )r/ )rB r r rc � s
z UnixEventLoopTestsMixin.tearDown)r< r= r> r\ rc r r )rB r r� � s r� �KqueueSelectorc s� e Z d Z d d � Z e j d d � e j e j j
d � d � � f d d � � � Z e j d d � � f d d
� � Z � S) �KqueueEventLoopTestsc C s t j t j � � S)N)r+ r� r� r� )r/ r r r r] � s z&KqueueEventLoopTests.create_event_loopr � Zopenbsdztest hangs on OpenBSDc s t � j � d S)N)r@ r� )r/ )rB r r r� � s z)KqueueEventLoopTests.test_read_pty_outputc s t � j � d S)N)r@ r� )r/ )rB r r r� � s z#KqueueEventLoopTests.test_write_pty)
r< r= r> r] r r� r� r� r r r� r� r� r r )rB r r� � s
r� �
EpollSelectorc @ s e Z d Z d d � Z d S)�EPollEventLoopTestsc C s t j t j � � S)N)r+ r� r� r� )r/ r r r r] � s z%EPollEventLoopTests.create_event_loopN)r< r= r> r] r r r r r� � s r� � PollSelectorc @ s e Z d Z d d � Z d S)�PollEventLoopTestsc C s t j t j � � S)N)r+ r� r� r� )r/ r r r r] � s z$PollEventLoopTests.create_event_loopN)r< r= r> r] r r r r r� � s r� c @ s e Z d Z d d � Z d S)r� c C s t j t j � � S)N)r+ r� r� ZSelectSelector)r/ r r r r] s z&SelectEventLoopTests.create_event_loopN)r< r= r> r] r r r r r� � s c G s d S)Nr )r� r r r �noop s r� c @ sp e Z 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 S)� HandleTestsc C s" t j � | _ d | j j _ d S)NT)r r0 r' � get_debug� return_value)r/ r r r r\ s zHandleTests.setUpc C s~ d d � } f } t j | | | j � } | j | j | � | j | j | � | j | j � | j � | j | j � d S)Nc W s | S)Nr )r� r r r rw s z)HandleTests.test_handle.<locals>.callback)
r+ �Handler' r� � _callback�_argsr� �
_cancelledr� re )r/ rw r� �hr r r � test_handle s
zHandleTests.test_handlec C sJ d d � } t j | f d | j �} | j t t j | f | j � d S)Nc W s | S)Nr )r� r r r rw s z5HandleTests.test_handle_from_handle.<locals>.callbackr' )r+ r� r' rj �AssertionError)r/ rw �h1r r r �test_handle_from_handle s z#HandleTests.test_handle_from_handlec C s� d d � } t j � | _ t j � | j _ t j | f | j � } | j � | j j j i t j d � d 6t j
d 6| d 6| j d 6� d S)Nc S s
t � � d S)N)r� r r r r rw % s z:HandleTests.test_callback_with_exception.<locals>.callbackzException in callback.*�messager ZhandleZsource_traceback) r r0 r' rZ r+ r� Z_run�assert_called_withr Z MockPattern�ANY�_source_traceback)r/ rw r� r r r � test_callback_with_exception$ s
z(HandleTests.test_callback_with_exceptionc C s8 t j � } t j d d � f | j � } | | d <d S)Nc S s d S)Nr r r r r r� 7 s z1HandleTests.test_handle_weakref.<locals>.<lambda>r� )�weakref�WeakValueDictionaryr+ r� r' )r/ Zwdr� r r r �test_handle_weakref5 s z HandleTests.test_handle_weakrefc C s� d | j j _ t j t d
| j � } t j t � \ } } | j t | � d | | f � | j
� | j t | � d � t j t � } t j | f | j � } | j t | � d | | f � t j
t d d � } t j | d | j � } d t j | � | f } | j t | � | � t j d k r�t j } t j | � } t j | � \ } } t j | f | j � } d
} d | } d | t j | � | f } | j t | � | � n d S)NFr# rQ z <Handle noop(1, 2) at %s:%s>z<Handle cancelled>z<Handle noop() at %s:%s>rO z%^<Handle noop\(1, 2\)\(3\) at %s:%s>$� z*<function HandleTests.test_handle_repr .*>z%functools.partialmethod\(%s, , \)\(\)z^<Handle %s at %s:%s>$)r# rQ )rO )rO r
)r' r� r� r+ r� r� r �get_function_sourcer{ �reprr� ri r r �re�escapeZ assertRegexr � version_infor� �test_handle_repr�
partialmethod)r/ r� r �linenors Zregex�methodZcb_regexr r r r : s:
z HandleTests.test_handle_reprc C s� d | j j _ t } t j � j d } t j t d | j � } t
j t � \ } } | j t
| � d | | | | f � | j � | j t
| � d | | | | f � | j � | j t
| � d | | | | f � d S)NTr# rQ z-<Handle noop(1, 2) at %s:%s created at %s:%s>z7<Handle cancelled noop(1, 2) at %s:%s created at %s:%s>)r# rQ )r' r� r� r r � _getframe�f_linenor+ r� r� r r r{ r r� )r/ �create_filename�
create_linenor� r r r r r �test_handle_repr_debugd s$
z"HandleTests.test_handle_repr_debugc s� t j � j � } | j d � � j | � � f d d � } | j t � } | | � | j t � } | | � | j d t � } | | � | j d t � } | | � d S)NTc sW t j d � j d } � j | j t � � j | j d d d � t | d f � d S)Nr# rO � test_handle_source_tracebackr� )r r r r� r �listr{ r )r� r )r/ r r �check_source_traceback� s zHHandleTests.test_handle_source_traceback.<locals>.check_source_tracebackr ) r+ �get_event_loop_policy�new_event_loopr� r^ r r� r� rx )r/ r' r r� r )r/ r r ~ s
z(HandleTests.test_handle_source_tracebackN) r< r= r> r\ r� r r r r r r r r r r r�
s
* r� 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)�
TimerTestsc C s t j � | _ d S)N)r r0 r' )r/ r r r r\ � s zTimerTests.setUpc C sP t j � } t j | d d � f t j � � } | j t | � t | � � d S)Nc S s d S)NFr r r r r r� � s z&TimerTests.test_hash.<locals>.<lambda>)rm ry r+ � TimerHandler r0 r{ �hash)r/ �whenr� r r r � test_hash� s zTimerTests.test_hashc C s� d d � } d } t j � } t j | | | t j � � } | j | j | � | j | j | � | j | j
� | j � | j | j
� | j
| j � | j
| j � | j t t j d | | | j � d S)Nc W s | S)Nr )r� r r r rw � s z'TimerTests.test_timer.<locals>.callbackr# rQ rO )r# rQ rO )rm ry r+ r r r0 r� r� r� r� r� r� re r_ rj r� r' )r/ rw r� r! r� r r r �
test_timer� s
zTimerTests.test_timerc C sw d | j j _ t j d t f | j � } t j t � } | j t | � d | � | j
� | j t | � d � d S)NF�{ z&<TimerHandle when=123 noop() at %s:%s>z <TimerHandle cancelled when=123>) r' r� r� r+ r r� r r r{ r r� )r/ r� �srcr r r �test_timer_repr� s
zTimerTests.test_timer_reprc C s� d | j j _ t } t j � j d } t j d t f | j � } t
j t � \ } } | j t
| � d | | | | f � | j � | j t
| � d | | | | f � d S)NTr# r$ z7<TimerHandle when=123 noop() at %s:%s created at %s:%s>zA<TimerHandle cancelled when=123 noop() at %s:%s created at %s:%s>)r' r� r� r r r r r+ r r� r r r{ r r� )r/ r r r� r r r r r �test_timer_repr_debug� s
z TimerTests.test_timer_repr_debugc C so d d � } t j � } t j | | f | j � } t j | | f | j � } | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j � | j | | k � t j | | f | j � } t j | d | f | j � } | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � | j | | k � t j | f | j � } | j t
| j | � � | j t
| j | � � d S)Nc W s | S)Nr )r� r r r rw � s z2TimerTests.test_timer_comparison.<locals>.callbackg $@)
rm ry r+ r r' r� re r� r� r� �NotImplemented�__eq__�__ne__)r/ rw r! r Zh2Zh3r r r �test_timer_comparison� s>
z TimerTests.test_timer_comparisonN) r< r= r> r\ r" r# r& r' r+ r r r r r � s r c @ s e Z d Z d d � Z d S)�AbstractEventLoopTestsc C s� t j � } t j � } | j t | j � | j t | j d � | j t | j � | j t | j � | j t | j
� | j t | j � | j t | j d � | j t | j
d d � | j t | j | | � | j t | j d � | j t | j � | j t | j d � | j t | j | | � | j t | j | � | j t | j d d � | j t | j d � | j t | j | � | j t | j | � | j t | j | � | j t | j d | � | j t | j d � | j t | j d | � | j t | j d � | j t | j | d � | j t | j | d � | j t | j | | � | j t | j | � | j t | j! d | � | j t | j" d � | j t | j" d � | j t | j# | t j$ j% � | j t | j& | t j$ j% � | j t | j' | t j$ � | j t | j( | � | j t | j) | � | j t | j* | � | j t | j+ | � | j t | j, � | j t | j- | � d S)Nr$ � r# r )z localhostr- ).r r0 r+ �AbstractEventLooprj �NotImplementedErrorrz rg rr rf r_ ra r� rx r� r rm r� r� Zset_default_executorr� Z getnameinfor� r2 r} r� r� r� r� r� r� r� r� r� r� r� �sentinelr� r� r� r� Zset_exception_handlerZdefault_exception_handlerrZ r� r� )r/ r� r' r r r �test_not_implemented s�
z+AbstractEventLoopTests.test_not_implementedN)r< r= r> r1 r r r r r, s r, c @ s e Z d Z d d � Z d S)�ProtocolsAbsTestsc C sU t j � } t j � } | j | j | � � | j | j | � � | j | j | � � | j | j � � t j � } | j | j | � � | j | j | � � | j | j
| � � | j | j | | � � t j � } | j | j | � � | j | j | � � | j | j
d | � � | j | j d | � � | j | j � � d S)Nr# )r r0 r+ r� r_ r3 r; r6 r8 �DatagramProtocolrG rF r� rX rY rZ )r/ r� r� ZdpZspr r r �
test_emptyc s" z ProtocolsAbsTests.test_emptyN)r< r= r> r4 r r r r r2 a s r2 c @ s� e Z 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)� PolicyTestsc C s{ t j � } | j t | j � | j t | j t � � | j t | j � | j t | j � | j t | j t � � d S)N)
r+ �AbstractEventLoopPolicyrj r/ �get_event_loopr^ r^ r Zget_child_watcherr� )r/ �policyr r r �test_event_loop_policy{ s z"PolicyTests.test_event_loop_policyc C sx t j � } | j | j j � | j � } | j | t j � | j | j j | � | j | | j � � | j � d S)N)
r+ �DefaultEventLoopPolicyr_ �_localZ_loopr7 r� r. r� ra )r/ r8 r' r r r �test_get_event_loop� s z PolicyTests.test_get_event_loopc C sZ t j � } t j j | d d | j �� } | j � } | j | � Wd QX| j � d S)Nr^ �wraps) r+ r: r r r^ r^ r7 r ra )r/ r8 Zm_set_event_loopr' r r r �(test_get_event_loop_calls_set_event_loop� s z4PolicyTests.test_get_event_loop_calls_set_event_loopc C s0 t j � } | j d � | j t | j � d S)N)r+ r: r^ rj rk r7 )r/ r8 r r r �"test_get_event_loop_after_set_none� s
z.PolicyTests.test_get_event_loop_after_set_nonez'asyncio.events.threading.current_threadc s< � f d d � } t j d | � } | j � | j � d S)Nc s# t j � } � j t | j � d S)N)r+ r: rj rk r7 )r8 )r/ r r r� � s z1PolicyTests.test_get_event_loop_thread.<locals>.fr� )r� r� r� r
)r/ Zm_current_threadr� Zthr )r/ r �test_get_event_loop_thread� s
z&PolicyTests.test_get_event_loop_threadc C s9 t j � } | j � } | j | t j � | j � d S)N)r+ r: r r� r. ra )r/ r8 r' r r r �test_new_event_loop� s z PolicyTests.test_new_event_loopc C s� t j � } | j � } | j t | j t � � | j � } | j | � | j | | j � � | j | | j � � | j
� | j
� d S)N) r+ r: r7 rj r� r^ r^ r r� � assertIsNotra )r/ r8 Zold_loopr' r r r �test_set_event_loop� s
z PolicyTests.test_set_event_loopc C s9 t j � } | j | t j � | j | t j � � d S)N)r+ r r� r6 r� )r/ r8 r r r �test_get_event_loop_policy� s z&PolicyTests.test_get_event_loop_policyc C sh | j t t j t � � t j � } t j � } t j | � | j | t j � � | j | | � d S)N) rj r� r+ Zset_event_loop_policyr^ r r: r� rB )r/ Z
old_policyr8 r r r �test_set_event_loop_policy� s
z&PolicyTests.test_set_event_loop_policyN)r< r= r> r9 r< r> r? r r r@ rA rC rD rE r r r r r5 y s
r5 �__main__)?�__doc__r rb r� r r r
r� r� r � ImportErrorr� r r� rm r! r� r r r+ r r r r r� r r r r" rR rS r[ rf rk r� r% r? r3 rC rH Z BaseProtocolrJ r� rK r[ r� ZTestCaser� r� r� r� r
r� r� r� r� r� r r, r2 r5 r< r� r r r r �<module> s�
" )� � � � � �� 7 �mX[