B 18\@ @sdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZddlmZddlmZmZy ddlZWnek rdZYnXddlmZddlmZddlmZdd lmZdd lmZdd lmZdd l m!Z!d dZ"e"dZ#e"dZ$e"dZ%e"dZ&ddddddddddd Z'ddZ(d d!d"d#Z)d$d%Z*d&d'Z+d`d)d*Z,d+d,Z-Gd-d.d.eZ.Gd/d0d0eZ/Gd1d2d2Z0Gd3d4d4e0e/Z1d5d6d7d8Z2e3ed9rdGd:d;d;e j4eZ5Gdd?d?e6Z7Gd@dAdAe0e7Z8dBdCZ9ej:dDdEZ;ej:d5d6dFdGZGdNdOdOej?Z@GdPdQdQejAZBdRdSZCGdTdUdUeDZEGdVdWdWZFdXdYZGGdZd[d[ejHZHej:d\d]ZIejJejKejLfd^d_ZMdS)azUtilities shared by tests.N)mock) HTTPServer)WSGIRequestHandler WSGIServer) base_events)events)format_helpers)futures)tasks)logger)supportcCs^ttdr*tjtj|}tj|r*|Stjtjtd|}tj|rR|St |dS)N TEST_HOME_DIRz..) hasattrr ospathjoinr isfiledirname__file__FileNotFoundError)filenamefullnamer[/oak/stanford/groups/akundaje/marinovg/programs/Python-3.7.3/Lib/test/test_asyncio/utils.py data_file&s   rz ssl_cert.pemz ssl_key.pemz keycert3.pemz pycacert.pem)z)http://testca.pythontest.net/testca/ocsp/)z0http://testca.pythontest.net/testca/pycacert.cer)z2http://testca.pythontest.net/testca/revocation.crl))) countryNameXY))organizationNamezPython Software Foundation CA)) commonNamez our-ca-serverzJul 7 14:23:16 2028 GMTzAug 29 14:23:16 2018 GMTZCB2D80995A69525C)))rr))Z localityNamezCastle Anthrax))rzPython Software Foundation))r localhost))ZDNSr) ZOCSPZ caIssuersZcrlDistributionPointsZissuerZnotAfterZ notBeforeZ serialNumbersubjectZsubjectAltNameversioncCs*ttj}|ttd|_tj|_|S)NF) ssl SSLContextZPROTOCOL_TLS_SERVERload_cert_chainONLYCERTONLYKEYcheck_hostname CERT_NONE verify_mode)Zserver_contextrrrsimple_server_sslcontextHs   r+T)disable_verifycCs"ttj}d|_|rtj|_|S)NF)r#r$ZPROTOCOL_TLS_CLIENTr(r)r*)r,Zclient_contextrrrsimple_client_sslcontextPs  r-cCstdkr dSttjSdS)N)r#r$Z PROTOCOL_TLSrrrrdummy_ssl_contextXsr.cCs<dd}|}||}d|_z||Wd|XdS)NcsdS)Nrrrrronce`szrun_briefly..onceF) create_task_log_destroy_pendingrun_until_completeclose)loopr/gentrrr run_briefly_s r7cCsTt|}xB|sN|dk r8|t}|dkr8t|tjd|dqWdS)NrgMbP?)r4)time monotonicr TimeoutErrorr2r sleep)r4predtimeoutdeadlinerrr run_untilms  r@cCs||j|dS)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) call_soonstop run_forever)r4rrrrun_oncews rDc@seZdZddZddZdS)SilentWSGIRequestHandlercCstS)N)ioStringIO)selfrrr get_stderrsz#SilentWSGIRequestHandler.get_stderrcGsdS)Nr)rHformatargsrrr log_messagesz$SilentWSGIRequestHandler.log_messageN)__name__ __module__ __qualname__rIrLrrrrrEsrEcs(eZdZdZfddZddZZS)SilentWSGIServercs"t\}}||j||fS)N)super get_request settimeoutrequest_timeout)rHrequest client_addr) __class__rrrSs zSilentWSGIServer.get_requestcCsdS)Nr)rHrVclient_addressrrr handle_errorszSilentWSGIServer.handle_error)rMrNrOrUrSrZ __classcell__rr)rXrrPs rPc@seZdZddZdS)SSLWSGIServerMixincCsVt}|tt|j|dd}y|||||Wntk rPYnXdS)NT) server_side) r#r$r%r&r'Z wrap_socketZRequestHandlerClassr3OSError)rHrVrYcontextZssockrrrfinish_requests  z!SSLWSGIServerMixin.finish_requestN)rMrNrOr`rrrrr\sr\c@s eZdZdS) SSLWSGIServerN)rMrNrOrrrrrasraF)use_sslc#svdd}|r|n|}||t|j_tjfddd}|z VWd| XdS)NcSsd}dg}|||dgS)Nz200 OK)z Content-typez text/plains Test messager)environZstart_responsestatusheadersrrrapps z_run_test_server..appcs jddS)Ng?)Z poll_interval) serve_foreverr)httpdrrz"_run_test_server..)target) rEZset_appZserver_addressaddress threadingThreadstartshutdownZ server_closer)rlrb server_clsserver_ssl_clsrfZ server_classZ server_threadr)rhr_run_test_servers    rsAF_UNIXc@seZdZddZdS)UnixHTTPServercCstj|d|_d|_dS)Nz 127.0.0.1P) socketserverUnixStreamServer server_bindZ server_nameZ server_port)rHrrrrys zUnixHTTPServer.server_bindN)rMrNrOryrrrrrusrucs(eZdZdZddZfddZZS)UnixWSGIServerrQcCst||dS)N)ruryZ setup_environ)rHrrrrys zUnixWSGIServer.server_bindcs"t\}}||j|dfS)N)z 127.0.0.1)rRrSrTrU)rHrVrW)rXrrrSs zUnixWSGIServer.get_request)rMrNrOrUryrSr[rr)rXrrzsrzc@seZdZddZdS)SilentUnixWSGIServercCsdS)Nr)rHrVrYrrrrZsz!SilentUnixWSGIServer.handle_errorN)rMrNrOrZrrrrr|sr|c@s eZdZdS)UnixSSLWSGIServerN)rMrNrOrrrrr}sr}c Cst}|jSQRXdS)N)tempfileNamedTemporaryFilename)filerrrgen_unix_socket_paths rccs<t}z |VWdyt|Wntk r4YnXXdS)N)rrunlinkr^)rrrrunix_socket_paths rc cs,t}t||ttdEdHWdQRXdS)N)rlrbrqrr)rrsr|r})rbrrrrrun_test_unix_serversrz 127.0.0.1)hostportrbccst||f|ttdEdHdS)N)rlrbrqrr)rsrPra)rrrbrrrrun_test_server s rcCsPi}x4t|D](}|dr(|dr(qtdd||<qWtd|f|j|S)N__) return_valueZ TestProtocol)dir startswithendswith MockCallbacktype __bases__)basedctrrrrmake_test_protocols rc@s6eZdZddZd ddZddZdd Zd d ZdS) TestSelectorcCs i|_dS)N)keys)rHrrr__init__szTestSelector.__init__NcCst|d||}||j|<|S)Nr) selectors SelectorKeyr)rHfileobjrdatakeyrrrregisters zTestSelector.registercCs |j|S)N)rpop)rHrrrr unregister$szTestSelector.unregistercCsgS)Nr)rHr>rrrselect'szTestSelector.selectcCs|jS)N)r)rHrrrget_map*szTestSelector.get_map)N)rMrNrOrrrrrrrrrrs  rcseZdZdZd.fdd ZddZddZfd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zfd%d&Zdd'fd(d) Zd*d+Zd,d-ZZS)/TestLoopaLoop 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. Ncsvt|dkr"dd}d|_nd|_||_t|jd|_d|_g|_t|_ i|_ i|_ | t |_dS)Ncss dVdS)Nrrrrrr5EszTestLoop.__init__..genFTrg& .>)rRr_check_on_close_gennext_time_clock_resolution_timersr _selectorreaderswritersreset_countersweakrefWeakValueDictionary _transports)rHr5)rXrrrAs  zTestLoop.__init__cCs|jS)N)r)rHrrrr9Xsz TestLoop.timecCs|r|j|7_dS)zMove test time forward.N)r)rHadvancerrr advance_time[szTestLoop.advance_timecsBt|jr>y|jdWntk r4Yn XtddS)NrzTime generator is not finished)rRr3rrsend StopIterationAssertionError)rH)rXrrr3`s zTestLoop.closecGst|||d|j|<dS)N)rHandler)rHfdcallbackrKrrr _add_readerjszTestLoop._add_readercCs0|j|d7<||jkr(|j|=dSdSdS)NTF)remove_reader_countr)rHrrrr_remove_readerms  zTestLoop._remove_readercGsh||jkrtd|d|j|}|j|krDtd|jd||j|krdtd|jd|dS)Nzfd z is not registeredzunexpected callback: z != zunexpected callback args: )rr _callback_args)rHrrrKhandlerrr assert_readerus    zTestLoop.assert_readercCs||jkrtd|ddS)Nzfd z is registered)rr)rHrrrrassert_no_readers zTestLoop.assert_no_readercGst|||d|j|<dS)N)rrr)rHrrrKrrr _add_writerszTestLoop._add_writercCs0|j|d7<||jkr(|j|=dSdSdS)NrTF)remove_writer_countr)rHrrrr_remove_writers  zTestLoop._remove_writercGs^||jkstd||j|}|j|ks>td|j||j|ksZtd|j|dS)Nzfd {} is not registeredz {!r} != {!r})rrrJrr)rHrrrKrrrr assert_writers   zTestLoop.assert_writerc Cs~t|tsFyt|}Wn*tttfk rDtd|dYnXy|j|}Wntk rhYnXt d||dS)NzInvalid file object: {!r}z.File descriptor {!r} is used by transport {!r}) isinstanceintfilenoAttributeError TypeError ValueErrorrJrKeyError RuntimeError)rHr transportrrr_ensure_fd_no_transports z TestLoop._ensure_fd_no_transportcGs|||j||f|S)zAdd a reader callback.)rr)rHrrrKrrr add_readers zTestLoop.add_readercCs||||S)zRemove a reader callback.)rr)rHrrrr remove_readers zTestLoop.remove_readercGs|||j||f|S)zAdd a writer callback..)rr)rHrrrKrrr add_writers zTestLoop.add_writercCs||||S)zRemove a writer callback.)rr)rHrrrr remove_writers zTestLoop.remove_writercCstt|_tt|_dS)N) collections defaultdictrrr)rHrrrrs zTestLoop.reset_counterscs:tx$|jD]}|j|}||qWg|_dS)N)rR _run_oncerrrr)rHwhenr)rXrrrs    zTestLoop._run_once)r_cs&|j|tj||f|d|iS)Nr_)rappendrRcall_at)rHrrr_rK)rXrrrs zTestLoop.call_atcCsdS)Nr)rH event_listrrr_process_eventsszTestLoop._process_eventscCsdS)Nr)rHrrr_write_to_selfszTestLoop._write_to_self)N)rMrNrO__doc__rr9rr3rrrrrrrrrrrrrrrrrr[rr)rXrr.s,   rcKstjfddgi|S)Nspec__call__)rMock)kwargsrrrrsrc@seZdZdZddZdS) 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')) cCsttt||tjS)N)boolresearchstrS)rHotherrrr__eq__szMockPattern.__eq__N)rMrNrOrrrrrrrsrc@seZdZddZddZdS)MockInstanceOfcCs ||_dS)N)_type)rHrrrrrszMockInstanceOf.__init__cCs t||jS)N)rr)rHrrrrrszMockInstanceOf.__eq__N)rMrNrOrrrrrrrsrcCs$t|}|dkr td|f|S)Nzunable to get the source of %r)r_get_function_sourcer)funcsourcerrrget_function_sources rc@sHeZdZeddZddddZddd Zd d Zd d ZddZ dS)TestCasecCs&|j}|dk r|jdd|dS)NT)wait)_default_executorrpr3)r4executorrrr close_loops zTestCase.close_loopT)cleanupcCs,|dk s ttd|r(||j|dS)N)rrset_event_loop addCleanupr)rHr4rrrrrs  zTestCase.set_event_loopNcCst|}|||S)N)rr)rHr5r4rrr new_test_loops zTestCase.new_test_loopcCs |jt_dS)N)_get_running_loopr)rHrrrunpatch_get_running_loopsz!TestCase.unpatch_get_running_loopcCs tj|_ddt_t|_dS)NcSsdS)Nrrrrrri rjz TestCase.setUp..)rrr threading_setup_thread_cleanup)rHrrrsetUp s zTestCase.setUpcCsB|td|td|tj|j t dS)N)NNN) rrr assertEqualsysexc_info doCleanupsr threading_cleanupr reap_children)rHrrrtearDowns   zTestCase.tearDown)N) rMrNrO staticmethodrrrrrrrrrrrs   rc cs2tj}zttjddVWdt|XdS)zrContext manager to disable asyncio logger. For example, it can be used to ignore warnings in debug mode. rN)r levelsetLevelloggingCRITICAL) old_levelrrrdisable_loggers  r cCs*ttj}||_||_||_d|j_|S)z'Create a mock of a non-blocking socket.g)r MagicMocksocketprotorfamily gettimeoutr)r rr sockrrrmock_nonblocking_socket*s  r)r8)Nrr contextlibrFrrrrr rwrr~rmr9unittestrrZ http.serverrZwsgiref.simple_serverrrr# ImportErrorasynciorrrr r Z asyncio.logr testr rr&r'ZSIGNED_CERTFILEZ SIGNING_CAZPEERCERTr+r-r.r7r@rDrErPr\rarsrrxrurzr|r}rcontextmanagerrrrr BaseSelectorr BaseEventLooprrrrrrrr  IPPROTO_TCP SOCK_STREAMAF_INETrrrrrs                   % *