← Back
Editing: test_http.cpython-38.pyc
U s�@gv � @ s d Z ddlmZmZ ddlZddlZddlZddlZzddlmZm Z m Z W n( e k rt ddl mZm Z m Z Y nX ddl mZ ddlmZ ddlmZ ddlmZmZmZmZmZmZ dd lmZ dd l m Z dd l m Z dd l m!Z! dd l"m#Z#m$Z$m%Z% ddl&m'Z'm(Z( ddl&m)Z) ddl*m+Z+ ddl,m-Z- ddl.m/Z/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5m6Z6 ddl7m8Z8 ddl9m:Z: ddlm;Z;m<Z< ddl=m>Z> G dd� dee#j?��Z@dd � ZAG d d � d e#jB�ZCeAeC�ZDG d d � d e j!�ZEG d!d"� d"e#jB�ZFeAeF�ZGee%jH�G d#d$� d$eF��ZIeAeI�ZJG d%d&� d&eF�ZKeAeK�ZLG d'd(� d(e#jM�ZNd)d*� ZOG d+d,� d,eP�ZQG d-d.� d.e j!eQ�ZRG d/d0� d0eR�ZSG d1d2� d2eR�ZTG d3d4� d4eR�ZUG d5d6� d6e j!eQ�ZVG d7d8� d8e j!�ZWG d9d:� d:e j!�ZXG d;d<� d<e j!�ZYG d=d>� d>e j!�ZZd?d@� Z[G dAdB� dBe j!�Z\G dCdD� dDe!�Z]G dEdF� dFe j!�Z^G dGdH� dHe j!eQ�Z_G dIdJ� dJe j!�Z`G dKdL� dLe j!�ZaG dMdN� dNe#jM�ZbG dOdP� dPe j!�ZcG dQdR� dRe j!eQ�ZdG dSdT� dTe j!�ZeG dUdV� dVe j!eQ�ZfdWdX� ZgG dYdZ� dZe j!�ZhG d[d\� d\e j!�ZidS )]z Test HTTP support. � )�absolute_import�divisionN)�urlparse� urlunsplit� clear_cache)�BytesIO)�provider)� verifyObject)�_PY3� iterbytes�long� networkString�unicode� intToBytes)�proxyForInterface)�Failure)�unittest)�TestCase)�http� http_headers�iweb)�PotentialDataLoss� _DataLoss)�_IdentityTransferDecoder)�address)�Clock)�ConnectionLost�ConnectionDone)�loopback)�StringTransport�NonStreamingProducer�EventLoggingObserver)� DummyProducer)� DummyChannel��directlyProvides� providedBy)�globalLogPublisherc @ s e Zd ZdZdS )�._IDeprecatedHTTPChannelToRequestInterfaceProxyz� Proxy L{_IDeprecatedHTTPChannelToRequestInterface}. Used to assert that the interface matches what L{HTTPChannel} expects. N)�__name__� __module__� __qualname__�__doc__� r- r- �</usr/lib/python3/dist-packages/twisted/web/test/test_http.pyr( , s r( c s � fdd�}t |t� �� |S )a� Return a callable that proxies instances of C{clsToWrap} via L{_IDeprecatedHTTPChannelToRequestInterface}. @param clsToWrap: The class whose instances will be proxied. @type cls: L{_IDeprecatedHTTPChannelToRequestInterface} implementer. @return: A factory that returns L{_IDeprecatedHTTPChannelToRequestInterface} proxies. @rtype: L{callable} whose interface matches C{clsToWrap}'s constructor. c s � | |�}t |�S �N)r( )�args�kwargs�instance�� clsToWrapr- r. �_makeRequestProxyC s z3_makeRequestProxyFactory.<locals>._makeRequestProxyr$ )r4 r5 r- r3 r. �_makeRequestProxyFactory5 s r6 c @ s e Zd ZdZdd� ZdS )�DummyPullProducerHandlerz� An HTTP request handler that registers a dummy pull producer to serve the body. The owner must call C{finish} to complete the response. c C s& t | �| _| �d� | �| jd� d S )N�� F)r �_actualProducer�setResponseCode�registerProducer��selfr- r- r. �processT s z DummyPullProducerHandler.processN�r) r* r+ r, r> r- r- r- r. r7 M s r7 c @ s e Zd ZdZdd� Zdd� ZdS )� DateTimeTestsz Test date parsing functions.c C s> t d�D ]0}t�dd�}t�|�}t�|�}| �||� qd S )Ni' r i �5w)�range�randomZrandintr ZdatetimeToString�stringToDatetime� assertEqual)r= �i�timeZtimestrZtime2r- r- r. � testRoundtripb s zDateTimeTests.testRoundtripc C sV ddddddddg}t �d �}|D ]}| �t�|�|� q"| �t�d �t �d �� d S ) Ns Sun, 06 Nov 1994 08:49:37 GMTs 06 Nov 1994 08:49:37 GMTs Sunday, 06-Nov-94 08:49:37 GMTs 06-Nov-94 08:49:37 GMTs Sunday, 06-Nov-1994 08:49:37 GMTs 06-Nov-1994 08:49:37 GMTs Sun Nov 6 08:49:37 1994s Nov 6 08:49:37 1994) i� � � � �1 �% rI rI r s Thursday, 29-Sep-16 17:15:29 GMT) i� � � � � rN � i r )�calendarZtimegmrD r rC )r= Z dateStringsZdateIntZ dateStringr- r- r. �testStringToDatetimej s � �z"DateTimeTests.testStringToDatetimeN)r) r* r+ r, rG rS r- r- r- r. r@ _ s r@ c @ s e Zd Zdd� ZdS )�DummyHTTPHandlerc C s� | j �dd� | j �� }| �d�}|d kr6t|��� }d| d | d }| �d� | �d| j� | �d| j � | �d| j � | �d t t |��� | � |� | �� d S ) Nr � content-lengths ''' � r8 s Requests Commands Version� Content-Length)�content�seek�read� getHeader�str�encoder: � setHeader�uri�method� clientprotor �len�write�finish)r= �data�length�requestr- r- r. r> � s zDummyHTTPHandler.processN�r) r* r+ r> r- r- r- r. rT ~ s rT c @ s e Zd ZdZdd� ZdS )�DummyNewHTTPHandlerz� This is exactly like the DummyHTTPHandler but it takes only one argument in its constructor, with no default arguments. This exists to test an alternative code path in L{HTTPChannel}. c C s t �| |� d S r/ )rT �__init__)r= �channelr- r- r. rj � s z DummyNewHTTPHandler.__init__N)r) r* r+ r, rj r- r- r- r. ri � s ri c @ s e Zd ZdZdd� Zdd� ZdS )�DelayedHTTPHandlerzD Like L{DummyHTTPHandler}, but doesn't respond immediately. c C s d S r/ r- r< r- r- r. r> � s zDelayedHTTPHandler.processc C s t �| � d S r/ )rT r> r< r- r- r. �delayedProcess� s z!DelayedHTTPHandler.delayedProcessN)r) r* r+ r, r> rm r- r- r- r. rl � s rl c @ s e Zd Zdd� ZdS )�LoopbackHTTPClientc C s0 | � dd� | �dd� | �� | j�d� d S )N� GET� /foo/barrW � � 0123456789)Z sendCommandZ sendHeaderZ endHeaders� transportrc r< r- r- r. �connectionMade� s z!LoopbackHTTPClient.connectionMadeN)r) r* r+ rt r- r- r- r. rn � s rn c C s |j | j_ | S )a� Parametrizes the L{TimeoutMixin} so that it works with whatever reactor is being used by the test. @param protocol: A L{_GenericHTTPChannel} or something implementing a similar interface. @type protocol: L{_GenericHTTPChannel} @param reactor: An L{IReactorTime} implementation. @type reactor: L{IReactorTime} @return: The C{channel}, with its C{callLater} method patched. )� callLater�_channel)�protocol�reactorr- r- r. �parametrizeTimeoutMixin� s ry c @ s e Zd ZdZdd� ZdS )�ResponseTestMixinz� A mixin that provides a simple means of comparing an actual response string to an expected response string by performing the minimal parsing. c C s� |D ]�}|dd� |d }}|d }|dd� }|� dd�\}}|�� }|�d�} | �|| � | �t|�t|�� |dt|�� } |t|�d� }| �| |� qdS )a( Assert that the C{responses} matches the C{expected} responses. @type responses: C{bytes} @param responses: The bytes sent in response to one or more requests. @type expected: C{list} of C{tuple} of C{bytes} @param expected: The expected values for the responses. Each tuple element of the list represents one response. Each byte string element of the tuple is a full header line without delimiter, except for the last element which gives the full response body. N���r � � )�split� splitlines�poprD �setrb ) r= Z responses�expected�response�expectedHeadersZexpectedContentZexpectedStatus�headers�rest�statusrX r- r- r. �assertResponseEquals� s z&ResponseTestMixin.assertResponseEqualsN)r) r* r+ r, r� r- r- r- r. rz � s rz c @ s^ e Zd ZdZdgZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dS )� HTTP1_0Testss7 GET / HTTP/1.0 GET / HTTP/1.1 Accept: text/html )� HTTP/1.0 200 OK� Request: /� Command: GET� Version: HTTP/1.0� Content-Length: 13� ''' None ''' c C s` t � }t�� }t|_|�|� t| j�D ]}|�|� q(|� t d�� |� � }| � || j � dS )zZ Send requests over a channel and check responses match what is expected. zall oneN�r r � HTTPChannel�DummyHTTPHandlerProxy�requestFactory�makeConnectionr �requests� dataReceived�connectionLost�IOError�valuer� �expected_response�r= �b�a�byter� r- r- r. � test_buffer s zHTTP1_0Tests.test_bufferc C s� t � }t� }t�� }d|_|j|_|�|� |�d� |�d� | � |j � |�d� |�d� | � |j � |�d� | � t |j �d� dS )zq L{HTTPChannel} resets its timeout whenever data from a request body is delivered to it. �d �& POST / HTTP/1.0 Content-Length: 2 �c � xr| N)r r r r� �timeOutru r� r� �advance� assertFalse� disconnectingrD rb r� )r= �clockrs rw r- r- r. �test_requestBodyTimeout s z$HTTP1_0Tests.test_requestBodyTimeoutc C sl t � }t� }t�� }|�d�}t||�}|�|� |�d� |�d� | � |j � |�d� | � |j � dS )zA L{HTTPChannel}'s default timeout is 60 seconds. Nr� �; r| ) r r r � HTTPFactory� buildProtocolry r� r� r� r� r� � assertTrue�r= r� rs �factoryrw r- r- r. � test_requestBodyDefaultTimeout* s z+HTTP1_0Tests.test_requestBodyDefaultTimeoutc C s t �| t�}t� }t� }t�� }|�d�}t||�}|� |� |� d� | � |j � | � |j � |�d� | �|j � | � |j � | �dt|�� |d }| �d|d � |�d� | �|j � | � |j � |�d� | �|j � | �|j � | �d t|�� |d }| �d |d � dS ) �} If a timed out transport doesn't close after 15 seconds, the L{HTTPChannel} will forcibly close it. Nr� �< r| r zTiming out client: {peer}� log_format� � z"Forcibly timing out client: {peer})r! �createWithCleanupr' r r r r� r� ry r� r� r� r� � disconnectedr� r� � assertEqualsrb �assertIn)r= � logObserverr� rs r� rw �eventr- r- r. � test_transportForciblyClosed; s> � �z)HTTP1_0Tests.test_transportForciblyClosedc C s� t � }t� }t�� }|�d�}t||�}|�|� |�d� | �|j � | �|j � |� d� | � |j � | �|j � |� d� |� d� |� d� | � |j � | �|j � dS )z� If a timed out transport ends up calling C{connectionLost}, it prevents the force-closure of the transport. Nr� r� r� r| )r r r r� r� ry r� r� r� r� r� r� r� r� r� r- r- r. �+test_transportNotAbortedAfterConnectionLoste s" z8HTTP1_0Tests.test_transportNotAbortedAfterConnectionLostc C s� t � }t� }t�� }|�d�}d|j_t||�}|�|� |� d� | � |j � | � |j � |� d� | �|j � | � |j � |� d� | �|j � | � |j � dS )zh If the L{HTTPChannel} has its c{abortTimeout} set to L{None}, it never aborts. Nr� r� l )r r r r� r� rv Z abortTimeoutry r� r� r� r� r� r� r� r� r- r- r. �,test_transportNotAbortedWithZeroAbortTimeout� s z9HTTP1_0Tests.test_transportNotAbortedWithZeroAbortTimeoutc C s� t � }t� }t�� }|�d�}t||�}|�|� |�d� | �|j � | �|j � |� d� |� d� | � |j � | � |j � |� t� dS )r� Nr� r� rP )r r r r� r� ry r� r� r� r� r� r� r� r� r r� r- r- r. �"test_connectionLostAfterForceClose� s z/HTTP1_0Tests.test_connectionLostAfterForceClosec C s` t � }t�� }t|_|�|� t| j�D ]}|�|� q(|� t d�� |� � }| � || j � dS )zf Test that a L{http.Request} subclass with no queued kwarg works as expected. �all doneNr� r� r- r- r. �test_noPipeliningApi� s z!HTTP1_0Tests.test_noPipeliningApic C s� t � }t�� }t|_|�|� t| j�D ]}|�|� q(|� � }| � |d� | � dt |j�� |jr�| � dt |j�� |jd j }|� � q^|� � }| �|| j� dS )zW Test that pipelined requests get buffered, not processed in parallel. � r| r N)r r r� �DelayedHTTPHandlerProxyr� r� r r� r� r� rD rb �originalrm r� r� )r= r� r� r� r� rg r- r- r. �test_noPipelining� s z HTTP1_0Tests.test_noPipeliningN)r) r* r+ r� r� r� r� r� r� r� r� r� r� r� r- r- r- r. r� � s ��* r� c @ s e Zd ZdZddddgZdS )� HTTP1_1Testss� GET / HTTP/1.1 Accept: text/html POST / HTTP/1.1 Content-Length: 10 0123456789POST / HTTP/1.1 Content-Length: 10 0123456789HEAD / HTTP/1.1 )� HTTP/1.1 200 OKr� r� � Version: HTTP/1.1r� r� �r� r� � Command: POSTr� s Content-Length: 21� ''' 10 0123456789''' )r� r� s Command: HEADr� r� r� N�r) r* r+ r� r� r- r- r- r. r� � s ��r� c @ s e Zd ZdZdgZdS )�HTTP1_1_close_Tests�J GET / HTTP/1.1 Accept: text/html Connection: close GET / HTTP/1.0 )r� s Connection: closer� r� r� r� r� Nr� r- r- r- r. r� s � �r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� HTTP0_9Testss GET / � HTTP/1.1 400 Bad Request c C s | � ||� d S r/ )rD )r= r� ZexpectedResponser- r- r. r� . s z!HTTP0_9Tests.assertResponseEqualsc C s t �d��d S )NzHTTP/0.9 not supported)r ZSkipTestr< r- r- r. r� 2 s z HTTP0_9Tests.test_noPipeliningN)r) r* r+ r� r� r� r� r- r- r- r. r� & s �r� c @ sL e Zd ZdZdZddgZdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dS )�PipeliningBodyTestszl Pipelined requests get buffered and executed in the order received, not processed in parallel. st POST / HTTP/1.1 Content-Length: 10 0123456789POST / HTTP/1.1 Transfer-Encoding: chunked a 0123456789 0 r� )r� r� r� r� s Content-Length: 23s ''' None 0123456789''' c C s� t � }t�� }t|_|�|� t| j�D ]}|�|� q(| � |� � d� | � dt |j�� |jr�| � dt |j�� |jd j }|� � qZ| �|� � | j� dS )z� Imitate a slow connection that delivers one byte at a time. The request handler (L{DelayedHTTPHandler}) is puppeted to step through the handling of each request. r� r| r N)r r r� r� r� r� r r� r� rD r� rb r� rm r� �expectedResponses)r= r� r� r� rg r- r- r. �test_stepwiseTinyTubeX s z)PipeliningBodyTests.test_stepwiseTinyTubec C s� t � }t�� }t|_|�|� |�| j� | �|� � d� | �dt |j�� |jrz| �dt |j�� |jd j }|� � qL| � |� � | j� dS )z� Imitate a fast connection where several pipelined requests arrive in a single read. The request handler (L{DelayedHTTPHandler}) is puppeted to step through the handling of each request. r� r| r N)r r r� r� r� r� r� r� rD r� rb r� rm r� r� )r= r� r� rg r- r- r. �test_stepwiseDumpTruckr s z*PipeliningBodyTests.test_stepwiseDumpTruckc C s` t � }t�� }t|_|�|� t| j�D ] }|�|� | � t |j�d� q(| � |� � | j � dS )z� Imitate a slow connection that delivers one byte at a time. (L{DummyHTTPHandler}) immediately responds, but no more than one r| N)r r r� r� r� r� r r� r� ZassertLessEqualrb r� r� r� )r= r� r� r� r- r- r. �test_immediateTinyTube� s z*PipeliningBodyTests.test_immediateTinyTubec C s@ t � }t�� }t|_|�|� |�| j� | �|� � | j � dS )aR Imitate a fast connection where several pipelined requests arrive in a single read. The request handler (L{DummyHTTPHandler}) immediately responds. This doesn't check the at-most-one pending request invariant but exercises otherwise uncovered code paths. See GHSA-c8m8-j448-xjx7. N) r r r� r� r� r� r� r� r� r� r� )r= r� r� r- r- r. �test_immediateDumpTruck� s z+PipeliningBodyTests.test_immediateDumpTruckc C s\ t � }t�� }t|_|�|� |jt| j� d }|� | j| � | � |� � | j | � dS )a� Imitate a fast connection where a so many pipelined requests arrive in a single read that backpressure is indicated. The request handler (L{DummyHTTPHandler}) immediately responds. This doesn't check the at-most-one pending request invariant but exercises otherwise uncovered code paths. See GHSA-c8m8-j448-xjx7. @see: L{http.HTTPChannel._optimisticEagerReadSize} rq N) r r r� r� r� r� �_optimisticEagerReadSizerb r� r� r� r� r� )r= r� r� ZoverLimitCountr- r- r. �test_immediateABiggerTruck� s z.PipeliningBodyTests.test_immediateABiggerTruckc C s� t � }t�� }t|_|�|� |jt| j� }t d|d �D ],}|� | j� | � |j ddj |j |d�� q<|� | j� | �|j d� dS )z� When pipelined requests are received, we will optimistically continue receiving data up to a specified limit, then pause the transport. @see: L{http.HTTPChannel._optimisticEagerReadSize} r| � producingz(state was {state!r} after {x} iterations)�state�x�pausedN)r r r� r� r� r� r� rb r� rA r� rD � producerState�formatr� )r= r� r� Z underLimitr� r- r- r. �test_pipeliningReadLimit� s �� z,PipeliningBodyTests.test_pipeliningReadLimitN) r) r* r+ r, r� r� r� r� r� r� r� r� r- r- r- r. r� 7 s ��r� c @ s. e Zd ZdZG dd� dej�ZdZdd� ZdS )� ShutdownTestszM Tests that connections can be shut down by L{http.Request} objects. c @ s e Zd ZdZdd� ZdS )z!ShutdownTests.ShutdownHTTPHandlerzL A HTTP handler that just immediately calls loseConnection. c C s | � � d S r/ )�loseConnectionr< r- r- r. r> � s z)ShutdownTests.ShutdownHTTPHandler.processNr? r- r- r- r. �ShutdownHTTPHandler� s r� s1 POST / HTTP/1.1 Content-Length: 10 0123456789c C sT t � }t�� }t| j�|_|�|� |�| j� | � |j � |� � }| � |d� dS )zi Calling L{http.Request.loseConnection} causes the transport to be disconnected. r� N) r r r� r6 r� r� r� r� rg r� r� r� rD )r= r� r� r� r- r- r. �test_losingConnection� s z#ShutdownTests.test_losingConnectionN) r) r* r+ r, r �Requestr� rg r� r- r- r- r. r� � s �r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � SecurityTestsz] Tests that L{http.Request.isSecure} correctly takes the transport into account. c C s6 t �� }t�� }|�|� t�|�}| �|�� � dS )z� Calling L{http.Request.isSecure} when the channel is backed with a secure transport will return L{True}. N)r# �SSLr r� r� r� r� �isSecure�r= r� r� �reqr- r- r. � test_isSecure s zSecurityTests.test_isSecurec C s6 t �� }t�� }|�|� t�|�}| �|�� � dS )z� Calling L{http.Request.isSecure} when the channel is not backed with a secure transport will return L{False}. N)r# �TCPr r� r� r� r� r� r� r- r- r. �test_notSecure s z SecurityTests.test_notSecurec C sT t �� }t�� }|�|� t�|�}|j�|� |�d� |� � | � |� � � dS )zt After a request is finished, calling L{http.Request.isSecure} will always return L{False}. r8 N) r# r� r r� r� r� r� �appendr: rd r� r� r� r- r- r. �test_notSecureAfterFinish' s z'SecurityTests.test_notSecureAfterFinishN)r) r* r+ r, r� r� r� r- r- r- r. r� s r� c @ s� e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z e j sDd e _ dd� Z e j rXde _ dd� Zdd� Zdd� Zdd� Ze j s�d e_ dd� Ze j s�d e_ dS ) �GenericHTTPChannelTestsz� Tests for L{http._genericHTTPChannelProtocol}, a L{HTTPChannel}-alike which can handle different HTTP protocol channels. r� c C sH t �d�}t|_|�|� t| j�D ]}|�|� q$|�t d�� |j S )z{ Run a request using the specific instance of a transport. Returns the negotiated protocol string. r� r� ) r �"_genericHTTPChannelProtocolFactoryr� r� r� r r� r� r� r� Z_negotiatedProtocol)r= �tr� r� r- r- r. �'_negotiatedProtocolForTransportInstanceF s z?GenericHTTPChannelTests._negotiatedProtocolForTransportInstancec C s t � }| �|�}| �|d� dS )z� If the transport has no support for protocol negotiation (no negotiatedProtocol attribute), HTTP/1.1 is assumed. � http/1.1N)r r� rD �r= r� �negotiatedProtocolr- r- r. �test_protocolUnspecifiedU s z0GenericHTTPChannelTests.test_protocolUnspecifiedc C s& t � }d|_| �|�}| �|d� dS )z� If the transport has no support for protocol negotiation (returns None for negotiatedProtocol), HTTP/1.1 is assumed. Nr� �r r� r� rD r� r- r- r. �test_protocolNone_ s z)GenericHTTPChannelTests.test_protocolNonec C s& t � }d|_| �|�}| �|d� dS )zi If the transport reports that HTTP/1.1 is negotiated, that's what's negotiated. r� Nr� r� r- r- r. � test_http11j s z#GenericHTTPChannelTests.test_http11c C s& t � }d|_| �|�}| �|d� dS )z} If the transport reports that HTTP/2 is negotiated and HTTP/2 is present, that's what's negotiated. � h2Nr� r� r- r- r. �test_http2_presentu s z*GenericHTTPChannelTests.test_http2_presentzHTTP/2 support not presentc C s t � }d|_| �t| j|� dS )z� If the transport reports that HTTP/2 is negotiated and HTTP/2 is not present, an error is encountered. r� N)r r� � assertRaises� ValueErrorr� �r= r� r- r- r. �test_http2_absent� s �z)GenericHTTPChannelTests.test_http2_absentzHTTP/2 support presentc C s t � }d|_| �t| j|� dS )z� If the transport reports that a protocol other than HTTP/1.1 or HTTP/2 is negotiated, an error occurs. s smtpN)r r� r� �AssertionErrorr� r� r- r- r. �test_unknownProtocol� s �z,GenericHTTPChannelTests.test_unknownProtocolc C s$ t �d�}d|j_| �|jd� dS )zK The C{factory} attribute is taken from the inner channel. r� � FooN)r r� rv r� rD )r= r� r- r- r. � test_factory� s z$GenericHTTPChannelTests.test_factoryc C sB t � }tj|d�}|�d�}| �|j|j� | �|jj|j� dS )z� If C{callLater} is patched onto the L{http._GenericHTTPChannelProtocol} then we need to propagate it through to the backing channel. �rx N)r r r� r� rD ru rv )r= r� r� rw r- r- r. �*test_GenericHTTPChannelPropagatesCallLater� s zBGenericHTTPChannelTests.test_GenericHTTPChannelPropagatesCallLaterc C s� t � }tj|d�}|�d�}| �|j|j� | �|jj|j� t� }d|_t |_ |� |� |� d� | �|j|j� | �|jj|j� dS )z� If C{callLater} is patched onto the L{http._GenericHTTPChannelProtocol} then we need to propagate it across onto a new backing channel after upgrade. r Nr� � P) r r r� r� rD ru rv r r� rT r� r� r� �r= r� r� rw rs r- r- r. �'test_genericHTTPChannelCallLaterUpgrade� s z?GenericHTTPChannelTests.test_genericHTTPChannelCallLaterUpgradec C sf t � }d|_t�d�}t|_|�|� |j}| �|j |� |� d� | � |j |� | �|j |j� dS )z� The L{_GenericHTTPChannelProtocol} will unregister its proxy channel from the transport if upgrade is negotiated. r� r� r N) r r� r r� r� r� r� rv �assertIs�producerr� Z assertIsNot)r= rs ZgenericProtocolZoriginalChannelr- r- r. �test_unregistersProducer� s z0GenericHTTPChannelTests.test_unregistersProducerN)r) r* r+ r, r� r� r� r� r� r� r Z H2_ENABLED�skipr� r r r r r r- r- r- r. r� 8 s. � � r� c @ sZ e Zd Zddddd�ZdZdZdZdZdd� Zd d � Z d d � Z d d� Z dd� Z dd� Z dS )�HTTPLoopbackTestsrp ro � HTTP/1.0s 21)s requests commands versionrU r c C s" d| _ | �|d� | �|d� d S )Nr| r s 200)� gotStatusrD �r= �versionr� �messager- r- r. � _handleStatus� s z HTTPLoopbackTests._handleStatusc C s d| _ | �|d� d S )Nr| r� )� gotResponserD )r= re r- r- r. �_handleResponse s z!HTTPLoopbackTests._handleResponsec C s& | j d | _ | �| j|�� |� d S )Nr| )� numHeadersrD r� �lower)r= �keyr� r- r- r. � _handleHeader s z HTTPLoopbackTests._handleHeaderc C s d| _ | �| jd� d S )Nr| � )� gotEndHeadersrD r r< r- r- r. �_handleEndHeaders s z#HTTPLoopbackTests._handleEndHeadersc C sP t �� }t|_t� }| j|_| j|_| j |_ | j |_ t �||�}|�| j� |S r/ )r r� r� r� rn r ZhandleResponser � handleHeaderr �handleEndHeadersr � handleStatusr Z loopbackAsync� addCallback�_cbTestLoopback)r= Zserver�client�dr- r- r. � testLoopback s z HTTPLoopbackTests.testLoopbackc C s> | j r| jr| js*td| j | j| jg ��| `| `| ` | `d S )Nzdidn't got all callbacks %s)r r r � RuntimeErrorr )r= Zignoredr- r- r. r! s ��z!HTTPLoopbackTests._cbTestLoopbackN)r) r* r+ r� r r r r r r r r r$ r! r- r- r- r. r � s � r c K s6 t �t� d�}| �� D ]\}}|j�t|�|� q|S )zR Make a request with the given request headers for the persistence tests. F)r r� r# �items�requestHeaders� setRawHeadersr )r� rg Z headerName�vr- r- r. � _prequest( s r* c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �PersistenceTestsz0 Tests for persistent HTTP connections. c C s t �� | _t� | _d S r/ )r r� rk r* rg r< r- r- r. �setUp7 s zPersistenceTests.setUpc C s6 | j �| jd�}| �|� | �g t| jj�� �� dS )zl After being used for an I{HTTP/0.9} request, the L{HTTPChannel} is not persistent. s HTTP/0.9N�rk �checkPersistencerg r� rD �list�responseHeaders�getAllRawHeaders�r= �persistr- r- r. � test_http09< s �z PersistenceTests.test_http09c C s6 | j �| jd�}| �|� | �g t| jj�� �� dS )zl After being used for an I{HTTP/1.0} request, the L{HTTPChannel} is not persistent. r Nr- r2 r- r- r. � test_http10G s �z PersistenceTests.test_http10c C s6 | j �| jd�}| �|� | �g t| jj�� �� dS )zh After being used for an I{HTTP/1.1} request, the L{HTTPChannel} is persistent. � HTTP/1.1N)rk r. rg r� rD r/ r0 r1 r2 r- r- r. r� R s �z PersistenceTests.test_http11c C sF t dgd�}| j�|d�}| �|� | �ddgfgt|j�� �� dS )z� After being used for an I{HTTP/1.1} request with a I{Connection: Close} header, the L{HTTPChannel} is not persistent. s close)Z connectionr6 s ConnectionN)r* rk r. r� rD r/ r0 r1 )r= rg r3 r- r- r. �test_http11Close] s �z!PersistenceTests.test_http11CloseN) r) r* r+ r, r, r4 r5 r� r7 r- r- r- r. r+ 3 s r+ c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )� IdentityTransferEncodingTestsz0 Tests for L{_IdentityTransferDecoder}. c C s. g | _ g | _d| _t| j| j j| jj�| _dS )z� Create an L{_IdentityTransferDecoder} with callbacks hooked up so that calls to them can be inspected. rq N)re rd � contentLengthr r� �decoderr< r- r- r. r, o s �z#IdentityTransferEncodingTests.setUpc C s< | j �d| j � | �| jd| j g� | �| jdg� dS )aC If L{_IdentityTransferDecoder.dataReceived} is called with a byte string with length equal to the content length passed to L{_IdentityTransferDecoder}'s initializer, the data callback is invoked with that string and the finish callback is invoked with a zero-length string. r� r� N�r: r� r9 rD re rd r< r- r- r. �test_exactAmountReceived{ s z6IdentityTransferEncodingTests.test_exactAmountReceivedc C sp | j �d� | �| jdg� | �| jg � | j �d| jd � | �| jdd| jd g� | �| jdg� dS )a3 If L{_IdentityTransferDecoder.dataReceived} is called multiple times with byte strings which, when concatenated, are as long as the content length provided, the data callback is invoked with each string and the finish callback is invoked only after the second call. r� � yr| r� N)r: r� rD re rd r9 r< r- r- r. �test_shortStrings� s z/IdentityTransferEncodingTests.test_shortStringsc C s@ | j �d| j d � | �| jd| j g� | �| jdg� dS )a3 If L{_IdentityTransferDecoder.dataReceived} is called with a byte string with length greater than the provided content length, only the prefix of that string up to the content length is passed to the data callback and the remainder is passed to the finish callback. r� r= Nr; r< r- r- r. �test_longString� s z-IdentityTransferEncodingTests.test_longStringc sj g �� �fdd�}t d| jj|�� � �d� | ��g � � �d� �d �t� | �t�d j�d� dS ) z� If data is passed to L{_IdentityTransferDecoder.dataReceived} after the finish callback has been invoked, C{RuntimeError} is raised. c s, z� � d� W n ��t� � Y nX d S )N� foo)r� r� r ��bytes�r: Zfailuresr- r. rd � s zJIdentityTransferEncodingTests.test_rejectDataAfterFinished.<locals>.finish� s xxxxr= r z;_IdentityTransferDecoder cannot decode data after finishingN) r re r� r� rD Ztrapr% r\ r� )r= rd r- rC r. � test_rejectDataAfterFinished� s �z:IdentityTransferEncodingTests.test_rejectDataAfterFinishedc C sZ g }g }t d|j|j�}|�d� | �|dg� |�d� | �|ddg� | �|g � dS )z� If L{_IdentityTransferDecoder} is constructed with L{None} for the content length, it passes all data delivered to it through to the data callback. Nr� r= )r r� r� rD )r= re rd r: r- r- r. �test_unknownContentLength� s z7IdentityTransferEncodingTests.test_unknownContentLengthc C s | � |jd� | � |jd� dS )z� Check the decoder's data and finish callbacks and make sure they are None in order to help avoid references cycles. N)�assertIdenticalZ dataCallbackZfinishCallback)r= r: r- r- r. � _verifyCallbacksUnreferenced� s z:IdentityTransferEncodingTests._verifyCallbacksUnreferencedc C s6 | j �d| jd � | �t| j j� | �| j � dS )z� L{_IdentityTransferDecoder.noMoreData} raises L{_DataLoss} if it is called and the content length is known but not enough bytes have been delivered. r� r| N)r: r� r9 r� r � noMoreDatarH r< r- r- r. �test_earlyConnectionLose� s z6IdentityTransferEncodingTests.test_earlyConnectionLosec C sN g }g }t d|j|j�}| �t|j� | �|g � | �|dg� | �|� dS )z� L{_IdentityTransferDecoder.noMoreData} calls the finish callback and raises L{PotentialDataLoss} if it is called and the content length is unknown. Nr� )r r� r� r rI rD rH )r= �body�finishedr: r- r- r. �'test_unknownContentLengthConnectionLose� s zEIdentityTransferEncodingTests.test_unknownContentLengthConnectionLosec C s, | j �d| j � | j �� | �| j � dS )z� L{_IdentityTransferDecoder.noMoreData} does not raise any exception if it is called when the content length is known and that many bytes have been delivered. r� N)r: r� r9 rI rH r< r- r- r. �test_finishedConnectionLose� s z9IdentityTransferEncodingTests.test_finishedConnectionLoseN)r) r* r+ r, r, r< r> r? rE rF rH rJ rM rN r- r- r- r. r8 k s r8 c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )� ChunkedTransferEncodingTestsz� Tests for L{_ChunkedTransferDecoder}, which turns a byte stream encoded using HTTP I{chunked} C{Transfer-Encoding} back into the original byte stream. c C s< g }t �|jd�}|�d� |�d� | �|dddg� dS )z� L{_ChunkedTransferDecoder.dataReceived} decodes chunked-encoded data and passes the result to the specified callback. Ns 3 abc 5 12345 s a 0123456789 � abcs 12345rr �r �_ChunkedTransferDecoderr� r� rD �r= �L�pr- r- r. � test_decoding� s z*ChunkedTransferEncodingTests.test_decodingc C s^ g }g }t �|j|j�}td�D ]}|�|� q | �|dddddddd g� | �|d g� d S ) z{ L{_ChunkedTransferDecoder.dataReceived} decodes chunks broken up and delivered in multiple calls. s 3 abc 5 12345 0 � a� b� c� 1� 2� 3� 4� 5r� N)r rR r� r r� rD )r= rT rL rU �sr- r- r. � test_short s z'ChunkedTransferEncodingTests.test_shortc C s. g }t �|jd�}|�d� | �|dg� dS )z L{_ChunkedTransferDecoder.dataReceived} doesn't treat CR LF pairs embedded in chunk bodies specially. Ns 2 � rQ rS r- r- r. � test_newlines s z*ChunkedTransferEncodingTests.test_newlinesc C s. g }t �|jd�}|�d� | �|dg� dS )zd L{_ChunkedTransferDecoder.dataReceived} disregards chunk-extension fields. Ns 3; x-foo=bar abc rP rQ rS r- r- r. �test_extensions" s z,ChunkedTransferEncodingTests.test_extensionsc C s. g }t �d|j�}|�d� | �|dg� dS )z� L{_ChunkedTransferDecoder.dataReceived} interprets a zero-length chunk as the end of the chunked data stream and calls the completion callback. N� 0 r� rQ �r= rL rU r- r- r. � test_finish- s z(ChunkedTransferEncodingTests.test_finishc C s. g }t �d|j�}|�d� | �|dg� dS )z� L{_ChunkedTransferDecoder.dataReceived} passes any bytes which come after the terminating zero-length chunk to the completion callback. Ns 0 hello� hellorQ re r- r- r. � test_extra9 s z'ChunkedTransferEncodingTests.test_extrac C s. t �ddd� �}|�d� | �t|jd� dS )z� L{_ChunkedTransferDecoder.dataReceived} raises C{RuntimeError} if it is called after it has seen the last chunk. Nc S s d S r/ r- rA r- r- r. �<lambda>I r� zAChunkedTransferEncodingTests.test_afterFinished.<locals>.<lambda>rd rg )r rR r� r� r% )r= rU r- r- r. �test_afterFinishedD s z/ChunkedTransferEncodingTests.test_afterFinishedc C s< t �ddd� �}|�d� | �t|j�}| �t|�d� dS )z� L{_ChunkedTransferDecoder.noMoreData} raises L{_DataLoss} if it is called and the end of the last trailer has not yet been received. Nc S s d S r/ r- rA r- r- r. ri S r� zGChunkedTransferEncodingTests.test_earlyConnectionLose.<locals>.<lambda>s 0 zYChunked decoder in 'TRAILER' state, still expecting more data to get to 'FINISHED' state.)r rR r� r� r rI rD r\ )r= �parser�excr- r- r. rJ N s �z5ChunkedTransferEncodingTests.test_earlyConnectionLosec C s& t �ddd� �}|�d� |�� dS )z� L{_ChunkedTransferDecoder.noMoreData} does not raise any exception if it is called after the terminal zero length chunk is received. Nc S s d S r/ r- rA r- r- r. ri a r� zJChunkedTransferEncodingTests.test_finishedConnectionLose.<locals>.<lambda>rd )r rR r� rI )r= rk r- r- r. rN \ s z8ChunkedTransferEncodingTests.test_finishedConnectionLosec sL g � g �� ��fdd�}t �d|����d� | �� g � | ��dg� dS )z� L{_ChunkedTransferDecoder.noMoreData} can be called from the finished callback without raising an exception. c s4 z �� � W n � �t� � Y n X ��d� d S �NT)rI r� r )Zextra��errorsrk Z successesr- r. rL m s zOChunkedTransferEncodingTests.test_reentrantFinishedNoMoreData.<locals>.finishedNrd T)r rR r� rD )r= rL r- rn r. � test_reentrantFinishedNoMoreDataf s z=ChunkedTransferEncodingTests.test_reentrantFinishedNoMoreDataN)r) r* r+ r, rV r` rb rc rf rh rj rJ rN rp r- r- r- r. rO � s rO c @ s4 e Zd ZddddddgZdd� Zd d � Zd d � Zd S )� ChunkingTestss abcvr� s fdfsd423s Ffasfas s 523523 fsdfs 4234c C sF | j D ]*}d�t�|��}| �|dft�|�� q| �ttjd� d S )Nr� s -5 malformed! )�strings�joinr �toChunkrD � fromChunkr� r� )r= r_ �chunkedr- r- r. � testChunks� s zChunkingTests.testChunksc C sx d� dd� | jD ��}g }d}t|�D ]>}|| }z t�|�\}}|�|� W q& tk rb Y q&X q&| �|| j� d S )Nr� c S s g | ]}d � t�|���qS )r� )rs r rt )�.0r� r- r- r. � <listcomp>� s z8ChunkingTests.testConcatenatedChunks.<locals>.<listcomp>)rs rr r r ru r� r� rD )r= rv �result�buffer�cre r- r- r. �testConcatenatedChunks� s z$ChunkingTests.testConcatenatedChunksc C sn t � }t�� }|�|� t�|d�}|�d� d|_|j�ddg� |� d� |� d� | � |� � dg� d S ) zV Test that the L{HTTPChannel} correctly chunks responses when needed. Fr8 r6 � test� lemur� Hello� World!�r� � Test: lemur� Transfer-Encoding: chunkeds 5 Hello 6 World! N) r r r� r� r� r: ra r0 r( rc r� r� )r= �transrk r� r- r- r. �test_chunkedResponses� s �z#ChunkingTests.test_chunkedResponsesN)r) r* r+ rr rw r} r� r- r- r- r. rq { s � rq c @ s e Zd ZdZdd� ZdEdd�Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zer�d(e_d)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%dS )F� ParsingTestsz7 Tests for protocol parsing in L{HTTPChannel}. c C s d| _ d S )NF�� didRequestr< r- r- r. r, � s zParsingTests.setUpNTc C s� |s t �� }|rt|�|_|�dd�}t� }|�|� t|�D ]}|jj rN qZ|� |� q>|� t d�� |rz| � | j� n | �| j� |S )a� Execute a web request based on plain text content. @param httpRequest: Content for the request which is processed. Each L{" "} will be replaced with L{" "}. @type httpRequest: C{bytes} @param requestFactory: 2-argument callable returning a Request. @type requestFactory: C{callable} @param success: Value to compare against I{self.didRequest}. @type success: C{bool} @param channel: Channel instance over which the request is processed. @type channel: L{HTTPChannel} @return: Returns the channel used for processing the request. @rtype: L{HTTPChannel} rV ra r� )r r� r6 r� �replacer r� r rs r� r� r� r� r� r� r� )r= � httpRequestr� �successrk rs r� r- r- r. � runRequest� s zParsingTests.runRequestc sd d� |�}g � G � fdd�dtj�}| j||dd�}| �|j�� d� | �|jj� | �� g � dS )aJ Execute a HTTP request and assert that it is rejected with a 400 Bad Response and disconnection. @param requestLines: Plain text lines of the request. These lines will be joined with newlines to form the HTTP request that is processed. @type requestLines: C{list} of C{bytes} rV c s e Zd Z� fdd�ZdS )z5ParsingTests.assertRequestRejected.<locals>.MyRequestc s � � | � | �� d S r/ �r� rd r< �� processedr- r. r> � s z=ParsingTests.assertRequestRejected.<locals>.MyRequest.processNrh r- r� r- r. � MyRequest� s r� F�r� r� N) rs r r� r� rD rs r� r� r� )r= � requestLinesr� r� rk r- r� r. �assertRequestRejected� s �z"ParsingTests.assertRequestRejectedc s\ g � G � fdd�dt j�}d}| �||d�}| �|j�� d� | �|jj� | �� g � dS )z� When client sends invalid HTTP method containing non-ascii characters HTTP 400 'Bad Request' status will be returned. c s e Zd Z� fdd�ZdS )z:ParsingTests.test_invalidNonAsciiMethod.<locals>.MyRequestc s � � | � | �� d S r/ r� r< r� r- r. r> s zBParsingTests.test_invalidNonAsciiMethod.<locals>.MyRequest.processNrh r- r� r- r. r� s r� s GE© / HTTP/1.1 r r� N)r r� r� rD rs r� r� r� )r= r� ZbadRequestLinerk r- r� r. �test_invalidNonAsciiMethod� s �z'ParsingTests.test_invalidNonAsciiMethodc sx g � G � fdd�dt j�}dD ]T\}}t�d�||f���� }d| d }| �||d� � �� }| �||f|j � q dS ) z� L{HTTPChannel} provides username and password information supplied in an I{Authorization} header to the L{Request} which makes it available via its C{getUser} and C{getPassword} methods. c s e Zd Z� fdd�ZdS )z,ParsingTests.test_basicAuth.<locals>.Requestc s | � � | �� f| _� �| � d S r/ �ZgetUserZ getPassword� credentialsr� r< �r� r- r. r> s z4ParsingTests.test_basicAuth.<locals>.Request.processNrh r- r� r- r. r� s r� ))r@ � bar)rg s there:z� :�$ GET / HTTP/1.0 Authorization: Basic � r N) r r� �base64� encodestringrs �stripr� r� rD r� )r= r� �urU r_ �fr� r- r� r. �test_basicAuth s zParsingTests.test_basicAuthc sv g � G � fdd�dt j�}ddddddg}| �d�|�|d � � \}| �|j�d �d g� | �|j�d �d dg� dS )zo Headers received by L{HTTPChannel} in a request are made available to the L{Request}. c s e Zd Z� fdd�ZdS )z,ParsingTests.test_headers.<locals>.MyRequestc s � � | � | �� d S r/ r� r< r� r- r. r> * s z4ParsingTests.test_headers.<locals>.MyRequest.processNrh r- r� r- r. r� ) s r� � GET / HTTP/1.0� Foo: bars baz: Quuxs baz: quuxr� rV r r@ r� s bAzs Quux� quuxN)r r� r� rs rD r' � getRawHeaders)r= r� r� rg r- r� r. � test_headers# s& � � �zParsingTests.test_headersc C sL dg}t tjjd �D ]}|�td|f �� q|�ddg� | �|� dS )z� C{HTTPChannel} enforces a limit of C{HTTPChannel.maxHeaders} on the number of headers received per request. r� r� z%s: foor� N)rA r r� � maxHeadersr� r �extendr� )r= r� rE r- r- r. �test_tooManyHeaders> s z ParsingTests.test_tooManyHeadersc C s | � ddddg� dS )z� If a I{Content-Length} header with a non-integer value is received, a 400 (Bad Request) response is sent to the client and the connection is closed. r� s Content-Length: xr� N�r� r< r- r- r. � test_invalidContentLengthHeaderK s z,ParsingTests.test_invalidContentLengthHeaderc C s | � ddddg� dS )z� If a header without colon is received a 400 (Bad Request) response is sent to the client and the connection is closed. r� s HeaderName r� Nr� r< r- r- r. �test_invalidHeaderNoColonT s z&ParsingTests.test_invalidHeaderNoColonc s� g � G � fdd�dt j�}| �t jdd� ddddddddg}| �d �|�|d �}� \}}| �|�d �d � | �|�d �d � | �|j� � d� dS )z� C{HTTPChannel} enforces the limit of C{HTTPChannel.maxHeaders} per request so that headers received in an earlier request do not count towards the limit when processing a later request. c s e Zd Z� fdd�ZdS )z:ParsingTests.test_headerLimitPerRequest.<locals>.MyRequestc s � � | � | �� d S r/ r� r< r� r- r. r> d s zBParsingTests.test_headerLimitPerRequest.<locals>.MyRequest.processNrh r- r� r- r. r� c s r� r� r| s GET / HTTP/1.1r� r� s Bar: bazrV r r@ r� � baz�h HTTP/1.1 200 OK Transfer-Encoding: chunked 0 HTTP/1.1 200 OK Transfer-Encoding: chunked 0 N) r r� �patchr� r� rs rD r[ rs r� )r= r� r� rk �first�secondr- r� r. �test_headerLimitPerRequest\ s( � �z'ParsingTests.test_headerLimitPerRequestc s` g � G � fdd�dt j�}t �� }d|_d}| j|||dd�}| �� g � | �|j�� d� dS ) z� Enforces a limit of C{HTTPChannel.totalHeadersSize} on the size of headers received per request starting from initial command line. c s e Zd Z� fdd�ZdS )z@ParsingTests.test_headersTooBigInitialCommand.<locals>.MyRequestc s � � | � | �� d S r/ r� r< r� r- r. r> � s zHParsingTests.test_headersTooBigInitialCommand.<locals>.MyRequest.processNrh r- r� r- r. r� � s r� rq s" GET /path/longer/than/10 HTTP/1.1 F�r� r� rk r� r� N�r r� r� �totalHeadersSizer� rD rs r� �r= r� rk r� r- r� r. � test_headersTooBigInitialCommand� s � �z-ParsingTests.test_headersTooBigInitialCommandc s` g � G � fdd�dt j�}t �� }d|_d}| j|||dd�}| �� g � | �|j�� d� dS ) z� Enforces a limit of C{HTTPChannel.totalHeadersSize} on the size of headers received per request counting first line and total headers. c s e Zd Z� fdd�ZdS )z>ParsingTests.test_headersTooBigOtherHeaders.<locals>.MyRequestc s � � | � | �� d S r/ r� r< r� r- r. r> � s zFParsingTests.test_headersTooBigOtherHeaders.<locals>.MyRequest.processNrh r- r� r- r. r� � s r� �( s5 GET /less/than/40 HTTP/1.1 Some-Header: less-than-40 Fr� r� Nr� r� r- r� r. � test_headersTooBigOtherHeaders� s" � � �z+ParsingTests.test_headersTooBigOtherHeadersc C sP G dd� dt j�}t �� }d|_||_d}| j||dd�}| �|j�� d� dS ) z� Enforces total size of headers per individual request and counter is reset at the end of each request. c @ s e Zd Zdd� ZdS )z@ParsingTests.test_headersTooBigPerRequest.<locals>.SimpleRequestc S s | � � d S r/ )rd r< r- r- r. r> � s zHParsingTests.test_headersTooBigPerRequest.<locals>.SimpleRequest.processNrh r- r- r- r. � SimpleRequest� s r� r� sZ GET / HTTP/1.1 Some-Header: total-less-than-60 GET / HTTP/1.1 Some-Header: less-than-60 F)r� rk r� r� N) r r� r� r� r� r� rD rs r� )r= r� rk r� r- r- r. � test_headersTooBigPerRequest� s � ��z)ParsingTests.test_headersTooBigPerRequestc sH d}i � | �G � �fdd�dt j�}| �||� | �� dddd�� dS ) z3 Test cookies parsing and reading. sK GET / HTTP/1.0 Cookie: rabbit="eat carrot"; ninja=secret; spam="hey 1=1!" c s e Zd Z� �fdd�ZdS )z+ParsingTests.testCookies.<locals>.MyRequestc s* dD ]}| � |�� |< qd�_| �� d S )N�s rabbits ninjas spamT)Z getCookier� rd )r= �name��cookies�testcaser- r. r> � s z3ParsingTests.testCookies.<locals>.MyRequest.processNrh r- r� r- r. r� � s r� s "eat carrot"s secrets "hey 1=1!"r� N�r r� r� rD �r= r� r� r- r� r. � testCookies� s ��zParsingTests.testCookiesc sb d}g �g � | �G � ��fdd�dt j�}| �||� | ��dg� | �� dgdgddgg� d S ) NsJ GET /?key=value&multiple=two+words&multiple=more%20words&empty= HTTP/1.0 c s e Zd Z� ��fdd�ZdS )z'ParsingTests.testGET.<locals>.MyRequestc s@ �� | j� � �| jd | jd | jd g� d�_| �� d S �Ns keys emptys multipleT)r� r` r� r0 r� rd r< �r0 r` r� r- r. r> s �z/ParsingTests.testGET.<locals>.MyRequest.processNrh r- r� r- r. r� s r� ro � valuer� � two words� more wordsr� r� r- r� r. �testGET s �zParsingTests.testGETc sp d}g �g �g � | �G � ���fdd�dt j�}| �||� | ��dg� | ��dg� | �� dgdgg� dS ) z� While only a single '?' is allowed in an URL, several other servers allow several and pass all after the first through as part of the query arguments. Test that we emulate this behavior. s" GET /foo?bar=?&baz=quux HTTP/1.0 c s e Zd Z� ���fdd�ZdS )z6ParsingTests.test_extraQuestionMark.<locals>.MyRequestc sD �� | j� �� | j� � �| jd | jd g� d�_| �� d S )Nr� r� T)r� r` �pathr� r0 r� rd r< �r0 r` r� r� r- r. r> ' s z>ParsingTests.test_extraQuestionMark.<locals>.MyRequest.processNrh r- r� r- r. r� & s r� ro s /foo� ?r� Nr� r� r- r� r. �test_extraQuestionMark s z#ParsingTests.test_extraQuestionMarkc s� d}t dt|�|f �}g �g � g �| �G � ���fdd�dtj�}| �||� | ��dg� | �� dgdgdd gg� | ��t |�g� d S ) a] The request body of a I{POST} request with a I{Content-Type} header of I{application/x-www-form-urlencoded} is parsed according to that content type and made available in the C{args} attribute of the request object. The original bytes of the request may still be read from the C{content} attribute. z9key=value&multiple=two+words&multiple=more%20words&empty=zVPOST / HTTP/1.0 Content-Length: %d Content-Type: application/x-www-form-urlencoded %sc s e Zd Z� ���fdd�ZdS )z4ParsingTests.test_formPOSTRequest.<locals>.MyRequestc sP �� | j� � �| jd | jd | jd g� �� | j�� � d�_| �� d S r� )r� r` r� r0 rX rZ r� rd r< �r0 rX r` r� r- r. r> I s �z<ParsingTests.test_formPOSTRequest.<locals>.MyRequest.processNrh r- r� r- r. r� H s r� s POSTr� r� r� r� N)r rb r r� r� rD )r= �queryr� r� r- r� r. �test_formPOSTRequest4 s � �z!ParsingTests.test_formPOSTRequestc C s, d}| j |tjdd�}| �|j�� d� dS )zr If the C{Content-Disposition} header is missing, the request is denied as a bad request. s� POST / HTTP/1.0 Content-Type: multipart/form-data; boundary=AaB03x Content-Length: 103 --AaB03x Content-Type: text/plain Content-Transfer-Encoding: quoted-printable abasdfg --AaB03x-- Fr� r� N�r� r r� rD rs r� �r= r� rk r- r- r. � test_missingContentDisposition[ s �z+ParsingTests.test_missingContentDispositionz<cgi.parse_multipart is much more error-tolerant on Python 3.c C s, d}| j |tjdd�}| �|j�� d� dS )zX When the multipart processing fails the client gets a 400 Bad Request. s POST / HTTP/1.0 Content-Type: multipart/form-data; boundary=☃ Content-Length: 103 --☃ Content-Type: text/plain Content-Length: 999999999999999999999999999999999999999999999999999999999999999 Content-Transfer-Encoding: quoted-printable abasdfg --☃-- Fr� r� Nr� r� r- r- r. � test_multipartProcessingFailurev s �z,ParsingTests.test_multipartProcessingFailurec sV g � G � fdd�dt j�}d}| j||dd�}| �|j�� d� | �� d ji � dS ) zS When the multipart does not contain a header is should be skipped c s e Zd Z� fdd�ZdS )zJParsingTests.test_multipartEmptyHeaderProcessingFailure.<locals>.MyRequestc s � � | � | �d� | �� d S �Ns done�r� rc rd r< r� r- r. r> � s zRParsingTests.test_multipartEmptyHeaderProcessingFailure.<locals>.MyRequest.processNrh r- r� r- r. r� � s r� sr POST / HTTP/1.0 Content-Type: multipart/form-data; boundary=AaBb1313 Content-Length: 14 --AaBb1313 --AaBb1313-- Fr� � HTTP/1.0 200 OK doner N)r r� r� rD rs r� r0 �r= r� r� rk r- r� r. �*test_multipartEmptyHeaderProcessingFailure� s z7ParsingTests.test_multipartEmptyHeaderProcessingFailurec sl g � G � fdd�dt j�}d}| j||dd�}| �|j�� d� | �t� �d� | �� d jd d gi� d S ) z� If the request has a Content-Type of C{multipart/form-data}, and the form data is parseable, the form arguments will be added to the request's args. c s e Zd Z� fdd�ZdS )z6ParsingTests.test_multipartFormData.<locals>.MyRequestc s � � | � | �d� | �� d S r� r� r< r� r- r. r> � s z>ParsingTests.test_multipartFormData.<locals>.MyRequest.processNrh r- r� r- r. r� � s r� s� POST / HTTP/1.0 Content-Type: multipart/form-data; boundary=AaB03x Content-Length: 149 --AaB03x Content-Type: text/plain Content-Disposition: form-data; name="text" Content-Transfer-Encoding: quoted-printable abasdfg --AaB03x-- Fr� r� r| r s text� abasdfgN)r r� r� rD rs r� rb r0 r� r- r� r. �test_multipartFormData� s �z#ParsingTests.test_multipartFormDatac s� g � G � fdd�dt j�}d}dtt|�dd��� d }| j|�d�| |d d �}| �|j� � d � | �t� �d � | �� d j ddgi� dS )z� If the request has a Content-Type of C{multipart/form-data}, and the form data is parseable and contains files, the file portions will be added to the request's args. c s e Zd Z� fdd�ZdS )z6ParsingTests.test_multipartFileData.<locals>.MyRequestc s � � | � | �d� | �� d S r� r� r< r� r- r. r> � s z>ParsingTests.test_multipartFileData.<locals>.MyRequest.processNrh r- r� r- r. r� � s r� s� -----------------------------738837029596785559389649595 Content-Disposition: form-data; name="uploadedfile"; filename="test" Content-Type: application/octet-stream abasdfg -----------------------------738837029596785559389649595-- z�POST / HTTP/1.0 Content-Type: multipart/form-data; boundary=---------------------------738837029596785559389649595 Content-Length: rV ra z �asciiFr� r� r| r s uploadedfiler� N) r r� r\ rb r� r� r] rD rs r� r0 )r= r� rK r� rk r- r� r. �test_multipartFileData� s ��� �z#ParsingTests.test_multipartFileDatac s� d}g �g �g � g �| �G � ����fdd�dt j�}| �||� | �� d t� | �� d d� | ��dg� | ��dg� | ��d g� d S ) z� If a request uses the I{chunked} transfer encoding, the request body is decoded accordingly before it is made available on the request. sh GET / HTTP/1.0 Content-Type: text/plain Transfer-Encoding: chunked 6 Hello, 14 spam,eggs spam spam 0 c s e Zd Z� ����fdd�ZdS )z4ParsingTests.test_chunkedEncoding.<locals>.MyRequestc sX � � | j�� � � � | j�� � �� | j� �� | j� �� | jj� d�_| � � d S rm ) r� rX �filenorZ r` r� rk Z_transferDecoderr� rd r< �rX r: r` r� r� r- r. r> s z<ParsingTests.test_chunkedEncoding.<locals>.MyRequest.processNrh r- r� r- r. r� s r� r r| s Hello, spam,eggs spam spamro � /N)r r� r� �assertIsInstance�intrD r� r- r� r. �test_chunkedEncoding� s z!ParsingTests.test_chunkedEncodingc s^ d}g � G � fdd�dt j�}| j||dd�}| �� d� | �|j�� d� | �|jj� dS ) z� If a request uses the I{chunked} transfer encoding, but provides an invalid chunk length value, the request fails with a 400 error. s� GET / HTTP/1.1 Content-Type: text/plain Transfer-Encoding: chunked MALFORMED_LINE_THIS_SHOULD_BE_'6' Hello, 14 spam,eggs spam spam 0 c s e Zd Z� fdd�ZdS )z=ParsingTests.test_malformedChunkedEncoding.<locals>.MyRequestc s � � d� d S rm �r� r< r� r- r. r> 2 s zEParsingTests.test_malformedChunkedEncoding.<locals>.MyRequest.processNrh r- r� r- r. r� 0 s r� Fr� �Request.process calledr� N) r r� r� r� rD rs r� r� r� )r= r� r� rk r- r� r. � test_malformedChunkedEncoding s �z*ParsingTests.test_malformedChunkedEncodingc s� t �| t�}g � G � fdd�dtj�}d}d}t�d�||f���� }d| d }| � tdd d � � | � ||d � � � � }| � d |j � | �d t|�� |d }|d }| �|jt� | �t� dS )z� A L{Request} that throws an exception processing basic authorization logs an error and uses an empty username and password. c s e Zd Z� fdd�ZdS )z5ParsingTests.test_basicAuthException.<locals>.Requestc s | � � | �� f| _� �| � d S r/ r� r< r� r- r. r> I s z=ParsingTests.test_basicAuthException.<locals>.Request.processNrh r- r� r- r. r� H s r� r@ r� r� r� r� Z decodestringc S s g S r/ r- )r� r- r- r. ri Q r� z6ParsingTests.test_basicAuthException.<locals>.<lambda>r )� r� r| � log_failureN)r! r� r' r r� r� r� rs r� r� r� r� rD r� r� rb r� r� �AttributeError�flushLoggedErrors) r= r� r� r� rU r_ r� r� r� r- r� r. �test_basicAuthException> s& � z$ParsingTests.test_basicAuthExceptionc sX G � fdd�dt j�� | j|� dd�}| �� jd� | �|j�� d� | �|jj � dS )z� Assert that the given request bytes fail with a 400 bad request without calling L{Request.process}. @param request: A raw HTTP request @type request: L{bytes} c s e Zd ZdZ� fdd�ZdS )zAParsingTests.assertDisconnectingBadRequest.<locals>.FailedRequestFc s d� _ d S rm r� r< �� FailedRequestr- r. r> f s zIParsingTests.assertDisconnectingBadRequest.<locals>.FailedRequest.processN�r) r* r+ r� r> r- r� r- r. r� d s r� Fr� r� r� N) r r� r� r� r� rD rs r� r� r� )r= rg rk r- r� r. � assertDisconnectingBadRequest\ s �z*ParsingTests.assertDisconnectingBadRequestc C s | � ddddddg� dS )z� A request which includes multiple C{content-length} headers fails with a 400 response without calling L{Request.process}. � GET /a HTTP/1.1� Content-Length: 56� Content-Length: 0� Host: host.invalidr� Nr� r< r- r- r. � test_duplicateContentLengthsq s �z)ParsingTests.test_duplicateContentLengthsc C s" | � ddddddddddg � dS )z� Two pipelined requests, the first of which includes multiple C{content-length} headers, trigger a 400 response without calling L{Request.process}. r� r� r� r� r� Nr� r< r- r- r. �1test_duplicateContentLengthsWithPipelinedRequests� s �z>ParsingTests.test_duplicateContentLengthsWithPipelinedRequestsc C s | � ddddddg� dS )z� A request that includes both C{content-length} and C{transfer-encoding} headers fails with a 400 response without calling L{Request.process}. r� r� r� r� r� Nr� r< r- r- r. �%test_contentLengthAndTransferEncoding� s �z2ParsingTests.test_contentLengthAndTransferEncodingc C s" | � ddddddddddg � dS )z� Two pipelined requests, the first of which includes both C{content-length} and C{transfer-encoding} headers, triggers a 400 response without calling L{Request.process}. r� r� r� r� r� Nr� r< r- r- r. �:test_contentLengthAndTransferEncodingWithPipelinedRequests� s �zGParsingTests.test_contentLengthAndTransferEncodingWithPipelinedRequestsc C s | � dddddg� dS )z� A request whose C{transfer-encoding} header includes a value other than C{chunked} or C{identity} fails with a 400 response without calling L{Request.process}. r� s Transfer-Encoding: unknownr� r� Nr� r< r- r- r. � test_unknownTransferEncoding� s �z)ParsingTests.test_unknownTransferEncodingc sP g � G � fdd�dt j�}d}| �||d�}| �� dg� | �|j�� d� dS )z� A request with a valid C{content-length} and a C{transfer-encoding} whose value is C{identity} succeeds. c s e Zd ZdZ� fdd�ZdS )zEParsingTests.test_transferEncodingIdentity.<locals>.SuccessfulRequestFc s( � � | j�� � | �dd� | �� d S )NrU � 0)r� rX rZ r^ rd r< �rK r- r. r> � s zMParsingTests.test_transferEncodingIdentity.<locals>.SuccessfulRequest.processNr� r- r� r- r. �SuccessfulRequest� s r� sT GET / HTTP/1.1 Host: host.invalid Content-Length: 2 Transfer-Encoding: identity ok Fs oks& HTTP/1.1 200 OK Content-Length: 0 N)r r� r� rD rs r� )r= r� rg rk r- r� r. � test_transferEncodingIdentity� s �z*ParsingTests.test_transferEncodingIdentity)NTN)&r) r* r+ r, r, r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r- r- r- r. r� � sL � - ) &'� $." r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�QueryArgumentsTestsc C sh | � t�d�t�d�� | jttjddd� | � tjddd�tjddd�� | � t�d�t�d�� d S )Ns a=b&d=c;+=fs blahT)Zstrict_parsings a=&b=cr| )Zkeep_blank_values)rD �cgiZparse_qsr r� r� r< r- r- r. � testParseqs� s$ � � ��z QueryArgumentsTests.testParseqsc s� dd� }� fdd�}t � |� D ]}||d� ||d� q t � |� D ]}||d� ||d� qF|� D ]$}t � ||d� t � ||d� qfdS )z� For a given URL, L{http.urlparse} should behave the same as L{urlparse}, except it should always return C{bytes}, never text. c s s\ dD ]R} dD ]H}dD ]>}dD ]4}|d k r |d t t|�� }t| ||ddf�V q qq qd S )N)s https https)� example.com)Nr� )r� s pathr� r� )r r\ r )�schemeZhostZportr� r- r- r. �urls� s z/QueryArgumentsTests.test_urlparse.<locals>.urlsc s� | }|r| � d�}t|�}t|jt�rh|j�d�|j�d�|j�d�|j�d�|j �d�|j �d�f}n |j|j|j|j|j |j f}t �| �\}}}}} } � � ||||| | f|� � � |t� � � |t� � � |t� � � |t� � � | t� � � | t� dS )z� Verify that C{url} is parsed into the same objects by both L{http.urlparse} and L{urlparse}. r� zutf-8N)�decoder � isinstancer r r] �netlocr� �paramsr� �fragmentr rD r� rB ) �urlr ZurlToStandardImplementationZstandardResultr� r r r� r r� r r< r- r. �assertSameParsing s< �� � z<QueryArgumentsTests.test_urlparse.<locals>.assertSameParsingTFN)r )r= r r r r- r< r. � test_urlparse� s ) z!QueryArgumentsTests.test_urlparsec C s | � ttjd� dS )zE L{http.urlparse} should reject unicode input early. zhttp://example.org/pathN)r� � TypeErrorr r r< r- r- r. �test_urlparseRejectsUnicode@ s z/QueryArgumentsTests.test_urlparseRejectsUnicodeN)r) r* r+ r r r r- r- r- r. r� � s Kr� c @ s e Zd Zdd� ZdS )� ClientDriverc C s || _ || _|| _d S r/ )r r� r r r- r- r. r I s zClientDriver.handleStatusN)r) r* r+ r r- r- r- r. r H s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ClientStatusParsingTestsc C s> t � }|�d� | �|jd� | �|jd� | �|jd� d S )Ns HTTP/1.0 201 foor � 201r@ �r � lineReceivedrD r r� r �r= r| r- r- r. � testBaselineO s z%ClientStatusParsingTests.testBaselinec C s> t � }|�d� | �|jd� | �|jd� | �|jd� d S )N� HTTP/1.0 201r r r� r r r- r- r. � testNoMessageV s z&ClientStatusParsingTests.testNoMessagec C s> t � }|�d� | �|jd� | �|jd� | �|jd� d S )Ns HTTP/1.0 201 r r r� r r r- r- r. �testNoMessage_trailingSpace] s z4ClientStatusParsingTests.testNoMessage_trailingSpaceN)r) r* r+ r r r r- r- r- r. r N s r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+dRdS� Z,dTdU� Z-dVdW� Z.dXdY� Z/dZd[� Z0d\d]� Z1d^d_� Z2d`da� Z3dbdc� Z4ddde� Z5dfdg� Z6dhdi� Z7djdk� Z8dldm� Z9dndo� Z:dpdq� Z;drds� Z<dtdu� Z=dvdw� Z>dxdy� Z?dzd{� Z@d|d}� ZAd~d� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�S )�� RequestTestsz# Tests for L{http.Request} c C s� t �t� d�}t||��ddg� | �t||�d d� t||ddi� | �tt||��� �ddgfg� t||t � � � | �t||�i � dS )a0 Verify that each of two different attributes which are associated with the same state properly reflect changes made through the other. This is used to test that the C{headers}/C{responseHeaders} and C{received_headers}/C{requestHeaders} pairs interact properly. Fr~ r r@ r� r N) r r� r# �getattrr( rD �setattrr/ r1 r ZHeaders)r= ZoldNameZnewNamer� r- r- r. �_compatHeadersTestj s �z RequestTests._compatHeadersTestc C s4 t �t� d�}|j�ddg� | �|�d�d� dS )zb L{http.Request.getHeader} returns the value of the named request header. Fr~ r N�r r� r# r' r( rD r[ �r= r� r- r- r. �test_getHeader} s zRequestTests.test_getHeaderc C s6 t �t� d�}|j�dddg� | �|�d�d� dS )z� When there are multiple values for a single request header, L{http.Request.getHeader} returns the last value. Fr~ r � pandaNr r r- r- r. � test_getHeaderReceivedMultiples� s z,RequestTests.test_getHeaderReceivedMultiplesc C s$ t �t� d�}| �|�d�d� dS )z� L{http.Request.getHeader} returns L{None} when asked for the value of a request header which is not present. Fr~ N)r r� r# rD r[ r r- r- r. �test_getHeaderNotFound� s z#RequestTests.test_getHeaderNotFoundc C s6 t �t� d�}|j�ddg� | �|�� ddi� dS )z� L{http.Request.getAllheaders} returns a C{dict} mapping all request header names to their corresponding values. Fr~ r N�r r� r# r' r( rD � getAllHeadersr r- r- r. �test_getAllHeaders� s z RequestTests.test_getAllHeadersc C s" t �t� d�}| �|�� i � dS )zq L{http.Request.getAllHeaders} returns an empty C{dict} if there are no request headers. FN)r r� r# rD r# r r- r- r. �test_getAllHeadersNoHeaders� s z(RequestTests.test_getAllHeadersNoHeadersc C s8 t �t� d�}|j�dddg� | �|�� ddi� dS )z� When there are multiple values for a single request header, L{http.Request.getAllHeaders} returns only the last value. Fr~ r r Nr" r r- r- r. �!test_getAllHeadersMultipleHeaders� s z.RequestTests.test_getAllHeadersMultipleHeadersc C sF t � }t�|d�}|�d� |�d� | �|jj�� � � d d� dS )z~ L{http.Request.setResponseCode} takes a status code and causes it to be used as the response status. F�� r� r s (no clientproto yet) 201 CreatedN� r# r r� r: rc rD rs �written�getvaluer �r= rk r� r- r- r. �test_setResponseCode� s �z!RequestTests.test_setResponseCodec C sH t � }t�|d�}|�dd� |�d� | �|jj�� � � d d� dS )z� L{http.Request.setResponseCode} takes a status code and a message and causes them to be used as the response status. F�� s happily acceptedr� r s) (no clientproto yet) 202 happily acceptedNr( r+ r- r- r. � test_setResponseCodeAndMessage� s �z+RequestTests.test_setResponseCodeAndMessagec C s( t � }t�|d�}| �t|jdd� dS )z� L{http.Request.setResponseCode} accepts C{bytes} for the message parameter and raises L{TypeError} if passed anything else. Fr- znot happily acceptedN)r# r r� r� r r: r+ r- r- r. �&test_setResponseCodeAndMessageNotBytes� s �z3RequestTests.test_setResponseCodeAndMessageNotBytesc C s, t �t� d�}|�d� | �t|jd� dS )z� L{http.Request.setResponseCode} accepts C{int} for the code parameter and raises L{TypeError} if passed anything else. Fr| �1N)r r� r# r: r� r r r- r- r. �#test_setResponseCodeAcceptsIntegers� s z0RequestTests.test_setResponseCodeAcceptsIntegersc C s t �t� d�}|�td�� dS )za L{http.Request.setResponseCode} accepts C{long} for the code parameter. Fr| N)r r� r# r: r r r- r- r. �'test_setResponseCodeAcceptsLongIntegers� s z4RequestTests.test_setResponseCodeAcceptsLongIntegersc C s* t �t� d�}|�d� | �|jd� dS )z� When no previous value was set and no 'if-modified-since' value was requested, L{http.Request.setLastModified} takes a timestamp in seconds since the epoch and sets the request's lastModified attribute. F�* N�r r� r# �setLastModifiedrD � lastModifiedr r- r- r. � test_setLastModifiedNeverSet� s z)RequestTests.test_setLastModifiedNeverSetc C s4 t �t� d�}|�d� |�d� | �|jd� dS )z� If the supplied timestamp is later than the lastModified attribute's value, L{http.Request.setLastModified} updates the lastModifed attribute. Fr r| Nr4 r r- r- r. �test_setLastModifiedUpdate� s z'RequestTests.test_setLastModifiedUpdatec C s4 t �t� d�}|�d� |�d� | �|jd� dS )z� If the supplied timestamp occurs earlier than the current lastModified attribute, L{http.Request.setLastModified} ignores it. Fr| r Nr4 r r- r- r. �test_setLastModifiedIgnore s z'RequestTests.test_setLastModifiedIgnorec C s> t �t� d�}|j�td�dg� |�d�}| �|t j� dS )z� If the resource is older than the if-modified-since date in the request header, L{http.Request.setLastModified} returns L{http.CACHED}. F�if-modified-sinces 02 Jan 1970 00:00:00 GMTr3 N� r r� r# r' r( r r5 rD ZCACHED�r= r� rz r- r- r. �test_setLastModifiedCached s � z'RequestTests.test_setLastModifiedCachedc C s< t �t� d�}|j�td�dg� |�d�}| �|d� dS )z� If the resource is newer than the if-modified-since date in the request header, L{http.Request.setLastModified} returns None Fr: s 01 Jan 1970 00:00:00 GMT�@B N�r r� r# r' r( r r5 rD r< r- r- r. � test_setLastModifiedNotCached) s � z*RequestTests.test_setLastModifiedNotCachedc C sF t �t� d�}|j�td�dg� |�d� |�d�}| �|d� dS )z� When L{http.Request.setLastModified} is called multiple times, the highest supplied value is honored. If that value is higher than the if-modified-since date in the request header, the method returns None. Fr: s 01 Jan 1970 00:00:01 GMTr> r Nr? r< r- r- r. �"test_setLastModifiedTwiceNotCached9 s � z/RequestTests.test_setLastModifiedTwiceNotCachedc C sH t �t� d�}|j�td�dg� |�d� |�d�}| �|t j� dS )a When L{http.Request.setLastModified} is called multiple times, the highest supplied value is honored. If that value is lower than the if-modified-since date in the request header, the method returns L{http.CACHED}. Fr: s 01 Jan 1999 00:00:01 GMTr| r Nr; r< r- r- r. � test_setLastModifiedTwiceCachedK s � z,RequestTests.test_setLastModifiedTwiceCachedc C s4 t �t� d�}|�dd� | �|j�d�dg� dS )�� L{http.Request.setHost} sets the value of the host request header. The port should not be added because it is the default. Fr �P � hostN�r r� r# �setHostrD r' r� r r- r- r. � test_setHost^ s �zRequestTests.test_setHostc C sB t � }t �� |_t�|d�}|�dd� | �|j�d�dg� dS )rC Fr i� rE N� r# r� rs r r� rG rD r' r� �r= r# r� r- r- r. �test_setHostSSLi s �z RequestTests.test_setHostSSLc C s4 t �t� d�}|�dd� | �|j�d�dg� dS �z� L{http.Request.setHost} sets the value of the host request header. The port should be added because it is not the default. Fr �Q rE s example.com:81NrF r r- r- r. �test_setHostNonDefaultPortv s �z'RequestTests.test_setHostNonDefaultPortc C sB t � }t �� |_t�|d�}|�dd� | �|j�d�dg� dS rL rI rJ r- r- r. � test_setHostSSLNonDefaultPort� s �z*RequestTests.test_setHostSSLNonDefaultPortc C s4 t �t� d�}|�dd� | �|j�d�dg� dS )zX L{http.Request.setHeader} sets the value of the given response header. Fr~ r N)r r� r# r^ rD r0 r� r r- r- r. �test_setHeader� s zRequestTests.test_setHeaderc O s� t � }t�|d�}|j||� | �|jd |� |�d� |jj� � � d�}dd� |D �}| �t |�d� | �|d d| � d S ) z� Call L{http.Request.setCookie} with C{*args} and C{**kwargs}, and check that the cookie value is equal to C{expectedCookieValue}. Fr r� ra c S s g | ]}|� d �r|�qS )s Set-Cookie)� startswith)rx r� r- r- r. ry � s �z-RequestTests._checkCookie.<locals>.<listcomp>r| s Set-Cookie: N) r# r r� Z addCookierD r� rc rs r) r* r~ rb )r= �expectedCookieValuer0 r1 rk r� Z writtenLinesZsetCookieLinesr- r- r. � _checkCookie� s �zRequestTests._checkCookiec C s d}| � |dd� dS )z� L{http.Request.setCookie} adds a new cookie to be sent with the response, and can be called with just a key and a value. L{unicode} arguments are encoded using UTF-8. � foo=bar�foo�barN�rS �r= rR r- r- r. �)test_addCookieWithMinimumArgumentsUnicode� s z6RequestTests.test_addCookieWithMinimumArgumentsUnicodec C s& d}| j |dddddddd d d � d S ) z� L{http.Request.setCookie} adds a new cookie to be sent with the response. L{unicode} arguments are encoded using UTF-8. �} foo=bar; Expires=Fri, 31 Dec 9999 23:59:59 GMT; Domain=.example.com; Path=/; Max-Age=31536000; Comment=test; Secure; HttpOnlyrU rV z Fri, 31 Dec 9999 23:59:59 GMTz .example.com�/Z31536000ZtestT�ZexpiresZdomainr� Zmax_ageZcommentZsecureZhttpOnlyNrW rX r- r- r. �%test_addCookieWithAllArgumentsUnicode� s � �z2RequestTests.test_addCookieWithAllArgumentsUnicodec C s d}| � |dd� dS )z� L{http.Request.setCookie} adds a new cookie to be sent with the response, and can be called with just a key and a value. L{bytes} arguments are not decoded. rT r@ r� NrW rX r- r- r. �'test_addCookieWithMinimumArgumentsBytes� s z4RequestTests.test_addCookieWithMinimumArgumentsBytesc C s& d}| j |dddddddd d d � d S ) z� L{http.Request.setCookie} adds a new cookie to be sent with the response. L{bytes} arguments are not decoded. rZ r@ r� s Fri, 31 Dec 9999 23:59:59 GMTs .example.comr� s 31536000r~ Tr\ NrW rX r- r- r. �#test_addCookieWithAllArgumentsBytes� s � �z0RequestTests.test_addCookieWithAllArgumentsBytesc C s\ d}| � |dd� | �| j g�}| �dt|�� | �|d d t� | �|d d d� d S ) z� L{http.Request.setCookie} will raise a L{DeprecationWarning} if non-string (not L{bytes} or L{unicode}) arguments are given, and will call C{str()} on it to preserve past behaviour. s foo=10r@ rq r| r �categoryr zSPassing non-bytes or non-unicode cookie arguments is deprecated since Twisted 16.1.N)rS � flushWarningsrD rb �DeprecationWarning)r= rR �warningsr- r- r. � test_addCookieNonStringArgument� s �z,RequestTests.test_addCookieNonStringArgumentc C sR | j ddddd� | j ddddd� | j ddddd� | jt| j d ddd d� d S ) zL L{http.Request.setCookie} supports a C{sameSite} argument. s foo=bar; SameSite=laxr@ r� Zlax)ZsameSiteZLaxs foo=bar; SameSite=strict�strictr� z anything-elseN)rS r� r� r< r- r- r. �test_addCookieSameSite� s4 � � � �z#RequestTests.test_addCookieSameSitec C s^ t � }t�|d�}t� }||_|�d� d|_|j�ddg� |� d� | � |� � dg� dS ) z� For an HTTP 1.0 request, L{http.Request.write} sends an HTTP 1.0 Response-Line and whatever response headers are set. Fr8 r r~ r r� )r� r� r� N� r# r r� r rs r: ra r0 r( rc r� r� �r= rk r� r� r- r- r. �test_firstWrite s �z RequestTests.test_firstWritec C s� t � }t�|d�}t� }||_|�d� d|_|j�ddg� |� d� | � |� � dg� | j | j gd�}| �d t|�� | �|d d t� | �|d d d � dS )zd L{http.Request.write} casts non-bytes header value to bytes transparently. Fr8 r r~ rq r� )r� s Test: 10r� �ZoffendingFunctionsr| r r` r zZPassing non-bytes header values is deprecated since Twisted 12.3. Pass only bytes instead.N)r# r r� r rs r: ra r0 r( rc r� r� ra �test_nonByteHeaderValuerD rb rb )r= rk r� r� rc r- r- r. rk s* �� �z$RequestTests.test_nonByteHeaderValuec C sh t � }t�|d�}t� }||_|�d� d|_|j�ddg� |� d� |� d� | � |� � dg� d S ) z� For an HTTP 1.1 request, L{http.Request.write} sends an HTTP 1.1 Response-Line, whatever response headers are set, and uses chunked encoding for the response body. Fr8 r6 r~ r r� r� r� Nrg rh r- r- r. � test_firstWriteHTTP11Chunked> s �z)RequestTests.test_firstWriteHTTP11Chunkedc C sd t � }t�|d�}t� }||_|�d� d|_d|_|j� ddg� |� d� | � |� � dg� d S ) z� For an HTTP 1.0 request for a resource with a known last modified time, L{http.Request.write} sends an HTTP Response-Line, whatever response headers are set, and a last-modified header with that time. Fr8 r r r~ r r� )r� r� �, Last-Modified: Thu, 01 Jan 1970 00:00:00 GMTr� N) r# r r� r rs r: ra r6 r0 r( rc r� r� rh r- r- r. �test_firstWriteLastModifiedX s �z(RequestTests.test_firstWriteLastModifiedc C s� t �| t�}t� }t�|d�}t� }||_|�d� d|_ d|_ |j � ddg� |� d� | �|�� dg� | �d t|�� |d }| �d |d � d S )z� If the last-modified header already exists in the L{http.Request} response headers, the lastModified attribute is ignored and a message is logged. Fr8 r i ʚ;s last-modifieds Thu, 01 Jan 1970 00:00:00 GMTr� )r� rm r� r| r zPWarning: last-modified specified both in header list and lastModified attribute.r� N)r! r� r' r# r r� r rs r: ra r6 r0 r( rc r� r� r� rb )r= r� rk r� r� r� r- r- r. � test_lastModifiedAlreadyWrittens s4 � � ��z,RequestTests.test_lastModifiedAlreadyWrittenc C s t �t� d�}| �|ji � dS )zP L{http.Request.received_cookies} defaults to an empty L{dict}. FN)r r� r# rD �received_cookiesr r- r- r. �test_receivedCookiesDefault� s z(RequestTests.test_receivedCookiesDefaultc C s> t �t� d�}|j�ddg� |�� | �|jddd�� dS )z� L{http.Request.parseCookies} extracts cookies from C{requestHeaders} and adds them to C{received_cookies}. F� cookies test="lemur"; test2="panda"� "lemur"� "panda"�r~ s test2N�r r� r# r' r( � parseCookiesrD rp r r- r- r. �test_parseCookies� s � �z RequestTests.test_parseCookiesc C s@ t �t� d�}|j�dddg� |�� | �|jddd�� dS ) zh L{http.Request.parseCookies} can extract cookies from multiple Cookie headers. Frr s test="lemur"s test2="panda"rs rt ru Nrv r r- r- r. � test_parseCookiesMultipleHeaders� s � �z-RequestTests.test_parseCookiesMultipleHeadersc C s( t �t� d�}|�� | �|ji � dS )zj L{http.Request.parseCookies} can be called on a request without a cookie header. FN)r r� r# rw rD rp r r- r- r. �test_parseCookiesNoCookie� s z&RequestTests.test_parseCookiesNoCookiec C s6 t �t� d�}|j�dg � |�� | �|ji � dS )zn L{http.Request.parseCookies} can be called on a request with an empty cookie header. Frr Nrv r r- r- r. � test_parseCookiesEmptyCookie� s �z)RequestTests.test_parseCookiesEmptyCookiec C s8 t �t� d�}|j�ddg� |�� | �|ji � dS )z` L{http.Request.parseCookies} ignores cookies which don't have a value. Frr s foo; bar; baz;Nrv r r- r- r. � test_parseCookiesIgnoreValueless� s � �z-RequestTests.test_parseCookiesIgnoreValuelessc C s< t �t� d�}|j�ddg� |�� | �|jddi� dS )zR L{http.Request.parseCookies} parses cookies with an empty value. Frr s foo=r@ r� Nrv r r- r- r. �test_parseCookiesEmptyValue� s � �z(RequestTests.test_parseCookiesEmptyValuec C s< t �t� d�}|j�ddg� |�� | �|jddi� dS )zf L{http.Request.parseCookies} leaves trailing whitespace in the cookie value. Frr s foo=bar r@ s bar Nrv r r- r- r. �!test_parseCookiesRetainRightSpace� s � �z.RequestTests.test_parseCookiesRetainRightSpacec C s< t �t� d�}|j�ddg� |�� | �|jddi� dS )zc L{http.Request.parseCookies} strips leading whitespace in the cookie key. Frr s foo=barr@ r� Nrv r r- r- r. � test_parseCookiesStripLeftSpace� s � �z,RequestTests.test_parseCookiesStripLeftSpacec C s> t �t� d�}|j�ddg� |�� | �|jddd�� dS )zr L{http.Request.parseCookies} parses valid cookies set before or after malformed cookies. Frr s0 12345; test="lemur"; 12345; test2="panda"; 12345rs rt ru Nrv r r- r- r. �-test_parseCookiesContinueAfterMalformedCookie s � �z:RequestTests.test_parseCookiesContinueAfterMalformedCookiec C sX t �t� d�}|�d� |j}|�d� |�ttd��� | � |j � | � |j d� dS )z� L{http.Request.connectionLost} closes L{Request.content} and drops the reference to the L{HTTPChannel} to assist with garbage collection. Frq rg ZFinishedN) r r� r# � gotLengthrX ZhandleContentChunkr� r r r� �closedrG rk )r= r� rX r- r- r. �test_connectionLost s z RequestTests.test_connectionLostc C s4 t �t� d�}|�t� d� | �t|jt� d� dS )zv Calling L{Request.registerProducer} when a producer is already registered raises ValueError. FTN)r r� r# r; r" r� r� r r- r- r. � test_registerProducerTwiceFails( s �z,RequestTests.test_registerProducerTwiceFailsc C s: t �t� d�}t� }|�|d� | �|dfg|jj� dS )z� Calling L{Request.registerProducer} with an IPushProducer when the request is not queued registers the producer as a push producer on the request's transport. FTN�r r� r# r" r; rD rs Z producers�r= r� r r- r- r. �7test_registerProducerWhenNotQueuedRegistersPushProducer3 s zDRequestTests.test_registerProducerWhenNotQueuedRegistersPushProducerc C s: t �t� d�}t� }|�|d� | �|dfg|jj� dS )z� Calling L{Request.registerProducer} with an IPullProducer when the request is not queued registers the producer as a pull producer on the request's transport. FNr� r� r- r- r. �7test_registerProducerWhenNotQueuedRegistersPullProducer? s zDRequestTests.test_registerProducerWhenNotQueuedRegistersPullProducerc C sF t � }t�|d�}|�� }|�ttd��� | �|jd� | � |t�S )z L{Request.connectionLost} triggers all finish notification Deferreds and cleans up per-request state. TzConnection doneN) r# r r� � notifyFinishr� r r rG rk Z assertFailure)r= r# rg rL r- r- r. � test_connectionLostNotificationK s z,RequestTests.test_connectionLostNotificationc C s, t �t� d�}|�� }|�d� |�� |S )zO L{Request.finish} triggers all finish notification Deferreds. Fr| )r r� r# r� r� rd �r= rg rL r- r- r. �test_finishNotificationX s z$RequestTests.test_finishNotificationc C sF t �t� d�}|�� }|�d� |�d� |�� | �t|jd� |S )z� Calling L{Request.write} after L{Request.finish} has been called results in a L{RuntimeError} being raised. Fr| s foobar) r r� r# r� r� rc rd r� r% r� r- r- r. �test_writeAfterFinishe s z"RequestTests.test_writeAfterFinishc C s6 t � }t�|d�}|�ttd��� | �t|j� dS )z� Calling L{Request.finish} after L{Request.connectionLost} has been called results in a L{RuntimeError} being raised. FzThe end.N) r# r r� r� r r r� r% rd r+ r- r- r. � test_finishAfterConnectionLostu s z+RequestTests.test_finishAfterConnectionLostc C s, t �t� d�}| �t|�dt|�f � dS )z� L{Request.__repr__} returns the class name, object address, and dummy-place holder values when used on a L{Request} which has not yet been initialized. FzZ<Request at 0x%x method=(no method yet) uri=(no uri yet) clientproto=(no clientproto yet)>N�r r� r# rD �repr�id�r= rg r- r- r. �test_reprUninitialized� s ��z#RequestTests.test_reprUninitializedc C s> t �t� d�}d|_d|_d|_| �t|�dt|�f � dS )a L{Request.__repr__} returns, as a L{str}, the class name, object address, and the method, uri, and client protocol of the HTTP request it represents. The string is in the form:: <Request at ADDRESS method=METHOD uri=URI clientproto=PROTOCOL> Fr ro rp z><Request at 0x%x method=GET uri=/foo/bar clientproto=HTTP/1.0>N) r r� r# ra r` r_ rD r� r� r� r- r- r. �test_reprInitialized� s ��z!RequestTests.test_reprInitializedc C s< G dd� dt j�}|t� d�}| �t|�dt|�f � dS )z� Subclasses of L{Request} inherit a C{__repr__} implementation which includes the subclass's name in place of the string C{"Request"}. c @ s e Zd ZdS )z1RequestTests.test_reprSubclass.<locals>.OtherwiseN)r) r* r+ r- r- r- r. � Otherwise� s r� Fz\<Otherwise at 0x%x method=(no method yet) uri=(no uri yet) clientproto=(no clientproto yet)>Nr� )r= r� rg r- r- r. �test_reprSubclass� s ��z RequestTests.test_reprSubclassc C sF t �t� d�}t� |_|�t� d� |�� | �d|j |jj f� dS )z� L{Request.unregisterProducer} unregisters a non-queued non-streaming producer from the request and the request's transport. F�NNN� r r� r# r rs r; r" �unregisterProducerrD r r r- r- r. �,test_unregisterNonQueuedNonStreamingProducer� s z9RequestTests.test_unregisterNonQueuedNonStreamingProducerc C sF t �t� d�}t� |_|�t� d� |�� | �d|j |jj f� dS )z� L{Request.unregisterProducer} unregisters a non-queued streaming producer from the request and the request's transport. FTr� Nr� r r- r- r. �)test_unregisterNonQueuedStreamingProducer� s z6RequestTests.test_unregisterNonQueuedStreamingProducerc C s� t �� }d|_d|_d|_|�� t� |_|�d�}ddg}t � }|� |� |D ]}|� |� qP|j j d �� | �d|j�� � dS )zj L{http.Request.finish} will call the channel's factory to produce a log message. N�sometimeT� GET /path HTTP/1.1 r} r s sometime "GET /path HTTP/1.1")r r� r� � _logDateTime�_logDateTimeCall� startFactoryr �logFiler� r r� r� rv r� rd r� r* �r= r� �proto�valr� r� r- r- r. �test_finishProducesLog� s$ � �z#RequestTests.test_finishProducesLogc C s� t � }tjd|d�}|�� |�d�}t� }t||�}| �|jd� |� |� |� d� |� d� | � |j � |� d� | �|j � dS )z� L{HTTPChannel} timeouts whenever data from a request body is not delivered to it in time, even when it gets built from a L{HTTPFactory}. r� )Ztimeoutrx Nr� r� r� )r r r� r� r� r ry rD r� r� r� r� r� r� r� r r- r- r. �"test_requestBodyTimeoutFromFactory� s z/RequestTests.test_requestBodyTimeoutFromFactoryc C s� t �� }d|_d|_d|_|�� t� |_|�d�}d|j _ ddg}t � }|� |� | � |jd� |D ]}|�|� qf|�d� | � |jd � |j jd �� | � |jd� dS ) z� L{http.Request.finish} will notify the channel that it is finished, and will put the transport back in the producing state so that the reactor can close the connection. Nr� Tr r� r} r� s GET r� )r r� r� r� r� r� r r� r� rv r� r r� rD r� r� r� rd r� r- r- r. �test_finishCleansConnection� s( � z(RequestTests.test_finishCleansConnectionc C s t �t� d�}tt j|� dS )z� L{http.Request} provides L{http._IDeprecatedHTTPChannelToRequestInterface}, which defines the interface used by L{http.HTTPChannel}. FN)r r� r# r �)_IDeprecatedHTTPChannelToRequestInterfacer r- r- r. �6test_provides_IDeprecatedHTTPChannelToRequestInterface s zCRequestTests.test_provides_IDeprecatedHTTPChannelToRequestInterfacec C s t �t� d�}| �||� dS )z7 A L{http.Request} is equal to itself. FN)r r� r# rD r r- r- r. �test_eq) s zRequestTests.test_eqc C s( t �t� d�}| �|t �t� d�� dS )zC A L{http.Request} is not equal to another object. FN)r r� r# �assertNotEqualr r- r- r. �test_ne1 s zRequestTests.test_nec C s t �t� d�}t|� dS )z0 A L{http.Request} is hashable. FN)r r� r# �hashr r- r- r. � test_hashable9 s zRequestTests.test_hashablec sF g � G � fdd�dt �}t�t� d�}| �||� � | �� |g� dS )z� A L{http.Request} on the left hand side of an equality comparison to an instance that is not a L{http.Request} hands the comparison off to that object's C{__eq__} implementation. c s e Zd Z� fdd�ZdS )z8RequestTests.test_eqWithNonRequest.<locals>._NotARequestc s � � |� dS rm r� �r= �other�ZeqCallsr- r. �__eq__K s z?RequestTests.test_eqWithNonRequest.<locals>._NotARequest.__eq__N)r) r* r+ r� r- r� r- r. � _NotARequestI s r� FN)�objectr r� r# rD �r= r� r� r- r� r. �test_eqWithNonRequestA s z"RequestTests.test_eqWithNonRequestc sF g � G � fdd�dt �}t�t� d�}| �||� � | �� |g� dS )z� A L{http.Request} on the left hand side of an inequality comparison to an instance that is not a L{http.Request} hands the comparison off to that object's C{__ne__} implementation. c s e Zd Z� fdd�ZdS )z8RequestTests.test_neWithNonRequest.<locals>._NotARequestc s � � |� dS rm r� r� r� r- r. �__ne___ s z?RequestTests.test_neWithNonRequest.<locals>._NotARequest.__ne__N)r) r* r+ r� r- r� r- r. r� ] s r� FN)r� r r� r# r� rD r� r- r� r. �test_neWithNonRequestU s z"RequestTests.test_neWithNonRequestc C sl t �| t�}t�t� d�}|�t� d� |�� | � dt |�� |d }|d }| � |j t � | �t � dS )zy A RuntimeError is logged if a producer is still registered when an L{http.Request} is finished. FTr| r r� N)r! r� r' r r� r# r; r" rd r� rb r� r� r% r� )r= r� rg r� r� r- r- r. �"test_finishProducerStillRegisteredi s �z/RequestTests.test_finishProducerStillRegisteredc C s. t �tt�ddd�d��}| �|�� d� dS )z} L{http.Request.getClientIP} returns the host part of the client's address when connected over IPv4. r� � 127.0.0.1�80 �ZpeerN�r r� r# r � IPv6AddressrD � getClientIPr� r- r- r. �test_getClientIPWithIPv4| s �z%RequestTests.test_getClientIPWithIPv4c C s. t �tt�ddd�d��}| �|�� d� dS )z} L{http.Request.getClientIP} returns the host part of the client's address when connected over IPv6. r� z::1r� r� Nr� r� r- r- r. �test_getClientIPWithIPv6� s �z%RequestTests.test_getClientIPWithIPv6c C s* t �tt�d�d��}| �|�� d� dS )z� L{http.Request.getClientIP} returns L{None} for the client's IP address when connected over a non-TCP transport. �/path/to/socketr� N)r r� r# r � UNIXAddressrD r� r� r- r- r. � test_getClientIPWithNonTCPPeer� s �z+RequestTests.test_getClientIPWithNonTCPPeerc C s. t �d�}t�t|d��}| �|�� |� dS )zs L{http.Request.getClientAddress} returns the client's address as an L{IAddress} provider. r� r� N)r r� r r� r# r ZgetClientAddress)r= r" rg r- r- r. �test_getClientAddress� s z"RequestTests.test_getClientAddressN)Mr) r* r+ r, r r r r! r$ r% r& r, r. r/ r1 r2 r7 r8 r9 r= r@ rA rB rH rK rN rO rP rS rY r] r^ r_ rd rf ri rk rl rn ro rq rx ry rz r{ r| r} r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r- r- r- r. r f s� ' # r c @ sL e Zd ZdZdd� Zdddd�Zdd � Zd d � Zd d � Zdd� Z dd� Z dS )�MultilineHeadersTestsa Tests to exercise handling of multiline headers by L{HTTPClient}. RFCs 1945 (HTTP 1.0) and 2616 (HTTP 1.1) state that HTTP message header fields can span multiple lines if each extra line is preceded by at least one space or horizontal tab. c C s d| _ d| _dS )zv Initialize variables used to verify that the header-processing functions are getting called. FN)�handleHeaderCalled�handleEndHeadersCalledr< r- r- r. r, � s zMultilineHeadersTests.setUps 10s line-0 line-1s line-2 line-3)rW s X-Multilines X-Multiline2c C s d| _ | �|| j| � dS )zE Dummy implementation of L{HTTPClient.handleHeader}. TN)r� rD r� )r= r r� r- r- r. �ourHandleHeader� s z%MultilineHeadersTests.ourHandleHeaderc C s d| _ dS )zI Dummy implementation of L{HTTPClient.handleEndHeaders}. TN)r� r< r- r- r. �ourHandleEndHeaders� s z)MultilineHeadersTests.ourHandleEndHeadersc C s� t � }| j|_| j|_|�d� |�d� | �|jd� | �| j � | �| j � |�d� | � | j � | � | j � | � |jd� dS )z� A header isn't processed by L{HTTPClient.extractHeader} until it is confirmed in L{HTTPClient.lineReceived} that the header has been received completely. r � Content-Length: 10Nr� rq ) r r� r r� r r rG rf r� r� r� r� rD r r- r- r. �test_extractHeader� s z(MultilineHeadersTests.test_extractHeaderc C sb t � }| j|_| j|_|�d� |�d� | �| j� | �| j � | � |j d� | � |j d� dS )z� An HTTP request with no headers will not cause any calls to L{handleHeader} but will cause L{handleEndHeaders} to be called on L{HTTPClient} subclasses. r r� r r N) r r� r r� r r r� r� r� r� rD r r� r r- r- r. �test_noHeaders� s z$MultilineHeadersTests.test_noHeadersc C s� t � }| j|_| j|_|�d� |�d� | �| j� |�d� |�d� | �| j� |�d� |�d� |�d� | �| j � | � |j d� | � |j d � | � |j d � d S ) z� L{HTTPClient} parses multiline headers by buffering header lines until an empty line or a line that does not start with whitespace hits lineReceived, confirming that the header has been received completely. r s X-Multiline: line-0s line-1s X-Multiline2: line-2s line-3r� r� r r rq N)r r� r r� r r r� r� r� r� rD r r� rf r r- r- r. �test_multilineHeaders� s z+MultilineHeadersTests.test_multilineHeadersN) r) r* r+ r, r, r� r� r� r� r� r� r- r- r- r. r� � s �r� c @ s e Zd ZdZdd� Zdd� ZdS )� Expect100ContinueServerTestsai Test that the HTTP server handles 'Expect: 100-continue' header correctly. The tests in this class all assume a simplistic behavior where user code cannot choose to deny a request. Once ticket #288 is implemented and user code can run before the body of a POST is processed this should be extended to support overriding this behavior. c C s� t � }t�� }t|_|�|� |�d� |�d� |�d� |�d� |�d� | �|�� d� |�d� | � |�� dg� d S ) z� HTTP/1.0 requests do not get 100-continue returned, even if 'Expect: 100-continue' is included (RFC 2616 10.1.1). s GET / HTTP/1.0 � Host: www.example.com � Content-Length: 3 � Expect: 100-continue ra r� rP )r� r� r� r� r� � ''' 3 abc''' N) r r r� r� r� r� r� rD r� r� )r= rs rk r- r- r. � test_HTTP10 s �z(Expect100ContinueServerTests.test_HTTP10c C s� t � }t�� }t|_|�|� |�d� |�d� |�d� |�d� | �|�� d� |�d� | �|�� d� |�d� |�� }| � |� d�� |t d�d � }| � |d g� d S ) a! If a HTTP/1.1 client sends a 'Expect: 100-continue' header, the server responds with a 100 response code before handling the request body, if any. The normal resource rendering code will then be called, which will send an additional response code. s GET / HTTP/1.1 r� r� r� r� ra s HTTP/1.1 100 Continue rP N)r� r� r� r� r� r� ) r r r� r� r� r� r� rD r� r� rQ rb r� )r= rs rk r� r- r- r. � test_expect100ContinueHeader9 s* ��z9Expect100ContinueServerTests.test_expect100ContinueHeaderN)r) r* r+ r, r� r� r- r- r- r. r� s r� c s t � fdd�| D ��S )a� Create a new dict containing only a subset of the items of an existing dict. @param keys: An iterable of the keys which will be added (with values from C{d}) to the result. @param d: The existing L{dict} from which to copy items. @return: The new L{dict} with keys given by C{keys} and values given by the corresponding values in C{d}. @rtype: L{dict} c s g | ]}|� | f�qS r- r- )rx �k�r# r- r. ry m s zsub.<locals>.<listcomp>)�dict)�keysr# r- r� r. �sub_ s r� c @ s e Zd ZdZdd� Zdd� ZdS )� DeprecatedRequestAttributesTestszI Tests for deprecated attributes of L{twisted.web.http.Request}. c C s~ t �tt�ddd�d��}|�d� |�ddd� |�� | j| j gd �}| � d t |�� | � t d d �t d dg|d �� dS )zg L{Request.getClientIP} is deprecated in favor of L{Request.getClientAddress}. r� r� i90 r� r ro r� r6 rj r| zjtwisted.web.http.Request.getClientIP was deprecated in Twisted 18.4.0; please use getClientAddress instead�r` r r` r N)r r� r# r r� r� ZrequestReceivedr� ra �test_getClientIPrD rb rb r� )r= rg rc r- r- r. r� v s � ���z1DeprecatedRequestAttributesTests.test_getClientIPc C s\ t � }t�|�}|�� | j| jgd�}| �dt|�� | �tdd�t ddg|d �� dS ) zs L{Request.noLongerQueued} is deprecated, as we no longer process requests simultaneously. rj r| zHtwisted.web.http.Request.noLongerQueued was deprecated in Twisted 16.3.0r� r` r r N) r# r r� ZnoLongerQueuedra �test_noLongerQueuedrD rb rb r� )r= rk rg rc r- r- r. r� � s ���z4DeprecatedRequestAttributesTests.test_noLongerQueuedN)r) r* r+ r, r� r� r- r- r- r. r� q s r� c @ s| e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd S ) �ChannelProductionTestsz= Tests for the way HTTPChannel manages backpressure. s# GET / HTTP/1.1 Host: localhost c C s( |}t �� }t|�|_|�|� ||fS )a Setup a L{HTTPChannel} and a transport and associate them. @param transport: A transport to back the L{HTTPChannel} @param requestFactory: An object that can construct L{Request} objects. @return: A tuple of the channel and the transport. )r r� r6 r� r� )r= rs r� rk r- r- r. �buildChannelAndTransport� s z/ChannelProductionTests.buildChannelAndTransportc C s0 | � t� t�\}}| �|j|� | �|j� dS )zu L{HTTPChannel} registers itself as a producer with its transport when a connection is made. N)r� r rT rD r r� � streaming�r= rk rs r- r- r. �test_HTTPChannelIsAProducer� s �z2ChannelProductionTests.test_HTTPChannelIsAProducerc C s: | � t� t�\}}|�� | �|jd� | �|jd� dS )zV L{HTTPChannel} unregisters itself when it has loseConnection called. N)r� r rT r� r r r� r� r- r- r. �8test_HTTPChannelUnregistersSelfWhenCallingLoseConnection� s �zOChannelProductionTests.test_HTTPChannelUnregistersSelfWhenCallingLoseConnectionc C s8 | � t� t�\}}|�t� d� | �t|jt� d� dS )z� If two producers are registered on a L{HTTPChannel} without the first being unregistered, a L{RuntimeError} is thrown. TN)r� r rT r; r" r� r% r� r- r- r. �(test_HTTPChannelRejectsMultipleProducers� s � �z?ChannelProductionTests.test_HTTPChannelRejectsMultipleProducersc C s, | � t� t�\}}|�� | �|jd� dS )zr If there is no producer, the L{HTTPChannel} can still have C{unregisterProducer} called. N�r� r rT r� r Z_requestProducerr� r- r- r. �+test_HTTPChannelCanUnregisterWithNoProducer� s �zBChannelProductionTests.test_HTTPChannelCanUnregisterWithNoProducerc C s, | � t� t�\}}|�� | �|jd� dS )zn If there is no request producer currently registered, C{stopProducing} does nothing. Nr� r� r- r- r. �,test_HTTPChannelStopWithNoRequestOutstanding� s �zCChannelProductionTests.test_HTTPChannelStopWithNoRequestOutstandingc C sf | � t� t�\}}|�| j� |jd j}t� }|�|d� | � |j g � |� � | � |j dg� dS )z� If there is a request producer registered with L{HTTPChannel}, calling C{stopProducing} causes that producer to be stopped as well. r T�stopN) r� r rl r� rg r� r� r" r; rD �eventsZ stopProducing�r= rk rs rg r r- r- r. �#test_HTTPChannelStopRequestProducer s � z:ChannelProductionTests.test_HTTPChannelStopRequestProducerc C sP | � t� t�\}}| �|jd� |�� | �|jd� |�� | �|jd� dS )z� When L{HTTPChannel} has C{pauseProducing} called on it by the transport it will call C{pauseProducing} on the transport. When unpaused, the L{HTTPChannel} will call C{resumeProducing} on its transport. r� r� N)r� r rT rD r� �pauseProducing�resumeProducingr� r- r- r. �;test_HTTPChannelPropagatesProducingFromTransportToTransport s �zRChannelProductionTests.test_HTTPChannelPropagatesProducingFromTransportToTransportc C s� | � t� t�\}}d|_|�| j� |�d� |jd j}t� }|� |d� | � |j d� | � |j g � |� � | � |j d� | � |j dg� |�� | � |j d� | � |j ddg� |�� |�� | � |j d� dS ) z� If a L{Request} object has registered itself as a producer with a L{HTTPChannel} object, and the L{HTTPChannel} object is paused, both the transport and L{Request} objects get paused. r s 123Tr� �pauseZresumer� N)r� r rl r� r� rg r� r� r" r; rD r� r� r� r� r� rm r� r- r- r. �3test_HTTPChannelPropagatesPausedProductionToRequest0 s* � zJChannelProductionTests.test_HTTPChannelPropagatesPausedProductionToRequestc C s� | � t� t�\}}d|_|�| j� |�d� |jd j}t� }|� |d� | � |j d� | � |j g � |� � | � |j d� | � |j dg� |�� |�� | � |j d� |�� | � |j d� dS )z� If a L{Request} object completes its response while the transport is paused, the L{HTTPChannel} does not resume the transport. r s extraTr� r� r� N)r� r rl r� r� rg r� r� r" r; rD r� r� r� r� rm r� r� r- r- r. �/test_HTTPChannelStaysPausedWhenRequestCompletes] s( � zFChannelProductionTests.test_HTTPChannelStaysPausedWhenRequestCompletesc C s� G dd� dt �}|� }|�� | �t|j� | �|� t�\}}| �|jd� |�� | �|jd� |�| j � | �|jd� | � |� � � d�� |� � | �|jd� dS )z� If the L{HTTPChannel} has paused the transport, it still tolerates receiving data, and does not attempt to pause the transport again. c @ s e Zd ZdZdd� ZdS )zgChannelProductionTests.test_HTTPChannelToleratesDataWhenTransportPaused.<locals>.NoDoublePauseTransportz| A version of L{StringTransport} that fails tests if it is paused while already paused. c S s | j dkrtd��t�| � d S )Nr� zTransport was paused twice!)r� r% r r� r< r- r- r. r� � s zvChannelProductionTests.test_HTTPChannelToleratesDataWhenTransportPaused.<locals>.NoDoublePauseTransport.pauseProducingN)r) r* r+ r, r� r- r- r- r. �NoDoublePauseTransport� s r� r� r� s HTTP/1.1 200 OK N)r r� r� r% r� rT rD r� r� rg r� r� rQ r� )r= r� rs rk r- r- r. �0test_HTTPChannelToleratesDataWhenTransportPaused� s � zGChannelProductionTests.test_HTTPChannelToleratesDataWhenTransportPausedc sh � � t� t�\}�t� �t�� }t|_|��� |�� j � |j d j }|j j }� �fdd�}|�|�S )z� If the L{HTTPChannel} has a L{IPullProducer} registered with it it can adapt that producer into an L{IPushProducer}. r c s( �� � �dd�d }d}� �||� d S )Nr} r| s< 1 0 1 1 1 2 1 3 1 4 1 5 1 6 1 7 1 8 1 9 )r� r~ rD )ZignZ responseBodyZexpectedResponseBody�r= rs r- r. �validate� s � zOChannelProductionTests.test_HTTPChannelToleratesPullProducers.<locals>.validate)r� r r7 r r� � DummyPullProducerHandlerProxyr� r� r� rg r� r� r9 rz r )r= rk rg ZresponseCompleter� r- r� r. �&test_HTTPChannelToleratesPullProducers� s � z=ChannelProductionTests.test_HTTPChannelToleratesPullProducersc C s| t � }t� }t�� }d|_|j|_|�|� |�d� | �|j |� | �|j d� |�d� | �|j d� | �|j d� dS )zS L{HTTPChannel} unregisters itself when it times out a connection. r� r� Tr| N) r r r r� r� ru r� r� r r r� )r= r� rs rk r- r- r. �,test_HTTPChannelUnregistersSelfWhenTimingOut� s zCChannelProductionTests.test_HTTPChannelUnregistersSelfWhenTimingOutN)r) r* r+ r, rg r� r� r� r� r� r� r� r� r� r� r� r� r� r- r- r- r. r� � s � -+,$r� )jr, Z __future__r r rB r� r� rR r r r � ImportErrorZ urllib.parse�ior Zzope.interfacer Zzope.interface.verifyr Ztwisted.python.compatr r r r r r Ztwisted.python.componentsr Ztwisted.python.failurer Z twisted.trialr Ztwisted.trial.unittestr Z twisted.webr r r Ztwisted.web.httpr r r Ztwisted.internetr Ztwisted.internet.taskr Ztwisted.internet.errorr r Ztwisted.protocolsr Ztwisted.test.proto_helpersr r r! Ztwisted.test.test_internetr" Z twisted.web.test.requesthelperr# r% r& Ztwisted.loggerr' r� r( r6 r� r7 r� r@ rT r� ZINonQueuedRequestFactoryri ZDummyNewHTTPHandlerProxyrl r� Z HTTPClientrn ry r� rz r� r� r� r� r� r� r� r� r r* r+ r8 rO rq r� r� r r r r� r� r� r� r� r- r- r- r. �<module> s� � � % r+ +). =4 8 1 @b GpJ2
Save File
Cancel