U AcjRF@sddZddlmZddlmZz ddlZWnek r@dZYnXeZddlm 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__rrA/tmp/pip-unpacked-wheel-nrvc_22y/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'src@seZdZdZdS)ArbitraryErrorz% An exception for this test. N)rrrr%rrrrr&7sr&c@seZdZdZdZdZdZddZddZddZ dd Z d d Z d d Z ddZ ddZddZddZddZddZddZdS) DistribTestsNcsttg|jdk r>|jjdk r>|jjfddndd|jdk r|jjdk r|jjjfdd|jjjj ndd|j dk r |j |jdk rΈ |j tS)z Clean up all the event sources left behind by either directly by test methods or indirectly via some distrib API. NcsddSNrcallbackrdlrrJz'DistribTests.tearDown..rcsddS)Nr)rr+rrr-Nr.r/)rDeferredf1r"notifyOnDisconnectr*subZ publisherZbroker transportZloseConnectionport1append stopListeningport2Z 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)rResourceputChildrDatarSiterrResourcePublisherr1r listenTCPr5ResourceSubscriptiongetHostportr3rr8rAgentencoderequest addCallbackreadBody assertEqual)r#Zr1Zsite1Zr2f2agenturldrrr 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;)Z interface)rr>r?rrArrrBr1rrCZ addCleanupr7rErDhostrFr3) r#childZ distribRootZ distribSiteZdistribFactoryZ distribPortr$ZmainRootZmainSitemainPortmainAddrrrr_setupDistribServerls       z DistribTests._setupDistribServercKsX||\}}tt}d|jd|jd}|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. http://:/childr<r=) rVrrGrrRrFrHrIrJrK)r#rSkwargsrTrUrNrOrPrrr _requestTests    zDistribTests._requestTestcKsZ||\}}d|jd|jd}|d}ttjd|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. rWrXrYr<r=cs8tt}_|fdd|S)NcsfSr r)_protocolresponserrr-r.zGDistribTests._requestAgentTest..cbCollectBody..)rZAccumulatingProtocolZ deliverBodyrr0ZclosedDeferredrJ)r_rPrr]r cbCollectBodys  z5DistribTests._requestAgentTest..cbCollectBody) rVrRrFrHrrGrrIrJ)r#rSrZrTrUrOrPr`rrr_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)Nrr.)updatedictrequestHeadersZgetAllRawHeadersr#rIreqrdrrrenderszEDistribTests.test_requestHeaders..ReportRequestHeaders.renderNrrrrhrrfrrReportRequestHeaderssrjcsDddD}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)assertInrZremoveObserver)Zmsgs) logObserverrgr#rr check_logss   z4DistribTests.test_requestHeaders..check_logsZfoobar)headerscs$jjddgdS)NsFoosbar)r1r"r2rLresult)rqrdr#rr cbRequestedsz5DistribTests.test_requestHeaders..cbRequested) rEventLoggingObserverrZ addObserverrr>r[rrJ)r#rjrIrvr)rqrprgrdr#rtest_requestHeaderss   z DistribTests.test_requestHeaderscs8Gdddtj}|}fdd}|||S)z The response code can be set by the request object passed to a distributed resource's C{render} method. c@seZdZddZdS)z>DistribTests.test_requestResponseCode..SetResponseCodecSs|ddS)NZsetResponseCodererrrrhs zEDistribTests.test_requestResponseCode..SetResponseCode.renderNrirrrrSetResponseCodesr|cs:|djd|djd|djddS)Nrr.r/rysOKrLdatacodephrasertr9rrrvsz:DistribTests.test_requestResponseCode..cbRequestedrr>rarJr#r|rIrvrr9rtest_requestResponseCodes    z%DistribTests.test_requestResponseCodecs8Gdddtj}|}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@seZdZddZdS)zEDistribTests.test_requestResponseCodeMessage..SetResponseCodecSs|dddS)Nry some-messagerzr{rerrrrhs zLDistribTests.test_requestResponseCodeMessage..SetResponseCode.renderNrirrrrr|sr|cs:|djd|djd|djddS)Nrr.r/ryrr}rtr9rrrvszADistribTests.test_requestResponseCodeMessage..cbRequestedrrrr9rtest_requestResponseCodeMessages    z,DistribTests.test_requestResponseCodeMessagecCs8Gdddtj}||}||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@seZdZddZdS)z0DistribTests.test_largeWrite..LargeWritecSs |dtd|tjSNxy)writer finishrZ NOT_DONE_YETrerrrrh sz7DistribTests.test_largeWrite..LargeWrite.renderNrirrrr LargeWrite srrrrr>r[rJrLr )r#rrIrrrtest_largeWrites zDistribTests.test_largeWritecCs8Gdddtj}||}||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@seZdZddZdS)z2DistribTests.test_largeReturn..LargeReturncSs dtdSrr rerrrrhsz9DistribTests.test_largeReturn..LargeReturn.renderNrirrrr LargeReturnsrrrr)r#rrIrrrtest_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;r.c s\jdtj}t|dddddddd d d dg }d |gjdS) Nir/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 )rL responseCodeflushLoggedErrorsr Z NoSuchMethodlenjoinwritten)ignorederrorsexpectedrIr#rr cbRendered2s z4DistribTests.test_connectionLost..cbRendered)rr ZRootr1rrCr5rrDrErFr3rrrJ)r#Z serverFactoryZ serverPortZ subscriptionrPrrrrtest_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. r.r/zFailure instancerrkN)rrwcreateWithCleanuprr Failurer&rrZIssuefailed assertEqualsrro)r#rpfrIissuerrrtest_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. r/rZ log_failureN)rrwrrr&r rrRequestrZfailrrrZassertIs)r#rperrrrgrrrtest_requestFail]s    zDistribTests.test_requestFail)rrrr5r8r3r1r:rQrVr[rarxrrrrrrrrrrrr'=s"",r'c@s$eZdZddZddZddZdS)_PasswordDatabasecCs ||_dSr )_users)r#Zusersrrr__init__osz_PasswordDatabase.__init__cCs t|jSr )iterrr9rrrgetpwallrsz_PasswordDatabase.getpwallcCs*|jD]}|d|kr|SqtdSr()rKeyError)r#usernameuserrrrgetpwnamus   z_PasswordDatabase.getpwnamN)rrrrrrrrrrrnsrc@steZdZdZddZddZedddd Zd d Zd d Z ddZ ddZ ddZ ddZ ee 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,,,)mktemprrrZdatabaser UserDirectory directoryr9rrrsetUpszUserDirectoryTests.setUpcCs|ttj|jdS)zK L{UserDirectory} instances provide L{resource.IResource}. N) assertTruerrZ IResourcerr9rrrtest_interfacesz!UserDirectoryTests.test_interfaceN)namereturncs>t|g}|j||}t||}|IdH||jddS)zr Verify that requesting the C{name} child of C{self.directory} results in a 404 response. Ni)rrgetChildrrLr)r#rrIrurPrrr_404Tests    zUserDirectoryTests._404Testcs|dIdHdS)z L{UserDirectory.getChild} returns a resource which renders a 404 response when passed a string which does not correspond to any known user. scarolNrr9rrrtest_getInvalidUsersz&UserDirectoryTests.test_getInvalidUsercs|dIdHdS)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. saliceNrr9rrrtest_getUserWithoutResourcesz.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_htmlrsbobN) r FilePathrrSmakedirsrrrassertIsInstancerZFilerLpath)r#homerrIrurrrtest_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-pbrs bob.twistdunixN)r rrrrSrrrrrrDrLrRrrFr)r#rwebrIrurrrtest_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. rzPOSTN)rmethod assertRaisesrZUnsupportedMethodrrhrerrrtest_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. rrrr.rzcstdj}|d\}}|jjd|jdd|jjjd|jjd|jdd|jjjddS) Nr.liahrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) rrrZgetElementsByTagNamerLZ firstChildZtagName getAttributer~)rdocumentrrrrrrsz2UserDirectoryTests.test_render..cbRendered) r rrrSrrZ setContentrrrrJ)r#rrrurrrr 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)rrZassertIdentical_pwdpwd)r#rrrrtest_passwordDatabasesz(UserDirectoryTests.test_passwordDatabase)rrrr%rrbytesrrrrrrrrrrrrrrr|s   r)1r%os.pathrZxml.dom.minidomrrr ImportErrorZunittestrZzope.interface.verifyrZtwisted.internetrrZtwisted.loggerrZtwisted.pythonr r Ztwisted.spreadr Ztwisted.spread.bananar Z twisted.testrZtwisted.trial.unittestrZ twisted.webrrrrrZtwisted.web.http_headersrZtwisted.web.test._utilrZtwisted.web.test.requesthelperrrrArr Exceptionr&r'rrrrrrs8             3