o Óó·iLNã@sÐdZddlmZmZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZGdd „d eƒZGd d „d ƒZGd d „d eƒZGdd„deƒZGdd„deƒZGdd„dƒZGdd„deƒZdS)z Test for L{twisted.web.proxy}. é)Ú MemoryReactorÚ StringTransportWithDisconnection)ÚTestCase)Ú ProxyClientÚProxyClientFactoryÚ ProxyRequestÚReverseProxyRequestÚReverseProxyResource)ÚResource)ÚSite©Ú DummyRequestc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚReverseProxyResourceTestsz, Tests for L{ReverseProxyResource}. cCsÂtƒ}tƒ}tddd|ƒ}| d|¡t|ƒ}tƒ}| d¡}| |¡| |j d¡|  d|d¡|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 ZputChildr rÚ buildProtocolÚmakeConnectionZ addCleanupÚconnectionLostÚ dataReceivedÚ tcpClientsÚ assertEqualÚassertIsInstancerÚrestÚheaders)ÚselfÚuriÚ expectedURIÚrootÚreactorÚresourceÚsiteÚ transportÚchannelÚhostÚportÚfactoryÚ_timeoutZ _bind_addr©r)úaC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/web/test/test_proxy.pyÚ _testRenders       z%ReverseProxyResourceTests._testRendercCó | dd¡S)z˜ Test that L{ReverseProxyResource.render} initiates a connection to the given server with a L{ProxyClientFactory} as parameter. s/indexrN©r+©rr)r)r*Ú test_render8ó z%ReverseProxyResourceTests.test_rendercCr,)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/page1Nr-r.r)r)r*Útest_render_subpage?ó z-ReverseProxyResourceTests.test_render_subpagecCr,)zr Test that L{ReverseProxyResource.render} passes query parameters to the created factory. s/index?foo=bars /path?foo=barNr-r.r)r)r*Útest_renderWithQueryGr0z.ReverseProxyResourceTests.test_renderWithQuerycCsjtƒ}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. rrrófooNs /path/foo) rr ÚgetChildrrÚpathr&r%ZassertIdenticalr )rr r!Úchildr)r)r*Ú test_getChildNs  z'ReverseProxyResourceTests.test_getChildcCs*tdddƒ}| dd¡}| |jd¡dS)zu The L{ReverseProxyResource} return by C{getChild} has a path which has already been quoted. rrrs /%Ns/path/%20%2F%25)r r5rr6)rr!r7r)r)r*Útest_getChildWithSpecial_s  z2ReverseProxyResourceTests.test_getChildWithSpecialN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r+r/r1r3r8r9r)r)r)r*rs rc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) Ú 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. cCs||_d|_dS)z4 Hold a reference to the transport. N)r#Ú lostReason)rr#r)r)r*Ú__init__rs zDummyChannel.__init__cCs ||_dS)z; Keep track of the connection lost reason. N)r?)rÚreasonr)r)r*ryó zDummyChannel.connectionLostcCó |j ¡S)z: Get peer information from the transport. N)r#ÚgetPeerr.r)r)r*rDrBzDummyChannel.getPeercCrC)z: Get host information from the transport. N)r#ÚgetHostr.r)r)r*rE…rBzDummyChannel.getHostN)r:r;r<r=r@rrDrEr)r)r)r*r>is  r>c@sšeZdZdZdd„Zdd„Zd&d d „Zd d „Zd d„Zdd„Z dd„Z   d'dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„ZdS)(ÚProxyClientTestsz# Tests for L{ProxyClient}. cCs:| d¡\}}| d¡}| d¡}|tdd„|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 ó rcss|]}| d¡VqdS)ó: N)Úsplit)Ú.0Úheaderr)r)r*Ú s€z4ProxyClientTests._parseOutHeaders..N)rIÚpopÚdict)rÚcontentrÚbodyÚ requestLiner)r)r*Ú_parseOutHeaders‘s  z!ProxyClientTests._parseOutHeaderscCst|ƒS)z Make a dummy request object for the URL path. @param path: A URL path, beginning with a slash. @return: A L{DummyRequest}. Nr )rr6r)r)r*Ú makeRequestŸszProxyClientTests.makeRequestóGETNócCs,|durddi}d|j}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} Nóacceptó text/htmló/óHTTP/1.0)Zpostpathr)rÚrequestÚmethodrÚ requestBodyr6r)r)r*ÚmakeProxyClient¨s  z ProxyClientTests.makeProxyClientcCstƒ}||_| |¡|S)z» Connect a proxy client to a L{StringTransportWithDisconnection}. @param proxyClient: A L{ProxyClient}. @return: The L{StringTransportWithDisconnection}. N)rÚprotocolr)rÚ proxyClientZclientTransportr)r)r*Ú connectProxy¹s zProxyClientTests.connectProxycCs@| |¡|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. N)r`r#ÚvaluerRr)rr_rQrZrequestContentZ receivedLineÚreceivedHeadersrPr)r)r*ÚassertForwardsHeadersÅs   z&ProxyClientTests.assertForwardsHeadersc Cs^dt|ƒ d¡d|g}|D]\}}|D] }| |d|¡qq| d|g¡d |¡S)Ns HTTP/1.0 Úasciió rHrUrG)ÚstrÚencodeÚappendÚextendÚjoin) rÚcodeÚmessagerrPÚlinesrKÚvaluesrar)r)r*ÚmakeResponseBytesÖs ÿ z"ProxyClientTests.makeResponseBytescCsj| |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. NrU) rÚ responseCodeÚresponseMessageÚlistÚresponseHeadersÚgetAllRawHeadersÚsortrjÚwritten)rrZrkrlrrPrbÚexpectedHeadersr)r)r*ÚassertForwardsResponseÞs   z'ProxyClientTests.assertForwardsResponseTc Cs˜| d¡}| ||ddi|¡} | | |ddddœ¡} | | |¡|  | ||||¡¡| |||||¡|r<| j ¡|  | jj ¡| |j d¡dS) z‹ Build a fake proxy connection, and send C{data} over it, checking that it's forwarded to the originating request. r4rVrWs /foo HTTP/1.0óclose)ó connectionrVéN) rSr]rcrrrorxr#ÚloseConnectionZ assertFalseÚ connectedÚfinished) rrkrlrrPr[r\r|rZÚclientZ receivedBodyr)r)r*Ú_testDataForwardñs  ÿý  z!ProxyClientTests._testDataForwardcCs| dddddgfgd¡S)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. éÈóOKóFooóbarsbazó Some data N©r€r.r)r)r*Ú test_forwardsÿzProxyClientTests.test_forwardcCs| ddddgfgddd¡S) z~ Try to post content in the request, and check that the proxy client forward the body of the request. rr‚rƒr„r…óPOSTó Some contentNr†r.r)r)r*Ú test_postData'sÿzProxyClientTests.test_postDatacCs| ddgd¡S)z’ If the response contains a status with a message, it should be forwarded to the parent request with all the information. i”s Not FoundrUNr†r.r)r)r*Útest_statusWithMessage0óz'ProxyClientTests.test_statusWithMessagecCs*d}| dddtt|ƒƒ d¡gfg|¡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 bazrr‚óContent-LengthrdN©r€rfÚlenrg©rÚdatar)r)r*Útest_contentLength7sÿz#ProxyClientTests.test_contentLengthcCs.d}|jdddtt|ƒƒ d¡gfg|ddS) zŸ If the response contains a I{Content-Length} header, the outgoing connection is closed when all response body data has been received. rrr‚rŽrdF)r|Nrr‘r)r)r*Útest_losesConnectionAsûz%ProxyClientTests.test_losesConnectioncCs0tddddddœddƒ}| |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. rTó/foorYrWr4)rVsproxy-connectionrUNry©rVrz)rrr)rrr)r)r*Útest_headersCleanupsOsú ÿz%ProxyClientTests.test_headersCleanupscCsFddddœ}| ¡}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. rWs300ó keep-alive)rVr˜rzryrzrTr•rYrUNsGET /foo HTTP/1.0)Úcopyrrc)rrrwrr)r)r*Útest_keepaliveNotForwardedasýz+ProxyClientTests.test_keepaliveNotForwardedcCs | d¡}|j ddg¡|j ddg¡|j ddg¡|j|dd id }| |¡d gd gd gdœ}| | dd| ¡d¡¡| |ddt | ¡ƒd¡dS)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. r4sserversold-barsdatesold-bazs content-typesold/quxrVrW)rr„s 2010-01-01sapplication/x-baz)sServersDateó Content-Typerr‚rUN) rSrsZ setRawHeadersr]r`rroÚitemsrxrr)rrZrrr)r)r*Útest_defaultHeadersOverriddenrs  ýz.ProxyClientTests.test_defaultHeadersOverridden)rTNrU)rTrUT)r:r;r<r=rRrSr]r`rcrorxr€r‡rŠr‹r“r”r—ršrr)r)r)r*rFŒs*   ø,    rFc@s eZdZdZdd„Zdd„ZdS)ÚProxyClientFactoryTestsz* Tests for L{ProxyClientFactory}. cCsŒtdgƒ}tdddddid|ƒ}| dd¡| |jd ¡| |jd ¡| t|j ¡ƒd dgfg¡| d   |j ¡d ¡| |j d¡dS)zƒ Check that L{ProxyClientFactory.clientConnectionFailed} produces a B{501} response to the parent request. r4rTr•rYrVrWÚNiõs Gateway errorr›rUs

Could not connect

r{) r rZclientConnectionFailedrrprqrrrsrtrjrvr~)rrZr'r)r)r*Útest_connectionFailedŒs ÿ   þz-ProxyClientFactoryTests.test_connectionFailedcCsntdddddiddƒ}| 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). rTr•rYrVrWs Some dataNryr–) rrrrrÚcommandrr’r)rr'Úprotor)r)r*Útest_buildProtocol sÿ   ÿz*ProxyClientFactoryTests.test_buildProtocolN)r:r;r<r=r r£r)r)r)r*rž‡s ržc@s:eZdZdZddd„Zdd„Zdd „Zd d „Zd d „ZdS)ÚProxyRequestTestsz$ Tests for L{ProxyRequest}. rTrUc Csþtƒ}t|ƒ}tƒ}t|d|ƒ}| t|ƒ¡| |¡| |d|d¡| 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.comrYr{rú example.coméPéró example.comN)rr>rrÚ gotLengthrZhandleContentChunkÚrequestReceivedrrrrr¡Úversionrr’rZfather) rrrr[r’r#r$r rZr'r)r)r*Ú _testProcess·s$   zProxyRequestTests._testProcesscCs | dd¡S)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/barN©r¬r.r)r)r*Ú test_processÑs zProxyRequestTests.test_processcCr,)z£ If the incoming request doesn't contain a slash, L{ProxyRequest.process} should add one when instantiating L{ProxyClientFactory}. rUrXNr®r.r)r)r*Ú test_processWithoutTrailingSlashÝr2z2ProxyRequestTests.test_processWithoutTrailingSlashcCs| dddd¡S)zl L{ProxyRequest.process} should be able to retrieve request body and to forward it. r­rˆr‰Nr®r.r)r)r*Útest_processWithDataårŒz&ProxyRequestTests.test_processWithDatacCsztƒ}t|ƒ}tƒ}t|d|ƒ}| d¡| ddd¡| t|jƒd¡| |jddd¡| |jddd¡d S) z˜ Check that L{ProxyRequest.process} correctly parse port in the incoming URL, and create an outgoing connection with this port. FrrTshttp://example.com:1234/foo/barrYr{r¥rN) rr>rrr©rªrrr)rr#r$r rZr)r)r*Útest_processWithPortìs  z&ProxyRequestTests.test_processWithPortN)rTrU) r:r;r<r=r¬r¯r°r±r²r)r)r)r*r¤²s   r¤c@óeZdZdZdd„ZdS)Ú DummyFactoryz> A simple holder for C{host} and C{port} information. cCs||_||_dS)N)r%r&)rr%r&r)r)r*r@s zDummyFactory.__init__N)r:r;r<r=r@r)r)r)r*r´þó r´c@r³)ÚReverseProxyRequestTestsz+ Tests for L{ReverseProxyRequest}. cCs²tƒ}t|ƒ}tƒ}t|d|ƒ}tddƒ|_| d¡| ddd¡| t |j ƒd¡| |j ddd¡| |j ddd¡|j dd }|  |t ¡| |j d d i¡d 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. Fr¥rrrTr­rYr{r§rr¨N)rr>rrr´r'r©rªrrrrrr)rr#r$r rZr'r)r)r*r¯ s    z%ReverseProxyRequestTests.test_processN)r:r;r<r=r¯r)r)r)r*r¶rµr¶N)r=Ztwisted.internet.testingrrZtwisted.trial.unittestrZtwisted.web.proxyrrrrr Ztwisted.web.resourcer Ztwisted.web.serverr Ztwisted.web.test.test_webr rr>rFržr¤r´r¶r)r)r)r*Ús    S#|+L