o h.@sddlmZddlmZddlmZmZmZmZm 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@ddlAmBZBddlCZCddlDZDddlEZEddlFZFddlGZGddlHZHddlIZIddlJZKddZLGdddeZMGdddeMZNGddde'ZOGddde'ZPGdddeIjQZRGdddeIjQZSGd d!d!eMZTGd"d#d#eMZUGd$d%d%e'ZVGd&d'd'eMZWGd(d)d)e'ZXGd*d+d+eMZYGd,d-d-e'ZZGd.d/d/eMZ[Gd0d1d1e'Z\Gd2d3d3e'Z]Gd4d5d5e'Z^Gd6d7d7e'Z_Gd8d9d9e'Z`Gd:d;d;e'ZaGdd?d?e'ZbGd@dAdAe'ZcGdBdCdCe'ZdGdDdEdEe'ZeGdFdGdGeMZfGdHdIdIeMZgGdJdKdKeMZhGdLdMdMeMZiGdNdOdOeMZjGdPdQdQeMZkGdRdSdSeMZlGdTdUdUeMZmGdVdWdWeMZnGdXdYdYeMZoGdZd[d[eMZpGd\d]d]eNZqGd^d_d_eNZrGd`dadaeNZsGdbdcdceNZtGdddedeeNZuGdfdgdgeNZvGdhdidieMZwGdjdkdkeNZxGdldmdmeMZyGdndodoeNZzGdpdqdqe{Z|GdrdsdseNZ}GdtdudueNZ~GdvdwdweNZGdxdydyeNZGdzd{d{eNZGd|d}d}eMZGd~ddeNZGdddeNZGdddeNZGdddeNZGdddeNZGdddeMZGdddeMZGdddeMZGdddeMZGdddeMZe6Gddde'ZGdddeZGdddeeMZGdddeeMZGdddeNZGdddeNZGdddeIjQZGdddeNZGdddeNZGdddeNZGdddeNZGdddeMZGdddeMZGdddeNZGdddeIjQZGdddeZGdddeIjQZGdddeMZGdddeMZdS))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)BytesIOcGstjjtjtg|RSN)ospathjoindirname__file__)ar7I/var/www/vscode/kcb/lib/python3.10/site-packages/tornado/test/web_test.pyrelpath=r9c@s(eZdZdZddZddZddZdS) 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. cCs t|fi||_|jSr0)r get_handlersget_app_kwargsappselfr7r7r8get_appHszWebTestCase.get_appcCtr0NotImplementedErrorr?r7r7r8r<LzWebTestCase.get_handlerscCsiSr0r7r?r7r7r8r=OzWebTestCase.get_app_kwargsN)__name__ __module__ __qualname____doc__rAr<r=r7r7r7r8r;As  r;c@seZdZdZdZddZdS)SimpleHandlerTestCasezSimplified base class for tests that work with a single handler class. To use, define a nested class named ``Handler``. NcCs d|jfgS)N/Handlerr?r7r7r8r<[ z"SimpleHandlerTestCase.get_handlers)rGrHrIrJrNr<r7r7r7r8rKSs rKc@eZdZddZdS) HelloHandlercC|ddSNhellowriter?r7r7r8get`zHelloHandler.getNrGrHrIrWr7r7r7r8rQ_ rQc@s(eZdZd ddZddZd ddZdS) CookieTestRequestHandler 0123456789NcCs<i|_|durtt|dd|_dStt||dd|_dS)N) cookie_secret)settingsr] key_version)_cookiesrdict application)r@r]r`r7r7r8__init__fs   z!CookieTestRequestHandler.__init__cCs |j|Sr0)rarW)r@namer7r7r8 get_cookierrOz#CookieTestRequestHandler.get_cookiecCs||j|<dSr0)ra)r@revalue expires_daysr7r7r8 set_cookieurXz#CookieTestRequestHandler.set_cookie)r\Nr0)rGrHrIrdrfrir7r7r7r8r[ds  r[c@$eZdZddZddZddZdS)SecureCookieV1TestcC0t}|jdddd||jdddddS)Nfoobarversion min_versionr[set_signed_cookie assertEqualget_signed_cookier@handlerr7r7r8test_round_trip{z"SecureCookieV1Test.test_round_tripcCst}|jdtddd|jd}td|}|dusJ|d}|d}|t |j j ddd|||t |j j ddd d ||t d t |t |f|jd<ttd ||jddd duWddS1suwYdS)Nrms d76df8e7aefcrorps12345678\|([0-9]+)\|([0-9a-f]+)r]123456781234s5678z1234|5678%s|%szCookie timestamp in futurerr)r[rubinasciia2b_hexrarematchgrouprvr%rcr^rr rr assertTruerw)r@rycookier timestampsigr7r7r8&test_cookie_tampering_future_timestamps>          "z9SecureCookieV1Test.test_cookie_tampering_future_timestampcCrl)Nrmrorprrrtrxr7r7r8test_arbitrary_bytessz'SecureCookieV1Test.test_arbitrary_bytesN)rGrHrIrzrrr7r7r7r8rkzs &rkc@s>eZdZdddZddZddZdd Zd d Zd d ZdS)SecureCookieV2Testajklasdf0ojaisdfaslkjasaolwkjsdf)rrocCrl)Nrmrnr|rprrrtrxr7r7r8rzr{z"SecureCookieV2Test.test_round_tripcC0t|jdd}|dd||dddS)Nrr_rmrnr[ KEY_VERSIONSrurvrwrxr7r7r8test_key_version_roundtrip  z-SecureCookieV2Test.test_key_version_roundtripcCr)Nror_rmrnrrxr7r7r8,test_key_version_roundtrip_differing_versionrz?SecureCookieV2Test.test_key_version_roundtrip_differing_versioncCsFt|jdd}|ddt|jdd}|j|_||dddSNrr_rmrnro)r[rrurarvrw)r@ry new_handlerr7r7r8"test_key_version_increment_versions z5SecureCookieV2Test.test_key_version_increment_versioncCsXt|jdd}|dd|j}|dt|dd}|j|_||dddSr)r[rrucopypoprarvrw)r@rynew_key_versionsrr7r7r8#test_key_version_invalidate_versions   z6SecureCookieV2Test.test_key_version_invalidate_versionN) rGrHrIrrzrrrrr7r7r7r8rs  rc@s0eZdZdZddZddZddZdd ZdS) FinalReturnTestNcs<|Gfdddt}Gfdddt}d|fd|fgS)Ncs0eZdZejfddZejfddZdS)z3FinalReturnTest.get_handlers..FinishHandlerc3s|_jVdSr0)finish final_returnr?testr7r8rWs  z7FinalReturnTest.get_handlers..FinishHandler.getc3s.|d|V|d_jVdS)Nzhello,world)rVflushrrr?rr7r8posts     z8FinalReturnTest.get_handlers..FinishHandler.postN)rGrHrIr coroutinerWrr7rr7r8 FinishHandlers rcs&eZdZddZejfddZdS)z3FinalReturnTest.get_handlers..RenderHandlercSs tddiS)Nfoo.htmlhirr@r2r7r7r8create_template_loaderrOzJFinalReturnTest.get_handlers..RenderHandler.create_template_loadercs|d_dSNr)renderrr?rr7r8rWsz7FinalReturnTest.get_handlers..RenderHandler.getN)rGrHrIrrrrWr7rr7r8 RenderHandlersr/finish/renderr)r@rrr7rr8r<s zFinalReturnTest.get_handlerscC tddS)Nr) template_pathrbr?r7r7r8r= zFinalReturnTest.get_app_kwargscCs||d}||jd||jt||j|j|dddd}||jd||jt||jdS)NrPOSTmethodbody) fetchget_urlrvcodeassertIsInstancerrrdoner@responser7r7r8 test_finish_method_return_futuresz0FinalReturnTest.test_finish_method_return_futurecCs0||d}||jd||jtdS)Nrr)rrrvrrrrrr7r7r8 test_render_method_return_future sz0FinalReturnTest.test_render_method_return_future)rGrHrIrr<r=rrr7r7r7r8rs  rc@s\eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS) CookieTestc CsGdddt}Gdddt}Gdddt}Gdddt}Gd d d t}Gd d d t}Gd ddt}Gdddt}Gdddt} d|fd|fd|fd|fd|fd|fd|fd|fd| fg S)Nc@rP)z1CookieTest.get_handlers..SetCookieHandlercS(|dd|dd|dddS)Nstrasdfunicodeqwerbytesszxcvrir?r7r7r8rWs  z5CookieTest.get_handlers..SetCookieHandler.getNrYr7r7r7r8SetCookieHandlerrZrc@rP)z1CookieTest.get_handlers..GetCookieHandlercSs&|dd}|dus J||dS)Nrmdefault)rfrV)r@rr7r7r8rWs  z5CookieTest.get_handlers..GetCookieHandler.getNrYr7r7r7r8GetCookieHandlerrZrc@rP)z7CookieTest.get_handlers..SetCookieDomainHandlercS|jddddddS)N unicode_argsblahzfoo.com/foo)domainr2rr?r7r7r8rW sz;CookieTest.get_handlers..SetCookieDomainHandler.getNrYr7r7r7r8SetCookieDomainHandlerrZrc@rP)z.SetCookieSpecialCharHandlercSr)Nequalsa=b semicolona;bquotea"brr?r7r7r8rW&s  z@CookieTest.get_handlers..SetCookieSpecialCharHandler.getNrYr7r7r7r8SetCookieSpecialCharHandler%rZrc@rP)z:CookieTest.get_handlers..SetCookieOverwriteHandlercSs0|jdddd|jdddd|dddS)Nr6bz example.com)rcderr?r7r7r8rW,sz>CookieTest.get_handlers..SetCookieOverwriteHandler.getNrYr7r7r7r8SetCookieOverwriteHandler+rZrc@rP)z7CookieTest.get_handlers..SetCookieMaxAgeHandlercS|jdddddS)Nrmbar )max_agerr?r7r7r8rW4z;CookieTest.get_handlers..SetCookieMaxAgeHandler.getNrYr7r7r7r8SetCookieMaxAgeHandler3rZrc@rP)z.SetCookieExpiresDaysHandlercSr)Nrmrr)rhrr?r7r7r8rW8rz@CookieTest.get_handlers..SetCookieExpiresDaysHandler.getNrYr7r7r7r8SetCookieExpiresDaysHandler7rZrc@rP)z4CookieTest.get_handlers..SetCookieFalsyFlagscSsD|jdddd|jdddd|jdddd|jd ddddS) Nr61T)securerFr)httponlyrrr?r7r7r8rW<sz8CookieTest.get_handlers..SetCookieFalsyFlags.getNrYr7r7r7r8SetCookieFalsyFlags;rZrc@rP)z8CookieTest.get_handlers..SetCookieDeprecatedArgscSr)Nr6rTr)HttpOnlypATHrr?r7r7r8rWCsz.SetCookieDeprecatedArgs.getNrYr7r7r7r8SetCookieDeprecatedArgsBrZr/set/get /set_domain /special_char/set_overwrite /set_max_age/set_expires_days/set_falsy_flags/set_deprecatedr) r@rrrrrrrrrr7r7r8r<s&zCookieTest.get_handlerscCs*|d}|t|jdgddS)Nr Set-Cookie)zbytes=zxcv; Path=/zstr=asdf; Path=/zunicode=qwer; Path=/)rrvsortedheadersget_listrr7r7r8test_set_cookieSs zCookieTest.test_set_cookiecCsd|jdddid}||jd|jdddid}||jd|jdddid}||jddS) NrCookiefoo=barrrnz foo="bar"z /=exception;defaultrrvrrr7r7r8test_get_cookieZs zCookieTest.test_get_cookiecCs$|d}||jddgdS)Nrrz,unicode_args=blah; Domain=foo.com; Path=/foorrvrrrr7r7r8test_set_cookie_domainds  z!CookieTest.test_set_cookie_domaincCs|d}t|jd}|t|d||dd||dd||dd v|dgd }|D]\}}td ||jd d |id}||j t |q7dS)Nrrrzequals="a=b"; Path=/rozquote="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 %rrrr) rrrrrvlenrloggingdebugrr)r@rrdataheaderexpectedr7r7r8test_cookie_special_charks    z#CookieTest.test_cookie_special_charcCs.|d}|jd}|t|ddgdS)Nrrz a=e; Path=/zc=d; Domain=example.com; Path=/rrrrvrr@rrr7r7r8test_set_cookie_overwrites   z$CookieTest.test_set_cookie_overwritecCs,|d}|jd}|t|dgdS)Nrrzfoo=bar; Max-Age=10; Path=/r r r7r7r8test_set_cookie_max_ages  z"CookieTest.test_set_cookie_max_agecCs|d}|jd}|dusJtd|}|dusJtjtjjtj dd}t j | d}|t||dkdS)Nrrz(foo=bar; expires=(?P.+); Path=/rdaysexpires)rrrWrrdatetimenowtimezoneutc timedeltaemailutilsparsedate_to_datetime groupdictrabs total_secondsr@rrrrheader_expiresr7r7r8test_set_cookie_expires_dayss     z'CookieTest.test_set_cookie_expires_dayscCsn|d}t|jd}||dd||dd||dd||d d dS) Nrrrza=1; path=/; secureroz b=1; path=/r|zc=1; httponly; path=/rz d=1; path=/)rrrrrvlowerr r7r7r8test_set_cookie_false_flagss z&CookieTest.test_set_cookie_false_flagscCsLt |d}Wdn1swY|jd}||ddS)Nrrza=b; HttpOnly; Path=/foo)rrrrWrv)r@rrr7r7r8test_set_cookie_deprecateds   z%CookieTest.test_set_cookie_deprecatedN) rGrHrIr<rrrr r rrr!r"r7r7r7r8rsC  rc@s(eZdZddZddZeddZdS)AuthRedirectRequestHandlercC ||_dSr0 login_url)r@r&r7r7r8 initializerz%AuthRedirectRequestHandler.initializecC|jSr0r%r?r7r7r8 get_login_urlrEz(AuthRedirectRequestHandler.get_login_urlcCrR)N) send_errorr?r7r7r8rWzAuthRedirectRequestHandler.getN)rGrHrIr'r)r#rWr7r7r7r8r#s r#c@rj)AuthRedirectTestcCs dttddfdttddfgS)N /relativez/loginr% /absolutezhttp://example.com/login)r#rbr?r7r7r8r<s zAuthRedirectTest.get_handlerscCs8|j|ddd}||jd||jdddS)Nr.Ffollow_redirects.Locationz/login?next=%2Frelative)rrrvrrrr7r7r8test_relative_auth_redirectsz,AuthRedirectTest.test_relative_auth_redirectcCsF|j|ddd}||jd|td|jd|jddS)Nr/Fr0r2zIhttp://example.com/login\?next=http%3A%2F%2F127.0.0.1%3A[0-9]+%2Fabsoluter3)rrrvrrrrrrr7r7r8test_absolute_auth_redirectsz,AuthRedirectTest.test_absolute_auth_redirectN)rGrHrIr<r4r5r7r7r7r8r-s r-c@s*eZdZddZejddZddZdS)ConnectionCloseHandlercCr$r0rr@rr7r7r8r'rz!ConnectionCloseHandler.initializeccs|j|jjVdSr0)ron_handler_waiting cleanup_eventwaitr?r7r7r8rWs zConnectionCloseHandler.getcCs|jdSr0)ron_connection_closer?r7r7r8r;rXz*ConnectionCloseHandler.on_connection_closeN)rGrHrIr'rrrWr;r7r7r7r8r6s   r6c@,eZdZddZddZddZddZd S) ConnectionCloseTestcCst|_dtt|dfgS)NrLr)rr9r6rbr?r7r7r8r<sz ConnectionCloseTest.get_handlerscCs`ttjtjd}|d|ft||_|jd||j |j dddS)Nr 127.0.0.1sGET / HTTP/1.0 cSs tdSNr)rsleepr7r7r7r8s z;ConnectionCloseTest.test_connection_close..) socketAF_INET SOCK_STREAMconnect get_http_portr streamrVr:r9setio_looprun_sync)r@sr7r7r8test_connection_closes   z)ConnectionCloseTest.test_connection_closecCstd|jdS)Nzhandler waiting)rrrGcloser?r7r7r8r8 z&ConnectionCloseTest.on_handler_waitingcCstd|dS)Nzconnection closed)rrstopr?r7r7r8r;  z'ConnectionCloseTest.on_connection_closeN)rGrHrIr<rLr8r;r7r7r7r8r=s  r=c@rP) EchoHandlercGs|jjD]=}t|tkrtdt||jj|D]}t|tkr*tdt|q||D]}t|tkr@tdt|q0q|D]}t|tkrTtdt|qD|t |jj |t |jjddS)Nzincorrect type for key: %rzincorrect type for value: %rzincorrect type for path arg: %rr2 path_argsargs) request argumentstyper Exceptionr get_argumentsrrVrbr2r)r@rSkeybvaluesvalueargr7r7r8rWs.      zEchoHandler.getNrYr7r7r7r8rQrZrQc@s<eZdZddZddZddZddZd d Zd d Zd S)RequestEncodingTestcCsdtfdtfgS)Nz /group/(.*)z/slashes/([^/]*)/([^/]*))rQr?r7r7r8r<z RequestEncodingTest.get_handlerscCst||jSr0)rrrrr7r7r8 fetch_jsonr_zRequestEncodingTest.fetch_jsoncCsF||dtddgid||dtddgddgiddS)Nz /group/%3F?rRz/group/%3F?%3F=%3Frvr`rbr?r7r7r8test_group_question_marksz,RequestEncodingTest.test_group_question_markcCs&||dddgddgiddS)Nz/group/%C3%A9?arg=%C3%A9z /group/%C3%A9ér]rRrvr`r?r7r7r8test_group_encoding)sz'RequestEncodingTest.test_group_encodingcCsD||dtdddgid||dtdddgiddS)Nz/slashes/foo/barrmrrRz/slashes/a%2Fb/c%2Fdza/bzc/drbr?r7r7r8 test_slashes4sz RequestEncodingTest.test_slashescC:ttd|dWddS1swYdS)Nz.*Invalid unicodez/group/?arg=%25%e9)rrrr?r7r7r8 test_error@  "zRequestEncodingTest.test_errorN) rGrHrIr<r`rcrfrgrir7r7r7r8r^s  r^c@r<) TypeCheckHandlercCsi|_|d|t|d|dt|dt|jdt |dt|j dj t t|jdgkrFt d|j|d | dt|d |dt |d |jt|d |t |d |ddt |d|t dS)Nstatusargumentrm cookie_keyr cookie_valuerz%unexpected values for cookie keys: %rrwrf xsrf_tokenxsrf_form_html reverse_url typecheckrequest_summary)errors check_type get_statusint get_argumentrlistcookieskeysrvaluesrgrXrwrrfrprqrr_request_summaryr?r7r7r8prepareHs zTypeCheckHandler.preparecC|d|t||jdSNpath_componentrvrrVrur@rr7r7r8rWcszTypeCheckHandler.getcCrrrrr7r7r8riszTypeCheckHandler.postcCs*t|}||krd||f|j|<dSdS)Nzexpected %s, got %s)rWru)r@reobj expected_type actual_typer7r7r8rvmszTypeCheckHandler.check_typeN)rGrHrIrrWrrvr7r7r7r8rkGs  rkc@seZdZdddZddZdS)DecodeArgHandlerNcCsFt|tkrtdt|d|jjvr!|t|jjddS|S)Nzunexpected type for value: %rencodingr)rWrrXrUrVdecoder)r@rgrer7r7r8decode_argumentts  z DecodeArgHandler.decode_argumentcCs*dd}|||||dddS)NcSs:t|tkrdtt|gSt|tkrd|gStd)Nrrz unknown type)rWrrrb2a_hexrrX)rKr7r7r8describe~s  z&DecodeArgHandler.get..describermr2query)rVry)r@r]rr7r7r8rW}s"zDecodeArgHandler.getr0)rGrHrIrrWr7r7r7r8rss  rc@rP)LinkifyHandlercCs|jddddS)N linkify.htmlzhttp://example.com)messagerr?r7r7r8rWzLinkifyHandler.getNrYr7r7r7r8rrZrc@rP)UIModuleResourceHandlercCs|jdddgddS)N page.htmlror|)entriesrr?r7r7r8rWszUIModuleResourceHandler.getNrYr7r7r7r8rrZrc@rP)OptionalPathHandlercCs|d|idS)Nr2rUrr7r7r8rWrzOptionalPathHandler.getNrYr7r7r7r8rrZrc@rP)MultiHeaderHandlercCs4|dd|dd|dd|dddS) N x-overwriterz X-Overwriter|x-multirzX-Multi4) set_header add_headerr?r7r7r8rWs   zMultiHeaderHandler.getNrYr7r7r7r8rrZrc@rP)rcCsd|dddur|jdtt|dddS|dddur.|jdt|dddStd)N permanentrL)rrl)rlz(didn't get permanent or status arguments)ryredirectboolrxrXr?r7r7r8rWs  zRedirectHandler.getNrYr7r7r7r8rrZrc@eZdZejddZdS)EmptyFlushCallbackHandlerccsB|V|V|d|V|V|ddS)Nok)rrVrr?r7r7r8rWs     zEmptyFlushCallbackHandler.getN)rGrHrIrrrWr7r7r7r8rrc@rP)HeaderInjectionHandlerc CsVz |ddtdty*}zdt|vr|dnWYd}~dSd}~ww)NzX-Foozfoo X-Bar: bazzDidn't get expected exceptionzUnsafe header valueok)rrX ValueErrorrrr@rr7r7r8rWs   zHeaderInjectionHandler.getNrYr7r7r7r8rrZrc@rP)GetArgumentHandlercCsJ|dddkr |j}n|dddkr|j}n|j}||dddS)Nsourcerrrmr)ryget_query_argumentget_body_argumentrr@rr7r7r8rs zGetArgumentHandler.prepareNrGrHrIrr7r7r7r8rrZrc@rP)GetArgumentsHandlercCs*|t|d|d|dddS)Nrmrrr)rrbrYget_query_argumentsget_body_argumentsr?r7r7r8rszGetArgumentsHandler.prepareNrr7r7r7r8rrZrcseZdZdZddZfddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+ZZS),WSGISafeWebTestzWebTest.COOKIE_SECRETcCs tdddd}t|d|jdS)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='