U ˆjLNã@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)ú?/tmp/pip-unpacked-wheel-j5h_c8_q/twisted/web/test/test_proxy.pyÚ _testRenders       z%ReverseProxyResourceTests._testRendercCs | dd¡S)z˜ Test that L{ReverseProxyResource.render} initiates a connection to the given server with a L{ProxyClientFactory} as parameter. s/indexr©r+©rr)r)r*Ú test_render8sz%ReverseProxyResourceTests.test_rendercCs | dd¡S)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/page1r,r-r)r)r*Útest_render_subpage?sz-ReverseProxyResourceTests.test_render_subpagecCs | dd¡S)zr Test that L{ReverseProxyResource.render} passes query parameters to the created factory. s/index?foo=bars /path?foo=barr,r-r)r)r*Útest_renderWithQueryGsz.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 r2rr3)rr!r4r)r)r*Útest_getChildWithSpecial_s  z2ReverseProxyResourceTests.test_getChildWithSpecialN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r+r.r/r0r5r6r)r)r)r*rsrc@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__rszDummyChannel.__init__cCs ||_dS)z; Keep track of the connection lost reason. N)r<)rÚreasonr)r)r*ryszDummyChannel.connectionLostcCs |j ¡S)z: Get peer information from the transport. )r#ÚgetPeerr-r)r)r*r?szDummyChannel.getPeercCs |j ¡S)z: Get host information from the transport. )r#ÚgetHostr-r)r)r*r@…szDummyChannel.getHostN)r7r8r9r:r=rr?r@r)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*Ú sz4ProxyClientTests._parseOutHeaders..)rDÚ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}. r )rr3r)r)r*Ú makeRequestŸszProxyClientTests.makeRequestóGETNócCs,|dkrddi}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Ú requestBodyr3r)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}. )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. )r[r#ÚvaluerMr)rrZrLrZrequestContentZ receivedLineÚreceivedHeadersrKr)r)r*ÚassertForwardsHeadersÅs    z&ProxyClientTests.assertForwardsHeadersc Cs^dt|ƒ d¡d|g}|D]$\}}|D]}| |d|¡q,q | d|g¡d |¡S)Ns HTTP/1.0 Úasciió rCrPrB)ÚstrÚencodeÚappendÚextendÚjoin) rÚcodeÚmessagerrKÚlinesrFÚvaluesr\r)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. NrP) rÚ responseCodeÚresponseMessageÚlistÚresponseHeadersÚgetAllRawHeadersÚsortreÚwritten)rrUrfrgrrKr]ÚexpectedHeadersr)r)r*ÚassertForwardsResponseÞs   z'ProxyClientTests.assertForwardsResponseTc Cs˜| d¡}| ||ddi|¡} | | |ddddœ¡} | | |¡|  | ||||¡¡| |||||¡|rx| 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. r1rQrRs /foo HTTP/1.0óclose)ó connectionrQéN) rNrXr^rrrjrsr#ÚloseConnectionZ assertFalseÚ connectedÚfinished) rrfrgrrKrVrWrwrUÚ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 ©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. r|r}r~rr€óPOSTó Some contentrr-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 FoundrPrr-r)r)r*Útest_statusWithMessage0sz'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 bazr|r}óContent-Lengthr_©r{raÚlenrb©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. r‡r|r}rˆr_F)rwr‰r‹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. rOó/foorTrRr1)rQsproxy-connectionrPNrt©rQru)rrr)rrzr)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. rRs300ó keep-alive)rQr’rurtrurOrrTrPNsGET /foo HTTP/1.0)Úcopyrr^)rrrrrzr)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. r1sserversold-barsdatesold-bazs content-typesold/quxrQrR)rrs 2010-01-01sapplication/x-baz)sServersDateó Content-Typer|r}rPN) rNrnZ setRawHeadersrXr[rrjÚitemsrsrm)rrUrzrr)r)r*Útest_defaultHeadersOverriddenrs  ýz.ProxyClientTests.test_defaultHeadersOverridden)rONrP)rOrPT)r7r8r9r:rMrNrXr[r^rjrsr{r‚r…r†rrŽr‘r”r—r)r)r)r*rAŒs(  ø ,   rAc@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. r1rOrrTrQrRÚNiõs Gateway errorr•rPs

Could not connect

rv) r rZclientConnectionFailedrrkrlrmrnrorerqry)rrUr'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). rOrrTrQrRs Some dataNrtr) rrrrrÚcommandrrŒr)rr'Úprotor)r)r*Útest_buildProtocol s"ÿ  ÿz*ProxyClientFactoryTests.test_buildProtocolN)r7r8r9r:ršrr)r)r)r*r˜‡sr˜c@s:eZdZdZddd„Zdd„Zdd „Zd d „Zd d „ZdS)ÚProxyRequestTestsz$ Tests for L{ProxyRequest}. rOrPc 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.comrTrvrú example.coméPéró example.comN)rr;rrÚ gotLengthrŠZhandleContentChunkÚrequestReceivedrrrrr›ÚversionrrŒrZfather) rrrrVrŒr#r$r rUr'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/bar©r¦r-r)r)r*Ú test_processÑs zProxyRequestTests.test_processcCs | dd¡S)z£ If the incoming request doesn't contain a slash, L{ProxyRequest.process} should add one when instantiating L{ProxyClientFactory}. rPrSr¨r-r)r)r*Ú test_processWithoutTrailingSlashÝsz2ProxyRequestTests.test_processWithoutTrailingSlashcCs| dddd¡S)zl L{ProxyRequest.process} should be able to retrieve request body and to forward it. r§rƒr„r¨r-r)r)r*Útest_processWithDataåsz&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. FrrOshttp://example.com:1234/foo/barrTrvrŸrN) rr;rrr£r¤rrŠr)rr#r$r rUr)r)r*Útest_processWithPortìs  z&ProxyRequestTests.test_processWithPortN)rOrP) r7r8r9r:r¦r©rªr«r¬r)r)r)r*rž²s   ržc@seZdZdZdd„ZdS)Ú DummyFactoryz> A simple holder for C{host} and C{port} information. cCs||_||_dS)N)r%r&)rr%r&r)r)r*r=szDummyFactory.__init__N)r7r8r9r:r=r)r)r)r*r­þsr­c@seZdZdZdd„ZdS)Ú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ŸrrrOr§rTrvr¡rr¢N)rr;rrr­r'r£r¤rrŠrrrr)rr#r$r rUr'r)r)r*r© s    z%ReverseProxyRequestTests.test_processN)r7r8r9r:r©r)r)r)r*r®sr®N)r:Ztwisted.internet.testingrrZtwisted.trial.unittestrZtwisted.web.proxyrrrrr Ztwisted.web.resourcer Ztwisted.web.serverr Ztwisted.web.test.test_webr rr;rAr˜ržr­r®r)r)r)r*Ús    S#|+L