o ¯bNNã@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 ÚputChildr rÚ buildProtocolÚmakeConnectionÚ addCleanupÚconnectionLostÚ dataReceivedÚ tcpClientsÚ assertEqualÚassertIsInstancerÚrestÚheaders)ÚselfÚuriÚ expectedURIÚrootÚreactorÚresourceÚsiteÚ transportÚchannelÚhostÚportÚfactoryÚ_timeoutÚ _bind_addr©r,ú=/usr/lib/python3/dist-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/indexr©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/page1r0r1r,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=barr0r1r,r,r-Útest_renderWithQueryGr3z.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'ÚassertIdenticalr")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 r8rr9)rr#r;r,r,r-Útest_getChildWithSpecial_s  z2ReverseProxyResourceTests.test_getChildWithSpecialN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r.r2r4r6r<r=r,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)rC)rÚreasonr,r,r-ryó zDummyChannel.connectionLostcCó |j ¡S)z: Get peer information from the transport. )r%ÚgetPeerr1r,r,r-rHrFzDummyChannel.getPeercCrG)z: Get host information from the transport. )r%ÚgetHostr1r,r,r-rI…rFzDummyChannel.getHostN)r>r?r@rArDrrHrIr,r,r,r-rBis  rBc@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..)rMÚ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 )rr9r,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)Úpostpathr)rÚrequestÚmethodrÚ requestBodyr9r,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Ú proxyClientÚclientTransportr,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. )rfr%ÚvaluerVr)rrdrUrÚrequestContentÚ receivedLineÚreceivedHeadersrTr,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ó rLrYrK)ÚstrÚencodeÚappendÚextendÚjoin) rÚcodeÚmessagerrTÚlinesrOÚvaluesrgr,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. NrY) rÚ responseCodeÚresponseMessageÚlistÚresponseHeadersÚgetAllRawHeadersÚsortrrÚwritten)rr_rsrtrrTrjÚ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. r7rZr[s /foo HTTP/1.0óclose)ó connectionrZéN) rWrbrkrrrwr€r%ÚloseConnectionÚ assertFalseÚ connectedÚfinished) rrsrtrrTr`rar„r_ÚclientÚ 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Šr1r,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ŒrrŽróPOSTó Some contentrr1r,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 FoundrYrr1r,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 bazr‹rŒóContent-Lengthrl©rŠrnÚlenro©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˜rlF)r„r™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. rXó/foor]r[r7)rZsproxy-connectionrYNr©rZr‚)rrr)rrˆr,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. r[s300ó keep-alive)rZr¢r‚rr‚rXrŸr]rYNsGET /foo HTTP/1.0)Úcopyrrk)rrrrˆr,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. r7sserversold-barsdatesold-bazs content-typesold/quxrZr[)rrŽs 2010-01-01sapplication/x-baz)sServersDateó Content-Typer‹rŒrYN) rWr{Ú setRawHeadersrbrfrrwÚitemsr€rz)rr_rˆrr,r,r-Útest_defaultHeadersOverriddenrs  ýz.ProxyClientTests.test_defaultHeadersOverridden)rXNrY)rXrYT)r>r?r@rArVrWrbrfrkrwr€rŠr‘r”r•rržr¡r¤r¨r,r,r,r-rJŒs*   ø,    rJc@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. r7rXrŸr]rZr[ÚNiõs Gateway errorr¥rYs

Could not connect

rƒ) r rÚclientConnectionFailedrrxryrzr{r|rrr~r‡)rr_r)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). rXrŸr]rZr[s Some dataNrr ) rrrrrÚcommandrrœr)rr)Úprotor,r,r-Útest_buildProtocol sÿ   ÿz*ProxyClientFactoryTests.test_buildProtocolN)r>r?r@rAr¬r¯r,r,r,r-r©‡s r©c@s:eZdZdZddd„Zdd„Zdd „Zd d „Zd d „ZdS)ÚProxyRequestTestsz$ Tests for L{ProxyRequest}. rXrYc 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.comr]rƒrú example.coméPéró example.comN)rrBrrÚ gotLengthršÚhandleContentChunkÚrequestReceivedrrrrr­ÚversionrrœrÚfather) rrr r`rœr%r&r"r_r)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ºr1r,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}. rYr\r¼r1r,r,r-Ú test_processWithoutTrailingSlashÝr5z2ProxyRequestTests.test_processWithoutTrailingSlashcCs| dddd¡S)zl L{ProxyRequest.process} should be able to retrieve request body and to forward it. r»r’r“r¼r1r,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. FrrXshttp://example.com:1234/foo/barr]rƒr±rN) rrBrrrµr·rršr)rr%r&r"r_r,r,r-Útest_processWithPortìs  z&ProxyRequestTests.test_processWithPortN)rXrY) r>r?r@rArº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-rDs zDummyFactory.__init__N)r>r?r@rArDr,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±rrrXr»r]rƒr³rr´N)rrBrrrÂr)rµr·rršrrrr)rr%r&r"r_r)r,r,r-r½ s    z%ReverseProxyRequestTests.test_processN)r>r?r@rAr½r,r,r,r-rÄrÃrÄN)rAÚtwisted.test.proto_helpersrrÚtwisted.trial.unittestrÚtwisted.web.proxyrrrrr Útwisted.web.resourcer Útwisted.web.serverr Útwisted.web.test.test_webr rrBrJr©r°rÂrÄr,r,r,r-Ús    S#|+L