Mhy 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'ZeGdEdFe'ZfGdGdHeMZgGdIdJeMZhGdKdLeMZiGdMdNeMZjGdOdPeMZkGdQdReMZlGdSdTeMZmGdUdVeMZnGdWdXeMZoGdYdZeMZpGd[d\eMZqGd]d^eNZrGd_d`eNZsGdadbeNZtGdcddeNZuGdedfeNZvGdgdheNZwGdidjeMZxGdkdleNZyGdmdneMZzGdodpeNZ{Gdqdre|Z}GdsdteNZ~GdudveNZGdwdxeNZGdydzeNZGd{d|eNZGd}d~eMZGddeNZGddeNZGddeNZGddeNZGddeNZGddeMZGddeMZGddeMZGddeMZGddeMZe6Gdde'ZGddZGddeeMZGddeeMZGddeNZGddeNZGddeIjQZGddeNZGddeNZGddeNZGddeNZGddeMZGddeMZGddeNZGddeIjQZGddeZGddeIjQZGddeMZGddeMZdS))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 U/var/www/html/test/jupyter/venv/lib/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<tt d 5||ddddddS#1swxYwYdS)Nrts d76df8e7aefcrvrws12345678\|([0-9]+)\|([0-9a-f]+)rb123456781234s5678z 1234|5678|zCookie timestamp in futurery)r_r|binasciia2b_hexrfrematchassertIsNotNonegroupr}r&rhrcrr rr assertIsNoner~)rCrcookier timestampsigs r9&test_cookie_tampering_future_timestampz9SecureCookieV1Test.test_cookie_tampering_future_timestamps*,,!!%)9/)J)JTU!VVV!%(.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;r9SetCookieDomainHandlerr s( 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;r ceZdZdZdS)7CookieTest.get_handlers..SetCookieMaxAgeHandlerc6|ddddS)Nrtbar )max_agerrBs r9r\z;CookieTest.get_handlers..SetCookieMaxAgeHandler.get5s ub99999r;Nr]rIr;r9SetCookieMaxAgeHandlerr 4s# : : : : :r;rceZdZdZdS).SetCookieExpiresDaysHandlerc6|ddddS)Nrtr r)rorrBs r9r\z@CookieTest.get_handlers..SetCookieExpiresDaysHandler.get9s u2>>>>>r;Nr]rIr;r9SetCookieExpiresDaysHandlerr8s# ? ? ? ? ?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.getDs"S4fEEEEEr;Nr]rIr;r9SetCookieDeprecatedArgsrCs( 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) rCrrrrr rrrr!s 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_cookieTsZ::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_cookie[s::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_domainesO::m,,    % %l 3 3 ; <     r;c>|d}t|jd}|t |d||dd||dd||dd |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#r2r4) rr-r.r/r}lenassertInloggingdebugrr)rCrr.dataheaderexpecteds r9test_cookie_special_charz#CookieTest.test_cookie_special_charls(::o..)22<@@AA Wq))) %;<<< %<=== AJ F AJ      !% < < FH M+v . . .zz&8V2DzEEH   X]DNN ; ; ; ; < .+); Path=/rdaysexpires)rr.r\rrrdatetimenowtimezoneutc timedeltaemailutilsparsedate_to_datetime groupdict assertLessabs total_secondsrCrrBrrNheader_expiress r9test_set_cookie_expires_daysz'CookieTest.test_set_cookie_expires_dayss::122!%%l33 V$$$!!!CVLL U###   #''(9(=>>ASB B B  ::5??;L;LY;WXX W~5DDFFGGLLLLLr;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}rGs r9test_set_cookie_false_flagsz&CookieTest.test_set_cookie_false_flagss::011)22<@@AA %:;;; ]333 %<=== ]33333r;ct5|d}dddn #1swxYwY|jd}||ddS)Nr*r,za=b; HttpOnly; Path=/foo)rrr.r\r})rCrrBs 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[rrsetio_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_mark s  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_slashes6s  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_errorBsw 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.prepareJs  $//"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.getes5 (.,GGG 4;r;cr|d|t||jdSrrrs r9rzTypeCheckHandler.postks3 (.,GGG 4;r;cTt|}||krd|d||j|<dSdS)Nz expected z, got )rr)rCrkobj expected_type actual_types r9rzTypeCheckHandler.check_typeosB3ii K ' ' NM N N N NDK    ( 'r;N)rKrLrMrr\rrrIr;r9rrIsXIII6      OOOOOr;rceZdZddZdZdS)DecodeArgHandlerNct|turtdt|zd|jjvr8|t |jjddS|S)Nzunexpected type for value: %rencodingr)rrrrrdecoder)rCrnrks r9decode_argumentz DecodeArgHandler.decode_argumentvsk ;;e # #;d5kkIJJ J / / /<< 4<+A*+Ma+P Q QRR RLr;cd}|||||dddS)Nct|tur#dttj|gSt|t urd|gSt d)Nrrz unknown type)rrr rb2a_hexrr)rs r9describez&DecodeArgHandler.get..describesZAww%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)NX-Foozfoo X-Bar: bazDidn't get expected exceptionUnsafe header valueok)rr ValueErrorrrrCrs r9r\zHeaderInjectionHandler.gets  OOG%8 9 9 9;<< <   $A.. E""""#""""" s%' A$(AA$Nr]rIr;r9r r s#r;r ceZdZdZdS)SetHeaderHandlercdtddD}|td|d|D]V} |dd|zdzt d #t $r}d t|vrYd}~Od}~wwxYw|dd |d dS) Nc,g|]}t|SrI)chr).0rs r9 z(SetHeaderHandler.get..s888AQ888r;r  r rtr r rr) rangeappendrremoverrrrr)rC illegal_charscharrs r9r\zSetHeaderHandler.gets88q$888 SYY'''T"""!  D )=>>> ?@@@   (A6676666  $$$ Es+B B( B##B(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;Nr)rIr;r9r+r+s#     r;r+ceZdZdZdZfdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZxZS)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='