o btF@sbdZddlmZddlmZzddlZWn eydZYnweZddlm Z ddl m Z ddl m Z mZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlmZmZmZm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(Gddde j)Z*Gdddej+Z+Gddde,Z-GdddeZ.GdddZ/GdddeZ0dS)z# Tests for L{twisted.web.distrib}. )abspath) parseStringN)skipIf) verifyObject)deferreactor)globalLogPublisher)failurefilepath)pb SIZE_LIMIT) proto_helpers)TestCase)clientdistribresourceserverstatic)Headers)_render) DummyChannel DummyRequestc@s eZdZdS)MySiteN)__name__ __module__ __qualname__rr?/usr/lib/python3/dist-packages/twisted/web/test/test_distrib.pyr#src@seZdZdZdZddZdS)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|||_|jSN)r r buildProtocolproto)selfaddrrrrr!2szPBServerFactory.buildProtocol)rrr__doc__r"r!rrrrr's rc@seZdZdZdS)ArbitraryErrorz% An exception for this test. N)rrrr%rrrrr&7sr&c@seZdZdZdZdZdZddZddZddZ dd Z d d Z d d Z ddZ ddZddZddZddZddZddZdS) DistribTestsNcsttg|jdur|jjdur|jjfddndd|jdurF|jjdurF|jjjfdd|jjjj ndd|j durZ |j |jdurg |j tS)z Clean up all the event sources left behind by either directly by test methods or indirectly via some distrib API. NcddSNrcallbackrdlrrJz'DistribTests.tearDown..rcr()Nr*rr,rrr.Nr/r0)rDeferredf1r"notifyOnDisconnectr+sub publisherbroker transportloseConnectionport1append stopListeningport2 gatherResultsr#rr,rtearDownCs   zDistribTests.tearDowncCst}|dtddt|}tt ||_ t d|j |_ td|j j|_t}|d|jt|}t d||_tt }d|jjd}|d }|d |}|tj||jd|S) Nstheresrootz text/plainr 127.0.0.1sherezhttp://127.0.0.1:z /here/thereasciiGET)rResourceputChildrDatarSiterrResourcePublisherr2r listenTCPr9ResourceSubscriptiongetHostportr4rr<rAgentencoderequest addCallbackreadBody assertEqual)r#r1site1r2f2agenturldrrr testDistribXs"     zDistribTests.testDistribc Cst}|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. schildrr@) interface)rrCrDrrFrrrGr2rrH addCleanupr;rJrIhostrKr4) r#child distribRoot distribSitedistribFactory distribPortr$mainRootmainSitemainPortmainAddrrrr_setupDistribServerls      z DistribTests._setupDistribServercKs\||\}}tt}d|jd|jd}|d}|jd|fi|}|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. http://:/childrArB) rfrrLrr\rKrMrNrOrP)r#r]kwargsrdrerVrWrXrrr _requestTests   zDistribTests._requestTestcKs^||\}}d|jd|jd}|d}ttjd|fi|}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. rgrhrirArBcs8tt}_|fdd|S)NcsfSr r)_protocolresponserrr.szGDistribTests._requestAgentTest..cbCollectBody..)rAccumulatingProtocol deliverBodyrr1closedDeferredrO)rorXrrmr cbCollectBodys  z5DistribTests._requestAgentTest..cbCollectBody) rfr\rKrMrrLrrNrO)r#r]rjrdrerWrXrsrrr_requestAgentTests   zDistribTests._requestAgentTestcs|ittdgGfdddtj}fddj|tddgid}fd d }|||S) z The request headers are available on the request object passed to a distributed resource's C{render} method. NcseZdZfddZdS)z>DistribTests.test_requestHeaders..ReportRequestHeaderscs |d<t|jdS)Nr)updatedictrequestHeadersgetAllRawHeadersr#rNreqrxrrrenderszEDistribTests.test_requestHeaders..ReportRequestHeaders.renderNrrrr}rr{rrReportRequestHeaderssrcsDddD}d|d|d|tdS)NcSsg|]}|dqS) log_formatr).0errr szHDistribTests.test_requestHeaders..check_logs..zconnected to publisherz3could not connect to distributed web service: {msg}r)assertInrremoveObserver)msgs) logObserverr|r#rr check_logss   z4DistribTests.test_requestHeaders..check_logsfoobar)headerscs$jjddgdS)NsFoosbar)r2r"r3rQresult)rrxr#rr cbRequestedsz5DistribTests.test_requestHeaders..cbRequested) rEventLoggingObserverr addObserverrrCrkrrO)r#rrNrr)rrr|rxr#rtest_requestHeaderss  z DistribTests.test_requestHeadersc8Gdddtj}|}fdd}|||S)z The response code can be set by the request object passed to a distributed resource's C{render} method. c@eZdZddZdS)z>DistribTests.test_requestResponseCode..SetResponseCodecSs|ddS)NsetResponseCoderzrrrr}s zEDistribTests.test_requestResponseCode..SetResponseCode.renderNr~rrrrSetResponseCode rc:|djd|djd|djddS)Nrrur0rsOKrQdatacodephraserr>rrrz:DistribTests.test_requestResponseCode..cbRequestedrrCrtrOr#rrNrrr>rtest_requestResponseCode    z%DistribTests.test_requestResponseCodecr)z The response code and message can be set by the request object passed to a distributed resource's C{render} method. c@r)zEDistribTests.test_requestResponseCodeMessage..SetResponseCodecSs|dddS)Nr some-messagerrrzrrrr}s zLDistribTests.test_requestResponseCodeMessage..SetResponseCode.renderNr~rrrrrrrcr)Nrrur0rrrrr>rrrrzADistribTests.test_requestResponseCodeMessage..cbRequestedrrrr>rtest_requestResponseCodeMessagerz,DistribTests.test_requestResponseCodeMessagecC8Gdddtj}||}||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@r)z0DistribTests.test_largeWrite..LargeWritecSs |dtd|tjSNxy)writer finishr NOT_DONE_YETrzrrrr} sz7DistribTests.test_largeWrite..LargeWrite.renderNr~rrrr LargeWrite rrrrrrCrkrOrQr )r#rrNrrrtest_largeWrites zDistribTests.test_largeWritecCr)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@r)z2DistribTests.test_largeReturn..LargeReturncSs dtdSrr rzrrrr}s z9DistribTests.test_largeReturn..LargeReturn.renderNr~rrrr LargeReturnrrrrr)r#rrNrrrtest_largeReturns zDistribTests.test_largeReturncsltt_}td|_}td| j _ }t dgt |}fdd}|||S)zz If there is an error issuing the request to the remote publisher, an error response is returned. rr@rucsLjdtj}t|dgd}d|gjdS)Nir0) russ: 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 sru )rQ responseCodeflushLoggedErrorsr NoSuchMethodlenjoinwritten)ignorederrorsexpectedrNr#rr cbRendered2s  z4DistribTests.test_connectionLost..cbRendered)rr Rootr2rrHr9rrIrJrKr4rrrO)r# serverFactory serverPort subscriptionrXrrrrtest_connectionLost"s    z DistribTests.test_connectionLostcCs`tj|t}tt}tdg}t |}| || dt || d|dddS)zQ When a request fails, the string form of the failure is logged. rur0zFailure instancerrN)rrcreateWithCleanuprr Failurer&rrIssuefailed assertEqualsrr)r#rfrNissuerrrtest_logFailedNs    zDistribTests.test_logFailedcCsftj|t}t}t|}tt }| || t| dt |||dd|dS)zf When L{twisted.web.distrib.Request}'s fail is called, the failure is logged. r0r log_failureN)rrrrr&r rrRequestrfailrrrassertIs)r#rerrrr|rrrtest_requestFail]s    zDistribTests.test_requestFail)rrrr9r<r4r2r?rYrfrkrtrrrrrrrrrrrrr'=s$", r'c@s$eZdZddZddZddZdS)_PasswordDatabasecCs ||_dSr )_users)r#usersrrr__init__o z_PasswordDatabase.__init__cCs t|jSr )iterrr>rrrgetpwallrrz_PasswordDatabase.getpwallcCs&|jD] }|d|kr|Sqtr))rKeyError)r#usernameuserrrrgetpwnamus  z_PasswordDatabase.getpwnamN)rrrrrrrrrrrns rc@sleZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ e e dddZdS)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) Nalicex{izAlice,,,z/bin/shbobi7zBob,,,)mktemprrrdatabaser UserDirectory directoryr>rrrsetUpszUserDirectoryTests.setUpcCs|ttj|jdS)zK L{UserDirectory} instances provide L{resource.IResource}. N) assertTruerr IResourcerr>rrrtest_interfacesz!UserDirectoryTests.test_interfacecs>t|gj|}t|}fdd}|||S)zr Verify that requesting the C{name} child of C{self.directory} results in a 404 response. csjddS)Ni)rQr)rrrrrsz/UserDirectoryTests._404Test..cbRendered)rrgetChildrrO)r#namerrXrrrr_404Tests   zUserDirectoryTests._404TestcC |dS)z L{UserDirectory.getChild} returns a resource which renders a 404 response when passed a string which does not correspond to any known user. carolrr>rrrtest_getInvalidUser z&UserDirectoryTests.test_getInvalidUsercCr)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. rrr>rrrtest_getUserWithoutResourcerz.UserDirectoryTests.test_getUserWithoutResourcecCs\t|jd}|d}|tdg}|jd|}||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_htmlrN) r FilePathrr]makedirsrrrassertIsInstancerFilerQpath)r#homerrNrrrrtest_getPublicHTMLChilds  z*UserDirectoryTests.test_getPublicHTMLChildcCsnt|jd}||d}tdg}|jd|}||t j | |j d| t |j|jdS)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-pbrz bob.twistdunixN)r rrrr]rrrrrrIrQr\rrKr)r#rwebrNrrrrtest_getDistribChilds  z'UserDirectoryTests.test_getDistribChildcCs(tdg}d|_|tj|jj|dS)zr L{UserDirectory.render} raises L{UnsupportedMethod} in response to a non-I{GET} request. rPOSTN)rmethod assertRaisesrUnsupportedMethodrr}rzrrrtest_invalidMethods z%UserDirectoryTests.test_invalidMethodcsxtjdd}|tjd}||ddtdgtj }fdd}| ||S)z} L{UserDirectory} renders a list of links to available user content in response to a I{GET} request. rrrrurcstdj}|d\}}|jjd|jdd|jjjd|jjd|jdd|jjjddS) Nruliahrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) rrrgetElementsByTagNamerQ firstChildtagName getAttributer)rdocumentrrrrrrsz2UserDirectoryTests.test_render..cbRendered) r rrr]rr setContentrrrrO)r#rrrrrrr test_renders   zUserDirectoryTests.test_renderzpwd module requiredcCst}||jtdS)z If L{UserDirectory} is instantiated with no arguments, it uses the L{pwd} module as its password database. N)rrassertIdentical_pwdpwd)r#rrrrtest_passwordDatabasesz(UserDirectoryTests.test_passwordDatabaseN)rrrr%rrrrrrrrrrrrrrrrr|s r)1r%os.pathrxml.dom.minidomrrr ImportErrorunittestrzope.interface.verifyrtwisted.internetrrtwisted.loggerrtwisted.pythonr r twisted.spreadr twisted.spread.bananar twisted.testrtwisted.trial.unittestr twisted.webrrrrrtwisted.web.http_headersrtwisted.web.test._utilrtwisted.web.test.test_webrrrFrr Exceptionr&r'rrrrrrs<             3