o Q_@sddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZmZmZdd l m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7ddl8Z8ddl9Z9ddl:Z:ddl;Z;ddlZ>ddl?m@Z@ddlAZAddlBZBddlCZCddlDZDddlEZEddlFZFddlGZGddlHZIddZJGdddeZKGdddeKZLGddde%ZMGddde%ZNGdddeGjOZPGdddeGjOZQGdd d eKZRGd!d"d"eKZSGd#d$d$e%ZTGd%d&d&eKZUGd'd(d(e%ZVGd)d*d*eKZWGd+d,d,e%ZXGd-d.d.eKZYGd/d0d0e%ZZGd1d2d2e%Z[Gd3d4d4e%Z\Gd5d6d6e%Z]Gd7d8d8e%Z^Gd9d:d:e%Z_Gd;d<dd>e%Z`Gd?d@d@e%ZaGdAdBdBe%ZbGdCdDdDe%ZcGdEdFdFeKZdGdGdHdHeKZeGdIdJdJeKZfGdKdLdLeKZgGdMdNdNeKZhGdOdPdPeKZiGdQdRdReKZjGdSdTdTeKZkGdUdVdVeKZlGdWdXdXeKZmGdYdZdZeLZnGd[d\d\eLZoGd]d^d^eLZpGd_d`d`eLZqGdadbdbeLZrGdcddddeLZsGdedfdfeKZtGdgdhdheLZuGdidjdjeKZvGdkdldleLZwGdmdndnexZyGdodpdpeLZzGdqdrdreLZ{GdsdtdteLZ|GdudvdveLZ}GdwdxdxeLZ~GdydzdzeKZGd{d|d|eLZGd}d~d~eLZGdddeLZGdddeLZGdddeLZGdddeKZGdddeKZGdddeKZGdddeKZGdddeKZe4Gddde%ZGdddeZGdddeeKZGdddeeKZGdddeLZGdddeLZGdddeGjOZGdddeLZGdddeLZGdddeLZGdddeKZGdddeKZGdddeLZGdddeGjOZGdddeZGdddeGjOZGdddeKZdS))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) ObjectDict unicode_type) ApplicationRequestHandlerStaticFileHandlerRedirectHandler HTTPErrorMissingArgumentError ErrorHandler authenticatedurl_create_signature_v1create_signed_valuedecode_signed_valueget_signature_key_versionUIModuleFinishstream_request_body removeslashaddslashGZipContentEncodingN)BytesIOcGstjjtjtg|RSN)ospathjoindirname__file__)ar67/usr/lib/python3/dist-packages/tornado/test/web_test.pyrelpath<r8c@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||_|jSr/)r get_handlersget_app_kwargsappselfr6r6r7get_appGszWebTestCase.get_appcCtr/NotImplementedErrorr>r6r6r7r;KzWebTestCase.get_handlerscCsiSr/r6r>r6r6r7r<NzWebTestCase.get_app_kwargsN)__name__ __module__ __qualname____doc__r@r;r<r6r6r6r7r:@s  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>r6r6r7r;Z z"SimpleHandlerTestCase.get_handlers)rFrGrHrIrMr;r6r6r6r7rJRs rJc@eZdZddZdS) HelloHandlercC|ddSNhellowriter>r6r6r7get_zHelloHandler.getNrFrGrHrVr6r6r6r7rP^ rPc@s(eZdZd ddZddZd ddZdS) CookieTestRequestHandler 0123456789NcCs<i|_|durtt|dd|_dStt||dd|_dS)N) cookie_secret)settingsr\ key_version)_cookiesrdict application)r?r\r_r6r6r7__init__es   z!CookieTestRequestHandler.__init__cCs |j|Sr/)r`rV)r?namer6r6r7 get_cookieqrNz#CookieTestRequestHandler.get_cookiecCs||j|<dSr/)r`)r?rdvalue expires_daysr6r6r7 set_cookietrWz#CookieTestRequestHandler.set_cookie)r[Nr/)rFrGrHrcrerhr6r6r6r7rZcs  rZc@$eZdZddZddZddZdS)SecureCookieV1TestcC0t}|jdddd||jdddddS)Nfoobarversion min_versionrZset_secure_cookie assertEqualget_secure_cookier?handlerr6r6r7test_round_tripzz"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)Nrls d76df8e7aefcrnros12345678\|([0-9]+)\|([0-9a-f]+)r\123456781234s5678z1234|5678%s|%szCookie timestamp in futurerq)rZrtbinasciia2b_hexr`rematchgrouprur$rbr]rr rr assertTruerv)r?rxcookier timestampsigr6r6r7&test_cookie_tampering_future_timestamps>          "z9SecureCookieV1Test.test_cookie_tampering_future_timestampcCrk)Nrlrnrorqrsrwr6r6r7test_arbitrary_bytessz'SecureCookieV1Test.test_arbitrary_bytesN)rFrGrHryrrr6r6r6r7rjys &rjc@s>eZdZdddZddZddZdd Zd d Zd d ZdS)SecureCookieV2Testajklasdf0ojaisdfaslkjasaolwkjsdf)rrncCrk)Nrlrmr{rorqrsrwr6r6r7ryrzz"SecureCookieV2Test.test_round_tripcC0t|jdd}|dd||dddS)Nrr^rlrmrZ KEY_VERSIONSrtrurvrwr6r6r7test_key_version_roundtrip  z-SecureCookieV2Test.test_key_version_roundtripcCr)Nrnr^rlrmrrwr6r6r7,test_key_version_roundtrip_differing_versionrz?SecureCookieV2Test.test_key_version_roundtrip_differing_versioncCsFt|jdd}|ddt|jdd}|j|_||dddSNrr^rlrmrn)rZrrtr`rurv)r?rx new_handlerr6r6r7"test_key_version_increment_versions z5SecureCookieV2Test.test_key_version_increment_versioncCsXt|jdd}|dd|j}|dt|dd}|j|_||dddSr)rZrrtcopypopr`rurv)r?rxnew_key_versionsrr6r6r7#test_key_version_invalidate_versions   z6SecureCookieV2Test.test_key_version_invalidate_versionN) rFrGrHrryrrrrr6r6r6r7rs  rc@s0eZdZdZddZddZddZdd ZdS) FinalReturnTestNcs<|Gfdddt}Gfdddt}d|fd|fgS)Ncs0eZdZejfddZejfddZdS)z3FinalReturnTest.get_handlers..FinishHandlerc3s|_jVdSr/)finish final_returnr>testr6r7rVs  z7FinalReturnTest.get_handlers..FinishHandler.getc3s.|d|V|d_jVdS)Nzhello,world)rUflushrrr>rr6r7posts     z8FinalReturnTest.get_handlers..FinishHandler.postN)rFrGrHr coroutinerVrr6rr6r7 FinishHandlers rcs&eZdZddZejfddZdS)z3FinalReturnTest.get_handlers..RenderHandlercSs tddiS)Nfoo.htmlhirr?r1r6r6r7create_template_loaderrNzJFinalReturnTest.get_handlers..RenderHandler.create_template_loadercs|d_dSNr)renderrr>rr6r7rVsz7FinalReturnTest.get_handlers..RenderHandler.getN)rFrGrHrrrrVr6rr6r7 RenderHandlersr/finish/renderr)r?rrr6rr7r;s zFinalReturnTest.get_handlerscC tddS)Nr) template_pathrar>r6r6r7r< zFinalReturnTest.get_app_kwargscCs||d}||jd||jt||j|j|dddd}||jd||jt||jdS)NrPOSTmethodbody) fetchget_urlrucodeassertIsInstancerrrdoner?responser6r6r7 test_finish_method_return_futuresz0FinalReturnTest.test_finish_method_return_futurecCs0||d}||jd||jtdS)Nrr)rrrurrrrrr6r6r7 test_render_method_return_futuresz0FinalReturnTest.test_render_method_return_future)rFrGrHrr;r<rrr6r6r6r7rs  rc@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS) CookieTestc CsGdddt}Gdddt}Gdddt}Gdddt}Gd d d t}Gd d d t}Gd ddt}Gdddt}d|fd|fd|fd|fd|fd|fd|fd|fgS)Nc@rO)z1CookieTest.get_handlers..SetCookieHandlercS(|dd|dd|dddS)Nstrasdfunicodeqwerbytesszxcvrhr>r6r6r7rVs  z5CookieTest.get_handlers..SetCookieHandler.getNrXr6r6r6r7SetCookieHandlerrYrc@rO)z1CookieTest.get_handlers..GetCookieHandlercSs&|dd}|dus J||dS)Nrldefault)rerU)r?rr6r6r7rVs  z5CookieTest.get_handlers..GetCookieHandler.getNrXr6r6r6r7GetCookieHandlerrYrc@rO)z7CookieTest.get_handlers..SetCookieDomainHandlercSs|jddddddS)N unicode_argsblahzfoo.com/foo)domainr1rr>r6r6r7rVsz;CookieTest.get_handlers..SetCookieDomainHandler.getNrXr6r6r6r7SetCookieDomainHandlerrYrc@rO)z.SetCookieSpecialCharHandlercSr)Nequalsa=b semicolona;bquotea"brr>r6r6r7rV%s  z@CookieTest.get_handlers..SetCookieSpecialCharHandler.getNrXr6r6r6r7SetCookieSpecialCharHandler$rYrc@rO)z:CookieTest.get_handlers..SetCookieOverwriteHandlercSs0|jdddd|jdddd|dddS)Nr5bz example.com)rcderr>r6r6r7rV+sz>CookieTest.get_handlers..SetCookieOverwriteHandler.getNrXr6r6r6r7SetCookieOverwriteHandler*rYrc@rO)z7CookieTest.get_handlers..SetCookieMaxAgeHandlercS|jdddddS)Nrlbar )max_agerr>r6r6r7rV3z;CookieTest.get_handlers..SetCookieMaxAgeHandler.getNrXr6r6r6r7SetCookieMaxAgeHandler2rYrc@rO)z.SetCookieExpiresDaysHandlercSr)Nrlrr)rgrr>r6r6r7rV7rz@CookieTest.get_handlers..SetCookieExpiresDaysHandler.getNrXr6r6r6r7SetCookieExpiresDaysHandler6rYrc@rO)z4CookieTest.get_handlers..SetCookieFalsyFlagscSsD|jdddd|jdddd|jdddd|jd ddddS) Nr51T)securerFr)httponlyrrr>r6r6r7rV;sz8CookieTest.get_handlers..SetCookieFalsyFlags.getNrXr6r6r6r7SetCookieFalsyFlags:rYr/set/get /set_domain /special_char/set_overwrite /set_max_age/set_expires_days/set_falsy_flagsr) r?rrrrrrrrr6r6r7r;s"zCookieTest.get_handlerscCs*|d}|t|jdgddS)Nr Set-Cookie)zbytes=zxcv; Path=/zstr=asdf; Path=/zunicode=qwer; Path=/)rrusortedheadersget_listrr6r6r7test_set_cookieLs zCookieTest.test_set_cookiecCsd|jdddid}||jd|jdddid}||jd|jdddid}||jddS) NrCookiefoo=barrrmz foo="bar"z /=exception;defaultrrurrr6r6r7test_get_cookieSs zCookieTest.test_get_cookiecCs$|d}||jddgdS)Nrrz,unicode_args=blah; Domain=foo.com; Path=/foorrurrrr6r6r7test_set_cookie_domain]s  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=/rnzquote="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) rrrrrulenrloggingdebugrr)r?rrdataheaderexpectedr6r6r7test_cookie_special_chards    z#CookieTest.test_cookie_special_charcCs.|d}|jd}|t|ddgdS)Nrrz a=e; Path=/zc=d; Domain=example.com; Path=/rrrrurr?rrr6r6r7test_set_cookie_overwrite}s   z$CookieTest.test_set_cookie_overwritecCs,|d}|jd}|t|dgdS)Nrrzfoo=bar; Max-Age=10; Path=/rrr6r6r7test_set_cookie_max_ages  z"CookieTest.test_set_cookie_max_agecCs|d}|jd}|dusJtd|}|dusJtjtjdd}tj | d}|dus8Jtj|dd}| t ||dkdS)Nrrz(foo=bar; expires=(?P.+); Path=/rdaysexpires)rrrVrrdatetimeutcnow timedeltaemailutils parsedate groupdictrabs total_secondsr?rrrr parsedheader_expiresr6r6r7test_set_cookie_expires_dayss      z'CookieTest.test_set_cookie_expires_dayscCsn|d}t|jd}||dd||dd||dd||d d dS) Nrrrza=1; path=/; securernz b=1; path=/r{zc=1; httponly; path=/rz d=1; path=/)rrrrrulowerrr6r6r7test_set_cookie_false_flagss z&CookieTest.test_set_cookie_false_flagsN) rFrGrHr;rrrrrrrrr6r6r6r7rs=  rc@s(eZdZddZddZeddZdS)AuthRedirectRequestHandlercC ||_dSr/ login_url)r?rr6r6r7 initializerz%AuthRedirectRequestHandler.initializecC|jSr/rr>r6r6r7 get_login_urlrDz(AuthRedirectRequestHandler.get_login_urlcCrQ)N) send_errorr>r6r6r7rVzAuthRedirectRequestHandler.getN)rFrGrHr r"r"rVr6r6r6r7rs rc@ri)AuthRedirectTestcCs dttddfdttddfgS)N /relativez/loginr /absolutezhttp://example.com/login)rrar>r6r6r7r;s zAuthRedirectTest.get_handlerscCs8|j|ddd}||jd||jdddS)Nr'Ffollow_redirects.Locationz/login?next=%2Frelative)rrrurrrr6r6r7test_relative_auth_redirectsz,AuthRedirectTest.test_relative_auth_redirectcCsF|j|ddd}||jd|td|jd|jddS)Nr(Fr)r+zIhttp://example.com/login\?next=http%3A%2F%2F127.0.0.1%3A[0-9]+%2Fabsoluter,)rrrurrrrrrr6r6r7test_absolute_auth_redirectsz,AuthRedirectTest.test_absolute_auth_redirectN)rFrGrHr;r-r.r6r6r6r7r&s r&c@s*eZdZddZejddZddZdS)ConnectionCloseHandlercCrr/rr?rr6r6r7r rz!ConnectionCloseHandler.initializeccs|j|jjVdSr/)ron_handler_waiting cleanup_eventwaitr>r6r6r7rVs zConnectionCloseHandler.getcCs|jdSr/)ron_connection_closer>r6r6r7r4rWz*ConnectionCloseHandler.on_connection_closeN)rFrGrHr rrrVr4r6r6r6r7r/s   r/c@,eZdZddZddZddZddZd S) ConnectionCloseTestcCst|_dtt|dfgS)NrKr)rr2r/rar>r6r6r7r;sz ConnectionCloseTest.get_handlerscCs`ttjtjd}|d|ft||_|jd||j |j dddS)Nr 127.0.0.1sGET / HTTP/1.0 cSs tdSNr)rsleepr6r6r6r7s z;ConnectionCloseTest.test_connection_close..) socketAF_INET SOCK_STREAMconnect get_http_portr streamrUr3r2setio_looprun_sync)r?sr6r6r7test_connection_closes   z)ConnectionCloseTest.test_connection_closecCstd|jdS)Nzhandler waiting)rrr@closer>r6r6r7r1 z&ConnectionCloseTest.on_handler_waitingcCstd|dS)Nzconnection closed)rrstopr>r6r6r7r4  z'ConnectionCloseTest.on_connection_closeN)rFrGrHr;rEr1r4r6r6r6r7r6s  r6c@rO) 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: %rr1 path_argsargs) request argumentstyper Exceptionr get_argumentsrrUrar1r)r?rLkeybvaluesvalueargr6r6r7rVs.      zEchoHandler.getNrXr6r6r6r7rJrYrJc@s<eZdZddZddZddZddZd d Zd d Zd S)RequestEncodingTestcCsdtfdtfgS)Nz /group/(.*)z/slashes/([^/]*)/([^/]*))rJr>r6r6r7r; z RequestEncodingTest.get_handlerscCst||jSr/)rrrrr6r6r7 fetch_jsonrXzRequestEncodingTest.fetch_jsoncCsF||dtddgid||dtddgddgiddS)Nz /group/%3F?rKz/group/%3F?%3F=%3FrurYrar>r6r6r7test_group_question_marksz,RequestEncodingTest.test_group_question_markcCs&||dddgddgiddS)Nz/group/%C3%A9?arg=%C3%A9z /group/%C3%A9érVrKrurYr>r6r6r7test_group_encodingsz'RequestEncodingTest.test_group_encodingcCsD||dtdddgid||dtdddgiddS)Nz/slashes/foo/barrlrrKz/slashes/a%2Fb/c%2Fdza/bzc/dr[r>r6r6r7 test_slashes'sz RequestEncodingTest.test_slashescC:ttd|dWddS1swYdS)Nz.*Invalid unicodez/group/?arg=%25%e9)rrrr>r6r6r7 test_error3  "zRequestEncodingTest.test_errorN) rFrGrHr;rYr\r_r`rbr6r6r6r7rW s  rWc@r5) 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)Nstatusargumentrl cookie_keyr cookie_valuerz%unexpected values for cookie keys: %rrvre xsrf_tokenxsrf_form_html reverse_url typecheckrequest_summary)errors check_type get_statusint get_argumentrlistcookieskeysrvaluesrfrQrvrrerirjrk_request_summaryr>r6r6r7prepare;s zTypeCheckHandler.preparecC|d|t||jdSNpath_componentrorrUrnr?r{r6r6r7rVVszTypeCheckHandler.getcCryrzr|r}r6r6r7r\szTypeCheckHandler.postcCs*t|}||krd||f|j|<dSdS)Nzexpected %s, got %s)rPrn)r?rdobj expected_type actual_typer6r6r7ro`szTypeCheckHandler.check_typeN)rFrGrHrxrVrror6r6r6r7rd:s  rdc@seZdZdddZddZdS)DecodeArgHandlerNcCsFt|tkrtdt|d|jjvr!|t|jjddS|S)Nzunexpected type for value: %rencodingr)rPrrQrNrOdecoder)r?rfrdr6r6r7decode_argumentgs  z DecodeArgHandler.decode_argumentcCs*dd}|||||dddS)NcSs:t|tkrdtt|gSt|tkrd|gStd)Nrrz unknown type)rPrrr~b2a_hexrrQ)rDr6r6r7describeqs  z&DecodeArgHandler.get..describerlr1query)rUrr)r?rVrr6r6r7rVps"zDecodeArgHandler.getr/)rFrGrHrrVr6r6r6r7rfs  rc@rO)LinkifyHandlercCs|jddddS)N linkify.htmlzhttp://example.com)messagerr>r6r6r7rV|zLinkifyHandler.getNrXr6r6r6r7r{rYrc@rO)UIModuleResourceHandlercCs|jdddgddS)N page.htmlrnr{)entriesrr>r6r6r7rVszUIModuleResourceHandler.getNrXr6r6r6r7rrYrc@rO)OptionalPathHandlercCs|d|idS)Nr1rTrr6r6r7rVrzOptionalPathHandler.getNrXr6r6r6r7rrYrc@rO)MultiHeaderHandlercCs4|dd|dd|dd|dddS) N x-overwriterz X-Overwriter{x-multirzX-Multi4) set_header add_headerr>r6r6r7rVs   zMultiHeaderHandler.getNrXr6r6r6r7rrYrc@rO)rcCsd|dddur|jdtt|dddS|dddur.|jdt|dddStd)N permanentrK)rre)rez(didn't get permanent or status arguments)rrredirectboolrqrQr>r6r6r7rVs  zRedirectHandler.getNrXr6r6r6r7rrYrc@eZdZejddZdS)EmptyFlushCallbackHandlerccsB|V|V|d|V|V|ddS)Nok)rrUrr>r6r6r7rVs     zEmptyFlushCallbackHandler.getN)rFrGrHrrrVr6r6r6r7rrc@rO)HeaderInjectionHandlerc CsVz |ddtdty*}zdt|vr|dnWYd}~dSd}~ww)NzX-Foozfoo X-Bar: bazzDidn't get expected exceptionzUnsafe header valueok)rrQ ValueErrorrrr?rr6r6r7rVs   zHeaderInjectionHandler.getNrXr6r6r6r7rrYrc@rO)GetArgumentHandlercCsJ|dddkr |j}n|dddkr|j}n|j}||dddS)Nsourcerrrlr)rrget_query_argumentget_body_argumentrr?rr6r6r7rxs zGetArgumentHandler.prepareNrFrGrHrxr6r6r6r7rrYrc@rO)GetArgumentsHandlercCs*|t|d|d|dddS)Nrlrrr)rrarRget_query_argumentsget_body_argumentsr>r6r6r7rxszGetArgumentsHandler.prepareNrr6r6r6r7rrYrcseZdZdZddZfddZddZdd 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='