File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_wsgi.cpython-310.pyc
o
�b�* � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZmZ ddlmZmZ dd lmZ dd
lmZmZ dd lmZ dd lmZ dd
l m Z ddl m Z ddl m!Z!m"Z" ddl#m$Z$ ddl%m&Z&m'Z' ddl(m)Z)m*Z*m+Z+ ddl,m-Z- ddl.m/Z/ G dd� d�Z0G dd� d�Z1G dd� de"�Z2G dd � d �Z3G d d � d e3e"�Z4G d d � d e3�Z5G d!d"� d"e5e"�Z6G d#d$� d$e5e"�Z7G d%d&� d&e5e"�Z8G d'd(� d(e5e"�Z9G d)d*� d*e3e"�Z:G d+d,� d,e3e"�Z;dS )-z
Tests for L{twisted.web.wsgi}.
� N)�exc_info)�quote)� verifyObject)�reactor)� IPv4Address� IPv6Address)�Deferred�
gatherResults)�ConnectionLost)�Logger�globalLogPublisher)�Failure)� getThreadID)�
ThreadPool)�EventLoggingObserver)�SkipTest�TestCase)�http)� IResource�Resource)�Request�Site�version)� DummyChannel)� WSGIResourcec @ s e Zd ZdZe� Zdd� ZdS )�SynchronousThreadPoola
A single-threaded implementation of part of the L{ThreadPool} interface.
This implementation calls functions synchronously rather than running
them in a thread pool. It is used to make the tests which are not
directly for thread-related behavior deterministic.
c O s6 z
||i |�� W dS t y | j�d� Y dS w )zp
Call C{f(*a, **kw)} in this thread rather than scheduling it to be
called in a thread.
z<Callable passed to SynchronousThreadPool.callInThread failedN)�
BaseException�_log�failure��self�f�a�kw� r$ �</usr/lib/python3/dist-packages/twisted/web/test/test_wsgi.py� callInThread+ s
��z"SynchronousThreadPool.callInThreadN)�__name__�
__module__� __qualname__�__doc__r r r& r$ r$ r$ r% r ! s r c @ � e Zd ZdZdd� ZdS )�SynchronousReactorThreadsa�
A single-threaded implementation of part of the L{IReactorThreads}
interface. This implementation assumes that it will only be invoked
from the reactor thread, so it calls functions synchronously rather than
trying to schedule them to run in the reactor thread. It is used in
conjunction with L{SynchronousThreadPool} to make the tests which are
not directly for thread-related behavior deterministic.
c O s ||i |�� dS )ze
Call C{f(*a, **kw)} in this thread which should also be the reactor
thread.
Nr$ r r$ r$ r% �callFromThreadE s z(SynchronousReactorThreads.callFromThreadN�r' r( r) r* r- r$ r$ r$ r% r, ; s r, c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�WSGIResourceTestsc C s t t� t� dd� �| _dS )z�
Create a L{WSGIResource} with synchronous threading objects and a no-op
application object. This is useful for testing certain things about
the resource implementation which are unrelated to WSGI.
c S � d S �Nr$ ��environ�
startResponser$ r$ r% �<lambda>W � z)WSGIResourceTests.setUp.<locals>.<lambda>N)r r, r �resource�r r$ r$ r% �setUpN s
�zWSGIResourceTests.setUpc C s t t| j� | �| jj� dS )zW
L{WSGIResource} implements L{IResource} and stops resource traversal.
N)r r r7 �
assertTrue�isLeafr8 r$ r$ r% �test_interfacesZ s z!WSGIResourceTests.test_interfacesc C s6 | � t| jjdtt� d�� | � t| jjdt� � dS )z�
A L{WSGIResource} cannot have L{IResource} children. Its
C{getChildWithDefault} and C{putChild} methods raise L{RuntimeError}.
� fooFN)� assertRaises� RuntimeErrorr7 �getChildWithDefaultr r �putChildr r8 r$ r$ r% �test_unsupporteda s
�z"WSGIResourceTests.test_unsupportedc s� t �| t�}G dd� dt�� G dd� dt�}� fdd�}G � fdd�d�}t|� t� |�| _| j�|t � d �� | �
d
t |�� |d d }| � |j
� � | �� � d
S )z�
If an exception is thrown by the application, and then in the
exception handling code, verify it should be propagated to the
provided L{ThreadPool}.
c @ s e Zd ZdZdS )zIWSGIResourceTests.test_applicationAndRequestThrow.<locals>.ArbitraryErrorz;
An arbitrary error for this class
N)r' r( r) r* r$ r$ r$ r% �ArbitraryErrorv s rC c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) zPWSGIResourceTests.test_applicationAndRequestThrow.<locals>.FinishThrowingRequestzL
An L{IRequest} request whose finish method throws.
c _ s. t j| g|�R i |�� d| _d| _d| _d S )N� s www.example.com/stuff)r �__init__�prepath�postpath�uri)r �args�kwargsr$ r$ r% rE � s
zYWSGIResourceTests.test_applicationAndRequestThrow.<locals>.FinishThrowingRequest.__init__c S � dS )zi
Return loopback address.
@return: loopback ip address.
� 127.0.0.1r$ r8 r$ r$ r% � getClientIP� s z\WSGIResourceTests.test_applicationAndRequestThrow.<locals>.FinishThrowingRequest.getClientIPc S s t ddd�S )z`
Return a fake Address
@return: A fake address
�TCPrL i0u )r r8 r$ r$ r% �getHost� s zXWSGIResourceTests.test_applicationAndRequestThrow.<locals>.FinishThrowingRequest.getHostN)r' r( r) r* rE rM rO r$ r$ r$ r% �FinishThrowingRequest{ s
rP c � � � �)z�
An application object that throws an exception.
@param environ: unused
@param startResponse: unused
r$ r2 �rC r$ r% � application� s zFWSGIResourceTests.test_applicationAndRequestThrow.<locals>.applicationc s e Zd ZdZ� fdd�ZdS )zQWSGIResourceTests.test_applicationAndRequestThrow.<locals>.ThrowingReactorThreadszt
An L{IReactorThreads} implementation whose callFromThread raises
an exception.
c rQ )z�
Raise an exception to the caller.
@param f: unused
@param a: unused
@param kw: unused
r$ r rR r$ r% r- � s
z`WSGIResourceTests.test_applicationAndRequestThrow.<locals>.ThrowingReactorThreads.callFromThreadNr. r$ rR r$ r% �ThrowingReactorThreads� s rT F� r � log_failureN)r �createWithCleanupr � Exceptionr r r r7 �renderr � assertEquals�len�assertIsInstance�value�flushLoggedErrors)r � logObserverrP rS rT r! r$ rR r% � test_applicationAndRequestThrown s
� z1WSGIResourceTests.test_applicationAndRequestThrowN)r' r( r) r9 r< rB r` r$ r$ r$ r% r/ M s
r/ c @ sT e Zd ZdZeZdd� Zdg ddfdd�Zdd � Ze fd
d �Z
d d
� Z ddd�Z dS )�WSGITestsMixinz�
@ivar channelFactory: A no-argument callable which will be invoked to
create a new HTTP channel to associate with request objects.
c C s t � | _t� | _d S r1 )r �
threadpoolr, r r8 r$ r$ r% r9 � s zWSGITestsMixin.setUpNrD c s dd� } t | j| j|� �}
|�� |D ]}t� }|�| |�|
� |}
q|� }t|
�|_||d�}| D ]\}}|j� | |�| |�� q2|�
d� |
rU|j � |
� |j �
d� dd�� fdd�|D �� }|durw|d d
�� fd d�|D �� 7 }|�| |�| |�d | |� � |S )
a�
@param method: A C{str} giving the request method to use.
@param version: A C{str} like C{'1.1'} giving the request version.
@param resourceSegments: A C{list} of unencoded path segments which
specifies the location in the resource hierarchy at which the
L{WSGIResource} will be placed, eg C{['']} for I{/}, C{['foo',
'bar', '']} for I{/foo/bar/}, etc.
@param requestSegments: A C{list} of unencoded path segments giving the
request URI.
@param query: A C{list} of two-tuples of C{str} giving unencoded query
argument keys and values.
@param headers: A C{list} of two-tuples of C{str} giving request header
names and corresponding values.
@param safe: A C{str} giving the bytes which are to be considered
I{safe} for inclusion in the request URI and not quoted.
@return: A L{Deferred} which will be called back with a two-tuple of
the arguments passed which would be passed to the WSGI application
object for this configuration and request (ie, the environment and
start_response callable).
c S s t | t�r| S | �d�S )Nz
iso-8859-1)�
isinstance�bytes�encode)�stringr$ r$ r% �
_toByteString� s
z4WSGITestsMixin.lowLevelRender.<locals>._toByteStringFr �/c s g | ]}t |� ��qS r$ )�urlquote)�.0�seg��safer$ r% �
<listcomp> s z1WSGITestsMixin.lowLevelRender.<locals>.<listcomp>N�?�&c s* g | ]\}}d � t|� �t|� �g��qS )�=)�joinri )rj �k�vrl r$ r% rn s * s HTTP/)r r rb �reverser rA r �site�requestHeaders� addRawHeader� gotLength�content�write�seekrr �requestReceived)r �requestFactory�applicationFactory�channelFactory�methodr �resourceSegments�requestSegments�query�headers�bodyrm rg �rootrk �tmp�channel�requestrs rt rH r$ rl r% �lowLevelRender� s2 *
��z WSGITestsMixin.lowLevelRenderc s@ t � � � fdd�}|�d| j�}| jt||g|�R i |�� � S )Nc � � fdd�} | S )Nc s$ | \}}� � | � |dg � td�S �N�200 OKr$ ��callback�iter�rI r3 r4 ��resultr$ r% rS s
zFWSGITestsMixin.render.<locals>.applicationFactory.<locals>.applicationr$ �rS r� r$ r% r s z1WSGITestsMixin.render.<locals>.applicationFactoryr� )r �popr� r� r )r r" r# r r� r$ r� r% rY s
zWSGITestsMixin.renderc s t � � � �fdd�}� |fS )Nc s �| i |��}|� � �� � |S r1 )� notifyFinish�
chainDeferred)r" r# r� ��d� requestClassr$ r% r~ ) s z<WSGITestsMixin.requestFactoryFactory.<locals>.requestFactory�r )r r� r~ r$ r� r% �requestFactoryFactory& � z$WSGITestsMixin.requestFactoryFactoryc C s |� dd�d S )N�
rU )�split)r �responser$ r$ r% �getContentFromResponse4 s z%WSGITestsMixin.getContentFromResponsec sv t � �� �fdd�}t| j| j|�}t� }|�d|� | �� }t|�|_G dd� dt �}||dd�}|�
d� |�fS ) aj
Prepare a L{Request} which, when a request is received, captures the
C{environ} and C{start_response} callable passed to a WSGI app.
@param application: An optional WSGI application callable that accepts
the familiar C{environ} and C{start_response} args and returns an
iterable of body content. If not supplied, C{start_response} will
be called with a "200 OK" status and no headers, and no content
will be yielded.
@return: A two-tuple of (C{request}, C{deferred}). The former is a
Twisted L{Request}. The latter is a L{Deferred} which will be
called back with a two-tuple of the arguments passed to a WSGI
application (i.e. the C{environ} and C{start_response} callable),
or will errback with any error arising within the WSGI app.
c sj z� d u r|dg � t d�}n� | |�}W n ty+ ��� |dg � t d� Y S w ��| |f� |S )Nr� r$ z 500 Error)r� r �errbackr� )r3 r4 rz �rS r� r$ r% �outerApplicationJ s
�
�z7WSGITestsMixin.prepareRequest.<locals>.outerApplications resc @ s e Zd ZdZddd�ZdS ) z4WSGITestsMixin.prepareRequest.<locals>.CannedRequestz�
Convenient L{Request} derivative which has canned values for all
of C{requestReceived}'s arguments.
� GET� /res� 1.1c S s t j| |||d�S )N)�command�pathr )r r} )r r� r� r r$ r$ r% r} g s �zDWSGITestsMixin.prepareRequest.<locals>.CannedRequest.requestReceivedN)r� r� r� )r' r( r) r* r} r$ r$ r$ r% �
CannedRequesta s r� F)�queuedr ) r r r rb r rA r� r rv r ry )r rS r� r7 r� r� r� r� r$ r� r% �prepareRequest7 s
z WSGITestsMixin.prepareRequestr1 )
r' r( r) r* r r� r9 r� rY r r� r� r� r$ r$ r$ r% ra � s
�M ra c @ s 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>S )?� EnvironTestsz�
Tests for the values in the C{environ} C{dict} passed to the application
object by L{twisted.web.wsgi.WSGIResource}.
c s � ��fdd�}|S )Nc s | \}}�� |� �� �S r1 �� assertEqual�r� r3 r4 ��keyr r] r$ r% �assertEnvironKeyEqualy s z;EnvironTests.environKeyEqual.<locals>.assertEnvironKeyEqualr$ )r r� r] r� r$ r� r% �environKeyEqualx s z EnvironTests.environKeyEqualc s, � � ddg dg�}� fdd�}|�|� |S )z�
L{WSGIResource} calls the application object with an C{environ}
parameter which is exactly of type C{dict}.
�GET�1.1rD c s2 | \}}� � t|�t� |D ]}� �|t� qd S r1 )�assertIdentical�type�dictr\ �str)r� r3 r4 �namer8 r$ r% �
cbRendered� s
�z3EnvironTests.test_environIsDict.<locals>.cbRendered)rY � addCallback)r r� r� r$ r8 r% �test_environIsDict� s
z EnvironTests.test_environIsDictc C sT | � ddg dg�}|�| �dd�� | � ddg dg�}|�| �dd�� t||g�S )z�
The C{'REQUEST_METHOD'} key of the C{environ} C{dict} passed to the
application contains the HTTP method in the request (RFC 3875, section
4.1.12).
r� r� rD �REQUEST_METHOD�POST�rY r� r� r )r �get�postr$ r$ r% �test_requestMethod� s
z EnvironTests.test_requestMethodc C sH dD ] }| � � \}}|�|� |�| �dd�� | �| �|�t� qdS )z�
The C{'REQUEST_METHOD'} key of the C{environ} C{dict} passed to the
application is always a native string.
)r� r� r� r� N�r� r} r� r� r\ �successResultOfr� )r r� r� r� r$ r$ r% � test_requestMethodIsNativeString� s
�z-EnvironTests.test_requestMethodIsNativeStringc C s | � ddg dg�}|�| �dd�� | � dddgdg�}|�| �dd�� | � dddgdg�}|�| �dd�� | � ddddgddg�}|�| �dd�� | � dddgdd g�}|�| �dd�� | � ddg d
�g d
��}|�| �dd �� t||||||g�S ) z�
The C{'SCRIPT_NAME'} key of the C{environ} C{dict} passed to the
application contains the I{abs_path} (RFC 2396, section 3) to this
resource (RFC 3875, section 4.1.13).
r� r� rD � SCRIPT_NAMErh �fooz/fooz/foo/�bar�r� rh s bar�� /foo///barÿr� )r r� �
emptyChild�leaf� container�internal� unencodedr$ r$ r% �test_scriptName� s �z EnvironTests.test_scriptNamec C s^ | � � \}}|jdd� |�| �dd�� | �| �|�t� | � � \}}| jt|jdd� dS )z�
The C{'SCRIPT_NAME'} key of the C{environ} C{dict} passed to the
application is always a native string.
r� �r� r� z/resN� r� r} r� r� r\ r� r� r> � TypeError�r r� r� r$ r$ r% � test_scriptNameIsNativeString� � z*EnvironTests.test_scriptNameIsNativeStringc C s | � dd�}| �ddg dg�}|�| � dd�� | �dddgdg�}|�|� | �dddgdg�}|�|� | �ddddgddg�}|�|� | �dddgddg�}|�| � dd�� | �dddgddg�}|�| � dd�� | �ddg g d ��}|�| � dd
�� t||||||g�S ) z�
The C{'PATH_INFO'} key of the C{environ} C{dict} passed to the
application contains the suffix of the request URI path which is not
included in the value for the C{'SCRIPT_NAME'} key (RFC 3875, section
4.1.5).
� PATH_INFOrD r� r� rh r� r� z/barr� r� )r� rY r� r ) r �assertKeyEmptyr� r� r� r� � internalLeaf�internalContainerr� r$ r$ r% �
test_pathInfo� s$
�zEnvironTests.test_pathInfoc C �^ | � � \}}|jdd� |�| �dd�� | �| �|�t� | � � \}}| jt|jdd� dS )z
The C{'PATH_INFO'} key of the C{environ} C{dict} passed to the
application is always a native string.
s /res/foo/barr� r� z/foo/barz /res/foo/barNr� r� r$ r$ r% �test_pathInfoIsNativeString� r� z(EnvironTests.test_pathInfoIsNativeStringc C s� | � ddg dgd�}|�| �dd�� | � ddg dgg �}|�| �dd�� | � ddg dgdg�}|�| �dd�� | � ddg dgdg�}|�| �dd �� | j ddg dgd
gd d �}|�| �dd
�� t|||||g�S )z�
The C{'QUERY_STRING'} key of the C{environ} C{dict} passed to the
application contains the portion of the request URI after the first
I{?} (RFC 3875, section 4.1.7).
r� r� rD N� QUERY_STRING�r� r� �foo=bar)rh rh z%2F=%2F)r� z?barro rl zfoo=?barr� )r �missing�empty�presentr� �doubleQuestionr$ r$ r% �test_queryString
s �z EnvironTests.test_queryStringc C r� )z�
The C{'QUERY_STRING'} key of the C{environ} C{dict} passed to the
application is always a native string.
s /res?foo=barr� r� r� z /res?foo=barNr� r� r$ r$ r% � test_queryStringIsNativeString& r� z+EnvironTests.test_queryStringIsNativeStringc C �Z | � ddg dg�}|�| �dd�� | � ddg dgddg�}|�| �dd�� t||g�S )z�
The C{'CONTENT_TYPE'} key of the C{environ} C{dict} passed to the
application contains the value of the I{Content-Type} request header
(RFC 3875, section 4.1.3).
r� r� rD � CONTENT_TYPEN)� content-type� x-foo/barr� r� �r r� r� r$ r$ r% �test_contentType5 � � z EnvironTests.test_contentTypec C �T dD ]%}| � � \}}|j�d|� |�� |�| �dd�� | �| �|�t� qdS )z�
The C{'CONTENT_TYPE'} key of the C{environ} C{dict} passed to the
application is always a native string.
)s x-foo/barr� s Content-Typer� r� N� r� rw rx r} r� r� r\ r� r� )r � contentTyper� r� r$ r$ r% � test_contentTypeIsNativeStringE � �z+EnvironTests.test_contentTypeIsNativeStringc C r� )z�
The C{'CONTENT_LENGTH'} key of the C{environ} C{dict} passed to the
application contains the value of the I{Content-Length} request header
(RFC 3875, section 4.1.2).
r� r� rD �CONTENT_LENGTHN)�content-length�1234r� r� r� r$ r$ r% �test_contentLengthQ r� z EnvironTests.test_contentLengthc C r� )z�
The C{'CONTENT_LENGTH'} key of the C{environ} C{dict} passed to the
application is always a native string.
)s 1234r� s Content-Lengthr� r� Nr� )r �
contentLengthr� r� r$ r$ r% � test_contentLengthIsNativeStringa r� z-EnvironTests.test_contentLengthIsNativeStringc C sZ | � ddg dg�}|�| �dd�� | � ddg dgddg�}|�| �dd�� t||g�S ) a_
The C{'SERVER_NAME'} key of the C{environ} C{dict} passed to the
application contains the best determination of the server hostname
possible, using either the value of the I{Host} header in the request
or the address the server is listening on if that header is not
present (RFC 3875, section 4.1.14).
r� r� rD � SERVER_NAMEz10.0.0.1N)�host� example.orgr� r� r� r$ r$ r% �test_serverNamem s
z EnvironTests.test_serverNamec sT dD ]%� | � � \}}� fdd�|_|�� |�| �dd�� | �| �|�t� qdS )z�
The C{'SERVER_NAME'} key of the C{environ} C{dict} passed to the
application is always a native string.
)s host.example.com�host.example.comc � � S r1 r$ r$ ��
serverNamer$ r% r5 � r6 z<EnvironTests.test_serverNameIsNativeString.<locals>.<lambda>r� r N)r� �getRequestHostnamer} r� r� r\ r� r� r� r$ r r% � test_serverNameIsNativeString s �z*EnvironTests.test_serverNameIsNativeStringc sB d� � fdd�}|| _ | �ddg dg�}|�| �dt� ��� |S )z�
The C{'SERVER_PORT'} key of the C{environ} C{dict} passed to the
application contains the port number of the server which received the
request (RFC 3875, section 4.1.15).
iB0 c s t � } t �� | _� | j_| S r1 )r rN � transport�port�r� ��
portNumberr$ r% � makeChannel� s
z1EnvironTests.test_serverPort.<locals>.makeChannelr� r� rD � SERVER_PORT)r� rY r� r� r� )r r r� r$ r r% �test_serverPort� s z EnvironTests.test_serverPortc C s< | � � \}}|�� |�| �dd�� | �| �|�t� dS )z�
The C{'SERVER_PORT'} key of the C{environ} C{dict} passed to the
application is always a native string.
r �80Nr� r� r$ r$ r% � test_serverPortIsNativeString� s z*EnvironTests.test_serverPortIsNativeStringc C sT | � ddg dg�}|�| �dd�� | � ddg dg�}|�| �dd�� t||g�S )z�
The C{'SERVER_PROTOCOL'} key of the C{environ} C{dict} passed to the
application contains the HTTP version number received in the request
(RFC 3875, section 4.1.16).
r� z1.0rD �SERVER_PROTOCOLzHTTP/1.0r� zHTTP/1.1r� )r �old�newr$ r$ r% �test_serverProtocol� s
z EnvironTests.test_serverProtocolc C sT dD ]%}| � � \}}dd� |_|jdd� |�| �dd�� | �| �|�t� qdS ) z�
The C{'SERVER_PROTOCOL'} key of the C{environ} C{dict} passed to the
application is always a native string.
)r� r� c S r0 r1 r$ )�datar$ r$ r% r5 � r6 z@EnvironTests.test_serverProtocolIsNativeString.<locals>.<lambda>r� )r r r� N)r� r{ r} r� r� r\ r� r� )r �serverProtocolr� r� r$ r$ r% �!test_serverProtocolIsNativeString� s
�z.EnvironTests.test_serverProtocolIsNativeStringc C �( | � ddg dg�}|�| �dd�� |S )z�
The C{'REMOTE_ADDR'} key of the C{environ} C{dict} passed to the
application contains the address of the client making the request.
r� r� rD � REMOTE_ADDRz 192.168.1.1�rY r� r� �r r� r$ r$ r% �test_remoteAddr� s z EnvironTests.test_remoteAddrc C s4 dd� }| j ddg dg|d�}|�| �dd�� |S ) z�
The C{'REMOTE_ADDR'} key of the C{environ} C{dict} passed to
the application contains the address of the client making the
request when connecting over IPv6.
c S s t tddd�d�S )NrN �::1i� )�peer)r r r$ r$ r$ r% r� � � z8EnvironTests.test_remoteAddrIPv6.<locals>.channelFactoryr� r� rD )r� r r r )r r� r� r$ r$ r% �test_remoteAddrIPv6� s z EnvironTests.test_remoteAddrIPv6c s� � � ddg dgdddg�}� fdd�}|�|� � � ddg dgddd g�}|�� �d
d �� � � ddg dgdd g�}|�� �d
d�� � � ddg dgddg�}|�� �d
d�� t||||g�S )z�
HTTP request headers are copied into the C{environ} C{dict} passed to
the application with a C{HTTP_} prefix added to their names.
r� r� rD Nr� ��baz�quuxc s, | \}}� � |d d� � � |d d� d S )N�HTTP_FOOr� �HTTP_BAZr" r� r� r8 r$ r% r� � s z-EnvironTests.test_headers.<locals>.cbRendered)r� r! r# zbar,baz)zfoo-barr! � HTTP_FOO_BARr! )r� zbar
bazzbar bazr� )r � singleValuer� �
multiValue�
withHyphen� multiLiner$ r8 r% � test_headers� s �
�zEnvironTests.test_headersc C r )z�
The C{'wsgi.version'} key of the C{environ} C{dict} passed to the
application has the value C{(1, 0)} indicating that this is a WSGI 1.0
container.
r� r� rD z wsgi.version)rU r r )r �versionDeferredr$ r$ r% �test_wsgiVersion� s z EnvironTests.test_wsgiVersionc C r )z|
The C{'wsgi.run_once'} key of the C{environ} C{dict} passed to the
application is set to C{False}.
r� r� rD z
wsgi.run_onceFr )r �oncer$ r$ r% �test_wsgiRunOnce � z EnvironTests.test_wsgiRunOncec C r )z~
The C{'wsgi.multithread'} key of the C{environ} C{dict} passed to the
application is set to C{True}.
r� r� rD zwsgi.multithreadTr )r �threadr$ r$ r% �test_wsgiMultithread r/ z!EnvironTests.test_wsgiMultithreadc C r )z�
The C{'wsgi.multiprocess'} key of the C{environ} C{dict} passed to the
application is set to C{False}.
r� r� rD zwsgi.multiprocessFr )r �processr$ r$ r% �test_wsgiMultiprocess r/ z"EnvironTests.test_wsgiMultiprocessc C sh dd� }t | _| �ddg dg�}|�| �dd�� || _| �ddg dg�}|�| �dd�� t||g�S ) z�
The C{'wsgi.url_scheme'} key of the C{environ} C{dict} passed to the
application has the request URL scheme.
c S s t � } t �� | _| S r1 )r �SSLr r r$ r$ r% r� + s
z7EnvironTests.test_wsgiURLScheme.<locals>.channelFactoryr� r� rD zwsgi.url_schemer �https)r r� rY r� r� r )r r� � httpDeferred�
httpsDeferredr$ r$ r% �test_wsgiURLScheme$ s z EnvironTests.test_wsgiURLSchemec s: t ��t�� ��ddg dg�}� �fdd�}|�|� |S )a`
The C{'wsgi.errors'} key of the C{environ} C{dict} passed to the
application is a file-like object (as defined in the U{Input and Errors
Streams<http://www.python.org/dev/peps/pep-0333/#input-and-error-streams>}
section of PEP 333) which converts bytes written to it into events for
the logging system.
r� r� rD c s� | \}}|d }|� d� |�dg� |�� ��� d d d� ��� d d d� ��� d d � ��� d
d d � ��� d
d d� ��� d
d � ��t� �d � d S )
N� wsgi.errors�
some message
�another
message
r �message)r: �system�wsgi�isErrorrU )r; � )r{ �
writelines�flushr� r: r[ )r� r3 �startApplication�errors��eventsr r$ r% �cbErrorsF s
z.EnvironTests.test_wsgiErrors.<locals>.cbErrors)r rW r rY r� )r rD rG r$ rE r% �test_wsgiErrors: s
z EnvironTests.test_wsgiErrorsc C sN | � � \}}|�� | �|�\}}|d }| �t|jd�}| �dt|�� dS )z�
The C{'wsgi.errors'} file-like object from the C{environ} C{dict}
permits writes of only native strings in Python 3, and raises
C{TypeError} for writes of non-native strings.
r9 s fredz1write() argument must be str, not b'fred' (bytes)N)r� r} r� r> r� r{ r� r� )r r� r� r3 �_rD �errorr$ r$ r% �0test_wsgiErrorsAcceptsOnlyNativeStringsInPython3W s �z=EnvironTests.test_wsgiErrorsAcceptsOnlyNativeStringsInPython3N)"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. r1 r3 r8 rH rK r$ r$ r$ r% r� r s@ "
r� c @ s� 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.S )/�InputStreamTestMixinz�
A mixin for L{TestCase} subclasses which defines a number of tests against
L{_InputStream}. The subclass is expected to create a file-like object to
be wrapped by an L{_InputStream} under test.
c C s t | jj� d���)Nz .getFile must be implemented)�NotImplementedError� __class__r' r8 r$ r$ r% � getFileTypeo s �z InputStreamTestMixin.getFileTypec sR | � � � G � fdd�dt�}dd� }||�\}}| �||tddg dgd g |�
|S )Nc � e Zd Z� fdd�ZdS )zLInputStreamTestMixin._renderAndReturnReaderResult.<locals>.CustomizedRequestc s � � | _ d S r1 )rz )r �length�r� r$ r% ry x s zVInputStreamTestMixin._renderAndReturnReaderResult.<locals>.CustomizedRequest.gotLengthN)r' r( r) ry r$ rR r$ r% �CustomizedRequestw � rS c s t � �� �fdd�}�|fS )Nc � � �fdd�} | S )Nc s, | \}}�� � |d �� |dg � td�S )Nz
wsgi.inputr� r$ r� r� ��readerr� r$ r% rS � s
z}InputStreamTestMixin._renderAndReturnReaderResult.<locals>.appFactoryFactory.<locals>.applicationFactory.<locals>.applicationr$ r� rV r$ r% r � s zhInputStreamTestMixin._renderAndReturnReaderResult.<locals>.appFactoryFactory.<locals>.applicationFactoryr� )rW r r$ rV r% �appFactoryFactory~ r� zLInputStreamTestMixin._renderAndReturnReaderResult.<locals>.appFactoryFactory�PUTr� rD )rO r r� r )r rW rz rS rX r� �
appFactoryr$ rR r% � _renderAndReturnReaderResultt s" � z1InputStreamTestMixin._renderAndReturnReaderResultc C �& d}| � dd� |�}|�| j|� |S )zi
Calling L{_InputStream.read} with no arguments returns the entire input
stream.
� some bytes are herec S � | � � S r1 ��read��inputr$ r$ r% r5 � � z3InputStreamTestMixin.test_readAll.<locals>.<lambda>�r[ r� r� �r rd r� r$ r$ r% � test_readAll� � z!InputStreamTestMixin.test_readAllc C �& d}| � dd� |�}|�| jd� |S )z�
Calling L{_InputStream.read} with an integer returns that many bytes
from the input stream, as long as it is less than or equal to the total
number of bytes available.
s
hello, world.c S �
| � d�S �N� r_ ra r$ r$ r% r5 � �
z4InputStreamTestMixin.test_readSome.<locals>.<lambda>s helrd re r$ r$ r% �
test_readSome� s z"InputStreamTestMixin.test_readSomec s* d� | � � fdd�� �}|�| j� � |S )z�
Calling L{_InputStream.read} with an integer that is greater than the
total number of bytes in the input stream returns all bytes in the
input stream.
r] c s | � t� �d �S rj )r` r[ ra �rd r$ r% r5 � s z8InputStreamTestMixin.test_readMoreThan.<locals>.<lambda>rd r r$ rn r% �test_readMoreThan� s �z&InputStreamTestMixin.test_readMoreThanc C s2 d}dd� }| � ||�}|�| j|dd� � |S )z�
Calling L{_InputStream.read} a second time returns bytes starting from
the position after the last byte returned by the previous read.
s some bytes, helloc S s | � d� | � � S rj r_ ra r$ r$ r% r` � �
z1InputStreamTestMixin.test_readTwice.<locals>.readrk Nrd )r rd r` r� r$ r$ r% �test_readTwice� s
z#InputStreamTestMixin.test_readTwicec C r\ )zy
Calling L{_InputStream.read} with L{None} as an argument returns all
bytes in the input stream.
s the entire streamc S �
| � d �S r1 r_ ra r$ r$ r% r5 � rl z4InputStreamTestMixin.test_readNone.<locals>.<lambda>rd re r$ r$ r% �
test_readNone� rg z"InputStreamTestMixin.test_readNonec C r\ )z�
Calling L{_InputStream.read} with a negative integer as an argument
returns all bytes in the input stream.
s all of the inputc S ri �N���r_ ra r$ r$ r% r5 � rl z8InputStreamTestMixin.test_readNegative.<locals>.<lambda>rd re r$ r$ r% �test_readNegative� rg z&InputStreamTestMixin.test_readNegativec C rh )zs
Calling L{_InputStream.readline} with no argument returns one line from
the input stream.
s hello
worldc S r^ r1 ��readlinera r$ r$ r% r5 � rc z4InputStreamTestMixin.test_readline.<locals>.<lambda>s hello
rd re r$ r$ r% �
test_readline� rg z"InputStreamTestMixin.test_readlinec C rh )a
Calling L{_InputStream.readline} with an integer returns at most that
many bytes, even if it is not enough to make up a complete line.
COMPATIBILITY NOTE: the size argument is excluded from the WSGI
specification, but is provided here anyhow, because useful libraries
such as python stdlib's cgi.py assume their input file-like-object
supports readline with a size argument. If you use it, be aware your
application may not be portable to other conformant WSGI servers.
s
goodbye
worldc S ri rj rw ra r$ r$ r% r5 � rl z8InputStreamTestMixin.test_readlineSome.<locals>.<lambda>s goord re r$ r$ r% �test_readlineSome� s z&InputStreamTestMixin.test_readlineSomec C rh )z�
Calling L{_InputStream.readline} with an integer which is greater than
the number of bytes in the next line returns only the next line.
s some lines
of textc S ri )N� rw ra r$ r$ r% r5 rl z<InputStreamTestMixin.test_readlineMoreThan.<locals>.<lambda>s some lines
rd re r$ r$ r% �test_readlineMoreThan� rg z*InputStreamTestMixin.test_readlineMoreThanc C s* d}dd� }| � ||�}|�| jd� |S )z�
Calling L{_InputStream.readline} a second time returns the line
following the line returned by the first call.
s first line
second line
last linec S s | � � | � � S r1 rw ra r$ r$ r% rx s z9InputStreamTestMixin.test_readlineTwice.<locals>.readlines second line
rd )r rd rx r� r$ r$ r% �test_readlineTwice s
z'InputStreamTestMixin.test_readlineTwicec C rh )z~
Calling L{_InputStream.readline} with L{None} as an argument returns
one line from the input stream.
s% this is one line
this is another linec S rr r1 rw ra r$ r$ r% r5 rl z8InputStreamTestMixin.test_readlineNone.<locals>.<lambda>s this is one line
rd re r$ r$ r% �test_readlineNone rg z&InputStreamTestMixin.test_readlineNonec C rh )z�
Calling L{_InputStream.readline} with a negative integer as an argument
returns one line from the input stream.
s input stream line one
line twoc S ri rt rw ra r$ r$ r% r5 # rl z<InputStreamTestMixin.test_readlineNegative.<locals>.<lambda>s input stream line one
rd re r$ r$ r% �test_readlineNegative rg z*InputStreamTestMixin.test_readlineNegativec C �* d}| � dd� |�}|�| jg d�� |S )z�
Calling L{_InputStream.readlines} with no arguments returns a list of
all lines from the input stream.
s alice
bob
carolc S r^ r1 �� readlinesra r$ r$ r% r5 - rc z5InputStreamTestMixin.test_readlines.<locals>.<lambda>)s alice
s bob
s carolrd re r$ r$ r% �test_readlines' � z#InputStreamTestMixin.test_readlinesc s. d}� � dd� |�}� fdd�}|�|� |S )z�
Calling L{_InputStream.readlines} with an integer as an argument
returns a list of lines from the input stream with the argument serving
as an approximate bound on the total number of bytes to read.
s
123
456
789
0c S ri )N� r� ra r$ r$ r% r5 8 rl z9InputStreamTestMixin.test_readlinesSome.<locals>.<lambda>c s � � | d d� ddg� d S )Nr@ s 123
s 456
r� )�linesr8 r$ r% �cbLines: s z8InputStreamTestMixin.test_readlinesSome.<locals>.cbLines)r[ r� )r rd r� r� r$ r8 r% �test_readlinesSome1 s
z'InputStreamTestMixin.test_readlinesSomec C r� )z�
Calling L{_InputStream.readlines} with an integer which is greater than
the total number of bytes in the input stream returns a list of all
lines from the input.
s" one potato
two potato
three potatoc S ri )N�d r� ra r$ r$ r% r5 I rl z=InputStreamTestMixin.test_readlinesMoreThan.<locals>.<lambda>)s one potato
s two potato
s three potatord re r$ r$ r% �test_readlinesMoreThanB s
�z+InputStreamTestMixin.test_readlinesMoreThanc C �. d}dd� }| � ||�}|�| jddg� |S )z�
Calling L{_InputStream.readlines} after a call to L{_InputStream.read}
returns lines starting at the byte after the last byte returned by the
C{read} call.
s hello
world
fooc S s | � d� | �� S )N� )r` r� ra r$ r$ r% r� W rp z?InputStreamTestMixin.test_readlinesAfterRead.<locals>.readliness orld
r= rd )r rd r� r� r$ r$ r% �test_readlinesAfterReadO �
z,InputStreamTestMixin.test_readlinesAfterReadc C �* d}| � dd� |�}|�| jddg� |S )zy
Calling L{_InputStream.readlines} with L{None} as an argument returns
all lines from the input.
s one fish
two fish
c S rr r1 r� ra r$ r$ r% r5 f rl z9InputStreamTestMixin.test_readlinesNone.<locals>.<lambda>s one fish
s two fish
rd re r$ r$ r% �test_readlinesNone_ s �z'InputStreamTestMixin.test_readlinesNonec C r� )z�
Calling L{_InputStream.readlines} with a negative integer as an
argument returns a list of all lines from the input.
s red fish
blue fish
c S ri rt r� ra r$ r$ r% r5 q rl z=InputStreamTestMixin.test_readlinesNegative.<locals>.<lambda>s red fish
s
blue fish
rd re r$ r$ r% �test_readlinesNegativek r� z+InputStreamTestMixin.test_readlinesNegativec C r� )zV
Iterating over L{_InputStream} produces lines from the input stream.
� green eggs
and ham
c S s t | �S r1 )�listra r$ r$ r% r5 z rc z4InputStreamTestMixin.test_iterable.<locals>.<lambda>s green eggs
� and ham
rd re r$ r$ r% �
test_iterableu s z"InputStreamTestMixin.test_iterablec C r� )z�
Iterating over L{_InputStream} after calling L{_InputStream.read}
produces lines from the input stream starting from the first byte after
the last byte returned by the C{read} call.
r� c S s | � d� t| �S rj )r` r� ra r$ r$ r% �iterate� rp z<InputStreamTestMixin.test_iterableAfterRead.<locals>.iterates en eggs
r� rd )r rd r� r� r$ r$ r% �test_iterableAfterRead~ r� z+InputStreamTestMixin.test_iterableAfterReadN)r' r( r) r* rO r[ rf rm ro rq rs rv ry rz r| r} r~ r r� r� r� r� r� r� r� r� r$ r$ r$ r% rL h s0 '
rL c @ r+ )�InputStreamStringIOTestsz�
Tests for L{_InputStream} when it is wrapped around a
L{StringIO.StringIO}.
This is only available in Python 2.
c C s* z ddl m } W |S ty td��w )Nr ��StringIOz#StringIO.StringIO is not available.)r� � ImportErrorr �r r� r$ r$ r% rO � � ��z$InputStreamStringIOTests.getFileTypeN�r' r( r) r* rO r$ r$ r$ r% r� � � r� c @ r+ )�InputStreamCStringIOTestsz�
Tests for L{_InputStream} when it is wrapped around a
L{cStringIO.StringIO}.
This is only available in Python 2.
c C s* z ddl m} W |S ty td��w )Nr r� z$cStringIO.StringIO is not available.)� cStringIOr� r� r r� r$ r$ r% rO � r� z%InputStreamCStringIOTests.getFileTypeNr� r$ r$ r$ r% r� � r� r� c @ r+ )�InputStreamBytesIOTestszO
Tests for L{_InputStream} when it is wrapped around an L{io.BytesIO}.
c C s ddl m} |S )Nr )�BytesIO)�ior� )r r� r$ r$ r% rO � s z#InputStreamBytesIOTests.getFileTypeNr� r$ r$ r$ r% r� � � r� c @ r+ )� InputStreamTemporaryFileTestszZ
Tests for L{_InputStream} when it is wrapped around a L{tempfile.TemporaryFile}.
c C s t jS r1 )�tempfile�
TemporaryFiler8 r$ r$ r% rO � s z)InputStreamTemporaryFileTests.getFileTypeNr� r$ r$ r$ r% r� � r� r� c @ s� 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,S )-�StartResponseTestszl
Tests for the I{start_response} parameter passed to the application object
by L{WSGIResource}.
c �Z t � � dd� }��� \}}� �fdd�}|�|� ��||� fdd�ddg d gd
g � |S ) z�
The response status passed to the I{start_response} callable is written
as the status of the response to the request.
c S � dd� } | S )Nc S � |dg � t d�S )Nz107 Strange messager$ �r� r2 r$ r$ r% rS � rp zOStartResponseTests.test_status.<locals>.applicationFactory.<locals>.applicationr$ r� r$ r$ r% r � � z:StartResponseTests.test_status.<locals>.applicationFactoryc � �� � jj�� �d�� d S )Ns HTTP/1.1 107 Strange message�r: r �written�getvalue�
startswith��ignored�r� r r$ r% r� � s
��z2StartResponseTests.test_status.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 � r6 z0StartResponseTests.test_status.<locals>.<lambda>r� r� rD N�r r� r� r� �r r r� r~ r� r$ r� r% � test_status� s"
� z StartResponseTests.test_statusc sD d��fdd�}� � |�\}}|�� � fdd�}� �|t��|�S )z�
The response status passed to the I{start_response} callable MUST be a
native string in Python 2 and Python 3.
s 200 OKc s |� g � t d�S )Nr$ r� r2 )�statusr$ r% rS � rp zEStartResponseTests.test_statusMustBeNativeString.<locals>.applicationc � � � dt| �� d S )Nz)status must be str, not b'200 OK' (bytes)�r� r� �rJ r8 r$ r% � checkMessage s zFStartResponseTests.test_statusMustBeNativeString.<locals>.checkMessage�r� r} �
assertFailurer� r� �r rS r� r� r� r$ )r r� r% � test_statusMustBeNativeString� s z0StartResponseTests.test_statusMustBeNativeStringc sr �� tddd� � t� �� fdd�}��� \}}���fdd�}|�|� ��||�fdd�d d
g d gd g � |S )
a
Verify that if the response headers given by C{appHeaders} are passed
to the I{start_response} callable, then the response header lines given
by C{expectedHeaders} plus I{Server} and I{Date} header lines are
included in the response.
�datetimeToStringc S rK )N�Tuesdayr$ r$ r$ r$ r% r5
r6 z1StartResponseTests._headersTest.<locals>.<lambda>c r� )Nc s |d� � t d�S r� r� r2 ��
appHeadersr$ r% rS rp zPStartResponseTests._headersTest.<locals>.applicationFactory.<locals>.applicationr$ r� r� r$ r% r � z;StartResponseTests._headersTest.<locals>.applicationFactoryc s` � j j�� }|�dd�\}}|�d�dd � }|�� �ddt dg }|�� ��||� d S )Nr� rU s
s
Date: Tuesdays Server: s Transfer-Encoding: chunked)r r� r� r� �sortr r� )r� r� r� �rest� headerLines�allExpectedHeaders)r� �expectedHeadersr r$ r% r� s �z3StartResponseTests._headersTest.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 , r6 r� r� rD N)�patchr r r� r� r� )r r� r� r r� r~ r� r$ )r� r� r� r r% � _headersTest s$
� z StartResponseTests._headersTestc C s | � ddgddg�S )z�
The headers passed to the I{start_response} callable are included in
the response as are the required I{Date} and I{Server} headers and the
necessary connection (hop to hop) header I{Transfer-Encoding}.
r� r s Baz: quuxs Foo: bar�r� r8 r$ r$ r% r* 6 s �z StartResponseTests.test_headersc sF dg� � fdd�}�� |�\}}|�� �fdd�}��|t��|�S )zb
The headers passed to the I{start_response} callable MUST be a
sequence.
�r� r] c s |dt � �� t d�S r� r� r2 )r� r$ r% rS G s zBStartResponseTests.test_headersMustBeSequence.<locals>.applicationc � � � t| �d� d S )NzKheaders must be a list, not <(list_?|sequence)iterator .+> [(]\1iterator[)]�� assertRegexr� r� r8 r$ r% r� N � �zCStartResponseTests.test_headersMustBeSequence.<locals>.checkMessager� r� r$ )r� r r% �test_headersMustBeSequence@ s z-StartResponseTests.test_headersMustBeSequencec C � dd� }| � |�\}}tjdd��}|�� | �|�}W d � n1 s%w Y | �dt|�� | �t|d j� | �dt |d j
�� dS ) a�
According to PEP-3333, the headers passed to the I{start_response}
callable MUST be a plain list:
The response_headers argument ... must be a Python list; i.e.
type(response_headers) is ListType
However, for bug-compatibility, any sequence is accepted. In both
Python 2 and Python 3, only a warning is issued when a sequence other
than a list is encountered.
c S s |dd� t d�S )Nr� ))�notr� r$ r� r2 r$ r$ r% rS d rp zEStartResponseTests.test_headersShouldBePlainList.<locals>.applicationT��recordNrU r z8headers should be a list, not (('not', 'list'),) (tuple)� r� �warnings�catch_warningsr} r� r� r[ �RuntimeWarning�categoryr� r< �r rS r� r� �caughtr$ r$ r% � test_headersShouldBePlainListW �
� �z0StartResponseTests.test_headersShouldBePlainListc sD d� � fdd�}�� |�\}}|�� �fdd�}��|t��|�S )zb
Each header passed to the I{start_response} callable MUST be a
sequence.
r� c s |dt � �g� t d�S r� r� r2 )�headerr$ r% rS { � zFStartResponseTests.test_headersMustEachBeSequence.<locals>.applicationc r� )Nz[header must be a [(]str, str[)] tuple, not <(tuple_?|sequence)iterator .+> [(]\1iterator[)]r� r� r8 r$ r% r� � r� zGStartResponseTests.test_headersMustEachBeSequence.<locals>.checkMessager� r� r$ )r� r r% � test_headersMustEachBeSequencet s z1StartResponseTests.test_headersMustEachBeSequencec C r� ) a�
According to PEP-3333, each header passed to the I{start_response}
callable should be a tuple:
The response_headers argument is a list of (header_name,
header_value) tuples
However, for bug-compatibility, any 2 element sequence is also
accepted. In both Python 2 and Python 3, only a warning is issued when
a sequence other than a tuple is encountered.
c S s |dddgg� t d�S )Nr� r� �tupler$ r� r2 r$ r$ r% rS � r� zEStartResponseTests.test_headersShouldEachBeTuple.<locals>.applicationTr� NrU r z@header should be a (str, str) tuple, not ['not', 'tuple'] (list)r� r� r$ r$ r% � test_headersShouldEachBeTuple� r� z0StartResponseTests.test_headersShouldEachBeTuplec �< dd� }� � |�\}}|�� � fdd�}� �|t��|�S )z�
Each header passed to the I{start_response} callable MUST hold a key
and a value, and ONLY a key and a value.
c S s |ddg� t d�S )Nr� )�too�many�cooksr$ r� r2 r$ r$ r% rS � s zMStartResponseTests.test_headersShouldEachHaveKeyAndValue.<locals>.applicationc r� )Nz?header must be a (str, str) tuple, not ('too', 'many', 'cooks')r� r� r8 r$ r% r� � s �zNStartResponseTests.test_headersShouldEachHaveKeyAndValue.<locals>.checkMessager� r� r$ r8 r% �%test_headersShouldEachHaveKeyAndValue� s
z8StartResponseTests.test_headersShouldEachHaveKeyAndValuec sF d� � fdd�}�� |�\}}|�� � �fdd�}��|t��|�S )z�
Each header key passed to the I{start_response} callable MUST be at
native string in Python 2 and Python 3.
s keyc s |d� dfg� t d�S )Nr� r] r$ r� r2 )r� r$ r% rS � r� zHStartResponseTests.test_headerKeyMustBeNativeString.<locals>.applicationc s �� d� �d�t| �� d S )Nz&header must be (str, str) tuple, not (z
, 'value')r� r� �r� r r$ r% r� � �
�zIStartResponseTests.test_headerKeyMustBeNativeString.<locals>.checkMessager� r� r$ r� r% � test_headerKeyMustBeNativeString� � z3StartResponseTests.test_headerKeyMustBeNativeStringc sF d��fdd�}� � |�\}}|�� � �fdd�}� �|t��|�S )z�
Each header value passed to the I{start_response} callable MUST be at
native string in Python 2 and Python 3.
s valuec s |dd� fg� t d�S )Nr� r� r$ r� r2 )r] r$ r% rS � r� zJStartResponseTests.test_headerValueMustBeNativeString.<locals>.applicationc s � � d��d�t| �� d S )Nz-header must be (str, str) tuple, not ('key', �)r� r� �r r] r$ r% r� � r� zKStartResponseTests.test_headerValueMustBeNativeString.<locals>.checkMessager� r� r$ r� r% �"test_headerValueMustBeNativeString� r� z5StartResponseTests.test_headerValueMustBeNativeStringc C s | � dgdg�S )z�
If I{Content-Type} is included in the headers passed to the
I{start_response} callable, one I{Content-Type} header is included in
the response.
)r� zmonkeys are greats Content-Type: monkeys are greatr� r8 r$ r$ r% �#test_applicationProvidedContentType� s �z6StartResponseTests.test_applicationProvidedContentTypec C s | � g d�g �S )z�
If either I{Server} or I{Date} is included in the headers passed to the
I{start_response} callable, they are disregarded.
))�serverr� )�Serverr� )�dater� )�dATEr� r� r8 r$ r$ r% �%test_applicationProvidedServerAndDate� s �z8StartResponseTests.test_applicationProvidedServerAndDatec �p t � � g �� �fdd���fdd�}��� \}}��fdd�}|�|� ��||� fdd�d d
g d gd g � |S )
a
Nothing is written in response to a request when the I{start_response}
callable is invoked. If the iterator returned by the application
object produces only empty strings, the response is written after the
last element is produced.
c � �� � jj�� � d S r1 ��appendr r� r� r$ �r� �intermediateValuesr$ r% r� � z:StartResponseTests.test_delayedUntilReturn.<locals>.recordc r� )Nc 3 s � |dddg� dV � � d S )Nr� r� r � r$ r2 r� r$ r% rS s �
z[StartResponseTests.test_delayedUntilReturn.<locals>.applicationFactory.<locals>.applicationr$ r� r� r$ r% r s zFStartResponseTests.test_delayedUntilReturn.<locals>.applicationFactoryc s �� � dg� d S )Nr r� r� �r� r r$ r% r� r z>StartResponseTests.test_delayedUntilReturn.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 r6 z<StartResponseTests.test_delayedUntilReturn.<locals>.<lambda>r� r� rD Nr� r� r$ �r� r� r� r r% �test_delayedUntilReturn� s&
� z*StartResponseTests.test_delayedUntilReturnc r� )
a
Nothing is written in response to a request when the I{start_response}
callable is invoked. Once a non-empty string has been produced by the
iterator returned by the application object, the response status and
headers are written.
c r� r1 r� r$ r� r$ r% r� 3 r z;StartResponseTests.test_delayedUntilContent.<locals>.recordc r� )Nc 3 �* � |ddg� dV � � dV � � d S )Nr� r� r r= r$ r2 r� r$ r% rS 7 � �
z\StartResponseTests.test_delayedUntilContent.<locals>.applicationFactory.<locals>.applicationr$ r� r� r$ r% r 6 � zGStartResponseTests.test_delayedUntilContent.<locals>.applicationFactoryc s �� � d � ��� d � d S )Nr rU )� assertFalser: r� r r$ r% r� B s z?StartResponseTests.test_delayedUntilContent.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 K r6 z=StartResponseTests.test_delayedUntilContent.<locals>.<lambda>r� r� rD Nr� r� r$ r r% �test_delayedUntilContent( s&
� z+StartResponseTests.test_delayedUntilContentc r� )
z�
Content produced by the iterator returned by the application object is
written to the request as it is produced.
c r� r1 r� r$ r� r$ r% r� _ r z/StartResponseTests.test_content.<locals>.recordc r� )Nc 3 r )Nr� �r� �6r= � barr$ r2 r� r$ r% rS c r zPStartResponseTests.test_content.<locals>.applicationFactory.<locals>.applicationr$ r� r� r$ r% r b r z;StartResponseTests.test_content.<locals>.applicationFactoryc �0 �� ��� d �d� �� ��� d �d� d S �Nr r= rU s foobar�r� r� r� r r$ r% r� n � �z3StartResponseTests.test_content.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 y r6 z1StartResponseTests.test_content.<locals>.<lambda>r� r� rD Nr� r� r$ r r% � test_contentV s&
� z StartResponseTests.test_contentc r� ) z�
If the I{start_response} callable is invoked multiple times before a
data for the response body is produced, the values from the last call
are used.
c S r� )Nc S s |dg � |dg � t d�S )N�100 Fooz200 Barr$ r� r2 r$ r$ r% rS � s
z^StartResponseTests.test_multipleStartResponse.<locals>.applicationFactory.<locals>.applicationr$ r� r$ r$ r% r � � zIStartResponseTests.test_multipleStartResponse.<locals>.applicationFactoryc r� )Ns HTTP/1.1 200 Bar
r� r� r� r$ r% r� � � �zAStartResponseTests.test_multipleStartResponse.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 � r6 z?StartResponseTests.test_multipleStartResponse.<locals>.<lambda>r� r� rD Nr� r� r$ r� r% �test_multipleStartResponse� s"
� z-StartResponseTests.test_multipleStartResponsec r� ) z�
If the I{start_response} callable is invoked with a third positional
argument before the status and headers have been written to the
response, the status and headers become the newly supplied values.
c S r� )Nc S s |dg t t d�d f� td�S )Nr r� r$ )rX r� r2 r$ r$ r% rS � s zcStartResponseTests.test_startResponseWithException.<locals>.applicationFactory.<locals>.applicationr$ r� r$ r$ r% r � r� zNStartResponseTests.test_startResponseWithException.<locals>.applicationFactoryc r� )Ns HTTP/1.1 100 Foo
r� r� r� r$ r% r� � r zFStartResponseTests.test_startResponseWithException.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 � r6 zDStartResponseTests.test_startResponseWithException.<locals>.<lambda>r� r� rD Nr� r� r$ r� r% � test_startResponseWithException� s"
� z2StartResponseTests.test_startResponseWithExceptionc s� t � � G dd� dt�}z|� � ty t� �Y nw g ���fdd�}��� \}}� ���fdd�}|�|� ��||� fdd�d d
g d gd g � |S )
z�
If the I{start_response} callable is invoked with a third positional
argument after the status and headers have been written to the
response, the supplied I{exc_info} values are re-raised to the
application.
c @ s e Zd ZdS )zPStartResponseTests.test_startResponseWithExceptionTooLate.<locals>.SomeExceptionN)r' r( r) r$ r$ r$ r% �
SomeException� s r c rU )Nc 3 sF � |dg � dV z |dg � � W d S t y" ��t� � Y d S w )Nr� r= z500 ERR)r r� r r2 ��excInfo�reraisedr$ r% rS � s �
�zjStartResponseTests.test_startResponseWithExceptionTooLate.<locals>.applicationFactory.<locals>.applicationr$ r� r r$ r% r � s zUStartResponseTests.test_startResponseWithExceptionTooLate.<locals>.applicationFactoryc s� �� � jj�� �d�� ���d d �d � ���d d �d � �d d j}�d }��t�|�d t�|�d � d S )Ns HTTP/1.1 200 OK
r rU r@ ) r: r r� r� r� r� �tb_next� traceback�
extract_tb)r� �tb1�tb2�r� r r r r$ r% r� � s �$zMStartResponseTests.test_startResponseWithExceptionTooLate.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 r6 zKStartResponseTests.test_startResponseWithExceptionTooLate.<locals>.<lambda>r� r� rD N)r rX r r r� r� r� )r r r r� r~ r� r$ r r% �&test_startResponseWithExceptionTooLate� s0
�
� z9StartResponseTests.test_startResponseWithExceptionTooLatec r� )
z�
I{start_response} returns the I{write} callable which can be used to
write bytes to the response body without buffering.
c r� r1 r� r$ r� r$ r% r� r z-StartResponseTests.test_write.<locals>.recordc r� )Nc s0 |ddg�}|d� � � |d� � � t d�S )Nr r
r= r r$ r� �r3 r4 r{ r� r$ r% rS s zNStartResponseTests.test_write.<locals>.applicationFactory.<locals>.applicationr$ r� r� r$ r% r s z9StartResponseTests.test_write.<locals>.applicationFactoryc r
r r r� r r$ r% r� & r z1StartResponseTests.test_write.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 1 r6 z/StartResponseTests.test_write.<locals>.<lambda>r� r� rD Nr� r� r$ r r% �
test_write
s&
� z StartResponseTests.test_writec r� )zj
The C{write} callable returned from C{start_response} only accepts
byte strings.
c S � |dg �}|d� t d�S )Nr� �bogusr$ r� r" r$ r$ r% rS B �
zHStartResponseTests.test_writeAcceptsOnlyByteStrings.<locals>.applicationc r� )Nz0Can only write bytes to a transport, not 'bogus'r� r� r8 r$ r% r� J s �zIStartResponseTests.test_writeAcceptsOnlyByteStrings.<locals>.checkMessager� r� r$ r8 r% � test_writeAcceptsOnlyByteStrings< s
z3StartResponseTests.test_writeAcceptsOnlyByteStringsN)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� � s. '1
+..'&< /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 S ) �ApplicationTestszd
Tests for things which are done to the application object and the iterator
it returns.
c C s* t | _ t� | _| j�� | �| jj� d S r1 )r r rb �start�
addCleanup�stopr8 r$ r$ r% �
enableThreadsX s
z ApplicationTests.enableThreadsc sp t � � G dd� d�}|� ��fdd�}��� \}}� ��fdd�}|�|� ��||� fdd�d d
g d g� |S ) z�
If the application object returns an iterator which also has a I{close}
method, that method is called after iteration is complete.
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z+ApplicationTests.test_close.<locals>.Resultc S �
d| _ d S )NT��openr8 r$ r$ r% rE f �
z4ApplicationTests.test_close.<locals>.Result.__init__c s s& � t d�D ] }| jrd|f V qd S �Nrk s %d)�ranger/ )r �ir$ r$ r% �__iter__i s � ��z4ApplicationTests.test_close.<locals>.Result.__iter__c S r- )NFr. r8 r$ r$ r% �closen r0 z1ApplicationTests.test_close.<locals>.Result.closeN�r' r( r) rE r4 r5 r$ r$ r$ r% �Resulte s r7 c r� )Nc s |ddg� � S )Nr� �r� �3r$ r2 r� r$ r% rS t s zLApplicationTests.test_close.<locals>.applicationFactory.<locals>.applicationr$ r� r� r$ r% r s r� z7ApplicationTests.test_close.<locals>.applicationFactoryc s* �� ��� jj�� �d� ���j� d S )Ns 012)r� r� r r� r� r r/ r� �r� r� r r$ r% r� | s
�z/ApplicationTests.test_close.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 � r6 z-ApplicationTests.test_close.<locals>.<lambda>r� r� rD r� )r r7 r r� r~ r� r$ r: r% �
test_close^ s
�zApplicationTests.test_closec sX �� � g � � fdd�}��� \}}� �fdd�}|�|� ��||tddg dg� |S )zu
The application object is invoked and iterated in a thread which is not
the reactor thread.
c r� )Nc s* � fdd�}� � t� � |ddg� |� S )Nc 3 s, � t d�D ]} � �t� � d| f V qd S r1 )r2 r� r )r3 ��invokedr$ r% r� � s
� �zpApplicationTests.test_applicationCalledInThread.<locals>.applicationFactory.<locals>.application.<locals>.resultr� r8 )r� r )r3 r4 r� r< r$ r% rS � s z`ApplicationTests.test_applicationCalledInThread.<locals>.applicationFactory.<locals>.applicationr$ r� r< r$ r% r � s
zKApplicationTests.test_applicationCalledInThread.<locals>.applicationFactoryc s& �� t� � � ��tt� ��d� d S )NrU )� assertNotInr r� r[ �setr� �r= r r$ r% r� � s zCApplicationTests.test_applicationCalledInThread.<locals>.cbRenderedr� r� rD )r, r� r� r� r r� r$ r@ r% � test_applicationCalledInThread� s
�z/ApplicationTests.test_applicationCalledInThreadc �j �� � g � G � fdd�dt�}dd� }��|�\}}� �fdd�}|�|� ��||tddg d g� |S )
z�
The I{write} callable returned by I{start_response} calls the request's
C{write} method in the reactor thread.
c rP )zCApplicationTests.test_writeCalledFromThread.<locals>.ThreadVerifierc � � � t� � t�| |�S r1 �r� r r r{ �r rd r< r$ r% r{ � � zIApplicationTests.test_writeCalledFromThread.<locals>.ThreadVerifier.writeN�r' r( r) r{ r$ r< r$ r% �ThreadVerifier� rT rH c S r� )Nc S r$ )Nr� r= r$ r� r" r$ r$ r% rS � r& z\ApplicationTests.test_writeCalledFromThread.<locals>.applicationFactory.<locals>.applicationr$ r� r$ r$ r% r � r zGApplicationTests.test_writeCalledFromThread.<locals>.applicationFactoryc � �� t� �t� h� d S r1 �r� r? r r� r@ r$ r% r� � � z?ApplicationTests.test_writeCalledFromThread.<locals>.cbRenderedr� r� rD �r, r r� r� r� r �r rH r r� r~ r� r$ r@ r% �test_writeCalledFromThread� s
�z+ApplicationTests.test_writeCalledFromThreadc rB )
z�
Strings produced by the iterator returned by the application object are
written to the request in the reactor thread.
c rP )zMApplicationTests.test_iteratedValuesWrittenFromThread.<locals>.ThreadVerifierc rC r1 rD rE r< r$ r% r{ � rF zSApplicationTests.test_iteratedValuesWrittenFromThread.<locals>.ThreadVerifier.writeNrG r$ r< r$ r% rH � rT rH c S r� )Nc s s � |dg � dV d S )Nr� r= r$ r2 r$ r$ r% rS � s �
zfApplicationTests.test_iteratedValuesWrittenFromThread.<locals>.applicationFactory.<locals>.applicationr$ r� r$ r$ r% r � r� zQApplicationTests.test_iteratedValuesWrittenFromThread.<locals>.applicationFactoryc rI r1 rJ r� r@ r$ r% r� � rK zIApplicationTests.test_iteratedValuesWrittenFromThread.<locals>.cbRenderedr� r� rD rL rM r$ r@ r% �$test_iteratedValuesWrittenFromThread� s
�z5ApplicationTests.test_iteratedValuesWrittenFromThreadc rB )
zY
The response status is set on the request object in the reactor thread.
c rP )zEApplicationTests.test_statusWrittenFromThread.<locals>.ThreadVerifierc s � � t� � t�| ||�S r1 )r� r r �setResponseCode)r �coder< r< r$ r% rP � s zUApplicationTests.test_statusWrittenFromThread.<locals>.ThreadVerifier.setResponseCodeN)r' r( r) rP r$ r< r$ r% rH � rT rH c S r� )Nc S r� r� r� r2 r$ r$ r% rS � rp z^ApplicationTests.test_statusWrittenFromThread.<locals>.applicationFactory.<locals>.applicationr$ r� r$ r$ r% r � r� zIApplicationTests.test_statusWrittenFromThread.<locals>.applicationFactoryc rI r1 rJ r� r@ r$ r% r� rK zAApplicationTests.test_statusWrittenFromThread.<locals>.cbRenderedr� r� rD rL rM r$ r@ r% � test_statusWrittenFromThread� s
�z-ApplicationTests.test_statusWrittenFromThreadc sv G dd� dt �}d�_�fdd�� � fdd�}��|�\}}�fdd �}|�|� ��||td
d g d g� ��|t�S )
z�
If the request connection is lost while the application object is being
iterated, iteration is stopped.
c @ r+ )zSApplicationTests.test_connectionClosedDuringIteration.<locals>.UnreliableConnectionz�
This is a request which pretends its connection is lost immediately
after the first write is done to it.
c S s | � ttd��� d S )NzNo more connection)�connectionLostr
r
rE r$ r$ r% r{ r zYApplicationTests.test_connectionClosedDuringIteration.<locals>.UnreliableConnection.writeN)r' r( r) r* r{ r$ r$ r$ r% �UnreliableConnection s rT Fc 3 s � dV d� _ td��)Nr= TzShould not have gotten here)�badIterrX r$ r8 r$ r% �appIter" s �zFApplicationTests.test_connectionClosedDuringIteration.<locals>.appIterc r� )Nc s |dg � � � S �Nr� r$ r2 �rV r$ r% rS ( s
zfApplicationTests.test_connectionClosedDuringIteration.<locals>.applicationFactory.<locals>.applicationr$ r� rX r$ r% r ' r� zQApplicationTests.test_connectionClosedDuringIteration.<locals>.applicationFactoryc s � � � jd� d S )Nz!Should not have resumed iteration)r rU r� r8 r$ r% r� 0 r zIApplicationTests.test_connectionClosedDuringIteration.<locals>.cbRenderedr� r� rD )r rU r� r� r� r r� r
)r rT r r� r~ r� r$ )rV r r% �$test_connectionClosedDuringIteration s
� z5ApplicationTests.test_connectionClosedDuringIterationc s^ t � �� fdd�}��� \}}��fdd�}|�|� ��||�fdd�ddg d gd g � |S )
Nc r r1 r$ r$ r� r$ r% r > � zEApplicationTests._internalServerErrorTest.<locals>.applicationFactoryc s6 �� t�}��t|�d� ��� jj�� �d�� d S )NrU s" HTTP/1.1 500 Internal Server Error) r^ r? r� r[ r: r r� r� r� )r� rD r� r$ r% r� C s
��z=ApplicationTests._internalServerErrorTest.<locals>.cbRenderedc r r1 r$ r$ r r$ r% r5 R r6 z;ApplicationTests._internalServerErrorTest.<locals>.<lambda>r� r� rD r� )r rS r r� r~ r� r$ )rS r� r r% �_internalServerErrorTest; s"
� z)ApplicationTests._internalServerErrorTestc C � dd� }| � |�S )z�
If the application raises an exception before calling I{start_response}
then the response status is I{500} and the exception is logged.
c S � t d���N� This application had some error.�r? r2 r$ r$ r% rS c � zRApplicationTests.test_applicationExceptionBeforeStartResponse.<locals>.application�r[ �r rS r$ r$ r% �,test_applicationExceptionBeforeStartResponse] s
z=ApplicationTests.test_applicationExceptionBeforeStartResponsec C r\ )z�
If the application calls I{start_response} but then raises an exception
before any data is written to the response then the response status is
I{500} and the exception is logged.
c S s |dg � t d���Nr� r_ r` r2 r$ r$ r% rS o rp zQApplicationTests.test_applicationExceptionAfterStartResponse.<locals>.applicationrb rc r$ r$ r% �+test_applicationExceptionAfterStartResponseh s
z<ApplicationTests.test_applicationExceptionAfterStartResponsec s� t � �t��t��� fdd�}��� \}�g ���fdd�}����fdd�}|�|� ��||�fdd�d d
g d gd g � ���jj � �d �
t t d
��� |S )Nc r r1 r$ r$ r� r$ r% r z rZ zBApplicationTests._connectionClosedTest.<locals>.applicationFactoryc s �� � | i |��� �d S rt )r� )r" r# )r~ �requestsr$ r% �requestFactoryWrapper� s zEApplicationTests._connectionClosedTest.<locals>.requestFactoryWrapperc sd �� dt��� �d }|d }��|jt� ��t� � jj�� }�� |�
d�� �� �|� d S )NrU r rV s HTTP/1.1 200 OK) rZ r[ r\ r] r? r^ r r� r� r: r� �assertIn)r� �eventr! r� )r� r_ �responseContentr r$ r% �
ebRendered� s
z:ApplicationTests._connectionClosedTest.<locals>.ebRenderedc r r1 r$ r$ r r$ r% r5 � r6 z8ApplicationTests._connectionClosedTest.<locals>.<lambda>r� r� rD r zAll gone)
r r rW r r� �
addErrbackr� r: r � disconnectedrS r
r
)r rS rk r r� rh rl r$ )rS r� r_ r~ rg rk r r% �_connectionClosedTestu s,
�
z&ApplicationTests._connectionClosedTestc s d� � fdd�}| � |� �S )z�
If the application raises an exception after the response status has
already been sent then the connection is closed and the exception is
logged.
s? Some bytes, triggering the server to start sending the responsec 3 s � |dg � � V t d��re r` r2 �rk r$ r% rS � s �
zIApplicationTests.test_applicationExceptionAfterWrite.<locals>.application�ro rc r$ rp r% �#test_applicationExceptionAfterWrite� s � z4ApplicationTests.test_applicationExceptionAfterWritec s" d� G � fdd�d�}| � |� �S )z�
If the application returns a closeable iterator and the C{close} method
raises an exception when called then the connection is still closed and
the exception is logged.
r= c s( e Zd Zdd� Z� fdd�Zdd� ZdS )zDApplicationTests.test_applicationCloseException.<locals>.Applicationc S s |dg � d S rW r$ )r r3 r4 r$ r$ r% rE � s zMApplicationTests.test_applicationCloseException.<locals>.Application.__init__c 3 s � � V d S r1 r$ r8 rp r$ r% r4 � s �
zMApplicationTests.test_applicationCloseException.<locals>.Application.__iter__c S r] r^ r` r8 r$ r$ r% r5 � ra zJApplicationTests.test_applicationCloseException.<locals>.Application.closeNr6 r$ rp r$ r% � Application� s rs rq )r rs r$ rp r% � test_applicationCloseException� s
z/ApplicationTests.test_applicationCloseExceptionN)r' r( r) r* r, r; rA rN rO rR rY r[ rd rf ro rr rt r$ r$ r$ r% r( R s -#"! *"
2 r( )<r* r� r r� �sysr � urllib.parser ri �zope.interface.verifyr �twisted.internetr �twisted.internet.addressr r �twisted.internet.deferr r �twisted.internet.errorr
�twisted.loggerr r �twisted.python.failurer
�twisted.python.threadabler �twisted.python.threadpoolr �twisted.test.proto_helpersr �twisted.trial.unittestr r � twisted.webr �twisted.web.resourcer r �twisted.web.serverr r r �twisted.web.test.test_webr �twisted.web.wsgir r r, r/ ra r� rL r� r� r� r� r� r( r$ r$ r$ r% �<module> sZ p 6 y )