tfLN&dZddlmZmZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZGdd eZGd d ZGd d eZGddeZGddeZGddZGddeZdS)z Test for L{twisted.web.proxy}. ) MemoryReactor StringTransportWithDisconnection)TestCase) ProxyClientProxyClientFactory ProxyRequestReverseProxyRequestReverseProxyResource)Resource)Site DummyRequestc6eZdZdZdZdZdZdZdZdZ dS) ReverseProxyResourceTestsz, Tests for L{ReverseProxyResource}. ct}t}tddd|}|d|t |}t }|d}||||j d| d|zdz|j \\} } } } } | | d| | d| | t| | j|| | jdd dS) z Check that a request pointing at C{uri} produce a new proxy connection, with the path of this request pointing at C{expectedURI}. 127.0.0.1/pathsindexNsGET s HTTP/1.1 Accept: text/html hosts127.0.0.1:1234)r rr putChildr r buildProtocolmakeConnection addCleanupconnectionLost dataReceived tcpClients assertEqualassertIsInstancerrestheaders)selfuri expectedURIrootreactorresourcesite transportchannelhostportfactory_timeout _bind_addrs Z/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/web/test/test_proxy.py _testRenderz%ReverseProxyResourceTests._testRendersF zz//' T8WMM h)))Dzz466 $$T**y))) .555Ws]-VVWWW8?8J5 4$gx {+++ t$$$ g'9::: {333 13DEEEEEc.|ddS)z Test that L{ReverseProxyResource.render} initiates a connection to the given server with a L{ProxyClientFactory} as parameter. s/indexrr0r!s r/ test_renderz%ReverseProxyResourceTests.test_render8s  8444r1c.|ddS)z Test that L{ReverseProxyResource.render} will instantiate a child resource that will initiate a connection to the given server requesting the apropiate url subpath. s /index/page1s /path/page1r3r4s r/test_render_subpagez-ReverseProxyResourceTests.test_render_subpage?s @@@r1c.|ddS)zr Test that L{ReverseProxyResource.render} passes query parameters to the created factory. s/index?foo=bars /path?foo=barr3r4s r/test_renderWithQueryz.ReverseProxyResourceTests.test_renderWithQueryGs  13CDDDr1ct}tddd|}|dd}||t||jd||jd||jd||j |j dS)a The L{ReverseProxyResource.getChild} method should return a resource instance with the same class as the originating resource, forward port, host, and reactor values, and update the path value with the value passed. rrrfooNs /path/foo) rr getChildrrpathr+r*assertIdenticalr%)r!r%r&childs r/ test_getChildz'ReverseProxyResourceTests.test_getChildNs //' T8WMM!!&$// e%9::: \222 T*** [111 U]H,<=====r1ctddd}|dd}||jddS)zu The L{ReverseProxyResource} return by C{getChild} has a path which has already been quoted. rrrs /%Ns/path/%20%2F%25)r r<rr=)r!r&r?s r/test_getChildWithSpecialz2ReverseProxyResourceTests.test_getChildWithSpecial_sI ( T8DD!!&$// %788888r1N) __name__ __module__ __qualname____doc__r0r5r7r9r@rBr1r/rrsFFF:555AAAEEE>>>"99999r1rc*eZdZdZdZdZdZdZdS) DummyChannelz A dummy HTTP channel, that does nothing but holds a transport and saves connection lost. @ivar transport: the transport used by the client. @ivar lostReason: the reason saved at connection lost. c"||_d|_dS)z4 Hold a reference to the transport. N)r( lostReason)r!r(s r/__init__zDummyChannel.__init__rs#r1c||_dS)z; Keep track of the connection lost reason. N)rK)r!reasons r/rzDummyChannel.connectionLostys!r1c4|jS)z: Get peer information from the transport. )r(getPeerr4s r/rPzDummyChannel.getPeer~%%'''r1c4|jS)z: Get host information from the transport. )r(getHostr4s r/rSzDummyChannel.getHostrQr1N)rCrDrErFrLrrPrSrGr1r/rIrIisZ!!! ((( (((((r1rIc|eZdZdZdZdZddZdZd Zd Z d Z dd Z dZ dZ dZdZdZdZdZdZdS)ProxyClientTestsz# Tests for L{ProxyClient}. c|d\}}|d}|d}|td|D|fS)aC Parse the headers out of some web content. @param content: Bytes received from a web server. @return: A tuple of (requestLine, headers, body). C{headers} is a dict of headers, C{requestLine} is the first line (e.g. "POST /foo ...") and C{body} is whatever is left. s  rc3@K|]}|dVdS): N)split).0headers r/ z4ProxyClientTests._parseOutHeaders..s.!L!L&&,,u"5"5!L!L!L!L!L!Lr1)rZpopdict)r!contentr body requestLines r/_parseOutHeadersz!ProxyClientTests._parseOutHeaderss] k22 --((kk!nn T!L!LG!L!L!LLLdSSr1c t|S)z Make a dummy request object for the URL path. @param path: A URL path, beginning with a slash. @return: A L{DummyRequest}. r )r!r=s r/ makeRequestzProxyClientTests.makeRequestsD!!!r1GETNr1cJ|ddi}d|jz}t||d|||S)a Make a L{ProxyClient} object used for testing. @param request: The request to use. @param method: The HTTP method to use, GET by default. @param headers: The HTTP headers to use expressed as a dict. If not provided, defaults to {'accept': 'text/html'}. @param requestBody: The body of the request. Defaults to the empty string. @return: A L{ProxyClient} Naccept text/html/HTTP/1.0)postpathr)r!requestmethodr requestBodyr=s r/makeProxyClientz ProxyClientTests.makeProxyClients8 ? ,/Gg&&64g{GTTTr1cZt}||_|||S)z Connect a proxy client to a L{StringTransportWithDisconnection}. @param proxyClient: A L{ProxyClient}. @return: The L{StringTransportWithDisconnection}. )rprotocolr)r! proxyClientclientTransports r/ connectProxyzProxyClientTests.connectProxys1;<<#. ""?333r1c|||j}||\}}}|||||||S)a` Assert that C{proxyClient} sends C{headers} when it connects. @param proxyClient: A L{ProxyClient}. @param requestLine: The request line we expect to be sent. @param headers: A dict of headers we expect to be sent. @return: If the assertion is successful, return the request body as bytes. )rur(valuercr)r!rsrbr requestContent receivedLinereceivedHeadersras r/assertForwardsHeadersz&ProxyClientTests.assertForwardsHeaderssw +&&&$.4466.2.C.CN.S.S+ ot {333 '222 r1cdt|dzdz|zg}|D]%\}}|D]}||dz|z&|d|gd|S)Ns HTTP/1.0 ascii rYr1rW)strencodeappendextendjoin) r!codemessager ralinesr\valuesrws r/makeResponseBytesz"ProxyClientTests.makeResponseBytessD 0 0 9 99D@7JK% 5 5NFF 5 5 Ve^e34444 5 c4[!!!||E"""r1c||j|||j|t|j}||dd}|||||d|j|dS)aK Assert that C{request} has forwarded a response from the server. @param request: A L{DummyRequest}. @param code: The expected HTTP response code. @param message: The expected HTTP message. @param headers: The expected HTTP headers. @param body: The expected response body. Nr1) r responseCoderesponseMessagelistresponseHeadersgetAllRawHeaderssortrwritten)r!rmrrr rarzexpectedHeaderss r/assertForwardsResponsez'ProxyClientTests.assertForwardsResponses -t444 0':::w6GGIIJJ!!!!* /::: '/22D99999r1Tc|d}|||ddi|} || |dzddd} || || ||||||||||||r| j| | jj ||j ddS) z Build a fake proxy connection, and send C{data} over it, checking that it's forwarded to the originating request. r;rhris /foo HTTP/1.0close) connectionrhN) rerpr{rrrrr(loseConnection assertFalse connectedfinished) r!rrr rarnrorrmclient receivedBodys r/_testDataForwardz!ProxyClientTests._testDataForwards"""6**%% Vi6   11  & &$ > >  {333 D224'4PPQQQ ##GT7GTJJJ  .   + + - - - )3444 )1-----r1c<|dddddgfgdS)z When connected to the server, L{ProxyClient} should send the saved request, with modifications of the headers, and then forward the result to the parent request. OKFoobarsbaz Some data rr4s r/ test_forwardzProxyClientTests.test_forwards3 $$ &66"2346F   r1c>|ddddgfgdddS)z~ Try to post content in the request, and check that the proxy client forward the body of the request. rrrrrPOST Some contentrr4s r/ test_postDatazProxyClientTests.test_postData's4 $$ &6(+,.>   r1c2|ddgdS)z If the response contains a status with a message, it should be forwarded to the parent request with all the information. is Not Foundr1rr4s r/test_statusWithMessagez'ProxyClientTests.test_statusWithMessage0s $$S,C@@@r1c d}|dddtt|dgfg|S)z If the response contains a I{Content-Length} header, the inbound request object should still only have C{finish} called on it once. foo bar bazrrContent-Lengthr}rrlenrr!datas r/test_contentLengthz#ProxyClientTests.test_contentLength7sQ $$ +c#d))nn.C.CG.L.L-MNOQU   r1c d}|dddtt|dgfg|dS)z If the response contains a I{Content-Length} header, the outgoing connection is closed when all response body data has been received. rrrrr}F)rrrs r/test_losesConnectionz%ProxyClientTests.test_losesConnectionAsZ $$  #c$ii.."7"7"@"@!A B C  %   r1cptdddddddd}||jdd d dS) z The headers given at initialization should be modified: B{proxy-connection} should be removed if present, and B{connection} should be added. rf/foorkrir;)rhsproxy-connectionr1Nrrhr)rrr )r!rs r/test_headersCleanupsz%ProxyClientTests.test_headersCleanupsOs`    $6 B B       N XNN     r1cdddd}|}d|d<|d=tddd |d d }||d |d S) z The proxy doesn't really know what to do with keepalive things from the remote server, so we stomp over any keepalive header we get from the client. ris300 keep-alive)rhrrrrrfrrkr1NsGET /foo HTTP/1.0)copyrr{)r!r rrs r/test_keepaliveNotForwardedz+ProxyClientTests.test_keepaliveNotForwardedasp$!(   ",,..)1 & M *VWk7CNN ""6+?QQQQQr1c 0|d}|jddg|jddg|jddg||dd i }||d gd gd gd}||dd|d||ddt|ddS)z L{server.Request} within the proxy sets certain response headers by default. When we get these headers back from the remote server, the defaults are overridden rather than simply appended. r;sserversold-barsdatesold-bazs content-typesold/quxrhri)r rs 2010-01-01sapplication/x-baz)sServersDate Content-Typerrr1N) rer setRawHeadersrprurritemsrr)r!rmrr s r/test_defaultHeadersOverriddenz.ProxyClientTests.test_defaultHeadersOverriddenrs ""6**--i*FFF--g |DDD--o |LLL%%g <7P%QQ &!!!x#_23   D223w}}PSTTUUU ##GS%gmmoo9N9NPSTTTTTr1)rfNr1)rfr1T)rCrDrErFrcrerprur{rrrrrrrrrrrrGr1r/rUrUs6 T T T"""UUUU"   "###:::2*.*.*.*.X      AAA          $RRR"UUUUUr1rUceZdZdZdZdZdS)ProxyClientFactoryTestsz* Tests for L{ProxyClientFactory}. ctdg}tdddddid|}|dd||jd ||jd |t |jd dgfg|d |j d ||j ddS)z Check that L{ProxyClientFactory.clientConnectionFailed} produces a B{501} response to the parent request. r;rfrrkrhriNis Gateway errorrr1s

Could not connect

r) rrclientConnectionFailedrrrrrrrrr)r!rmr,s r/test_connectionFailedz-ProxyClientFactoryTests.test_connectionFaileds x(($ G[9l*CR   &&tT222 -s333 02BCCC  (99;; < < ~ . /    '/224QRRR )1-----r1cptdddddidd}|d}||t||jd||jd||jd||jddd dS) z L{ProxyClientFactory.buildProtocol} should produce a L{ProxyClient} with the same values of attributes (with updates on the headers). rfrrkrhris Some dataNrr) rrrrrcommandrrr )r!r,protos r/test_buildProtocolz*ProxyClientFactoryTests.test_buildProtocols % G[9l*C\SW  %%d++ e[111 /// W--- \222  M|HMM     r1N)rCrDrErFrrrGr1r/rrs<...(     r1rc2eZdZdZd dZdZdZdZdZd S) ProxyRequestTestsz$ Tests for L{ProxyRequest}. rfr1ct}t|}t}t|d|}|t |||||d|zd|t |j d||j ddd||j ddd|j dd} | | t|| j ||| j d|| jd d i|| j||| j||| j|d S) z Build a request pointing at C{uri}, and check that a proxied request is created, pointing a C{expectedURI}. Fshttp://example.comrkrr example.comPr example.comN)rrIrr gotLengthrhandleContentChunkrequestReceivedrrrrrversionr rrfather) r!r"r#rnrr(r)r%rmr,s r/ _testProcesszProxyRequestTests._testProcesss 566 y))//ww77#d))$$$""4((((=(C[QQQ W/00!444 +A.q1=AAA +A.q12666$Q'* g'9::: &111 +666 7N*CDDD t,,, {333 11111r1c.|ddS)a L{ProxyRequest.process} should create a connection to the given server, with a L{ProxyClientFactory} as connection factory, with the correct parameters: - forward comment, version and data values - update headers with the B{host} value - remove the host from the URL - pass the request as parent request /foo/barrr4s r/ test_processzProxyRequestTests.test_processs  k:::r1c.|ddS)z If the incoming request doesn't contain a slash, L{ProxyRequest.process} should add one when instantiating L{ProxyClientFactory}. r1rjrr4s r/ test_processWithoutTrailingSlashz2ProxyRequestTests.test_processWithoutTrailingSlashs   d+++r1c2|ddddS)zl L{ProxyRequest.process} should be able to retrieve request body and to forward it. rrrrr4s r/test_processWithDataz&ProxyRequestTests.test_processWithDatas   k7OTTTr1ct}t|}t}t|d|}|d|ddd|t|jd||jddd||jdddd S) z Check that L{ProxyRequest.process} correctly parse port in the incoming URL, and create an outgoing connection with this port. Frrfshttp://example.com:1234/foo/barrkrrrN) rrIrrrrrrr)r!r(r)r%rms r/test_processWithPortz&ProxyRequestTests.test_processWithPorts 566 y))//ww77!(JKXXX W/00!444 +A.q1=AAA +A.q1488888r1N)rfr1) rCrDrErFrrrrrrGr1r/rrsq22224 ; ; ;,,,UUU99999r1rceZdZdZdZdS) DummyFactoryz> A simple holder for C{host} and C{port} information. c"||_||_dS)N)r*r+)r!r*r+s r/rLzDummyFactory.__init__s  r1N)rCrDrErFrLrGr1r/rrs-r1rceZdZdZdZdS)ReverseProxyRequestTestsz+ Tests for L{ReverseProxyRequest}. ct}t|}t}t|d|}t dd|_|d|ddd|t|j d||j ddd||j ddd|j dd }| |t||j d d id S) a L{ReverseProxyRequest.process} should create a connection to its factory host/port, using a L{ProxyClientFactory} instantiated with the correct parameters, and particularly set the B{host} header to the factory host. FrrrrfrrkrrrrN)rrIrr rr,rrrrrrrr )r!r(r)r%rmr,s r/rz%ReverseProxyRequestTests.test_process s*566 y))//%gug>>&}d;;! [AAA W/00!444 +A.q1=AAA +A.q14888$Q'* g'9::: 7N*CDDDDDr1N)rCrDrErFrrGr1r/rrs2EEEEEr1rN)rFtwisted.internet.testingrrtwisted.trial.unittestrtwisted.web.proxyrrrr r twisted.web.resourcer twisted.web.serverr twisted.web.test.test_webrrrIrUrrrrrGr1r/rsUTTTTTTT++++++*)))))######222222P9P9P9P9P9P9P9P9f ( ( ( ( ( ( ( (FxUxUxUxUxUxxUxUxUv( ( ( ( ( h( ( ( VI9I9I9I9I9I9I9I9XEEEEExEEEEEr1