hY dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZddlmZmZmZmZmZmZmZmZmZmZmZddlmZmZddlmZddl m!Z!m"Z"dd l#m$Z$ddl%Z%dd l&m'Z'm(Z(m)Z)d d l*m+Z+m,Z,d d l m-Z-d dl.m/Z/d dl0m1Z1d dl2m3Z3d dl4m5Z5m6Z6d dl7m8Z8m9Z9d dl:m;Z;d dlZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEd dlFmGZGerddlHmIZIndZIejJdkrerddlmKZKejJdkrddlmLZLneZLede=dZMede?ZNejOdko ejPdkZQejRfdeSd ejTd!ejfd"ZUdeSd#eVd ejTd!ejfd$ZWd!eVfd%ZXGd&d'e ZYGd(d)eYZZGd*d+eYZ[Gd,d-eeNeMfZ\Gd.d/eZ]d0ed1ed2ed!efd3Z^egej_fZ`ejaejbd4fd5e`d6ecd!eej_fd7Zdejbfd5e`d!ej_fd8ZedQd9ej_d6ecd!dfd:Zfd!ejgfd;ZhdRdd?d@ dAeSdBeSdCedDedEe5dFecdGedHedIedJedKe9dK|d{VdSrrrs r@ __aexit__zBaseTestServer.__aexit__. jjllrBrr=N)r=rT)$__name__ __module__ __qualname____test__r?rr asyncioAbstractEventLooprPboolrrE AddressFamilyrrarrr)rrr%rrpropertyrrr+rrrr BaseExceptionrrrrrorBr@rTrTjsH 48"!& ----w01 -  - sm --! #v+ ,fm ; -- ----0;?IIW67IJMI IIII@ 3 :   ^ 4X4#4444''''X'XX    &2222 4 ./ M* M*       4 ./M*M*   rBrTc ZeZdZdddddedededeed ef fd Zd ed e fd Z xZ S) TestServerrUrNNrVr;rCapprVr;rCrZc P||_tjd|||d|dSNrro)rsuperra)r`rrVr;rCrZ __class__s r@razTestServer.__init__s8GTGGGGGGGrBr=c*Kt|jfi|Sr)r'rrs r@rrzTestServer._make_runners,,V,,,rB) rrrr&rr rPrrar)rr __classcell__rs@r@rrs " H H H  H H  H sm H H H H H H H-3-:--------rBrc deZdZdddddedededeed ed df fd Zdd e d ed e fdZ xZ S) RawTestServerrUrNNrrrVr;rCrZr=c P||_tjd|||d|dSr)_handlerrra)r`rrVr;rCrZrs r@razRawTestServer.__init__s8  GTGGGGGGGrBTdebugcXKt|jf|j|d|}t|fd|i|S)N)rWrr)r+rr\r,)r`rrZsrvs r@rrzRawTestServer._make_runner sAT]K5KKFKKC77u7777rB)T) rrrr/rr rPrrarr,rrrrs@r@rrs " H H H  H H  H sm H H  H H H H H H888s8|88888888rBrc 8eZdZdZdZedddddedeed e d df d Z edddd de deed e d df d Z dddde deedee j d e d df dZ d6dZed efdZed eefdZed e fdZed efdZed efdZded efdZdeded e d efdZejdkre rdeded e!e"d e#fdZ$ded e!e"d e#fdZ%ded e!e"d e#fdZ&ded e!e"d e#fdZ'ded e!e"d e#fd Z(ded e!e"d e#fd!Z)ded e!e"d e#fd"Z*ded e!e"d e#fd#Z+nRdeded e d e#fd$Z$ded e d e#fd%Z%ded e d e#fd&Z(ded e d e#fd'Z&ded e d e#fd(Z'ded e d e#fd)Z)ded e d e#fd*Z*ded e d e#fd+Z+ded e d e,fd,Z-ded e d e.fd-Z/d6d.Z0d6d/Z1d0ee2e3d1ee3d2ee4d dfd3Z5d e6fd4Z7d0ee2e3d1ee3d2ee4d dfd5Z8dS)7 TestClientzb A test client implementation. To write functional tests for aiohttp based servers. FN) cookie_jarr`z TestClient[Request, Application]rrrZr=c dSrror`rrrZs r@razTestClient.__init__ srBzTestClient[_Request, None]c dSrrors r@razTestClient.__init__#rrB)rrWrWc $t|tstdt|z||_||_|t jd|}td||d||_ d|j _ d|_ g|_ g|_ dS)Nz2server must be TestServer instance, found type: %rT)unsaferW)rWrFro) isinstancerTrtyperzr\aiohttp CookieJarr_session_retry_connectionr_ _responses _websockets)r`rrrWrZs r@razTestClient.__init__+s&.11 DtF||S     *$TBBBJ%Q4JQQ&QQ */ ' 02:<rBcVK|j|jd{VdSr)rzrr\rs r@rzTestClient.start_serverAs9l''TZ'88888888888rBc|jjSr)rzr;rs r@r;zTestClient.hostD |  rBc|jjSr)rzrCrs r@rCzTestClient.portHrrBc|jSr)rzrs r@rzTestClient.serverLrrBc.t|jddS)Nr)getattrrzrs r@rzTestClient.appPst|UD111rBc|jS)zAn internal aiohttp.ClientSession. Unlike the methods on the TestClient, client session requests do not automatically include the host in the url queried, and will require an absolute path to the resource. )rrs r@sessionzTestClient.sessionTs }rBrc6|j|Sr)rzr)r`rs r@rzTestClient.make_url_s|$$T***rBmethodcK|jj|||fi|d{V}|j||Sr)rrequestrrappend)r`rrrZresps r@_requestzTestClient._requestbs_+T]*64==3F3FQQ&QQQQQQQQ t$$$ rBr1c dSrror`rrrZs r@rzTestClient.requestls%(SrBc dSrror`rrZs r@getzTestClient.getp &)SrBc dSrrors r@optionszTestClient.optionsvrrBc dSrrors r@headzTestClient.head|rrBc dSrrors r@postzTestClient.postrrBc dSrrors r@putzTestClient.putrrBc dSrrors r@patchzTestClient.patchrrBc dSrrors r@deletezTestClient.deleterrBc :t|j||fi|S)zRoutes a request to tested http server. The interface is identical to aiohttp.ClientSession.request, except the loop kwarg is overridden by the instance used by the test server. )rrrs r@rzTestClient.requests(*-$-*O*O*O*OPP PrBc Nt|jtj|fi|S)zPerform an HTTP GET request.)rrrMETH_GETrs r@rzTestClient.get()-$- t*V*Vv*V*VWW WrBc Nt|jtj|fi|S)zPerform an HTTP POST request.)rrr METH_POSTrs r@rzTestClient.post))-$-*W*WPV*W*WXX XrBc Nt|jtj|fi|S)z Perform an HTTP OPTIONS request.)rrr METH_OPTIONSrs r@rzTestClient.optionss0) d/@@@@ rBc Nt|jtj|fi|S)zPerform an HTTP HEAD request.)rrr METH_HEADrs r@rzTestClient.headrrBc Nt|jtj|fi|S)zPerform an HTTP PUT request.)rrrMETH_PUTrs r@rzTestClient.putrrBc Nt|jtj|fi|SzPerform an HTTP PATCH request.)rrr METH_PATCHrs r@rzTestClient.patchs/) dot>>v>> rBc Nt|jtj|fi|Sr )rrr METH_DELETErs r@rzTestClient.deletes0) d.???? rBc 8t|j|fi|S)zjInitiate websocket connection. The api corresponds to aiohttp.ClientSession.ws_connect. )r _ws_connectrs r@ ws_connectzTestClient.ws_connects( ((8(8(H(H(H(HIIIrBcK|jj||fi|d{V}|j||Sr)rrrrr)r`rrZwss r@rzTestClient._ws_connects^,4=+DMM$,?,?JJ6JJ J J J J J J ### rBc*K|js|jD]}||jD]}|d{V|jd{V|jd{Vd|_dSdS)aHClose 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 on exit when used as a(n) (asynchronous) context manager. NT)r_rrrrrz)r`rrs r@rzTestClient.closes|    & ! !hhjj        -%%'' ' ' ' ' ' ' ',$$&& & & & & & & &DLLL  rBc tdrrrs r@rzTestClient.__enter__rrBrexctbcdSrror`rrrs r@rzTestClient.__exit__rrBc>K|d{V|Sr)rrs r@rzTestClient.__aenter__s/!!!!!!!!! rBc>K|d{VdSrrrs r@rzTestClient.__aexit__rrBr)9rrr__doc__rrrr rrrarTrrrrrr;rPrCrr6rrrr%rrrrsys version_inforr4rrrrrrrrrrrrrrrrr rrrr5rrrorBr@rrsH 37 0./     X 37 *./     X3748 ====./ = w01 =  = ====,9999!c!!!X!!hsm!!!X!X2%222X2X+X+#++++!)58  7""}" ) )%- )9?9P ) # ) ) ) ) ) )_- )$  ) ) ) )  ) )_- )$  ) ) ) )  ) )_- )$  ) ) ) )  ) )_- )$  ) ) ) )  ) )_- )$  ) ) ) )  ) )_- )$  ) ) ) )  ) )_- )$  ) ) ) ) ) Q Q%- Q9< Q # Q Q Q Q XH X X8N X X X X YX Y Y9O Y Y Y Y  C tjdtd|S)zw A decorator dedicated to use with asynchronous AioHTTPTestCase test methods. In 3.8+, this does nothing. zBDecorator `@unittest_run_loop` is no longer needed in aiohttp 3.8+ri) stacklevel)warningswarnDeprecationWarning)r2r3rZs r@unittest_run_loopr9@s-  ML KrBF loop_factoryfastc#RKt|}|Vt||dS)zzA contextmanager that creates an event_loop, for test purposes. Handles the creation and cleanup of a test loop. )r;N)setup_test_loopteardown_test_loop)r:r;rWs r@ loop_contextr?Qs7 < ( (D JJJt$''''''rBcB|}tj||S)zCreate and return an asyncio.BaseEventLoop instance. The caller should also call teardown_test_loop, once they are done with the loop. )rset_event_loop)r:rWs r@r=r=^s% <>>D 4   KrBrWc|}|sB||j|||st jtjddS)z>Teardown and cleanup an event_loop created by setup_test_loop.N) is_closed call_soonstop run_foreverrgccollectrrA)rWr;rs r@r>r>ksv ^^  F  ty!!!    4     rBc4dtdtdtfd}dtdtdtddfd}tjt}i|_||_||_d|_t||_ |j |S) Nrkeyr=c|j|Sr __app_dict)rrJs r@get_dictz"_create_app_mock..get_dictzs~c""rBvaluec||j|<dSrrL)rrJrOs r@set_dictz"_create_app_mock..set_dict}s#srB)specF) rrr MagicMockr&rM __getitem__ __setitem___debugron_response_preparefreeze)rNrQrs r@_create_app_mockrYys#c######$c$$C$D$$$$ .k * * *CCNCOCOCJ$SkkC""$$$ JrB sslcontextctj}dtdttffd }||j_|S)NrJr=c|dkrSdS)NrZro)rJrZs r@get_extra_infoz)_create_transport..get_extra_infos ,   4rB)rMockrr r0r] side_effect)rZ transportr]s` r@_create_transportrasR ICHZ$8 ,:I( rBi.) match_inforvclosingrwriterprotocolr`payloadrZclient_max_sizerWrrheadersrbrvrcrrdrer`rfrgc ~tj}| durD tj} n/#t$r"tj} d| j_YnwxYw|tddkrd}|rHtt|}td| D}ntt}d}d| tjdv}t!||||||dd |t#| }|t%}| t&urt)| } |t&ur{tj}| |_tj| d  t1|_tj| t1|_|t&urjtj}t7d|_t7d|_t7d|_t7d|_| |_| |_||_ tC|| |||| | }tE|t&urin|tj}|#|||_$|S) zCreates 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. .rorTc3pK|]1\}}|d|dfV2dS)zutf-8N)encode).0kvs r@ z&make_mocked_request..sT  7;q!QXXg   1 1 2      rBchunkedrUNFpeername) return_value)rg)%rr^rr%r# create_futurerrr!rrtupleitemsrrTRANSFER_ENCODINGlowerr"rrYr rar` PropertyMockr]rrqrkmake_mocked_coro write_headerswrite write_eofdrainrdr*r.add_app _match_info)rrrhrbrvrcrrdrer`rfrZrgrWtaskraw_hdrsrpmessagereqs r@make_mocked_requestrs, 9;;D s{{  1+--DD 1 1 19;;D.0D  + + + 1Q""""";w#7#788  ?F}}     #;==117;;t'=rBBHHJJJG    D   G {  H%j11 89;;&"&"3"11*==# # # X&*%6J%O%O%OX" /55'-- +D11'-- $"HHO (FD$   C%H$$*dikkJs CO Js-)AArrraise_exceptionchdtdtdtffd }tj|S)zCreates a coroutine mock.r3rZr=c^KturtjsSd{VdSr)r inspect isawaitable)r3rZrrrs r@ mock_coroz#make_mocked_coro..mock_corosO ( * *! !"<00  rB)wraps)rrr^)rrrrs`` r@ryrysQ scc 99 % % %%rB)Fr)mrr contextlibrGrrtosrErr6abcrrtypesrtypingrrrr r r r r rrrunittestrr aiosignalr multidictrryarlrraiohttp.clientrrrrUrrr client_reqrepr client_wsrhelpersr rmr!r"streamsr#r$typedefsr%webr&r'r(r)r*r+r,r-r. web_protocolr/rcr0rr4r5r6r7nameplatformrGrOrrrArPr?rRrTrrrrr9r _LOOP_FACTORYcontextmanagernew_event_looprr?r=r>rSrYr^rarryrorBr@rs   ########                          3222222233333333 "!!!!!!!""""""))))))......0000000000000000                      *)))))Jw=w D7-{DAA 7:[ 1 1 17"?s|x'? /5n,, ,+, ],,,,      "("6   ]     -S----AAAAASAAAH--------"88888N888$sssss#334sssl6262626262-626262r C  s s    W667  ")"8u ( ( (@D ( g'( ( ( ( (#*"8         ! !W6 !d !t ! ! ! !$.$  (:"6 $)     _ &;q!,,)'+"___ _ __  _  __ _ ____$__ _  !____F! & & &36 & & & & & & &rB