tfRFdZddlmZddlmZ ddlZeZn #e$rdZYnwxYwddlm 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(Gdde j)Z*Gddej+Z+Gdde,Z-GddeZ.GddZ/GddeZ0dS)z# Tests for L{twisted.web.distrib}. )abspath) parseStringN)skipIf) verifyObject)deferreactor)globalLogPublisher)failurefilepath)pb SIZE_LIMIT) proto_helpers)TestCase)clientdistribresourceserverstatic)Headers)_render) DummyChannel DummyRequestceZdZdS)MySiteN)__name__ __module__ __qualname__\/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/web/test/test_distrib.pyrr#sDr rceZdZdZdZdZdS)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}. NcZtj|||_|jSN)r r# buildProtocolproto)selfaddrs r!r&zPBServerFactory.buildProtocol2s$'55dDAA zr )rrr__doc__r'r&rr r!r#r#'s4 Er r#ceZdZdZdS)ArbitraryErrorz% An exception for this test. N)rrrr*rr r!r,r,7sr r,cleZdZdZdZdZdZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdS) DistribTestsNctjtjg|j/|jj#|jjfdndd|j\|jjP|jjjfd|jjjj ndd|j , |j |j, |j tjS)z Clean up all the event sources left behind by either directly by test methods or indirectly via some distrib API. Nc:ddSNrcallbackdlsr!z'DistribTests.tearDown..JsRU^^D5I5Ir rc:ddS)Nr2r4sr!r6z'DistribTests.tearDown..NsAPTAUAUr r8)rDeferredf1r'notifyOnDisconnectr3sub publisherbroker transportloseConnectionport1append stopListeningport2 gatherResults)r(r5s @r!tearDownzDistribTests.tearDownCs: n 0 0 1 7 47=#< GM , ,-I-I-I-I J J J J qENN4 8 DH$6$B H  % 8 89U9U9U9U V V V H  % / > > @ @ @ @ qENN4 : ! IIdj..00 1 1 1 : ! IIdj..00 1 1 1"2&&&r ctj}|dtjddt j|}ttj ||_ tj d|j |_ tjd|j j|_tj}|d|jt%|}tj d||_t)jt}d|jjd}|d }|d |}|t(j||jd|S) Nstheresrootz text/plainr 127.0.0.1sherezhttp://127.0.0.1:z /here/thereasciiGET)rResourceputChildrDatarSiter#rResourcePublisherr:r listenTCPrAResourceSubscriptiongetHostportr<rrDrAgentencoderequest addCallbackreadBody assertEqual)r(r1site1r2f2agenturlds r! testDistribzDistribTests.testDistribXsY    Hfk'<@@AAA B!'";E"B"BCC&q$'22 / TZ=O=O=Q=Q=VWW    GTX&&& BZZ&q"--  W%%H$*"4"4"6"6";HHHjj!! MM&# & & fo&&& d&000r c\tj}|d|tj|}t t j|x|_}tj d|d}| |j | }t j|j|jx|_}tj|}tj d|d} | | 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. schildrrH) interface)rrKrLrrNr#rrOr:rrP addCleanuprCrRrQhostrSr<) r(child distribRoot distribSitedistribFactory distribPortr)mainRootmainSitemainPortmainAddrs r!_setupDistribServerz DistribTests._setupDistribServerls')) Xu---k+.. #2  %k 2 2$ $  .'>[QQQ   1222""$$%:49diPPP8;x(($QKHHH .///##%%!!r c ||\}}tjt}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://:/childrIrJ) rorrTrrerSrUrVrWrX)r(rfkwargsrmrnr^r_r`s r! _requestTestzDistribTests._requestTests"55e<<( W%%= == ===jj!! EM&# 0 0 0 0 fo&&&r c ||\}}d|jd|jd}|d}t jt jd|fi|}d}|||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. rqrrrsrIrJctjtjx}_|fd|S)Nc fSr%r)_protocolresponses r!r6zGDistribTests._requestAgentTest..cbCollectBody..s Xx$8r )rAccumulatingProtocol deliverBodyrr9closedDeferredrW)r{r`rzs` @r! cbCollectBodyz5DistribTests._requestAgentTest..cbCollectBodys_$9;;H   * * **/.*:*: :A' MM88888 9 9 9Hr ) rorerSrUrrTrrVrW)r(rfrtrmrnr_r`rs r!_requestAgentTestzDistribTests._requestAgentTests"55e<<(= == ===jj!! )FL ! ! )&# @ @ @ @    m$$$r cBitjtjdgGfddtj}fd|tddgi}fd}|||S) z The request headers are available on the request object passed to a distributed resource's C{render} method. NceZdZfdZdS)>DistribTests.test_requestHeaders..ReportRequestHeadersc|d<t|jdS)Nrr )updatedictrequestHeadersgetAllRawHeaders)r(rVreqrs r!renderzEDistribTests.test_requestHeaders..ReportRequestHeaders.renders< A%%d7+A+R+R+T+T&U&UVVVsr Nrrrr)rrsr!ReportRequestHeadersrs3        r rcdD}d|d|d|tjdS)Ncg|] }|d S) log_formatr).0es r! zHDistribTests.test_requestHeaders..check_logs..s999AlO999r zconnected to publisherz3could not connect to distributed web service: {msg}r)assertInr removeObserver)msgs logObserverrr(s r! check_logsz4DistribTests.test_requestHeaders..check_logssp99[999D MM2D 9 9 9 MMOQU V V V MM#a&$ ' ' '  -k : : : : :r foobar)headerscjjddgdS)NsFoosbar)r:r'r;rY)resultrrr(s r! cbRequestedz5DistribTests.test_requestHeaders..cbRequesteds> GM , ,Z 8 8 8   ^F3fX > > > > >r ) rEventLoggingObserverr addObserverrrKrurrW)r(rrVrrrrrs` @@@@r!test_requestHeadersz DistribTests.test_requestHeaderss  #8:: &{333f        8#4     ; ; ; ; ; ; ;## " "GUUG4D,E,E$   ? ? ? ? ? ? ? K(((r cGddtj}|}fd}|||S)z The response code can be set by the request object passed to a distributed resource's C{render} method. ceZdZdZdS)>DistribTests.test_requestResponseCode..SetResponseCodec0|ddS)NsetResponseCoder(rVs r!rzEDistribTests.test_requestResponseCode..SetResponseCode.renders'',,,rr Nrrr r!SetResponseCoder#     r rc|djd|djd|djddS)Nrr r8rsOKrYdatacodephraserr(s r!rz:DistribTests.test_requestResponseCode..cbRequesteds^   VAY^S 1 1 1   VAY^S 1 1 1   VAY-u 5 5 5 5 5r rrKrrWr(rrVrs` r!test_requestResponseCodez%DistribTests.test_requestResponseCodes      h/    (():):;; 6 6 6 6 6 K(((r cGddtj}|}fd}|||S)z The response code and message can be set by the request object passed to a distributed resource's C{render} method. ceZdZdZdS)EDistribTests.test_requestResponseCodeMessage..SetResponseCodec2|dddS)Nr some-messagerrrs r!rzLDistribTests.test_requestResponseCodeMessage..SetResponseCode.renders''_===rr Nrrr r!rrrr rc|djd|djd|djddS)Nrr r8rrrrs r!rzADistribTests.test_requestResponseCodeMessage..cbRequesteds^   VAY^S 1 1 1   VAY^S 1 1 1   VAY- ? ? ? ? ?r rrs` r!test_requestResponseCodeMessagez,DistribTests.test_requestResponseCodeMessages      h/    (():):;; @ @ @ @ @ K(((r cGddtj}||}||jdt zdz|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. ceZdZdZdS)0DistribTests.test_largeWrite..LargeWritec|dtzdz|tjSNxy)writerfinishr NOT_DONE_YETrs r!rz7DistribTests.test_largeWrite..LargeWrite.render s7 dZ/$6777   **r Nrrr r! LargeWriter s# + + + + +r rrrrrKrurWrYr)r(rrVs r!test_largeWritezDistribTests.test_largeWritesr + + + + +* + + + ##JJLL11D,dZ.?$.FGGGr cGddtj}||}||jdt zdz|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}. ceZdZdZdS)2DistribTests.test_largeReturn..LargeReturncdtzdzSrr rs r!rz9DistribTests.test_largeReturn..LargeReturn.rendersj(4//r Nrrr r! LargeReturnrs# 0 0 0 0 0r rrrr)r(rrVs r!test_largeReturnzDistribTests.test_largeReturnsr  0 0 0 0 0(+ 0 0 0##KKMM22D,dZ.?$.FGGGr clttjx_}t jd|x_}tjd| j x_ }tdgt|}fd}|||S)zz If there is an error issuing the request to the remote publisher, an error response is returned. rrHr c(jdtj}t |dgd}d|gjdS)Nir8) 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 sr  )rY responseCodeflushLoggedErrorsr NoSuchMethodlenjoinwritten)ignorederrorsexpectedrVr(s r! cbRenderedz4DistribTests.test_connectionLost..cbRendered2s   W13 7 7 7++BO< ++--2# #  <u%% L' * * F F F F F F2 j!!!r ctj|t}t jt }tdg}tj |}| || dt|| d|dddS)zQ When a request fails, the string form of the failure is logged. r r8zFailure instancerrN)rrcreateWithCleanupr r Failurer,rrIssuefailed assertEqualsrr)r(rfrVissues r!test_logFailedzDistribTests.test_logFailedNs$8JJ $   ON,, - -u%% g&& Q !S--... (+a.*FGGGGGr ctj|t}t }t j|}tjt}| || t| dt|||dd|dS)zf When L{twisted.web.distrib.Request}'s fail is called, the failure is logged. r8r log_failureN)rrrr r,r rrRequestrfailrrrassertIs)r(rerrrrs r!test_requestFailzDistribTests.test_requestFail]s $8JJ $   OC olnn--   ~... !S--... k!n]3Q77777r )rrrrArDr<r:rFrarorurrrrrrrrrrr r!r.r.=s E E C B'''*("""6&:   D**"   ***X H H H88888r r.c eZdZdZdZdZdS)_PasswordDatabasec||_dSr%)_users)r(userss r!__init__z_PasswordDatabase.__init__os  r c*t|jSr%)iterrr(s r!getpwallz_PasswordDatabase.getpwallrsDK   r cR|jD]}|d|kr|cStr1)rKeyError)r(usernameusers r!getpwnamz_PasswordDatabase.getpwnamus:K  DAw("" #jjr N)rrrrrrrr r!rrnsA!!!r rc~eZdZdZdZdZdeddfdZdZd Z d Z d Z d Z d Z ee ddZdS)UserDirectoryTestszj Tests for L{UserDirectory}, a resource for listing all user resources available on a system. cddddd|df|_dddd d |df|_t|j|jg|_t j|j|_dS) Nalicex{izAlice,,,z/bin/shbobi7zBob,,,)mktemprrrdatabaser UserDirectory directoryrs r!setUpzUserDirectoryTests.setUpsnsCj$++--S 3S(DKKMM9M)4:tx*@AA  .t}==r cj|ttj|jdS)zK L{UserDirectory} instances provide L{resource.IResource}. N) assertTruerr IResourcer rs r!test_interfacez!UserDirectoryTests.test_interfaces+  X%7HHIIIIIr namereturnNcKt|g}|j||}t||}|d{V||jddS)zr Verify that requesting the C{name} child of C{self.directory} results in a 404 response. Ni)rr getChildrrYr)r(rrVrr`s r!_404TestzUserDirectoryTests._404Testsm v&&((w77 FG $ $ -s33333r c@K|dd{VdS)z L{UserDirectory.getChild} returns a resource which renders a 404 response when passed a string which does not correspond to any known user. scarolNrrs r!test_getInvalidUserz&UserDirectoryTests.test_getInvalidUser2 mmH%%%%%%%%%%%r c@K|dd{VdS)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. saliceNrrs r!test_getUserWithoutResourcez.UserDirectoryTests.test_getUserWithoutResourcerr cltj|jd}|d}|t dg}|jd|}||tj | |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 FilePathrrfmakedirsrr rassertIsInstancerFilerYpath)r(homerrVrs r!test_getPublicHTMLChildz*UserDirectoryTests.test_getPublicHTMLChilds  "..jj// w''((99 ffk222 k&677777r ctj|jd}||d}t dg}|jd|}||tj | |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 rrr rfrr rr!rrQrYrerrSr#)r(r$webrVrs r!test_getDistribChildz'UserDirectoryTests.test_getDistribChilds  ".. jj)**w''((@@ fg&BCCC f--- --sx88888r ctdg}d|_|tj|jj|dS)zr L{UserDirectory.render} raises L{UnsupportedMethod} in response to a non-I{GET} request. rPOSTN)rmethod assertRaisesrUnsupportedMethodr rrs r!test_invalidMethodz%UserDirectoryTests.test_invalidMethods@ t$$ &2DN4I7SSSSSr ctjjdd}|tjjd}||ddtdgtj }fd}| ||S)z} L{UserDirectory} renders a list of links to available user content in response to a I{GET} request. rrr'r rcRtdj}|d\}}|jjd|jdd|jjjd|jjd|jdd|jjjddS) Nr liahrefzalice/z Alice (file)z bob.twistd/z Bob (twistd)) rrrgetElementsByTagNamerY firstChildtagName getAttributer)rdocumentrrrVr(s r!rz2UserDirectoryTests.test_render..cbRendereds"388GO#<#<==H$88>>LUC   U-5s ; ; ;   U-::6BBH M M M   U-8=~ N N N   S^3S 9 9 9   S^88@@- P P P   S^6;^ L L L L Lr ) r rrrfr r setContentrrr rW)r(rr)rrrVs` @r! test_renderzUserDirectoryTests.test_renders ' 277==mLL  --  "##..s333t$$11 M M M M M M :&&& r zpwd module requiredcltj}||jtdS)z If L{UserDirectory} is instantiated with no arguments, it uses the L{pwd} module as its password database. N)rr assertIdentical_pwdpwd)r(r s r!test_passwordDatabasez(UserDirectoryTests.test_passwordDatabases0 )++  Y^S11111r )rrrr*r rbytesrrrr%r*r0r<rr@rArr r!rr|s >>> JJJ 45 4T 4 4 4 4&&&&&& 8 8 8 9 9 9TTT@ VG*++22,+222r r)1r*os.pathrxml.dom.minidomrr@r? ImportErrorunittestrzope.interface.verifyrtwisted.internetrrtwisted.loggerr twisted.pythonr r twisted.spreadr twisted.spread.bananar twisted.testrtwisted.trial.unittestr twisted.webrrrrrtwisted.web.http_headersrtwisted.web.test._utilrtwisted.web.test.requesthelperrrrNrr# Exceptionr,r.rrrr r!rTs'''''' CC CCC ......++++++++------,,,,,,,,,,,,,,&&&&&&++++++AAAAAAAAAAAAAA,,,,,,******EEEEEEEE     V[        b(    Y n8n8n8n8n88n8n8n8b        z2z2z2z2z2z2z2z2z2z2s !!