]8 @ s 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. e/ e d& rGd' d( d( e j0 e Z1 Gd) d* d* e1 e Z2 Gd+ d, d, e2 Z3 Gd- d. d. e, e3 Z4 d/ d0 Z5 e j6 d1 d2 Z7 e j6 d" d# d3 d4 Z8 e j6 d5 d6 d7 d d" d# d8 d9 Z9 d: d; Z: Gd< d= d= e j; Z< Gd> d? d? e j= Z> d@ dA Z? GdB dC dC e@ ZA dD dE ZB GdF dG dG e jC ZC e j6 dH dI ZD e jE e jF e jG dJ dK ZH dL dM ZI d S)N 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 //opt/alt/python35/lib64/python3.5/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 sleep)r Zpredtimeoutdeadliner r r run_untilC s r# c C s | j | j | j d S)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 r4 e s z SilentWSGIServer.get_requestc C s d S)Nr )r) r7 client_addressr r r handle_errorj s z SilentWSGIServer.handle_error)r. r/ r0 r6 r4 r; r r )r9 r r1 a s r1 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) r7 r: herer> r? Zssockr r r finish_requestp s $
z!SSLWSGIServerMixin.finish_requestN)r. r/ r0 rH r r r r r< n s r< c @ s e Z d Z d S)
SSLWSGIServerN)r. r/ r0 r r r r rI s rI 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)rK rL r )environZstart_responsestatusZheadersr 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 threadingThreadstartshutdownZ server_closerB )rS rJ
server_clsserver_ssl_clsrO Z server_classZ
server_threadr )rQ r _run_test_server s
rZ AF_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 r` s zUnixHTTPServer.server_bindN)r. r/ r0 r` r r r r r\ s r\ c s4 e Z d Z d Z d d Z f d d Z S)UnixWSGIServerr2 c C s t j | | j d S)N)r\ r` Z
setup_environ)r) r r r r` s
zUnixWSGIServer.server_bindc s/ t j \ } } | j | j | d f S)N 127.0.0.1 )rb rc )r3 r4 r5 r6 )r) r7 r8 )r9 r r r4 s zUnixWSGIServer.get_request)r. r/ r0 r6 r` r4 r r )r9 r ra s ra c @ s e Z d Z d d Z d S)SilentUnixWSGIServerc C s d S)Nr )r) r7 r: r r r r; s z!SilentUnixWSGIServer.handle_errorN)r. r/ r0 r; r r r r rd s rd c @ s e Z d Z d S)UnixSSLWSGIServerN)r. r/ r0 r r r r re s re c C s! t j } | j SWd QRXd S)N)tempfileNamedTemporaryFilename)filer r r gen_unix_socket_path s rj c c s@ t } z | VWd y t j | Wn t k
r: Yn XXd S)N)rj r@ unlinkrF )rA r r r unix_socket_path s
rl c c s; t + } t d | d | d t d t Ed HWd QRXd S)NrS rJ rX rY )rl rZ rd re )rJ rA r r r run_test_unix_server s rm hostz 127.0.0.1portc c s. t d | | f d | d t d t Ed Hd S)NrS rJ rX rY )rZ r1 rI )rn ro rJ r r r run_test_server s rp c C sn i } xH t | D]: } | j d r: | j d r: q t d d | | .genFTr g& .>)r3 r~ _check_on_close_gennext_timeZ_clock_resolution_timersr| _selectorreaderswritersreset_countersweakrefWeakValueDictionary _transports)r) r )r9 r r r~ 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)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)r3 r r r send
StopIterationAssertionError)r) )r9 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/ r0 r r r r r r EmptyCM s r r )r) r, r r r r r subTest s zTestCase.subTest) r. r/ r0 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)Nr )r levelsetLevelloggingCRITICAL)Z old_levelr r r disable_logger s
r c C s= t j t j } | | _ | | _ | | _ d | j _ | S)Ng )r Z MagicMocksocketprotorw family
gettimeoutrr )r rw r sockr r r mock_nonblocking_socket s r c C s t j d d d S)Nz'asyncio.sslproto._is_sslproto_availablerr F)r Zpatchr r r r force_legacy_ssl_support s r )Jr
contextlibr' r r@ r r r^ r rf rT r Zunittestr r Z http.serverr Zwsgiref.simple_serverr r r ImportErrorrc r r r r
r r Z
coroutinesr
logr r= r platformZ
windows_utilsr r r r# r% r& r1 r< rI rZ hasattrr_ r\ ra rd re rj contextmanagerrl rm rp r{ BaseSelectorr| Z
BaseEventLoopr rv r r r r r IPPROTO_TCP SOCK_STREAMAF_INETr r r r r r s~
-