← Back
Editing: requesthelper.cpython-38.pyc
U W[^4 � @ s6 d Z ddlmZmZ ddgZddlmZ ddlmZm Z ddl m Z ddl m Z dd lmZ dd lmZ dd lmZmZ dd lmZmZ dd lmZ ddlmZ ddl m Z ddl m Z m Z m!Z! ddl"m#Z# ee�G dd� de$��Z%G dd� d�Z&G dd� de$�Z'e edddd�dd�e'j(�e'_(G dd � d ej)�Z*d S ) z2 Helpers related to HTTP requests, used by tests. � )�division�absolute_import� DummyChannel� DummyRequest)�BytesIO)� implementer�verify)� intToBytes)� deprecated)�Version)�Deferred)� IPv4Address� IPv6Address)� ISSLTransport�IAddress)�unittest)�Headers)�Resource)� NOT_DONE_YET�Session�Site)�FOUNDc @ s e Zd ZdZdS )� NullAddressz/ A null implementation of L{IAddress}. N)�__name__� __module__� __qualname__�__doc__� r r �@/usr/lib/python3/dist-packages/twisted/web/test/requesthelper.pyr s r c @ s� e Zd ZG dd� d�Zee�G dd� de��Zee� �Z d dd�Z dd � Z d d � Z d d � Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � ZdS ) r c @ sV e Zd ZdZdZddd�Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )zDummyChannel.TCP�P FNc C s, |d krt ddd�}|| _t� | _g | _d S )N�TCPz 192.168.1.1i80 )r �_peerr �written� producers��selfZpeerr r r �__init__- s zDummyChannel.TCP.__init__c C s | j S �N)r! �r% r r r �getPeer4 s zDummyChannel.TCP.getPeerc C s( t |t�std|f ��| j�|� d S )Nz+Can only write bytes to a transport, not %r)� isinstance�bytes� TypeErrorr" �write�r% �datar r r r- 7 s zDummyChannel.TCP.writec C s |D ]}| � |� qd S r' )r- )r% �iovecr/ r r r � writeSequence< s z DummyChannel.TCP.writeSequencec C s t dd| j�S )Nr z10.0.0.1)r �portr( r r r �getHost@ s zDummyChannel.TCP.getHostc C s | j �||f� d S r' )r# �append�r% ZproducerZ streamingr r r �registerProducerC s z!DummyChannel.TCP.registerProducerc C s d S r' r r( r r r �unregisterProducerF s z#DummyChannel.TCP.unregisterProducerc C s d| _ d S )NT)� disconnectedr( r r r �loseConnectionI s z DummyChannel.TCP.loseConnection)N) r r r r2 r8 r& r) r- r1 r3 r6 r7 r9 r r r r r ) s r c @ s e Zd ZdS )zDummyChannel.SSLN)r r r r r r r �SSLM s r: Nc C s | � |�| _d S r' )r � transportr$ r r r r& S s zDummyChannel.__init__c C s d S r' r )r% �requestr r r � requestDoneW s zDummyChannel.requestDonec C sL |d | d | d }|g}|� dd� |D �� |�d� | j�|� d S )N� � c s s" | ]\}}|d | d V qdS )s : r? Nr )�.0�name�valuer r r � <genexpr>^ s z,DummyChannel.writeHeaders.<locals>.<genexpr>)�extendr4 r; r1 )r% �version�code�reason�headersZ response_lineZheaderSequencer r r � writeHeaders[ s � zDummyChannel.writeHeadersc C s | j �� S r' )r; r) r( r r r r) e s zDummyChannel.getPeerc C s | j �� S r' )r; r3 r( r r r r3 i s zDummyChannel.getHostc C s | j �||� d S r' )r; r6 r5 r r r r6 m s z DummyChannel.registerProducerc C s | j �� d S r' )r; r7 r( r r r r7 q s z DummyChannel.unregisterProducerc C s | j �|� d S r' )r; r- r. r r r r- u s zDummyChannel.writec C s | j �|� d S r' )r; r1 )r% r0 r r r r1 y s zDummyChannel.writeSequencec C s | j �� d S r' )r; r9 r( r r r r9 } s zDummyChannel.loseConnectionc C s d S r' r r( r r r � endRequest� s zDummyChannel.endRequestc C s t | j| j�S r' )r* r; r: r( r r r �isSecure� s zDummyChannel.isSecure)N)r r r r r r r: r r Zsiter& r= rI r) r3 r6 r7 r- r1 r9 rJ rK r r r r r ( s $ c @ s� e Zd ZdZdZdZdZdd� Zdd� Zd2d d �Z d d � Z d d� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd3d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd4d.d/�Zd0d1� Z dS )5r a Represents a dummy or fake request. See L{twisted.web.server.Request}. @ivar _finishedDeferreds: L{None} or a C{list} of L{Deferreds} which will be called back with L{None} when C{finish} is called or which will be errbacked if C{processingFailed} is called. @type requestheaders: C{Headers} @ivar requestheaders: A Headers instance that stores values for all request headers. @type responseHeaders: C{Headers} @ivar responseHeaders: A Headers instance that stores values for all response headers. @type responseCode: C{int} @ivar responseCode: The response code which was passed to C{setResponseCode}. @type written: C{list} of C{bytes} @ivar written: The bytes which have been written to the request. s http://dummy/s GETNc C s d| _ | j r|�� qdS )z� Call an L{IPullProducer}'s C{resumeProducing} method in a loop until it unregisters itself. @param prod: The producer. @type prod: L{IPullProducer} @param s: Whether or not the producer is streaming. � N)�goZresumeProducing)r% Zprod�sr r r r6 � s z DummyRequest.registerProducerc C s d| _ d S )Nr )rM r( r r r r7 � s z DummyRequest.unregisterProducerc C sf g | _ g | _d| _|| _g | _d | _|p0td| �| _i | _t � | _ t � | _ d | _ g | _ d| _d| _d S )Nr s dummys HTTP/1.0)Zsitepathr" �finished�postpathZprepath�sessionr � protoSession�argsr �requestHeaders�responseHeaders� responseCode�_finishedDeferreds� _serverNameZ clientproto)r% rP rQ �clientr r r r& � s zDummyRequest.__init__c C s, i }| j �� D ]\}}|d ||�� < q|S )ah Return dictionary mapping the names of all received headers to the last value received for each. Since this method does not return all header information, C{self.requestHeaders.getAllRawHeaders()} may be preferred. NOTE: This function is a direct copy of C{twisted.web.http.Request.getAllRawHeaders}. ���)rT ZgetAllRawHeaders�lower)r% rH �k�vr r r � getAllHeaders� s zDummyRequest.getAllHeadersc C s | j �|�� dg�d S )aH Retrieve the value of a request header. @type name: C{bytes} @param name: The name of the request header for which to retrieve the value. Header names are compared case-insensitively. @rtype: C{bytes} or L{None} @return: The value of the specified request header. Nr )rT Z getRawHeadersr[ )r% rA r r r � getHeader� s zDummyRequest.getHeaderc C s | j �||� dS )zJTODO: make this assert on write() if the header is content-length N)rU � addRawHeader�r% rA rB r r r � setHeader� s zDummyRequest.setHeaderc C s( | j r | j S | jrtd��| j| _ | j S )Nz8Session cannot be requested after data has been written.)rQ r" �AssertionErrorrR r( r r r � getSession� s zDummyRequest.getSessionc C s, |� | �}|tkrdS | �|� | �� dS )a� Render the given resource as a response to this request. This implementation only handles a few of the most common behaviors of resources. It can handle a render method that returns a string or C{NOT_DONE_YET}. It doesn't know anything about the semantics of request methods (eg HEAD) nor how to set any particular headers. Basically, it's largely broken, but sufficient for some tests at least. It should B{not} be expanded to do all the same stuff L{Request} does. Instead, L{DummyRequest} should be phased out and L{Request} (or some other real code factored in a different way) used. N)�renderr r- �finish)r% �resource�resultr r r re � s zDummyRequest.renderc C s" t |t�std��| j�|� d S )Nzwrite() only accepts bytes)r* r+ r, r" r4 r. r r r r- s zDummyRequest.writec C s t � }| j�|� |S )z� Return a L{Deferred} which is called back with L{None} when the request is finished. This will probably only work if you haven't called C{finish} yet. )r rW r4 )r% rO r r r � notifyFinish s zDummyRequest.notifyFinishc C s: | j d | _ | jdk r6| j}d| _|D ]}|�d� q&dS )z} Record that the request is finished and callback and L{Deferred}s waiting for notification of this. rL N)rO rW �callback)r% � observers�obsr r r rf s zDummyRequest.finishc C s. | j dk r*| j }d| _ |D ]}|�|� qdS )zK Errback and L{Deferreds} waiting for finish notification. N)rW Zerrback)r% rG rk rl r r r �processingFailed* s z DummyRequest.processingFailedc C s |g| j |< d S r' )rS ra r r r �addArg5 s zDummyRequest.addArgc C s* | j rtdd�| j � ��|| _|| _dS )z{ Set the HTTP status response code, but takes care that this is called before any data is written. z<Response code cannot be set after data has been written: %s.�@@@@N)r" rc �joinrV ZresponseMessage)r% rF �messager r r �setResponseCode9 s z DummyRequest.setResponseCodec C s | j rtdd�| j � ��d S )Nz<Last-Modified cannot be set after data has been written: %s.ro �r" rc rp )r% Zwhenr r r �setLastModifiedC s z DummyRequest.setLastModifiedc C s | j rtdd�| j � ��d S )Nz3ETag cannot be set after data has been written: %s.ro rs )r% �tagr r r �setETagG s zDummyRequest.setETagc C s t | jttf�r| jjS dS )z| Return the IPv4 address of the client which made this request, if there is one, otherwise L{None}. N)r* rY r r �hostr( r r r � getClientIPK s zDummyRequest.getClientIPc C s | j dkrt� S | j S )z� Return the L{IAddress} of the client that made this request. @return: an address. @rtype: an L{IAddress} provider. N)rY r r( r r r �getClientAddressU s z DummyRequest.getClientAddressc C s | j S )z� Get a dummy hostname associated to the HTTP request. @rtype: C{bytes} @returns: a dummy hostname )rX r( r r r �getRequestHostnamea s z DummyRequest.getRequestHostnamec C s t ddd�S )zz Get a dummy transport's host. @rtype: C{IPv4Address} @returns: a dummy transport's host r � 127.0.0.1r )r r( r r r r3 k s zDummyRequest.getHostr c C sH || _ | �� rd}nd}||kr&|}n|d t|� }| j�d|� dS )a_ Change the host and port the request thinks it's using. @type host: C{bytes} @param host: The value to which to change the host header. @type ssl: C{bool} @param ssl: A flag which, if C{True}, indicates that the request is considered secure (if C{True}, L{isSecure} will return C{True}). i� r � :s hostN)Z _forceSSLrK r rT r` )r% rw r2 Zssl�defaultZ hostHeaderr r r �setHostu s zDummyRequest.setHostc C s | � t� | �d|� dS )zu Utility function that does a redirect. The request should have finish() called after this. s locationN)rr r rb )r% Zurlr r r �redirect� s zDummyRequest.redirect)NN)N)r ) r r r r Zuri�methodrY r6 r7 r& r^ r_ rb rd re r- ri rf rm rn rr rt rv rx ry rz r3 r~ r r r r r r � s4 ZTwisted� � ry )Z replacementc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �DummyRequestTestsz$ Tests for L{DummyRequest}. c C s^ t g �}|�� | j| jgd�}| �dt|�� |\}| �|�d�t� | �|�d�d� dS )zp L{DummyRequest.getClientIP} is deprecated in favor of L{DummyRequest.getClientAddress} )ZoffendingFunctionsrL �categoryrq z}twisted.web.test.requesthelper.DummyRequest.getClientIP was deprecated in Twisted 18.4.0; please use getClientAddress insteadN)r rx Z flushWarnings�test_getClientIPDeprecated� assertEqual�len�get�DeprecationWarning)r% r< �warningsZwarningr r r r� � s ��z,DummyRequestTests.test_getClientIPDeprecatedc C s. t g �}tddd�}||_| �d|�� � dS )z� L{DummyRequest.getClientIP} supports IPv6 addresses, just like L{twisted.web.http.Request.getClientIP}. r z::1�90 N)r r rY r� rx )r% r< rY r r r � test_getClientIPSupportsIPv6� s z.DummyRequestTests.test_getClientIPSupportsIPv6c C s t g �}|�� }t�t|� dS )zu L{DummyRequest.getClientAddress} returns an L{IAddress} provider no C{client} has been set. N)r ry r Z verifyObjectr )r% r< Znullr r r �"test_getClientAddressWithoutClient� s z4DummyRequestTests.test_getClientAddressWithoutClientc C s2 t g �}tddd�}||_|�� }| �||� dS )zI L{DummyRequest.getClientAddress} returns the C{client}. r r{ r� N)r r rY ry ZassertIs)r% r< rY Zaddressr r r �test_getClientAddress� s z'DummyRequestTests.test_getClientAddressN)r r r r r� r� r� r� r r r r r� � s r� N)+r Z __future__r r �__all__�ior Zzope.interfacer r Ztwisted.python.compatr Ztwisted.python.deprecater Z incrementalr Ztwisted.internet.deferr Ztwisted.internet.addressr r Ztwisted.internet.interfacesr r Z twisted.trialr Ztwisted.web.http_headersr Ztwisted.web.resourcer Ztwisted.web.serverr r r Ztwisted.web._responsesr �objectr r r rx ZSynchronousTestCaser� r r r r �<module> s8 b ��
Save File
Cancel