File: //usr/lib/python3.5/asyncio/__pycache__/test_utils.cpython-35.pyc
Fa7 @ 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! 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& r
Gd' d( d( e j/ e Z0 Gd) d* d* e0 e Z1 Gd+ d, d, e1 Z2 Gd- d. d. e+ e2 Z3 d/ d0 Z4 e j5 d1 d2 Z6 e j5 d" d# d3 d4 Z7 e j5 d5 d6 d7 d d" d# d8 d9 Z8 d: d; Z9 Gd< d= d= e j: Z; Gd> d? d? e j< Z= d@ dA Z> GdB dC dC e? Z@ dD dE ZA GdF dG dG e
jB ZB e j5 dH dI ZC e jD e jE e jF dJ dK ZG dL dM ZH d S)NzUtilities shared by tests. N)mock)
HTTPServer)WSGIRequestHandler
WSGIServer ) base_events)compat)events)futures) selectors)tasks) coroutine)loggerZwin32)
socketpairc C s$ t d k r d St j t j Sd S)N)sslZ
SSLContextZPROTOCOL_SSLv23 r r (/usr/lib/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 once4 s zrun_briefly.<locals>.onceF)r
Z create_taskZ_log_destroy_pendingrun_until_completeclose)loopr gentr r r run_briefly3 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_untilB 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_onceL 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_stderrY 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" W 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 r0 d s z SilentWSGIServer.get_requestc C s d S)Nr )r% r3 client_addressr r r handle_errori s z SilentWSGIServer.handle_error)r* r+ r, r2 r0 r7 r r )r5 r r- ` 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..ZtestsZtestZ test_asyncioz ssl_key.pemz ssl_cert.pemkeyfilecertfileZ server_sideT) ospathjoindirname__file__isdirr Z wrap_socketZRequestHandlerClassr OSError)r% r3 r6 herer9 r: Zssockr r r finish_requesto s $
z!SSLWSGIServerMixin.finish_requestN)r* r+ r, rC r r r r r8 m s r8 c @ s e Z d Z d S)
SSLWSGIServerN)r* r+ r, r r r r rD s rD 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)rF rG r )environZstart_responseZstatusZheadersr r r app s
z _run_test_server.<locals>.apptargetc s j d d S)NZ
poll_intervalg?)Z
serve_foreverr )httpdr r <lambda> s z"_run_test_server.<locals>.<lambda>)
r" Zset_appZserver_addressaddress threadingZThreadstartZshutdownZ server_closer= )rM rE
server_clsserver_ssl_clsrI Z server_classZ
server_threadr )rK r _run_test_server s
rR 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 rW s zUnixHTTPServer.server_bindN)r* r+ r, rW r r r r rS s rS c s4 e Z d Z d Z d d Z f d d Z S)UnixWSGIServerr. c C s t j | | j d S)N)rS rW Z
setup_environ)r% r r r rW s
zUnixWSGIServer.server_bindc s/ t j \ } } | j | j | d f S)N 127.0.0.1 )rY rZ )r/ r0 r1 r2 )r% r3 r4 )r5 r r r0 s zUnixWSGIServer.get_request)r* r+ r, r2 rW r0 r r )r5 r rX s rX c @ s e Z d Z d d Z d S)SilentUnixWSGIServerc C s d S)Nr )r% r3 r6 r r r r7 s z!SilentUnixWSGIServer.handle_errorN)r* r+ r, r7 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 r` c c s@ t } z | VWd y t j | Wn t k
r: Yn XXd S)N)r` r; unlinkrA )r< r r r unix_socket_path s
rb c c s; t + } t d | d | d t d t Ed HWd QRXd S)NrM rE rP rQ )rb rR r[ r\ )rE r< r r r run_test_unix_server s rc hostz 127.0.0.1portc c s. t d | | f d | d t d t Ed Hd S)NrM rE rP rQ )rR r- rD )rd re rE r r r run_test_server s rf c C sn i } xH t | D]: } | j d r: | j d r: q t d d | | <q Wt d | f | j | S)N__ return_valueZ TestProtocol)dir
startswithendswith MockCallbacktype __bases__)baseZdctr^ r r r make_test_protocol s rp c @ sO e Z d Z d d Z d d d Z d d Z d d Z d
d Z d S) TestSelectorc C s
i | _ d S)N)keys)r% r r r __init__ s zTestSelector.__init__Nc C s) t j | d | | } | | j | <| S)Nr )r Z SelectorKeyrr )r% fileobjr datakeyr r r register s
zTestSelector.registerc C s | j j | S)N)rr pop)r% rt r r r
unregister s zTestSelector.unregisterc C s g S)Nr )r% r r r r select s zTestSelector.selectc C s | j S)N)rr )r% r r r get_map s zTestSelector.get_map)r* r+ r, rs rw ry rz r{ r r r r rq s
rq c s! e Z d Z d Z d f d d Z d d Z 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 f d# d$ Z f d% d& Z d' d( Z d) d* Z S)+TestLoopa Loop for unittests.
It manages self time directly.
If something scheduled to be executed later then
on next loop iteration after all ready handlers done
generator passed to __init__ is calling.
Generator should be like this:
def gen():
...
when = yield ...
... = yield time_advance
Value returned by yield is absolute time of next scheduled handler.
Value passed to yield is time advance to move loop's time forward.
Nc s t j | d k r1 d d } d | _ n d | _ | | _ t | j d | _ d | _ g | _ t | _ i | _
i | _ | j t
j | _ d S)Nc s s d Vd S)Nr r r r r r s z TestLoop.__init__.<locals>.genFTr g& .>)r/ rs _check_on_close_gennext_timeZ_clock_resolution_timersrq Z _selectorreaderswritersreset_countersweakrefWeakValueDictionary _transports)r% r )r5 r r rs s
zTestLoop.__init__c C s | j S)N)r )r% r r r r 3 s z
TestLoop.timec C s | r | j | 7_ d S)zMove test time forward.N)r )r% advancer r r advance_time6 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)r/ r r} r~ send
StopIterationAssertionError)r% )r5 r r r ; s
zTestLoop.closec G s t j | | | | j | <d S)N)r Handler )r% fdcallbackr( r r r _add_readerE s zTestLoop._add_readerc C s6 | j | d 7<| | j k r. | j | =d Sd Sd S)Nr TF)remove_reader_countr )r% r r r r _remove_readerH s
zTestLoop._remove_readerc G s | | j k s$ t d j | | j | } | j | k s[ t d j | j | | j | k s t d j | j | d S)Nzfd {} is not registeredz {!r} != {!r})r r r' _callback_args)r% r r r( handler r r
assert_readerP s $
zTestLoop.assert_readerc G s t j | | | | j | <d S)N)r r r )r% r r r( r r r _add_writerX s zTestLoop._add_writerc C s6 | j | d 7<| | j k r. | j | =d Sd Sd S)Nr TF)remove_writer_countr )r% r r r r _remove_writer[ s
zTestLoop._remove_writerc G s | | j k s$ t d j | | j | } | j | k s[ t d j | j | | j | k s t d j | j | d S)Nzfd {} is not registeredz {!r} != {!r})r r r' r r )r% r r r( r r r r
assert_writerc s $
zTestLoop.assert_writerc C sB y | j | } Wn t k
r% Yn Xt d j | | d S)Nz.File descriptor {!r} is used by transport {!r})r KeyError RuntimeErrorr' )r% r Z transportr r r _ensure_fd_no_transportk s
z TestLoop._ensure_fd_no_transportc G s | j | | j | | | S)zAdd a reader callback.)r r )r% r r r( r r r
add_readeru s
zTestLoop.add_readerc C s | j | | j | S)zRemove a reader callback.)r r )r% r r r r
remove_readerz s
zTestLoop.remove_readerc G s | j | | j | | | S)zAdd a writer callback..)r r )r% r r r( r r r
add_writer s
zTestLoop.add_writerc C s | j | | j | S)zRemove a writer callback.)r r )r% r r r r
remove_writer s
zTestLoop.remove_writerc C s( t j t | _ t j t | _ d S)N) collections defaultdictintr r )r% r r r r s zTestLoop.reset_countersc sM t j x0 | j D]% } | j j | } | j | q Wg | _ d S)N)r/ _run_oncer r~ r r )r% whenr )r5 r r r s
zTestLoop._run_oncec s& | j j | t j | | | S)N)r appendr/ call_at)r% r r r( )r5 r r r s zTestLoop.call_atc C s d S)Nr )r% Z
event_listr r r _process_events s zTestLoop._process_eventsc C s d S)Nr )r% r r r _write_to_self s zTestLoop._write_to_self)r* r+ r, __doc__rs r r r r r r r r r r r r r r r r r r r r r )r5 r r| s*
r| c K s t j d d g | S)Nspec__call__)r ZMock)kwargsr r r rl s rl c @ s" e Z d Z d Z d d Z d S) MockPatternzA regex based str with a fuzzy __eq__.
Use this helper with 'mock.assert_called_with', or anywhere
where a regex comparison between strings is needed.
For instance:
mock_call.assert_called_with(MockPattern('spam.*ham'))
c C s" t t j t | | t j S)N)boolresearchstrS)r% otherr r r __eq__ s zMockPattern.__eq__N)r* r+ r, r r r r r r r s r c C s2 t j | } | d k r. t d | f | S)Nz unable to get the source of %r)r Z_get_function_source
ValueError)funcsourcer r r get_function_source s r c @ s^ e Z d Z d d d d Z d d d Z d d Z d
d Z e j sZ d d
Z d S)TestCasecleanupTc C s9 | d k s t t j d | r5 | j | j d S)N)r r set_event_loopZ
addCleanupr )r% r r r r r r s
zTestCase.set_event_loopNc C s t | } | j | | S)N)r| r )r% r r r r r
new_test_loop s
zTestCase.new_test_loopc C s t j | _ d d t _ d S)Nc S s d S)Nr r r r r rL s z TestCase.setUp.<locals>.<lambda>)r _get_running_loop)r% r r r setUp s zTestCase.setUpc C s3 | j t _ t j d | j t j d d S)N)NNN)r r r Z assertEqualsysexc_info)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.<locals>.EmptyCMc S s d S)Nr )r% r r r __enter__ s z+TestCase.subTest.<locals>.EmptyCM.__enter__c W s d S)Nr )r% excr r r __exit__ s z*TestCase.subTest.<locals>.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 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 MagicMocksocketprotorm familyZ
gettimeoutrh )r rm r Zsockr r r mock_nonblocking_socket s r c C s t j d d d S)Nz'asyncio.sslproto._is_sslproto_availablerh F)r Zpatchr r r r force_legacy_ssl_support s r )Ir r
contextlibr# r r; r r rU r r] rN r Zunittestr r Z http.serverr Zwsgiref.simple_serverr r r ImportErrorrZ r r r r
r r Z
coroutinesr
logr platformZ
windows_utilsr r r r r! r" r- r8 rD rR hasattrrV rS rX r[ r\ r` contextmanagerrb rc rf rp Z BaseSelectorrq Z
BaseEventLoopr| rl r r r r r Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr r r r r r <module> s~
%