[4%e, ddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZmZmZdd l m!Z!ddl"m#Z#m$Z$ddl%m&Z&m'Z'm(Z(m)Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:Z:ddl;Z;ddlZ?ddl@Z@ddlAmBZBddlCZCddlDZDddlEZEddlFZFddlGZGddlHZHddlIZIddlJZKdZLGddeZMGddeMZNGdde'ZOGdde'ZPGddeIjQZRGddeIjQZSGdd eMZTGd!d"eMZUGd#d$e'ZVGd%d&eMZWGd'd(e'ZXGd)d*eMZYGd+d,e'ZZGd-d.eMZ[Gd/d0e'Z\Gd1d2e'Z]Gd3d4e'Z^Gd5d6e'Z_Gd7d8e'Z`Gd9d:e'ZaGd;de'ZbGd?d@e'ZcGdAdBe'ZdGdCdDe'ZeGdEdFeMZfGdGdHeMZgGdIdJeMZhGdKdLeMZiGdMdNeMZjGdOdPeMZkGdQdReMZlGdSdTeMZmGdUdVeMZnGdWdXeMZoGdYdZeNZpGd[d\eNZqGd]d^eNZrGd_d`eNZsGdadbeNZtGdcddeNZuGdedfeMZvGdgdheNZwGdidjeMZxGdkdleNZyGdmdnezZ{GdodpeNZ|GdqdreNZ}GdsdteNZ~GdudveNZGdwdxeNZGdydzeMZGd{d|eNZGd}d~eNZGddeNZGddeNZGddeNZGddeMZGddeMZGddeMZGddeMZGddeMZe6Gdde'ZGddeZGddeeMZGddeeMZGddeNZGddeNZGddeIjQZGddeNZGddeNZGddeNZGddeNZGddeMZGddeMZGddeNZGddeIjQZGddeZGddeIjQZGddeMZGddeMZdS))Future)gen) json_decodeutf8 to_unicoderecursive_unicode native_str to_basestring)HTTPClientError)format_timestamp)IOStream)locale)Event)app_loggen_logSimpleAsyncHTTPClient DictLoader)AsyncHTTPTestCase AsyncTestCase ExpectLoggen_test)ignore_deprecation) ObjectDict unicode_type) ApplicationRequestHandlerStaticFileHandlerRedirectHandler HTTPErrorMissingArgumentError ErrorHandler authenticatedurl_create_signature_v1create_signed_valuedecode_signed_valueget_signature_key_versionUIModuleFinishstream_request_body removeslashaddslashGZipContentEncodingN)BytesIOcttjjtjtg|RSN)ospathjoindirname__file__)as 5lib/python3.11/site-packages/tornado/test/web_test.pyrelpathr:=s* 7<11 6A 6 6 66c$eZdZdZdZdZdZdS) WebTestCasezBase class for web tests that also supports WSGI mode. Override get_handlers and get_app_kwargs instead of get_app. This class is deprecated since WSGI mode is no longer supported. c|t|fi||_|jSr2)r get_handlersget_app_kwargsappselfs r9get_appzWebTestCase.get_appHs8t0022LLd6I6I6K6KLLxr;ctr2NotImplementedErrorrBs r9r?zWebTestCase.get_handlersLs!###r;ciSr2rBs r9r@zWebTestCase.get_app_kwargsO r;N)__name__ __module__ __qualname____doc__rDr?r@rIr;r9r=r=AsK $$$r;r=ceZdZdZdZdZdS)SimpleHandlerTestCasezSimplified base class for tests that work with a single handler class. To use, define a nested class named ``Handler``. Ncd|jfgS)N/HandlerrBs r9r?z"SimpleHandlerTestCase.get_handlers[sdl#$$r;)rKrLrMrNrTr?rIr;r9rPrPSs4 G%%%%%r;rPceZdZdZdS) HelloHandlerc0|ddSNhellowriterBs r9getzHelloHandler.get`s 7r;NrKrLrMr\rIr;r9rVrV_s#r;rVc$eZdZddZdZddZdS)CookieTestRequestHandler 0123456789Nci|_|%tt||_dStt|||_dS)N) cookie_secret)settingsrb key_version)_cookiesrdict application)rCrbres r9__init__z!CookieTestRequestHandler.__init__fsi  )M:::   D    *M{SSS   D   r;c6|j|Sr2)rfr\)rCnames r9 get_cookiez#CookieTestRequestHandler.get_cookiers}  &&&r;c||j|<dSr2)rf)rCrkvalue expires_dayss r9 set_cookiez#CookieTestRequestHandler.set_cookieus# dr;)r`Nr2)rKrLrMrirlrprIr;r9r_r_dsK    '''$$$$$$r;r_c eZdZdZdZdZdS)SecureCookieV1Testct}|ddd||ddddS)Nfoobarversion min_versionr_set_signed_cookie assertEqualget_signed_cookierChandlers r9test_round_tripz"SecureCookieV1Test.test_round_trip{W*,,!!%!;;; 225a2HH&QQQQQr;c t}|dtjdd|jd}t jd|}|J|d}|d}|t|j j ddd|||t|j j ddd d |z|td t|d t||jd<ttd 5||ddduddddS#1swxYwYdS)Nrts d76df8e7aefcrvrws12345678\|([0-9]+)\|([0-9a-f]+)rb123456781234s5678z 1234|5678|zCookie timestamp in futurery)r_r|binasciia2b_hexrfrematchgroupr}r&rhrcrr rr assertTruer~)rCrcookier timestampsigs r9&test_cookie_tampering_future_timestampz9SecureCookieV1Test.test_cookie_tampering_future_timestamps*,,!!%)9/)J)JTU!VVV!%(>FFct}|ddd||ddddS)Nrtrvrwryr{rs r9test_arbitrary_bytesz'SecureCookieV1Test.test_arbitrary_bytessY+,,!!%!!<<< 225a2HH'RRRRRr;N)rKrLrMrrrrIr;r9rrrrzsMRRR $U$U$ULSSSSSr;rrc6eZdZdddZdZdZdZdZdZd S) SecureCookieV2Testajklasdf0ojaisdfaslkjasaolwkjsdf)rrvct}|ddd||ddddS)Nrtrurrwryr{rs r9rz"SecureCookieV2Test.test_round_triprr;ct|jd}|dd||dddS)Nrrdrtrur_ KEY_VERSIONSr|r}r~rs r9test_key_version_roundtripz-SecureCookieV2Test.test_key_version_roundtrip`*+    !!%000 225996BBBBBr;ct|jd}|dd||dddS)Nrvrdrtrurrs r9,test_key_version_roundtrip_differing_versionz?SecureCookieV2Test.test_key_version_roundtrip_differing_versionrr;ct|jd}|ddt|jd}|j|_||dddSNrrdrtrurv)r_rr|rfr}r~)rCr new_handlers r9"test_key_version_increment_versionz5SecureCookieV2Test.test_key_version_increment_versions*+    !!%000.+    '/  66u==vFFFFFr;cFt|jd}|dd|j}|dt|d}|j|_||dddSr)r_rr|copypoprfr}r~)rCrnew_key_versionsrs r9#test_key_version_invalidate_versionz6SecureCookieV2Test.test_key_version_invalidate_versions*+    !!%000,1133Q.*    '/  66u==tDDDDDr;N) rKrLrMrrrrrrrIr;r9rrs~).@AALRRR CCCCCC G G G E E E E Er;rc*eZdZdZdZdZdZdZdS)FinalReturnTestNcl|Gfddt}Gfddt}d|fd|fgS)NcXeZdZejfdZejfdZdS)3FinalReturnTest.get_handlers..FinishHandlerc3PK|_jVdSr2)finish final_returnrCtests r9r\z7FinalReturnTest.get_handlers..FinishHandler.gets,$(KKMM!''''''r;c3K|d|V|d_jVdS)Nzhello,world)r[flushrrrs r9postz8FinalReturnTest.get_handlers..FinishHandler.postsV 8$$$jjll"""$(KK$8$8!''''''r;N)rKrLrMr coroutiner\rrsr9 FinishHandlerrsa ] ( ( ( (] (] ( ( ( (] ( ( (r;rc:eZdZdZejfdZdS)3FinalReturnTest.get_handlers..RenderHandlerc$tddiS)Nfoo.htmlhirrCr4s r9create_template_loaderzJFinalReturnTest.get_handlers..RenderHandler.create_template_loaders!:t"4555r;c<|d_dSNr)renderrrs r9r\z7FinalReturnTest.get_handlers..RenderHandler.gets$(KK $;$;!!!r;N)rKrLrMrrrr\rsr9 RenderHandlerrsK 6 6 6] < < < <] < < .SetCookieHandlerc|dd|dd|dddS)NstrasdfunicodeqwerbytesszxcvrprBs r9r\z5CookieTest.get_handlers..SetCookieHandler.getsFv... 622211111r;Nr]rIr;r9SetCookieHandlerrs# 2 2 2 2 2r;rceZdZdZdS)1CookieTest.get_handlers..GetCookieHandlercd|dd}|J||dS)Nrtdefault)rlr[)rCrs r9r\z5CookieTest.get_handlers..GetCookieHandler.gets9 ::))) 6"""""r;Nr]rIr;r9GetCookieHandlerrs# # # # # #r;rceZdZdZdS)7CookieTest.get_handlers..SetCookieDomainHandlerc8|dddddS)N unicode_argsblahzfoo.com/foo)domainr4rrBs r9r\z;CookieTest.get_handlers..SetCookieDomainHandler.get s$yvVVVVVr;Nr]rIr;r9SetCookieDomainHandlerrs( W W W W Wr;rceZdZdZdS).SetCookieSpecialCharHandlerc|dd|dd|dddS)Nequalsa=b semicolona;bquotea"brrBs r9r\z@CookieTest.get_handlers..SetCookieSpecialCharHandler.get&sD%000 U333/////r;Nr]rIr;r9SetCookieSpecialCharHandlerr%s# 0 0 0 0 0r;rceZdZdZdS):CookieTest.get_handlers..SetCookieOverwriteHandlerc|ddd|ddd|dddS)Nr8bz example.com)rcderrBs r9r\z>CookieTest.get_handlers..SetCookieOverwriteHandler.get,sNS???S???S)))))r;Nr]rIr;r9SetCookieOverwriteHandlerr+s# * * * * *r;rceZdZdZdS)7CookieTest.get_handlers..SetCookieMaxAgeHandlerc6|ddddS)Nrtbar )max_agerrBs r9r\z;CookieTest.get_handlers..SetCookieMaxAgeHandler.get4s ub99999r;Nr]rIr;r9SetCookieMaxAgeHandlerr 3s# : : : : :r;rceZdZdZdS).SetCookieExpiresDaysHandlerc6|ddddS)Nrtr r )rorrBs r9r\z@CookieTest.get_handlers..SetCookieExpiresDaysHandler.get8s u2>>>>>r;Nr]rIr;r9SetCookieExpiresDaysHandlerr7s# ? ? ? ? ?r;rceZdZdZdS)4CookieTest.get_handlers..SetCookieFalsyFlagsc|ddd|ddd|ddd|d dddS) Nr81T)securerFr)httponlyrrrBs r9r\z8CookieTest.get_handlers..SetCookieFalsyFlags.get<shS666S777S4888S599999r;Nr]rIr;r9SetCookieFalsyFlagsr;s# : : : : :r;rceZdZdZdS)8CookieTest.get_handlers..SetCookieDeprecatedArgsc8|dddddS)Nr8rTr)HttpOnlypATHrrBs r9r\z.SetCookieDeprecatedArgs.getCs"S4fEEEEEr;Nr]rIr;r9SetCookieDeprecatedArgsrBs( F F F F Fr;r/set/get /set_domain /special_char/set_overwrite /set_max_age/set_expires_days/set_falsy_flags/set_deprecatedr) rCrrrrrrrrrs r9r?zCookieTest.get_handlerss 2 2 2 2 2~ 2 2 2 # # # # #~ # # #  W W W W W^ W W W  0 0 0 0 0. 0 0 0  * * * * * * * * : : : : :^ : : : ? ? ? ? ?. ? ? ? : : : : :. : : : F F F F Fn F F F % & % & 2 3 9 : 8 9 3 4 "= > !4 5  7 8  r;c|d}|t|jdgddS)Nr  Set-Cookie)zbytes=zxcv; Path=/zstr=asdf; Path=/zunicode=qwer; Path=/)rr}sortedheadersget_listrs r9test_set_cookiezCookieTest.test_set_cookieSsZ::f%%  8#,,\:: ; ; N N N     r;c>|dddi}||jd|dddi}||jd|dddi}||jddS) Nr!Cookiefoo=barr,ruz foo="bar"z /=exception;defaultrr}rrs r9test_get_cookiezCookieTest.test_get_cookieZs::fx.C:DD ///::fx.E:FF ///::fx.H:II  33333r;c|d}||jddgdS)Nr"r*z,unicode_args=blah; Domain=foo.com; Path=/foorr}r,r-rs r9test_set_cookie_domainz!CookieTest.test_set_cookie_domaindsO::m,,    % %l 3 3 ; <     r;c@|d}t|jd}|t |d||dd||dd||dd v|dgd }|D][\}}tjd ||d d |i}||j t|\dS)Nr#r*rzequals="a=b"; Path=/rvzquote="a\"b"; Path=/r)zsemicolon="a;b"; Path=/zsemicolon="a\073b"; Path=/))zfoo=a=br)z foo="a=b"r)z foo="a;b"z"a)z foo=a\073bza\073b)z foo="a\073b"r)z foo="a\"b"rz trying %rr!r0r2) rr+r,r-r}lenrloggingdebugrr)rCrr,dataheaderexpecteds r9test_cookie_special_charz#CookieTest.test_cookie_special_charks+::o..)22<@@AA Wq))) %;<<< %<===  AJT T AJ      !% < < FH M+v . . .zz&8V2DzEEH   X]DNN ; ; ; ; < .+); Path=/r daysexpires)rr,r\rrdatetimeutcnow timedeltaemailutils parsedate groupdictrabs total_secondsrCrr?rrKparsedheader_expiress r9test_set_cookie_expires_daysz'CookieTest.test_set_cookie_expires_dayss::122!%%l33!!!CVLL   #**,,x/Ar/J/J/JJ&&u'8'8'CDD!!!!*F2A2J7 W~5DDFFGG"LMMMMMr;c|d}t|jd}||dd||dd||dd||d d dS) Nr'r*rza=1; path=/; securervz b=1; path=/rzc=1; httponly; path=/r:z d=1; path=/)rr+r,r-r}lowerrDs r9test_set_cookie_false_flagsz&CookieTest.test_set_cookie_false_flagss::011)22<@@AA ))++-BCCC ))++];;; ))++-DEEE ))++];;;;;r;ct5|d}dddn #1swxYwY|jd}||ddS)Nr(r*za=b; HttpOnly; Path=/foo)rrr,r\r})rCrr?s r9test_set_cookie_deprecatedz%CookieTest.test_set_cookie_deprecateds  ! ! 5 5zz"344H 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5!%%l33 !;<<<<??r;ctjtjtjd}|d|ft ||_|jd||j |j ddS)Nr 127.0.0.1sGET / HTTP/1.0 c*tjdSNr)rsleeprIr;r9z;ConnectionCloseTest.test_connection_close..scillr;) socketAF_INET SOCK_STREAMconnect get_http_portr streamr[r}r|setio_looprun_sync)rCss r9test_connection_closez)ConnectionCloseTest.test_connection_closes M&.&*>        r;Nr]rIr;r9rrs#     r;rc2eZdZdZdZdZdZdZdZdS)RequestEncodingTestc&dtfdtfgS)Nz /group/(.*)z/slashes/([^/]*)/([^/]*))rrBs r9r?z RequestEncodingTest.get_handlerss ,/I;.WXXr;cPt||jSr2)rrrrs r9 fetch_jsonzRequestEncodingTest.fetch_jsons4::d++0111r;c ||dtddgi||dtddgddgidS)Nz /group/%3F?rz/group/%3F?%3F=%3Fr}rrgrBs r9test_group_question_markz,RequestEncodingTest.test_group_question_marks  OOL ) ) lse" = = =     OO0 1 1 lse3, G G G     r;ch||dddgddgiddS)Nz/group/%C3%A9?arg=%C3%A9z /group/%C3%A9érrr}rrBs r9test_group_encodingz'RequestEncodingTest.test_group_encoding)sP  OO6 7 7'&Z +       r;c||dtdddgi||dtdddgidS)Nz/slashes/foo/barrtr rz/slashes/a%2Fb/c%2Fdza/bzc/drrBs r9 test_slashesz RequestEncodingTest.test_slashes4s  OO. / / (UEN L L L     OO2 3 3 ,R P P P     r;cttd5|dddddS#1swxYwYdS)Nz.*Invalid unicodez/group/?arg=%25%e9)rrrrBs r9 test_errorzRequestEncodingTest.test_error@sw 3 4 4 - - JJ+ , , , - - - - - - - - - - - - - - - - - - 9==N) rKrLrMr?rrrrrrIr;r9rrsqYYY222            -----r;rc&eZdZdZdZdZdZdS)TypeCheckHandlerci|_|d|t|d|dt |dt |jdt|dt |j dj tt |jdgkr)td|jz|d | dt|d |dt|d |jt|d |t|d |ddt|d|tdS)Nstatusargumentrt cookie_keyr cookie_valuerz%unexpected values for cookie keys: %rr~rl xsrf_tokenxsrf_form_html reverse_url typecheckrequest_summary)errors check_type get_statusint get_argumentrlistcookieskeysrvaluesrnrr~rrlrrr_request_summaryrBs r9preparezTypeCheckHandler.prepareHs  $//"3"3S999  D$5$5e$<$+>&?&?&BCHHH T\-@-@-B-B(C(CA(F(LcRRR  !!## $ $ 0 07$,:K:K:M:MM  +T-C-CF-K-KUSSS  doof&=&=sCCC  dou=== ($*=*=*?*?EEE  t'7'7 U'K'KSQQQ )4+@+@+B+BCHHHHHr;cr|d|t||jdSNpath_componentrrr[rrCrs r9r\zTypeCheckHandler.getcs5 (.,GGG 4;r;cr|d|t||jdSrrrs r9rzTypeCheckHandler.postis3 (.,GGG 4;r;cTt|}||krd|d||j|<dSdS)Nz expected z, got )rr)rCrkobj expected_type actual_types r9rzTypeCheckHandler.check_typems@3ii K ' ' '9F TDK    ( 'r;N)rKrLrMrr\rrrIr;r9rrGsXIII6      UUUUUr;rceZdZddZdZdS)DecodeArgHandlerNct|tkrtdt|zd|jjvr8|t |jjddS|S)Nzunexpected type for value: %rencodingr)rrrrrdecoder)rCrnrks r9decode_argumentz DecodeArgHandler.decode_argumenttsk ;;%  ;d5kkIJJ J / / /<< 4<+A*+Ma+P Q QRR RLr;cd}|||||dddS)Nct|tkr#dttj|gSt|t krd|gSt d)Nrrz unknown type)rrr rb2a_hexrr)rs r9describez&DecodeArgHandler.get..describe~sZAww%H,"""""r;Nr]rIr;r9rrs######r;rceZdZdZdS)MultiHeaderHandlerc|dd|dd|dd|dddS) N x-overwriterz X-Overwriterx-multir:zX-Multi4) set_header add_headerrBs r9r\zMultiHeaderHandler.getsX  s+++  q)))  1%%%  3'''''r;Nr]rIr;r9rrs#(((((r;rceZdZdZdS)r c v|ddF|dtt|ddS|dd9|dt|ddSt d)N permanentrR)rr)rz(didn't get permanent or status arguments)rredirectboolrrrBs r9r\zRedirectHandler.gets   [$ / / ; MM#c$2C2CK2P2P.Q.Q)R)RM S S S S S   x . . : MM#c$*;*;H*E*E&F&FM G G G G GFGG Gr;Nr]rIr;r9r r s(HHHHHr;r c.eZdZejdZdS)EmptyFlushCallbackHandlerc#K|V|V|d|V|V|ddS)Nok)rr[rrBs r9r\zEmptyFlushCallbackHandler.gets jjlljjll 3jjlljjll Cr;N)rKrLrMrrr\rIr;r9rrs/]  ]   r;rceZdZdZdS)HeaderInjectionHandlerc |ddtd#t$r3}dt|vr|dnYd}~dSd}~wwxYw)NzX-Foozfoo X-Bar: bazzDidn't get expected exceptionzUnsafe header valueok)rr ValueErrorrrrCrs r9r\zHeaderInjectionHandler.gets  OOG%8 9 9 9;<< <   $A.. E""""#""""" s%' A$(AA$Nr]rIr;r9rrs#r;rceZdZdZdS)GetArgumentHandlerc|dddkr|j}n)|dddkr|j}n|j}||dddS)Nsourcerrrtr)rget_query_argumentget_body_argumentrrCrs r9rzGetArgumentHandler.preparesx   Xt , , 7 7,FF   x . .& 8 8+FF&F FF5),,-----r;NrKrLrMrrIr;r9r r s#.....r;r ceZdZdZdS)GetArgumentsHandlerc |t|d|d|ddS)Nrtrrr)rrgrget_query_argumentsget_body_argumentsrBs r9rzGetArgumentsHandler.preparesg **511..u55,,U33        r;NrrIr;r9rrs#     r;rceZdZdZdZfdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZxZS)WSGISafeWebTestzWebTest.COOKIE_SECRETcVtdddd}t|d|jS)Nz{% module linkify(message) %}zv {% for e in entries %} {% module Template("entry.html", entry=e) %} {% end %} aY{{ set_resources(embedded_css=".entry { margin-bottom: 1em; }", embedded_javascript="js_embed()", css_files=["/base.css", "/foo.css"], javascript_files="/common.js", html_head="", html_body='