← Back
Editing: test_proxy.cpython-38.pyc
U W[|N � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl m Z m Z ddl m Z mZmZ dd lmZ G d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )z Test for L{twisted.web.proxy}. � )�TestCase)� StringTransportWithDisconnection)� MemoryReactor)�Resource)�Site)�ReverseProxyResource�ProxyClientFactory)� ProxyClient� ProxyRequest�ReverseProxyRequest�� DummyRequestc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �ReverseProxyResourceTestsz, Tests for L{ReverseProxyResource}. c C s� t � }t� }tddd|�}|�d|� t|�}t� }|�d�}|�|� | �|j d� |� d| d � | � t |j �d� | � |j d d d� | � |j d d d� |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�� � /paths indexNs GET s HTTP/1.1 Accept: text/html � r � � hosts 127.0.0.1:1234)r r r ZputChildr r � 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� _testRender s* �� z%ReverseProxyResourceTests._testRenderc C s | � dd�S )z� Test that L{ReverseProxyResource.render} initiates a connection to the given server with a L{ProxyClientFactory} as parameter. s /indexr �r* �r r( r( r) � test_render9 s z%ReverseProxyResourceTests.test_renderc C s | � 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_renderWithQueryA s z.ReverseProxyResourceTests.test_renderWithQueryc C sj t � }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. r r r � fooNs /path/foo) r r �getChildr r �path�port�hostZassertIdenticalr# )r r# r$ �childr( r( r) � test_getChildI s z'ReverseProxyResourceTests.test_getChildc C s* t ddd�}|�dd�}| �|jd� dS )zu The L{ReverseProxyResource} return by C{getChild} has a path which has already been quoted. r r r s /%Ns /path/%20%2F%25)r r0 r r1 )r r$ r4 r( r( r) �test_getChildWithSpecial[ s z2ReverseProxyResourceTests.test_getChildWithSpecialN) �__name__� __module__� __qualname__�__doc__r* r- r. r5 r6 r( r( r( r) r s !r c @ s0 e Zd ZdZdd� 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. c C s || _ d| _dS )z4 Hold a reference to the transport. N)r% � lostReason)r r% r( r( r) �__init__o s zDummyChannel.__init__c C s || _ dS )z; Keep track of the connection lost reason. N)r< )r �reasonr( r( r) r w s zDummyChannel.connectionLostc C s | j �� S )z: Get peer information from the transport. )r% �getPeerr, r( r( r) r? ~ s zDummyChannel.getPeerc C s | j �� S )z: Get host information from the transport. )r% �getHostr, r( r( r) r@ � s zDummyChannel.getHostN)r7 r8 r9 r: r= r r? r@ r( r( r( r) r; f s r; c @ s� e Zd ZdZdd� Zdd� Zd&d d �Zd d � Zd d� Zdd� Z dd� Z d'dd�Z dd� Z dd� Z dd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� ZdS )(�ProxyClientTestsz# Tests for L{ProxyClient}. c C s: |� 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 � r c s s | ]}|� d �V qdS )� : N)�split)�.0�headerr( r( r) � <genexpr>� s z4ProxyClientTests._parseOutHeaders.<locals>.<genexpr>)rD �pop�dict)r Zcontentr �body� requestLiner( r( r) �_parseOutHeaders� s �z!ProxyClientTests._parseOutHeadersc C s 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 r1 r( r( r) � makeRequest� s z ProxyClientTests.makeRequest� GETN� c C s, |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 � requestBodyr1 r( r( r) �makeProxyClient� s �z ProxyClientTests.makeProxyClientc C s t � }||_|�|� |S )z� Connect a proxy client to a L{StringTransportWithDisconnection}. @param proxyClient: A L{ProxyClient}. @return: The L{StringTransportWithDisconnection}. )r Zprotocolr )r � proxyClientZclientTransportr( r( r) � connectProxy� s z ProxyClientTests.connectProxyc C s@ | � |� |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. )rY r% �valuerL r )r rX rK r ZrequestContentZ receivedLine�receivedHeadersrJ r( r( r) �assertForwardsHeaders� s � z&ProxyClientTests.assertForwardsHeadersc C s^ dt |��d� d | g}|D ]$\}}|D ]}|�|d | � q,q |�d|g� d�|�S )Ns HTTP/1.0 �ascii� rC rO rB )�str�encode�append�extend�join) r �code�messager rJ �linesrF �valuesrZ r( r( r) �makeResponseBytes� s z"ProxyClientTests.makeResponseBytesc C sj | � |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)r rT rd re r rJ r[ �expectedHeadersr( r( r) �assertForwardsResponse� s z'ProxyClientTests.assertForwardsResponseTc C s� | � d�}| �||ddi|�} | �| |d ddd��} | �| |� | �| �||||�� | �|||||� |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/ rP rQ s /foo HTTP/1.0� close)� connectionrP r N) rM rW r\ r r rh rq r% �loseConnectionZ assertFalseZ connected�finished) r rd re r rJ rU rV rt rT �clientZ receivedBodyr( r( r) �_testDataForward� s* � � � z!ProxyClientTests._testDataForwardc C s | � 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� bars baz� Some data �rw r, r( r( r) � test_forward s �z ProxyClientTests.test_forwardc C s | � ddddgfgddd�S )z~ Try to post content in the request, and check that the proxy client forward the body of the request. rx ry rz r{ r| � POST� Some contentr} r, r( r( r) � test_postData* s �z ProxyClientTests.test_postDatac C s | � ddg d�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 FoundrO r} r, r( r( r) �test_statusWithMessage3 s �z'ProxyClientTests.test_statusWithMessagec C s* 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 bazrx ry � Content-Lengthr] �rw r_ r r` �r �datar( r( r) �test_contentLength< s �z#ProxyClientTests.test_contentLengthc C s. d}| j dddtt|���d�gfg|dd�S )z� If the response contains a I{Content-Length} header, the outgoing connection is closed when all response body data has been received. r� rx ry r� r] F)rt r� r� r( r( r) �test_losesConnectionI s �z%ProxyClientTests.test_losesConnectionc C s0 t dddddd�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 � /foorS rQ r/ )rP s proxy-connectionrO Nrr �rP rs )r r r )r rv r( r( r) �test_headersCleanupsW s ��z%ProxyClientTests.test_headersCleanupsc C sF dddd�}|� � }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. rQ s 300� keep-alive)rP r� rs rr rs rN r� rS rO Ns GET /foo HTTP/1.0)�copyr r\ )r r rp rv r( r( r) �test_keepaliveNotForwardedc s � �z+ProxyClientTests.test_keepaliveNotForwardedc C s� | � 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/ s servers old-bars dates old-bazs content-types old/quxrP rQ )r r{ s 2010-01-01s application/x-baz)s Servers Date� Content-Typerx ry rO N) rM rl Z setRawHeadersrW rY r rh �itemsrq rk )r rT rv r r( r( r) � test_defaultHeadersOverriddenv s( �� �z.ProxyClientTests.test_defaultHeadersOverridden)rN NrO )rN rO T)r7 r8 r9 r: rL rM rW rY r\ rh rq rw r~ r� r� r� r� r� r� r� r( r( r( r) rA � s, � � # rA c @ s e Zd ZdZdd� Zdd� ZdS )�ProxyClientFactoryTestsz* Tests for L{ProxyClientFactory}. c C s� t dg�}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/ rN r� rS rP rQ � Ni� s Gateway errorr� rO s <H1>Could not connect</H1>r ) r r ZclientConnectionFailedr ri rj rk rl rm rc ro ru )r rT r' r( r( r) �test_connectionFailed� s$ � � �z-ProxyClientFactoryTests.test_connectionFailedc C sn t dddddidd�}|�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). rN r� rS rP rQ s Some dataNrr r� ) r r r r r �commandr r� r )r r' �protor( r( r) �test_buildProtocol� s � �z*ProxyClientFactoryTests.test_buildProtocolN)r7 r8 r9 r: r� r� r( r( r( r) r� � s r� c @ s: e Zd ZdZddd�Zdd� Zdd � Zd d � Zd d � ZdS )�ProxyRequestTestsz$ Tests for L{ProxyRequest}. rN rO c C s� t � }t|�}t� }t|d|�}|�t|�� |�|� |�|d| d� | �t|j �d� | �|j d d d� | �|j d d d� |j d d } | � | 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}. Fs http://example.comrS r r � example.com�P r r � example.comN)r r; r r � gotLengthr ZhandleContentChunk�requestReceivedr r r r r� �versionr r� r Zfather) r r r! rU r� r% r& r# rT r' r( r( r) � _testProcess� s( � z ProxyRequestTests._testProcessc C s | � 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 z ProxyRequestTests.test_processc C s | � dd�S )z� If the incoming request doesn't contain a slash, L{ProxyRequest.process} should add one when instantiating L{ProxyClientFactory}. rO rR r� r, r( r( r) � test_processWithoutTrailingSlash� s z2ProxyRequestTests.test_processWithoutTrailingSlashc C s | � 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� s �z&ProxyRequestTests.test_processWithDatac C sz t � }t|�}t� }t|d|�}|�d� |�ddd� | �t|j�d� | �|jd d d� | �|jd d d� d S ) z� Check that L{ProxyRequest.process} correctly parse port in the incoming URL, and create an outgoing connection with this port. Fr rN s http://example.com:1234/foo/barrS r r� r N) r r; r r r� r� r r r )r r% r& r# rT r( r( r) �test_processWithPort� s �z&ProxyRequestTests.test_processWithPortN)rN rO ) r7 r8 r9 r: r� r� r� r� r� r( r( r( r) r� � s r� c @ s e Zd ZdZdd� ZdS )� DummyFactoryz> A simple holder for C{host} and C{port} information. c C s || _ || _d S )N)r3 r2 )r r3 r2 r( r( r) r= s zDummyFactory.__init__N)r7 r8 r9 r: r= r( r( r( r) r� s r� c @ s e Zd ZdZdd� ZdS )�ReverseProxyRequestTestsz+ Tests for L{ReverseProxyRequest}. c C s� t � }t|�}t� }t|d|�}tdd�|_|�d� |�ddd� | �t |j �d� | �|j d d d� | �|j d d d� |j d d }| � |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� r r rN r� rS r r r r� N)r r; r r r� r' r� r� r r r r r r )r r% r& r# rT r' r( r( r) r� s z%ReverseProxyRequestTests.test_processN)r7 r8 r9 r: r� r( r( r( r) r� s r� N)r: Ztwisted.trial.unittestr Ztwisted.test.proto_helpersr r Ztwisted.web.resourcer Ztwisted.web.serverr Ztwisted.web.proxyr r r r r Ztwisted.web.test.test_webr r �objectr; rA r� r� r� r� r( r( r( r) �<module> s" S' ,T
Save File
Cancel