o iRF@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__rrcC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-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+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)rResourceputChildrZDatarSiterrResourcePublisherr2r listenTCPr6ResourceSubscriptiongetHostportr4rr9rAgentencoderequest 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 interfaceN)rr?r@rrArrrBr2rrCZ addCleanupr8rErDhostrFr4) r#childZ distribRootZ distribSiteZdistribFactoryZ distribPortr$ZmainRootZmainSitemainPortmainAddrrrr_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://:/childr=r>N) rVrrGrrRrFrHrIrJrK)r#rSkwargsrTrUrNrOrPrrr _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. rWrXrYr=r>cs8tt}_|fdd|S)NcsfSr r)_protocolresponserrr.szGDistribTests._requestAgentTest..cbCollectBody..)rZAccumulatingProtocolZ deliverBodyrr1ZclosedDeferredrJ)r_rPrr]r cbCollectBodys  z5DistribTests._requestAgentTest..cbCollectBodyN) 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)Nr)updatedictrequestHeadersZgetAllRawHeadersr#rIreqrerrrenderszEDistribTests.test_requestHeaders..ReportRequestHeaders.renderNrrrrirrgrrReportRequestHeaderssrkcsDddD}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) logObserverrhr#rr check_logss   z4DistribTests.test_requestHeaders..check_logsZfoobar)headerscs$jjddgdS)NsFoosbar)r2r"r3rLresult)rrrer#rr cbRequestedsz5DistribTests.test_requestHeaders..cbRequested) rEventLoggingObserverrZ addObserverrr?r[rrJ)r#rkrIrwr)rrrqrhrer#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)NZsetResponseCoderfrrrris zEDistribTests.test_requestResponseCode..SetResponseCode.renderNrjrrrrSetResponseCode rc:|djd|djd|djddS)Nrrbr0r|sOKrLdatacodephraserur:rrrwz:DistribTests.test_requestResponseCode..cbRequestedNrr?rarJr#rrIrwrr:rtest_requestResponseCode    z%DistribTests.test_requestResponseCodecrz)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-messager}r~rfrrrris zLDistribTests.test_requestResponseCodeMessage..SetResponseCode.renderNrjrrrrrrrcr)Nrrbr0r|rrrur:rrrwrzADistribTests.test_requestResponseCodeMessage..cbRequestedNrrrr: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 finishrZ NOT_DONE_YETrfrrrri sz7DistribTests.test_largeWrite..LargeWrite.renderNrjrrrr LargeWrite rrrrNrr?r[rJrLr )r#rrIrrrtest_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 rfrrrris z9DistribTests.test_largeReturn..LargeReturn.renderNrjrrrr LargeReturnrrrrNr)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<rbcsLjdtj}t|dgd}d|gjdS)Nir0) rbss: 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 srb )rL responseCodeflushLoggedErrorsr Z NoSuchMethodlenjoinwritten)ignorederrorsexpectedrIr#rr cbRendered2s  z4DistribTests.test_connectionLost..cbRenderedN)rr ZRootr2rrCr6rrDrErFr4rrrJ)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. rbr0zFailure instancerrlN)rrxcreateWithCleanuprr Failurer&rrZIssuefailed assertEqualsrrp)r#rqfrIissuerrrtest_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. r0rZ log_failureN)rrxrrr&r rrRequestrZfailrrrZassertIs)r#rqerrrrhrrrtest_requestFail]s    zDistribTests.test_requestFail)rrrr6r9r4r2r;rQrVr[raryrrrrrrrrrrrr'=s$", r'c@s$eZdZddZddZddZdS)_PasswordDatabasecCs ||_dSr )_users)r#Zusersrrr__init__o z_PasswordDatabase.__init__cCs t|jSr )iterrr:rrrgetpwallrrz_PasswordDatabase.getpwallcCs&|jD] }|d|kr|Sqtr))rKeyError)r#usernameuserrrrgetpwnamus  z_PasswordDatabase.getpwnamN)rrrrrrrrrrrns rc@sveZdZdZddZddZdeddfd d Zd d Zd dZ 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 directoryr:rrrsetUpszUserDirectoryTests.setUpcCs|ttj|jdS)zK L{UserDirectory} instances provide L{resource.IResource}. N) assertTruerrZ IResourcerr:rrrtest_interfacesz!UserDirectoryTests.test_interfacenamereturnNcs@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#rrIrvrPrrr_404Tests    zUserDirectoryTests._404Testc|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. scarolNrr:rrrtest_getInvalidUserz&UserDirectoryTests.test_getInvalidUsercr)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. saliceNrr: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_htmlrsbobN) r FilePathrrSmakedirsrrrassertIsInstancerFilerLpath)r#homerrIrvrrrtest_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#rwebrIrvrrrtest_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. r}POSTN)rmethod assertRaisesrZUnsupportedMethodrrirfrrrtest_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. rrrrbr}cstdj}|d\}}|jjd|jdd|jjjd|jjd|jdd|jjjddS) Nrbliahrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) rrrZgetElementsByTagNamerLZ firstChildZtagName getAttributer)rdocumentrrrrrrsz2UserDirectoryTests.test_render..cbRenderedN) r rrrSrrZ setContentrrrrJ)r#rrrvrrrr 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'rrrrrrs<             3