ÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂÂ U 'â7`Oã@s dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl mZmZddl mZddlmZmZmZmZmZmZmZmZddl mZddlmZmZddlmZddlZdd lm Z m!Z!m"Z"d dl#m$Z$m%Z%d dl m&Z&d d l'm Z d dl(m)Z)d dl*m+Z+d dl,m-Z-m.Z.d dl/m0Z0d dl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9d dl:m;Z;erxddldkoej?dkZ@eAejdœdd„ZBeAeCejdœdd„ZDeCdœdd„ZEGd d!„d!eƒZFGd"d#„d#eFƒZGGd$d%„d%eFƒZHGd&d'„d'ƒZIGd(d)„d)e jJƒZKeeeed*œd+d,„ZLegejMfZNejOejPd-feNeQeejMd.œd/d0„ƒZRejPfeNejMd1œd2d3„ZSdEejMeQdd4œd5d6„ZTejUdœd7d8„ZVdFee=ejWd9œd:d;„ZXdGe+e-d d ƒd-de+e+e+e+ddœeAeAeee-eQeeeeeee=eCee5d?œd@dA„ZYe+e+feeedBœdCdD„ZZdS)HzUtilities shared by tests.éN)ÚABCÚabstractmethod)Ú TracebackType)Ú TYPE_CHECKINGÚAnyÚCallableÚIteratorÚListÚOptionalÚTypeÚUnion)Úmock)ÚCIMultiDictÚCIMultiDictProxy)ÚURL)ÚClientResponseÚ_RequestContextManagerÚ_WSRequestContextManageré)Ú ClientSessionÚhdrs)ÚAbstractCookieJar)r)ÚClientWebSocketResponse)Úsentinel)ÚHttpVersionÚRawRequestMessage)ÚSignal)ÚApplicationÚ AppRunnerÚ BaseRunnerÚRequestÚServerÚServerRunnerÚSockSiteÚUrlMappingMatchInfo)Ú_RequestHandler)Ú SSLContextÚposixÚcygwin)ÚhostÚreturncCs t|dƒS)Nr)Úget_port_socket)r)©r,úE/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/test_utils.pyÚget_unused_port_socket7sr.)r)Úportr*cCs8t tjtj¡}tr&| tjtjd¡| ||f¡|S)Nr)ÚsocketÚAF_INETÚSOCK_STREAMÚ REUSE_ADDRESSÚ setsockoptÚ SOL_SOCKETÚSO_REUSEADDRÚbind)r)r/Úsr,r,r-r+;s r+©r*c Cs>t tjtj¡$}| d¡| ¡dW5QR£SQRXdS)z1Return a port that is unused on the current host.)ú 127.0.0.1rrN)r0r1r2r7Úgetsockname)r8r,r,r-Úunused_portFs r<c @s0eZdZdZedddddœeeefee j eeeee ddœdd„Zd#ee j e ddœd d „Zee edœdd „ƒZeedœdd„Zeedœdd„ƒZeedœdd„ƒZeedœdd„ƒZddœdd„Zddœdd„Zeeeeeeeddœdd„Zddœdd „Z eeeeeeeddœd!d"„Z!dS)$ÚBaseTestServerFNr:)ÚschemeÚloopr)r/Úskip_url_asserts)r>r?r)r/r@Úkwargsr*cKs4||_d|_d|_||_||_d|_||_||_dS)NF)Ú_loopÚrunnerÚ_rootr)r/Ú_closedr>r@)Úselfr>r?r)r/r@rAr,r,r-Ú__init__Ps zBaseTestServer.__init__)r?rAr*cËs|jr dS||_| dd¡|_|jf|ŽIdH|_|j ¡IdH|jsNd|_t|j|jƒ}| ¡dd…\|_|_t |j||jd}| ¡IdH|j}|dk s¦t ‚|j}|dk s¸t ‚|d ¡d|_|jtkrê|jràd}nd}||_t|j›d|j›d |j›ƒ|_dS) NÚsslré)ÚsockZssl_contextrÚhttpsÚhttpz://ú:)rCrBÚpopZ_sslÚ_make_runnerÚsetupr/r+r)r;r#ÚstartÚ_serverÚAssertionErrorÚsocketsr>rrrD)rFr?rAÚ_sockÚsiteÚserverrTr>r,r,r-Ústart_servercs. zBaseTestServer.start_server©rAr*cËsdS©Nr,©rFrAr,r,r-rOszBaseTestServer._make_runner©Úpathr*cCsJ|jdk st‚t|ƒ}|js4| ¡r(t‚|j |¡Stt|jƒ|ƒSdSrZ)rDrSrr@Úis_absoluteÚjoinÚstr)rFr]Úurlr,r,r-Úmake_urlƒszBaseTestServer.make_urlr9cCs |jdk SrZ)rC©rFr,r,r-ÚstartedŒszBaseTestServer.startedcCs|jSrZ)rErcr,r,r-ÚclosedszBaseTestServer.closedcCs&|j}|dk st‚|jdk s t‚|jSrZ)rCrSrW)rFrCr,r,r-Úhandler”szBaseTestServer.handlercÃs@|jr<|js<|jdk st‚|j ¡IdHd|_d|_d|_dS)a`Close all fixtures created by the test client. After that point, the TestClient is no longer usable. This is an idempotent function: running close multiple times will not have any additional effects. close is also run when the object is garbage collected, and on exit when used as a context manager. NT)rdrerCrSÚcleanuprDr/rErcr,r,r-ÚcloseszBaseTestServer.closecCstdƒ‚dS©NzUse async with instead©Ú TypeErrorrcr,r,r-Ú __enter__°szBaseTestServer.__enter__)Úexc_typeÚ exc_valueÚ tracebackr*cCsdSrZr,©rFrmrnror,r,r-Ú__exit__³szBaseTestServer.__exit__cÃs|j|jdIdH|S©N©r?)rXrBrcr,r,r-Ú __aenter__¼szBaseTestServer.__aenter__cÃs| ¡IdHdSrZ©rhrpr,r,r-Ú __aexit__ÀszBaseTestServer.__aexit__)N)"Ú__name__Ú __module__Ú__qualname__Ú__test__rrr`Úobjectr ÚasyncioÚAbstractEventLoopÚintÚboolrrGrXrrrOrrbÚpropertyrdrer!rfrhrlrÚ BaseExceptionrrqrtrvr,r,r,r-r=MsVù ÷ÿþ û ûr=csPeZdZedddœeeeefeee e dœ‡fdd„Ze edœdd „Z ‡ZS) Ú TestServerr:N©r>r)r/)Úappr>r)r/rAcs$||_tƒjf|||dœ|—ŽdS©Nrƒ)r„ÚsuperrG)rFr„r>r)r/rA©Ú __class__r,r-rGÊs zTestServer.__init__rYcËst|jf|ŽSrZ)rr„r[r,r,r-rOÖszTestServer._make_runner)rwrxryrrrr`r{r r~rrGrrOÚ __classcell__r,r,r‡r-r‚Ésú ùr‚csVeZdZedddœeeeefeee e ddœ‡fdd„Zdee e dœd d „Z‡ZS)Ú RawTestServerr:Nrƒ)rfr>r)r/rAr*cs$||_tƒjf|||dœ|—ŽdSr…)Ú_handlerr†rG)rFrfr>r)r/rAr‡r,r-rGÛs zRawTestServer.__init__T)ÚdebugrAr*cËs.t|jf|j|dœ|—Ž}t|fd|i|—ŽS)N)r?rŒrŒ)r!r‹rBr")rFrŒrAZsrvr,r,r-rOçszRawTestServer._make_runner)T)rwrxryrr%rr`r{r r~rrGrr"rOr‰r,r,r‡r-rŠÚsú ørŠc@sôeZdZdZdZdddœeeeeej e ddœdd„Zddœd d „Ze edœdd„ƒZe eedœd d„ƒZe edœdd„ƒZe edœdd„ƒZe edœdd„ƒZeedœdd„Zeee edœdd„Zeee edœdd„Zee edœdd„Zee edœd d!„Zee edœd"d#„Zee edœd$d%„Z ee edœd&d'„Z!ee edœd(d)„Z"ee edœd*d+„Z#ee e$dœd,d-„Z%ee e&dœd.d/„Z'ddœd0d1„Z(ddœd2d3„Z)ee*e+ee+ee,dd4œd5d6„Z-ddœd7d8„Z.ee*e+ee+ee,dd4œd9d:„Z/dS);Ú TestClientzb A test client implementation. To write functional tests for aiohttp based servers. FN)Ú cookie_jarr?)rWrŽr?rAr*cKsht|tƒstdt|ƒƒ‚||_||_|dkr.new_func)Ú functoolsÚwrapsr)rºr»rAr¿r,r¾r-Úunittest_run_loopÓsrÂF)Úloop_factoryÚfastr*ccst|ƒ}|Vt||ddS)zzA contextmanager that creates an event_loop, for test purposes. Handles the creation and cleanup of a test loop. )rÄN)r°r¸)rÃrÄr?r,r,r-Úloop_contextåsrÅ)rÃr*c CsŠ|ƒ}z|jj}d|k}Wntk r2d}YnXt |¡tjdkr†|s†t ¡}t ¡}| |¡t t¡| |¡W5QRX|S)z˜Create and return an asyncio.BaseEventLoop instance. The caller should also call teardown_test_loop, once they are done with the loop. ZuvloopTÚwin32)rˆrxÚAttributeErrorr|Úset_event_loopÚsysÚplatformZget_event_loop_policyZSafeChildWatcherZattach_loopÚ contextlibÚsuppressÚNotImplementedErrorZset_child_watcher)rÃr?ÚmoduleZskip_watcherÚpolicyZwatcherr,r,r-r°òs r°)r?rÄr*cCsB| ¡}|s(| |j¡| ¡| ¡|s4t ¡t d¡dS)zHTeardown and cleanup an event_loop created by setup_test_loop. N) Z is_closedZ call_soonÚstopZrun_foreverrhÚgcZcollectr|rÈ)r?rÄrer,r,r-r¸sr¸cCs^tttdœdd„}tttddœdd„}t ¡}i|_||_||_d|_t|ƒ|_ |j ¡|S)N)r„Úkeyr*cSs |j|SrZ©Ú __app_dict)r„rÒr,r,r-Úget_dictsz"_create_app_mock..get_dict)r„rÒÚvaluer*cSs||j|<dSrZrÓ)r„rÒrÖr,r,r-Úset_dict!sz"_create_app_mock..set_dictF)rr`r Ú MagicMockrÔÚ__getitem__Ú__setitem__Z_debugrZon_response_prepareZfreeze)rÕr×r„r,r,r-Ú_create_app_mocks rÛ)Ú sslcontextr*cs,t ¡}tttdœ‡fdd„}||j_|S)N)rÒr*cs|dkrˆSdSdS)NrÜr,)rÒ©rÜr,r-Úget_extra_info2sz)_create_transport..get_extra_info)r ÚMockr`r r&rÞÚside_effect)rÜÚ transportrÞr,rÝr-Ú_create_transport/srâi.)Ú match_infoÚversionÚclosingr„ÚwriterÚprotocolráÚpayloadrÜÚclient_max_sizer?)r—r]Úheadersrãrärår„rærçrárèrÜrér?r*cCstt ¡}| dkr t ¡} d| j_|tddƒkr2d}|rZtt|ƒƒ}tdd„| ¡Dƒƒ}nttƒƒ}d}d| t jd¡ ¡k}t ||||||d d |t|ƒƒ }|d krªtƒ}| tkrºt|ƒ} |tkrÐt ¡}| |_|tkrt ¡}td ƒ|_td ƒ|_td ƒ|_td ƒ|_| |_| |_||_| tkr.t ¡} t|| |||| |d}t|tkrTin|t ¡ƒ}| |¡||_|S)z²Creates mocked web.Request testing purposes. Useful in unit tests, when spinning full web server is overkill or specific conditions and errors are hard to trigger. .r,rTcss&|]\}}| d¡| d¡fVqdS)zutf-8N)Úencode)Ú.0ÚkÚvr,r,r-Ú ^sz&make_mocked_request..ÚchunkedÚFN)ré)r rßZ create_futureÚreturn_valuerrrÚtupleÚitemsrrZTRANSFER_ENCODINGÚlowerrrrÛrrâráÚmake_mocked_coroZ write_headersÚwriteZ write_eofZdrainrær r$Zadd_appZ_match_info)r—r]rêrãrärår„rærçrárèrÜrér?ZtaskZraw_hdrsrðÚmessageÚreqr,r,r-Úmake_mocked_request<sxÿ ö ÿÿ rú)ròÚraise_exceptionr*cs$tttdœ‡‡fdd„}tj|dS)zCreates a coroutine mock.)r»rAr*cŸs(ˆtk rˆ‚t ˆ¡sˆSˆIdHdSrZ)rÚinspectÚisawaitable)r»rA©rûròr,r-Ú mock_coros z#make_mocked_coro..mock_coro)rÁ)rr rß)ròrûrÿr,rþr-rö˜srö)F)N)N)[r«r|rËrÀrÑrüÚosr0rÉZunittestÚabcrrÚtypesrÚtypingrrrrr r rrr Z multidictrrZyarlrr‘Zaiohttp.clientrrrrñrrrZ client_reqrepZ client_wsrZhelpersrrLrrZsignalsrZwebrrrr r!r"r#r$Zweb_protocolr%rHr&ÚnamerÊr3r`r.r~r+r<r=r‚rŠrZTestCaser¬rÂr}Z _LOOP_FACTORYÚcontextmanagerZnew_event_looprrÅr°r¸rØrÛrßrârúrör,r,r,r-Ús¼(( |(@ÿþ ÿþýñð]ÿþ