U  W[pGã@sXdZddlmZddlmZz ddlZWnek r@dZYnXddlmZddl m Z m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZmZmZmZmZdd lmZmZdd lmZddl m!Z!ddl"m#Z#ddl$m%Z%Gdd„dej&ƒZ'Gdd„dej(ƒZ(Gdd„de)ƒZ*Gdd„dej+ƒZ,Gdd„dƒZ-Gdd„dej+ƒZ.dS)z# Tests for L{twisted.web.distrib}. é)Úabspath)Ú parseStringN)Ú verifyObject)ÚfilepathÚfailure)ÚreactorÚdefer)Úunittest)Úpb©Ú SIZE_LIMIT)ÚdistribÚclientÚresourceÚstaticÚserver)Ú DummyRequestÚ DummyChannel)Ú_render)Ú proto_helpers)ÚHeaders)ÚglobalLogPublisherc@s eZdZdS)ÚMySiteN)Ú__name__Ú __module__Ú __qualname__©rrú?/usr/lib/python3/dist-packages/twisted/web/test/test_distrib.pyrsrc@seZdZdZdZdd„ZdS)ÚPBServerFactoryzÅ A PB server factory which keeps track of the most recent protocol it created. @ivar proto: L{None} or the L{Broker} instance most recently returned from C{buildProtocol}. NcCstj ||¡|_|jS©N)r rÚ buildProtocolÚproto)ÚselfÚaddrrrrr ,szPBServerFactory.buildProtocol)rrrÚ__doc__r!r rrrrr"src@seZdZdZdS)ÚArbitraryErrorz% An exception for this test. N)rrrr$rrrrr%2sr%c@s„eZdZdZdZdZdZdd„Zdd„Zdd„Z dd „Z d d „Z d d „Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS)Ú DistribTestsNcsØt ¡t ¡g‰|jdk r>|jjdk r>|jj ‡fdd„¡nˆd d¡|jdk rŒ|jjdk rŒ|jjj ‡fdd„¡|jjjj   ¡nˆd d¡|j dk r´ˆ  |j   ¡¡|jdk rΈ  |j  ¡¡t ˆ¡S)z‹ Clean up all the event sources left behind by either directly by test methods or indirectly via some distrib API. Ncsˆd d¡S©Nr©Úcallbackr©ZdlrrÚFóz'DistribTests.tearDown..rcsˆd d¡S)Nér(rr*rrr+Kr,r-)rÚDeferredÚf1r!ÚnotifyOnDisconnectr)ÚsubZ publisherZbrokerZ transportZloseConnectionÚport1ÚappendÚ stopListeningÚport2Z gatherResults©r"rr*rÚtearDown?s  ÿ  zDistribTests.tearDowncCsØt ¡}| dt dd¡¡t |¡}tt  |¡ƒ|_ t   d|j ¡|_ t d|j  ¡j¡|_t ¡}| d|j¡t|ƒ}t   d|¡|_t t ¡}d |j ¡j¡}| d¡}| d |¡}| tj¡| |jd¡|S) Nstheresrootz text/plainrú 127.0.0.1sherezhttp://127.0.0.1:{}/here/thereÚasciióGET)rÚResourceÚputChildrZDatarÚSiterr ÚResourcePublisherr/rÚ listenTCPr2ÚResourceSubscriptionÚgetHostÚportr1rr5rÚAgentÚformatÚencodeÚrequestÚ addCallbackÚreadBodyÚ assertEqual)r"Zr1Zsite1Zr2Úf2ÚagentÚurlÚdrrrÚ testDistribVs*  ÿ  ÿ   zDistribTests.testDistribc Cs¢t ¡}| d|¡t |¡}tt |¡ƒ|_}t j d|dd}|  |j ¡|  ¡}t |j|j¡|_}t |¡}t j d|dd} |  | j ¡|   ¡} | | fS)zû Set up a resource on a distrib site using L{ResourcePublisher}. @param child: The resource to publish using distrib. @return: A tuple consisting of the host and port on which to contact the created site. schildrr8)Z interface)rr;r<rr=rr r>r/rr?Z addCleanupr4rAr@ÚhostrBr1) r"ÚchildZ distribRootZ distribSiteZdistribFactoryZ distribPortr#ZmainRootZmainSiteÚmainPortÚmainAddrrrrÚ_setupDistribServerms,   ÿ ÿ ÿ   z DistribTests._setupDistribServercKsR| |¡\}}t t¡}d|j|jf}| d¡}|jd|f|Ž}| tj ¡|S)a¡ Set up a resource on a distrib site using L{ResourcePublisher} and then retrieve it from a L{ResourceSubscription} via an HTTP client. @param child: The resource to publish using distrib. @param **kwargs: Extra keyword arguments to pass to L{Agent.request} when requesting the resource. @return: A L{Deferred} which fires with the result of the request. zhttp://%s:%s/childr9r:) rSrrCrrOrBrErFrGrH)r"rPÚkwargsrQrRrKrLrMrrrÚ _requestTestŠs    zDistribTests._requestTestcKsT| |¡\}}d |j|j¡}| d¡}t t¡jd|f|Ž}dd„}|  |¡|S)a; Set up a resource on a distrib site using L{ResourcePublisher} and then retrieve it from a L{ResourceSubscription} via an HTTP client. @param child: The resource to publish using distrib. @param **kwargs: Extra keyword arguments to pass to L{Agent.request} when requesting the resource. @return: A L{Deferred} which fires with a tuple consisting of a L{twisted.test.proto_helpers.AccumulatingProtocol} containing the body of the response and an L{IResponse} with the response itself. zhttp://{}:{}/childr9r:cs8t ¡‰ˆ ˆ¡t ¡}ˆ_| ‡‡fdd„¡|S)NcsˆˆfSrr)Ú_©ZprotocolÚresponserrr+µr,zGDistribTests._requestAgentTest..cbCollectBody..)rZAccumulatingProtocolZ deliverBodyrr.ZclosedDeferredrG)rXrMrrWrÚ cbCollectBody±s  z5DistribTests._requestAgentTest..cbCollectBody) rSrDrOrBrErrCrrFrG)r"rPrTrQrRrLrMrYrrrÚ_requestAgentTestžs   zDistribTests._requestAgentTestcs|i‰t ¡‰t ˆ¡dg‰G‡‡fdd„dtjƒ}‡‡‡fdd„‰ˆj|ƒtddgiƒd}‡‡‡fd d „}| |¡|S) z† The request headers are available on the request object passed to a distributed resource's C{render} method. NcseZdZ‡‡fdd„ZdS)z>DistribTests.test_requestHeaders..ReportRequestHeaderscs |ˆd<ˆ t|j ¡ƒ¡dS)Nrr,)ÚupdateÚdictÚrequestHeadersZgetAllRawHeaders©r"rF©Úreqr]rrÚrenderÇs ÿzEDistribTests.test_requestHeaders..ReportRequestHeaders.renderN©rrrrarr_rrÚReportRequestHeadersÆsrccsDdd„ˆDƒ}ˆ d|¡ˆ d|¡ˆ ˆd|¡t ˆ¡dS)NcSsg|] }|d‘qS)Ú log_formatr)Ú.0ÚerrrÚ ÎszHDistribTests.test_requestHeaders..check_logs..zconnected to publisherz3could not connect to distributed web service: {msg}r)ÚassertInrZremoveObserver)Zmsgs)Ú logObserverr`r"rrÚ check_logsÍs þz4DistribTests.test_requestHeaders..check_logsZfooZbar)Zheaderscs$ˆjj ˆ¡ˆ ˆddg¡dS)NsFoosbar)r/r!r0rI©Úresult)rjr]r"rrÚ cbRequestedÙsz5DistribTests.test_requestHeaders..cbRequested) rÚEventLoggingObserverrZ addObserverrr;rUrrG)r"rcrFrmr)rjrir`r]r"rÚtest_requestHeaders»s   ÿ z DistribTests.test_requestHeaderscs8Gdd„dtjƒ}ˆ |ƒ¡}‡fdd„}| |¡|S)z The response code can be set by the request object passed to a distributed resource's C{render} method. c@seZdZdd„ZdS)z>DistribTests.test_requestResponseCode..SetResponseCodecSs| d¡dS)NéÈÚ©ZsetResponseCoder^rrrraçs zEDistribTests.test_requestResponseCode..SetResponseCode.renderNrbrrrrÚSetResponseCodeæsrscs:ˆ |djd¡ˆ |djd¡ˆ |djd¡dS)Nrr,r-rpsOK©rIÚdataÚcodeÚphraserkr6rrrmìsz:DistribTests.test_requestResponseCode..cbRequested©rr;rZrG©r"rsrFrmrr6rÚtest_requestResponseCodeás    z%DistribTests.test_requestResponseCodecs8Gdd„dtjƒ}ˆ |ƒ¡}‡fdd„}| |¡|S)z The response code and message can be set by the request object passed to a distributed resource's C{render} method. c@seZdZdd„ZdS)zEDistribTests.test_requestResponseCodeMessage..SetResponseCodecSs| dd¡dS)Nrpó some-messagerqrrr^rrrraús zLDistribTests.test_requestResponseCodeMessage..SetResponseCode.renderNrbrrrrrsùsrscs:ˆ |djd¡ˆ |djd¡ˆ |djd¡dS)Nrr,r-rpr{rtrkr6rrrmÿszADistribTests.test_requestResponseCodeMessage..cbRequestedrxryrr6rÚtest_requestResponseCodeMessageôs    z,DistribTests.test_requestResponseCodeMessagecCs8Gdd„dtjƒ}| |ƒ¡}| |jdtd¡|S)zÞ If a string longer than the Banana size limit is passed to the L{distrib.Request} passed to the remote resource, it is broken into smaller strings to be transported over the PB connection. c@seZdZdd„ZdS)z0DistribTests.test_largeWrite..LargeWritecSs | dtd¡| ¡tjS©Nóxóy)Úwriter ZfinishrZ NOT_DONE_YETr^rrrrasz7DistribTests.test_largeWrite..LargeWrite.renderNrbrrrrÚ LargeWrite srr~r©rr;rUrGrIr )r"rrFrrrÚtest_largeWrites zDistribTests.test_largeWritecCs8Gdd„dtjƒ}| |ƒ¡}| |jdtd¡|S)z  Like L{test_largeWrite}, but for the case where C{render} returns a long string rather than explicitly passing it to L{Request.write}. c@seZdZdd„ZdS)z2DistribTests.test_largeReturn..LargeReturncSs dtdSr}r r^rrrrasz9DistribTests.test_largeReturn..LargeReturn.renderNrbrrrrÚ LargeReturnsr„r~rr‚)r"r„rFrrrÚtest_largeReturns zDistribTests.test_largeReturncsltt ¡ƒˆ_}t d|¡ˆ_}t d|  ¡j ¡ˆ_ }t dgƒ‰t |ˆƒ}‡‡fdd„}| |¡|S)zz If there is an error issuing the request to the remote publisher, an error response is returned. rr8r,c s\ˆ ˆjd¡ˆ tj¡}ˆ t|ƒd¡ddddddd d d dg }ˆ d  |¡gˆj¡dS) Niôr-r,ss: 500 - Server Connection Losts s#

Server Connection Lost

s¦

Connection to distributed server lost:

[Failure instance: Traceback from remote host -- twisted.spread.flavors.NoSuchMethod: No such method: remote_requests]

s só )rIÚ responseCodeÚflushLoggedErrorsr Z NoSuchMethodÚlenÚjoinÚwritten)ÚignoredÚerrorsZexpected©rFr"rrÚ cbRendered4s òz4DistribTests.test_connectionLost..cbRendered)rr ZRootr/rr?r2r r@rArBr1rrrG)r"Z serverFactoryZ serverPortZ subscriptionrMrrrŽrÚtest_connectionLost&sÿ    z DistribTests.test_connectionLostcCs`tj |t¡}t tƒ¡}tdgƒ}t  |¡}|  |¡|  dt |ƒ¡|  d|dd¡dS)zQ When a request fails, the string form of the failure is logged. r,r-zFailure instancerrdN)rrnÚcreateWithCleanuprrÚFailurer%rr ZIssueZfailedÚ assertEqualsr‰rh)r"riÚfrFZissuerrrÚtest_logFailedQsþ     þzDistribTests.test_logFailedcCsftj |t¡}tƒ}t |¡}t t ƒ¡}|  |¡|  t¡|  dt |ƒ¡| |dd|¡dS)zf When L{twisted.web.distrib.Request}'s fail is called, the failure is logged. r-rZ log_failureN)rrnr‘rr%rr’r ZRequestrZfailrˆr“r‰ZassertIs)r"riÚerrr”r`rrrÚtest_requestFailesþ    zDistribTests.test_requestFail)rrrr2r5r1r/r7rNrSrUrZrorzr|rƒr…rr•r—rrrrr&9s"&+r&c@s$eZdZdd„Zdd„Zdd„ZdS)Ú_PasswordDatabasecCs ||_dSr)Ú_users)r"ZusersrrrÚ__init__ysz_PasswordDatabase.__init__cCs t|jƒSr)Úiterr™r6rrrÚgetpwall}sz_PasswordDatabase.getpwallcCs*|jD]}|d|kr|Sqtƒ‚dSr')r™ÚKeyError)r"ZusernameÚuserrrrÚgetpwnams   z_PasswordDatabase.getpwnamN)rrrršrœrŸrrrrr˜xsr˜c@sneZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z edkrjde _dS)ÚUserDirectoryTestszj Tests for L{UserDirectory}, a resource for listing all user resources available on a system. cCsTddddd| ¡df|_dddd d | ¡df|_t|j|jgƒ|_t |j¡|_dS) NÚaliceÚxé{iÈzAlice,,,z/bin/shÚbobéêi7zBob,,,)Zmktempr¡r¤r˜Zdatabaser Ú UserDirectoryÚ directoryr6rrrÚsetUpŽszUserDirectoryTests.setUpcCs| ttj|jƒ¡dS)zK L{UserDirectory} instances provide L{resource.IResource}. N)Z assertTruerrZ IResourcer§r6rrrÚtest_interface•sz!UserDirectoryTests.test_interfacecs>t|gƒ‰ˆj |ˆ¡}t|ˆƒ}‡‡fdd„}| |¡|S)zr Verify that requesting the C{name} child of C{self.directory} results in a 404 response. csˆ ˆjd¡dS)Ni”)rIr‡)rŒrŽrrr¤sz/UserDirectoryTests._404Test..cbRendered)rr§ÚgetChildrrG)r"ÚnamerlrMrrrŽrÚ_404Testœs    zUserDirectoryTests._404TestcCs | d¡S)z­ L{UserDirectory.getChild} returns a resource which renders a 404 response when passed a string which does not correspond to any known user. Zcarol©r¬r6rrrÚtest_getInvalidUserªsz&UserDirectoryTests.test_getInvalidUsercCs | d¡S)zÞ L{UserDirectory.getChild} returns a resource which renders a 404 response when passed a string which corresponds to a known user who has neither a user directory nor a user distrib socket. r¡r­r6rrrÚtest_getUserWithoutResource³sz.UserDirectoryTests.test_getUserWithoutResourcecCs\t |jd¡}| d¡}| ¡tdgƒ}|j d|¡}| |t j ¡|  |j |j ¡dS)z¹ L{UserDirectory.getChild} returns a L{static.File} instance when passed the name of a user with a home directory containing a I{public_html} directory. éþÿÿÿÚ public_htmlr¤N) rÚFilePathr¤rPÚmakedirsrr§rªÚassertIsInstancerZFilerIÚpath)r"Úhomer±rFrlrrrÚtest_getPublicHTMLChild¼s  z*UserDirectoryTests.test_getPublicHTMLChildcCsnt |jd¡}| ¡| d¡}tdgƒ}|j d|¡}| |t j ¡|  |j d¡|  t |jƒ|j¡dS)zà L{UserDirectory.getChild} returns a L{ResourceSubscription} instance when passed the name of a user suffixed with C{".twistd"} who has a home directory containing a I{.twistd-web-pb} socket. r°ú.twistd-web-pbr¤z bob.twistdZunixN)rr²r¤r³rPrr§rªr´r r@rIrOrrBrµ)r"r¶ÚwebrFrlrrrÚtest_getDistribChildËs  z'UserDirectoryTests.test_getDistribChildcCs(tdgƒ}d|_| tj|jj|¡dS)zr L{UserDirectory.render} raises L{UnsupportedMethod} in response to a non-I{GET} request. rqZPOSTN)rÚmethodZ assertRaisesrZUnsupportedMethodr§rar^rrrÚtest_invalidMethodÛs ÿz%UserDirectoryTests.test_invalidMethodcsxt ˆjd¡ d¡}| ¡t ˆjd¡}| ¡| d¡ d¡tdgƒ‰tˆj ˆƒ}‡‡fdd„}|  |¡|S)z} L{UserDirectory} renders a list of links to available user content in response to a I{GET} request. r°r±r¸r,rqcsŽtd ˆj¡ƒ}| d¡\}}ˆ |jjd¡ˆ |j d¡d¡ˆ |jjjd¡ˆ |jjd¡ˆ |j d¡d¡ˆ |jjjd¡dS) Nr,ZliÚaZhrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) rrŠr‹ZgetElementsByTagNamerIZ firstChildZtagNameZ getAttributeru)rŒZdocumentr¡r¤rŽrrrösz2UserDirectoryTests.test_render..cbRendered) rr²r¡rPr³r¤Z setContentrrr§rG)r"r±r¹rlrrrŽrÚ test_renderæs   zUserDirectoryTests.test_rendercCst ¡}| |jt¡dS)z„ If L{UserDirectory} is instantiated with no arguments, it uses the L{pwd} module as its password database. N)r r¦ZassertIdenticalZ_pwdÚpwd)r"r§rrrÚtest_passwordDatabasesz(UserDirectoryTests.test_passwordDatabaseNzpwd module required)rrrr$r¨r©r¬r®r¯r·rºr¼r¾rÀr¿Úskiprrrrr ‰s    r )/r$Úos.pathrZxml.dom.minidomrr¿Ú ImportErrorZzope.interface.verifyrZtwisted.pythonrrZtwisted.internetrrZ twisted.trialr Ztwisted.spreadr Ztwisted.spread.bananar Z twisted.webr rrrrZtwisted.web.test.test_webrrZtwisted.web.test._utilrZ twisted.testrZtwisted.web.http_headersrZtwisted.loggerrr=rrÚ Exceptionr%ZTestCaser&r˜r rrrrÚs4            A