Yf8 @ 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 Z d d l
m Z d d l m Z d d l m Z m Z y d d l Z Wn e k
rd Z Yn Xd d l m Z d d l m Z d d l m Z d d
l m Z d d l m Z d d l m Z d d
l m Z d d l m! Z! d d l" m# Z# e j$ d k rd d l% m& Z& n d d l m& Z& d d Z' d d Z( d d d Z) d d Z* Gd d d e Z+ Gd d d e Z, Gd d d Z- Gd! d" d" e- e, Z. d# d$ d% d& Z/ e0 e d' r Gd( d) d) e j1 e Z2 Gd* d+ d+ e2 e Z3 Gd, d- d- e3 Z4 Gd. d/ d/ e- e4 Z5 d0 d1 Z6 e j7 d2 d3 Z8 e j7 d# d$ d4 d5 Z9 e j7 d6 d7 d8 d d# d$ d9 d: Z: d; d< Z; Gd= d> d> e j< Z= Gd? d@ d@ e j> Z? dA dB Z@ GdC dD dD eA ZB dE dF ZC GdG dH dH e
jD ZD e j7 dI dJ ZE e jF e jG e jH dK dL ZI dM dN ZJ d S)OzUtilities shared by tests. N)mock)
HTTPServer)WSGIRequestHandler
WSGIServer ) base_events)compat)events)futures) selectors)tasks) coroutine)logger)supportwin32)
socketpairc C s$ t d k r d St j t j Sd S)N)sslZ
SSLContextZPROTOCOL_SSLv23 r r 7/opt/alt/python35/lib64/python3.5/asyncio/test_utils.pydummy_ssl_context- s r c
C sV t d d } | } | j | } d | _ z | j | Wd | j Xd S)Nc S s d S)Nr r r r r once5 s zrun_briefly..onceF)r
Z create_taskZ_log_destroy_pendingrun_until_completeclose)loopr gentr r r run_briefly4 s r c C st t j | } x] | so | d k rP | t j } | d k rP t j | j t j d d | q Wd S)Nr gMbP?r )timer
TimeoutErrorr r Zsleep)r ZpredtimeoutZdeadliner r r run_untilC s r! c C s | j | j | j d S)zLegacy API to run once through the event loop.
This is the recommended pattern for test code. It will poll the
selector once and run all callbacks scheduled in response to I/O
events.
N)Z call_soonstopZ run_forever)r r r r run_onceM s r# c @ s( e Z d Z d d Z d d Z d S)SilentWSGIRequestHandlerc C s
t j S)N)ioStringIO)selfr r r
get_stderrZ s z#SilentWSGIRequestHandler.get_stderrc G s d S)Nr )r' formatargsr r r log_message] s z$SilentWSGIRequestHandler.log_messageN)__name__
__module__ __qualname__r( r+ r r r r r$ X s r$ c s4 e Z d Z d Z f d d Z d d Z S)SilentWSGIServer c s/ t j \ } } | j | j | | f S)N)super get_request
settimeoutrequest_timeout)r' request client_addr) __class__r r r2 e s z SilentWSGIServer.get_requestc C s d S)Nr )r' r5 client_addressr r r handle_errorj s z SilentWSGIServer.handle_error)r, r- r. r4 r2 r9 r r )r7 r r/ a s r/ c @ s e Z d Z d d Z d S)SSLWSGIServerMixinc C s t j j t j j t d d } t j j | s] t j j t j j t j d d } t j j | d } t j j | d } t j | d | d | d d
} y! | j | | | | j Wn t
k
r Yn Xd S) Nz..ZteststestZ test_asyncioz ssl_key.pemz ssl_cert.pemkeyfilecertfileZ server_sideT) ospathjoindirname__file__isdirr Z wrap_socketZRequestHandlerClassr OSError)r' r5 r8 herer< r= Zssockr r r finish_requestp s $
z!SSLWSGIServerMixin.finish_requestN)r, r- r. rF r r r r r: n s r: c @ s e Z d Z d S)
SSLWSGIServerN)r, r- r. r r r r rG s rG use_sslFc
# s d d } | r | n | } | | t j | j _ t j d f d d } | j z VWd j j | j Xd S)Nc S s# d } d g } | | | d g S)Nz200 OK Content-type
text/plains Test message)rI rJ r )environZstart_responseZstatusZheadersr r r app s
z _run_test_server..apptargetc s j d d S)NZ
poll_intervalg?)Z
serve_foreverr )httpdr r s z"_run_test_server..)
r$ Zset_appZserver_addressaddress threadingZThreadstartZshutdownZ server_closer@ )rP rH
server_clsserver_ssl_clsrL Z server_classZ
server_threadr )rN r _run_test_server s
rU ZAF_UNIXc @ s e Z d Z d d Z d S)UnixHTTPServerc C s& t j j | d | _ d | _ d S)Nz 127.0.0.1P ) socketserverUnixStreamServer server_bindZ server_nameZ server_port)r' r r r rZ s zUnixHTTPServer.server_bindN)r, r- r. rZ r r r r rV s rV c s4 e Z d Z d Z d d Z f d d Z S)UnixWSGIServerr0 c C s t j | | j d S)N)rV rZ Z
setup_environ)r' r r r rZ s
zUnixWSGIServer.server_bindc s/ t j \ } } | j | j | d f S)N 127.0.0.1 )r\ r] )r1 r2 r3 r4 )r' r5 r6 )r7 r r r2 s zUnixWSGIServer.get_request)r, r- r. r4 rZ r2 r r )r7 r r[ s r[ c @ s e Z d Z d d Z d S)SilentUnixWSGIServerc C s d S)Nr )r' r5 r8 r r r r9 s z!SilentUnixWSGIServer.handle_errorN)r, r- r. r9 r r r r r^ s r^ c @ s e Z d Z d S)UnixSSLWSGIServerN)r, r- r. r r r r r_ s r_ c C s! t j } | j SWd QRXd S)N)tempfileZNamedTemporaryFilename)filer r r gen_unix_socket_path s rc c c s@ t } z | VWd y t j | Wn t k
r: Yn XXd S)N)rc r> unlinkrD )r? r r r unix_socket_path s
re c c s; t + } t d | d | d t d t Ed HWd QRXd S)NrP rH rS rT )re rU r^ r_ )rH r? r r r run_test_unix_server s rf hostz 127.0.0.1portc c s. t d | | f d | d t d t Ed Hd S)NrP rH rS rT )rU r/ rG )rg rh rH r r r run_test_server s ri c C sn i } xH t | D]: } | j d r: | j d r: q t d d | | .genFTr g& .>)r1 rv _check_on_close_gennext_timeZ_clock_resolution_timersrt Z _selectorreaderswritersreset_countersweakrefWeakValueDictionary _transports)r' r )r7 r r rv s
zTestLoop.__init__c C s | j S)N)r )r' r r r r 4 s z
TestLoop.timec C s | r | j | 7_ d S)zMove test time forward.N)r )r' advancer r r advance_time7 s zTestLoop.advance_timec sO t j | j rK y | j j d Wn t k
r> Yn
Xt d d S)Nr z Time generator is not finished)r1 r r r send
StopIterationAssertionError)r' )r7 r r r < s
zTestLoop.closec G s t j | | | | j | .)r r r Zthreading_setup_thread_cleanup)r' r r r setUp s zTestCase.setUpc C sU | j t j d | j t j d | j t j | j t j
d S)N)NNN) r r r Z assertEqualsysexc_infoZ
doCleanupsr Zthreading_cleanupr Z
reap_children)r' r r r tearDown s
zTestCase.tearDownc O s Gd d d } | S)Nc @ s( e Z d Z d d Z d d Z d S)z!TestCase.subTest..EmptyCMc S s d S)Nr )r' r r r __enter__ s z+TestCase.subTest..EmptyCM.__enter__c W s d S)Nr )r' excr r r __exit__ s z*TestCase.subTest..EmptyCM.__exit__N)r, r- r. r r r r r r EmptyCM s r r )r' r* r r r r r subTest s zTestCase.subTest) r, r- r. r r r r r r ZPY34r r r r r r s
r c c s; t j } z t j t j d d VWd t j | Xd S)zrContext manager to disable asyncio logger.
For example, it can be used to ignore warnings in debug mode.
r N)r levelZsetLevelloggingZCRITICAL)Z old_levelr r r disable_logger s
r c C s= t j t j } | | _ | | _ | | _ d | j _ | S)z'Create a mock of a non-blocking socket.g )r Z MagicMocksocketprotorp familyZ
gettimeoutrk )r rp r Zsockr r r mock_nonblocking_socket s r c C s t j d d d S)Nz'asyncio.sslproto._is_sslproto_availablerk F)r Zpatchr r r r force_legacy_ssl_support s r )Kr r
contextlibr% r r> r r rX r r` rQ r Zunittestr r Z http.serverr Zwsgiref.simple_serverr r r ImportErrorr] r r r r
r r Z
coroutinesr
logr r; r platformZ
windows_utilsr r r r! r# r$ r/ r: rG rU hasattrrY rV r[ r^ r_ rc contextmanagerre rf ri rs Z BaseSelectorrt Z
BaseEventLoopr ro r r r r r Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr r r r r r s
-