U  W[|Nã@sädZddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl m Z mZmZdd lmZGd d „d eƒZGd d „d eƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZGdd„deƒZdS)z Test for L{twisted.web.proxy}. é)ÚTestCase)Ú StringTransportWithDisconnection)Ú MemoryReactor)ÚResource)ÚSite)ÚReverseProxyResourceÚProxyClientFactory)Ú ProxyClientÚ ProxyRequestÚReverseProxyRequest©Ú DummyRequestc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚReverseProxyResourceTestsz, Tests for L{ReverseProxyResource}. c Csätƒ}tƒ}tddd|ƒ}| d|¡t|ƒ}tƒ}| d¡}| |¡| |j d¡|  d|d¡|  t |j ƒd¡|  |j d d d¡|  |j d dd¡|j d d } | | t¡|  | j|¡|  | jd d ¡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 éréóhosts127.0.0.1:1234)rrrZputChildrrÚ buildProtocolÚmakeConnectionZ addCleanupÚconnectionLostÚ dataReceivedÚ assertEqualÚlenÚ tcpClientsÚassertIsInstancerÚrestÚheaders) ÚselfÚuriÚ expectedURIÚrootÚreactorÚresourceZsiteÚ transportÚchannelÚfactory©r(ú=/usr/lib/python3/dist-packages/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_render9sz%ReverseProxyResourceTests.test_rendercCs | 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_renderWithQueryAsz.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ÚpathÚportÚhostZassertIdenticalr#)rr#r$Úchildr(r(r)Ú test_getChildIs  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)rr0rr1)rr$r4r(r(r)Útest_getChildWithSpecial[s  z2ReverseProxyResourceTests.test_getChildWithSpecialN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r*r-r.r5r6r(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__oszDummyChannel.__init__cCs ||_dS)z; Keep track of the connection lost reason. N)r<)rÚreasonr(r(r)rwszDummyChannel.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;fs 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)rZcontentrÚ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 )rr1r(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Ú requestBodyr1r(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}. )rZprotocolr)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. )rYr%ÚvaluerLr)rrXrKrZrequestContentZ receivedLineÚreceivedHeadersrJr(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ó rCrOrB)ÚstrÚencodeÚappendÚextendÚjoin) rÚcodeÚmessagerrJÚlinesrFÚvaluesrZr(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. NrO) rÚ responseCodeÚresponseMessageÚlistÚresponseHeadersÚgetAllRawHeadersÚsortrcÚwritten)rrTrdrerrJr[Ú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. r/rPrQs /foo HTTP/1.0óclose)ó connectionrPrN) rMrWr\rrrhrqr%ÚloseConnectionZ assertFalseZ connectedÚfinished) rrdrerrJrUrVrtrTÚ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 ©rwr,r(r(r)Ú test_forward s  ÿ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. rxryrzr{r|óPOSTó Some contentr}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 FoundrOr}r,r(r(r)Útest_statusWithMessage3s ÿ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 bazrxryóContent-Lengthr]©rwr_rr`©rÚdatar(r(r)Útest_contentLength<sü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ƒrxryr„r]F)rtr…r†r(r(r)Útest_losesConnectionIsû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. rNó/foorSrQr/)rPsproxy-connectionrONrr©rPrs)r rr)rrvr(r(r)Útest_headersCleanupsWsÿÿ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. rQs300ó keep-alive)rPrrsrrrsrNrŠrSrONsGET /foo HTTP/1.0)Úcopyr r\)rrrprvr(r(r)Útest_keepaliveNotForwardedcsýÿ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. r/sserversold-barsdatesold-bazs content-typesold/quxrPrQ)rr{s 2010-01-01sapplication/x-baz)sServersDateó Content-TyperxryrON) rMrlZ setRawHeadersrWrYrrhÚitemsrqrk)rrTrvrr(r(r)Útest_defaultHeadersOverriddenvs(  ýÿ ÿz.ProxyClientTests.test_defaultHeadersOverridden)rNNrO)rNrOT)r7r8r9r:rLrMrWrYr\rhrqrwr~rr‚rˆr‰rŒrr’r(r(r(r)rAs, ÿ   ÿ #     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. r/rNrŠrSrPrQÚNiõs Gateway errorrrOs

Could not connect

r) r rZclientConnectionFailedrrirjrkrlrmrcroru)rrTr'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). rNrŠrSrPrQs Some dataNrrr‹) rrrr rÚcommandrr‡r)rr'Úprotor(r(r)Útest_buildProtocol¨sþ  ÿz*ProxyClientFactoryTests.test_buildProtocolN)r7r8r9r:r•r˜r(r(r(r)r“Žsr“c@s:eZdZdZddd„Zdd„Zdd „Zd d „Zd d „ZdS)ÚProxyRequestTestsz$ Tests for L{ProxyRequest}. rNrOc 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.comrSrrú example.coméPrró example.comN)rr;rr Ú gotLengthrZhandleContentChunkÚrequestReceivedrrrrr–Úversionrr‡rZfather) rr r!rUr‡r%r&r#rTr'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}. rOrRr¢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¡rr€r¢r,r(r(r)Útest_processWithDatañs ÿ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. FrrNshttp://example.com:1234/foo/barrSrršrN) rr;rr rržrrr)rr%r&r#rTr(r(r)Útest_processWithPortús  ÿz&ProxyRequestTests.test_processWithPortN)rNrO) 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)r3r2)rr3r2r(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šrrrNr¡rSrrrrœN)rr;rr r§r'rržrrrrrr)rr%r&r#rTr'r(r(r)r£s    z%ReverseProxyRequestTests.test_processN)r7r8r9r:r£r(r(r(r)r¨sr¨N)r:Ztwisted.trial.unittestrZtwisted.test.proto_helpersrrZtwisted.web.resourcerZtwisted.web.serverrZtwisted.web.proxyrrr r r Ztwisted.web.test.test_webr rÚobjectr;rAr“r™r§r¨r(r(r(r)Ús"      S',T