File: //usr/local/lib/python3.7/test/__pycache__/test_wsgiref.cpython-37.pyc
B
��g�v � @ s� d dl mZ d dlmZ d dlmZ d dl mZ d dlmZ d dl m
Z
d dl m Z m
Z
mZ d dlmZ d d lmZ d d
lmZmZ d d lmZ d d lmZ d d
lmZmZm Z d dl m Z d dl m Z d dl!Z!d dl"Z"d dl#Z#d dl$Z$d dl%Z%d dl Z G dd� de�Z&G dd� de�Z'dd� Z(dd� Z)e(dfdd�Z*d d � Z+G d d � d e�Z,G d d!� d!e�Z-G d"d#� d#e�Z.G d$d%� d%e
�Z/G d&d'� d'e/�Z0G d(d)� d)e�Z1e2d*k�r�e �3� dS )+� )�mock)�support)�NoLogRequestHandler)�TestCase)�setup_testing_defaults)�Headers)� BaseHandler�BaseCGIHandler�
SimpleHandler)�util)� validator)�
WSGIServer�WSGIRequestHandler)� make_server)�HTTPConnection)�StringIO�BytesIO�BufferedReader)�
BaseServer)�python_implementationNc @ s e Zd ZdZdd� Zdd� ZdS )�
MockServerzNon-socket HTTP serverc C s t �| ||� | �� d S )N)r �__init__� server_bind)�self�server_addressZRequestHandlerClass� r �-/usr/local/lib/python3.7/test/test_wsgiref.pyr s zMockServer.__init__c C s" | j \}}|| _|| _| �� d S )N)r Z server_nameZ server_port�
setup_environ)r ZhostZportr r r r s
zMockServer.server_bindN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ s e Zd ZdZdd� Zdd� ZdS )� MockHandlerzNon-socket HTTP handlerc C s | j | _| j\| _| _d S )N)�requestZ
connectionZrfileZwfile)r r r r �setup) s zMockHandler.setupc C s d S )Nr )r r r r �finish- s zMockHandler.finishN)r r r r! r$ r% r r r r r" ' s r" c C s |dddg� dgS )Nz200 OK)z Content-Typez
text/plain)�Datez Mon, 05 Jun 2006 18:49:54 GMTs
Hello, world!r )�environ�start_responser r r � hello_app1 s r) c C s2 |dddg� d� | d | d | d g��d�gS ) Nz200 OK)z Content-Typez
text/plain)r& z Mon, 05 Jun 2006 18:49:54 GMT�;ZHTTP_X_TEST_HEADER� QUERY_STRING� PATH_INFOz
iso-8859-1)�join�encode)r' r( r r r �
header_app9 s r/ s GET / HTTP/1.0
c C sb t dd| tt�}tt|��}t� }tj}t� }t_z|�||fd� W d |t_X |� � |� � fS )N� �P )z 127.0.0.1i�" )
r r r" r r �sys�stderrr Zfinish_request�getvalue)�app�data�serverZinp�outZolderr�errr r r � run_amockD s r: c C s� | � }d}x$|D ] }|| |ks$t �|d7 }qW y || W n tk
rP Y n X t d|��y ttf W n tk
r| Y nbX | � }t|�|k r�t �x |D ]}t|�|ks�t �q�W y t|� W n tk
r� Y n X t d|��dS )a Utility to compare a generic 2.1/2.2+ iterator with an iterable
If running under Python 2.2+, this tests the iterator using iter()/next(),
as well as __getitem__. 'make_it' must be a function returning a fresh
iterator to be tested (since this may test the iterator twice).r � z Too many items from __getitem__z Too many items from .__next__()N)�AssertionError�
IndexError�iter�
StopIteration� NameError�next)�make_it�match�it�n�itemr r r �compare_generic_iterR s4
rG c @ sf e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )�IntegrationTestsTc C sF t � d tj�� d }| �|d| d |r0dp2d d �d�� d S ) N�/r z(HTTP/1.0 200 OK
Server: WSGIServer/0.2 zA
Content-Type: text/plain
Date: Mon, 05 Jun 2006 18:49:54 GMT
zContent-Length: 13
r0 z
Hello, world!z
iso-8859-1)r r2 �version�split� assertEqualr. )r r8 �
has_length�pyverr r r � check_helloy s z IntegrationTests.check_helloc C s t � \}}| �|� d S )N)r: rO )r r8 r9 r r r �test_plain_hello� s
z!IntegrationTests.test_plain_helloc C s* d}t t|�\}}| �|�� d d� d S )Nsm GET /p%61th/?query=test HTTP/1.0
X-Test-Header: Python test
X-Test-Header: Python test 2
Content-Length: 0
���s+ Python test,Python test 2;query=test;/path/)r: r/ rL �
splitlines)r r# r8 r9 r r r � test_environ� s
z IntegrationTests.test_environc C s2 t ddd d d�\}}| �|�� d d� d S )Ns GET � xi s HTTP/1.0
)r6 r s! HTTP/1.0 414 Request-URI Too Long)r: rL rR )r r8 r9 r r r �test_request_length� s z$IntegrationTests.test_request_lengthc C s" t tt��\}}| j|dd� d S )NF)rM )r: r r) rO )r r8 r9 r r r �test_validated_hello� s z%IntegrationTests.test_validated_helloc C s@ dd� }t t|��\}}| �|�d�� | �|�� d d� d S )Nc S s |dd� dgS )Nz200 OK)z Content-Typez
text/plainz
Hello, world!r )r' r( r r r �bad_app� s
z>IntegrationTests.test_simple_validation_error.<locals>.bad_apps; A server error occurred. Please contact the administrator.���z^AssertionError: Headers (('Content-Type', 'text/plain')) must be of type list: <class 'tuple'>)r: r �
assertTrue�endswithrL rR )r rW r8 r9 r r r � test_simple_validation_error� s
z-IntegrationTests.test_simple_validation_errorc
C st dd� }dddg}x\|D ]T\}}| j |d��: t||��\}}| �|�d�� | �|�� d |� W d Q R X qW d S ) Nc s � fdd�}|S )Nc s |� dg� dgS )N)z Content-Typeztext/plain; charset=utf-8s
Hello, world!r )r' r( )�statusr r rW � s zWIntegrationTests.test_status_validation_errors.<locals>.create_bad_app.<locals>.bad_appr )r\ rW r )r\ r �create_bad_app� s zFIntegrationTests.test_status_validation_errors.<locals>.create_bad_app)Z200z4AssertionError: Status must be at least 4 characters)z20X OKz8AssertionError: Status message must begin w/3-digit code)Z200OKz;AssertionError: Status message must have a space after code)r\ s; A server error occurred. Please contact the administrator.rX )�subTestr: rY rZ rL rR )r r] Ztestsr\ Z exc_messager8 r9 r r r � test_status_validation_errors� s z.IntegrationTests.test_status_validation_errorsc C s@ dd� }t t|��\}}| �|�d�� | �|�� d d� d S )Nc S s | d � � |ddg� dgS )Nz
wsgi.inputz200 OK)z Content-Typeztext/plain; charset=utf-8s data)�read)�e�sr r r rW � s z1IntegrationTests.test_wsgi_input.<locals>.bad_apps; A server error occurred. Please contact the administrator.rX r< )r: r rY rZ rL rR )r rW r8 r9 r r r �test_wsgi_input� s z IntegrationTests.test_wsgi_inputc C sl dd� }t t|��\}}| �|�d�� tj�� d �d�}t� �d�}|d | }| � d| d |� d S ) Nc S s |dddg� dgS )Nz200 OK)z Content-Typeztext/plain; charset=utf-8)r& z Wed, 24 Dec 2008 13:29:32 GMTs datar )ra rb r r r r5 � s z3IntegrationTests.test_bytes_validation.<locals>.appz"GET / HTTP/1.0" 200 4
r �ascii� /s( HTTP/1.0 200 OK
Server: WSGIServer/0.2 sV
Content-Type: text/plain; charset=utf-8
Date: Wed, 24 Dec 2008 13:29:32 GMT
data)
r: r rY rZ r2 rJ rK r. r rL )r r5 r8 r9 Zver�pyrN r r r �test_bytes_validation� s
z&IntegrationTests.test_bytes_validationc C s> dd� }t t|�dd�\}}| �dtjdddd g|�� � d S )
Nc S s |dddg� | d � d�gS )Nz200 OK)z Content-Typez
text/plain)r& z Wed, 24 Dec 2008 13:29:32 GMTr, �latin1)r. )ra rb r r r r5 � s z-IntegrationTests.test_cp1252_url.<locals>.apps GET /�%80 HTTP/1.0)r6 s HTTP/1.0 200 OKs Content-Type: text/plains# Date: Wed, 24 Dec 2008 13:29:32 GMT� s /��)r: r rL r �ANYrR )r r5 r8 r9 r r r �test_cp1252_url� s z IntegrationTests.test_cp1252_urlc s� t �td��dd� }G dd� dtt�}tt jd||d��| ��j� t �
� � � fdd �}t�tj |�}| �tjtj |� d �t � � �� ����fd
d �}t j
|d �}|�� ��� |�� | ��t jd
� d S )N� pthread_killc S s |dg � dt j gS )Nz200 OK� )r �
SOCK_MAX_SIZE)r' r( r r r r5 s
z4IntegrationTests.test_interrupted_write.<locals>.appc @ s e Zd ZdS )z<IntegrationTests.test_interrupted_write.<locals>.WsgiHandlerN)r r r r r r r � WsgiHandler s ro r )Z
handler_classc s � � � d S )N)�set)Zsignum�frame)� interruptedr r �signal_handler s z?IntegrationTests.test_interrupted_write.<locals>.signal_handlerc sp t �j� } | �dd� | �� �@}|�d� x"��tj� � jtd�d�r,P q,W t |�� ��W d Q R X | �
� d S )N�GETrI �d r; )Ztimeout) r r r# Z getresponser` �signal�SIGUSR1�wait�float�len�close)�httpZresponse)rr � main_threadrl �receivedr7 r r �
run_client s
z;IntegrationTests.test_interrupted_write.<locals>.run_client)�targetru )r Z
get_attributerv r r r ZHOSTZ
addCleanupZ server_close� threadingZEventrw � get_identZThread�startZhandle_requestr- rL rn )r r5 ro rs Zoriginalr Z
backgroundr )rr r} rl r~ r7 r �test_interrupted_write� s" z'IntegrationTests.test_interrupted_writeN)T)r r r rO rP rS rU rV r[ r_ rc rg rk r� r r r r rH w s
rH c @ s� e Zd Zdd� Zd!dd�Zdd� Zdd � Zd"d d �Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � ZdS )#� UtilityTestsc C sJ ||d�}t �|� | �t �|�|� | �|d |� | �|d |� |S )N)� SCRIPT_NAMEr, r, r� )r r rL Zshift_path_info)r Zsn_inZpi_in�partZsn_outZpi_out�envr r r �
checkShift1 s
zUtilityTests.checkShiftNc C s| i }t �|� t|t�r*| �|| t� n,t|t�rF| �|| t� n| �|| |� ||i}t �|� | �|| |� d S )N)r r �
isinstancer ZassertIsInstancer rL ZassertIs)r �key�value�altr� r r r � checkDefault9 s
zUtilityTests.checkDefaultc K s t �|� | �|| |� d S )N)r r rL )r r� r� �kwr r r �checkCrossDefaultI s
z UtilityTests.checkCrossDefaultc K s t �|� | �t �|�|� d S )N)r r rL Zapplication_uri)r �urir� r r r � checkAppURIM s
zUtilityTests.checkAppURIr; c K s" t �|� | �t �||�|� d S )N)r r rL Z request_uri)r r� Zqueryr� r r r � checkReqURIQ s
zUtilityTests.checkReqURIc C sb ||fdd�}t ||� |� }| �|jj� x |D ]}q2W | �|jj� |�� | �|jj� d S )Nc S s t �t| �|�S )N)r � FileWrapperr )�text�sizer r r rB W s z%UtilityTests.checkFW.<locals>.make_it)rG � assertFalseZfilelike�closedr{ rY )r r� r� rC rB rD rF r r r �checkFWU s
zUtilityTests.checkFWc C s^ | � ddddd� | � ddddd� | � ddd dd� | � ddddd� | � dd ddd� d S )
Nr0 rI z/x�xz/az/x/yz/a/xz/yz/x/)r� )r r r r �testSimpleShiftsg s
z UtilityTests.testSimpleShiftsc C s� | � ddddd� | � ddddd� | � dddd d� | � dd
dd d � | � dd dd d� | � dd
dd d � | � ddddd� | � ddddd� | � ddddd� | � ddddd � | � ddd dd� d S )Nz/a/bz/../yz..z/az/yr0 z//y�yz/a/b/yz//y/rI z/./yz/./y/z///./..//y/.//z/y/z///z/a/b/z/.//z/x//r� z/a/b/xz/.)r� )r r r r �testNormalizedShiftsn s z!UtilityTests.testNormalizedShiftsc C sJ xDddddddddd d
d d t � fd
t� fdgD ]\}}| �||� q.W d S )N)� SERVER_NAMEz 127.0.0.1)� SERVER_PORT�80)�SERVER_PROTOCOLzHTTP/1.0)� HTTP_HOSTz 127.0.0.1)ZREQUEST_METHODrt )r� r0 )r, rI )z wsgi.version)r; r )z
wsgi.run_oncer )zwsgi.multithreadr )zwsgi.multiprocessr z
wsgi.inputz wsgi.errors)zwsgi.url_schemer| )r r r� )r r� r� r r r � testDefaults{ s zUtilityTests.testDefaultsc C st | j dddd� | j dddd� | j dddd� | j ddd d� | j dd
d d� | j d d
d d� | j d ddd� d S )Nr� zfoo.bar)r� zwsgi.url_scheme�https�on)�HTTPS�1�yesr| �foor� r� Z443)r� )r r r r �testCrossDefaults� s z UtilityTests.testCrossDefaultsc C sn | � t�i �d� | � t�ddi�d� | � t�ddi�d� | � t�ddi�d� | � t�ddi�d� d S )Nr| r� r� r� r� r� r� )rL r Z guess_scheme)r r r r �testGuessScheme� s
z UtilityTests.testGuessSchemec C sv | � d� | j ddd� | j ddd� | j ddd d
� | j d d d
� | j ddd d� | j ddd� | j ddd d� d S )Nzhttp://127.0.0.1/zhttp://127.0.0.1/spamz/spam)r� zhttp://127.0.0.1/sp%E4mu /spämz http://spam.example.com:2071/zspam.example.com:2071Z2071)r� r� zhttp://spam.example.com/zspam.example.com)r� z 127.0.0.1)r� r� zhttps://127.0.0.1/r� )r� zhttp://127.0.0.1:8000/Z8000)r� r� )r� )r r r r � testAppURIs� s
zUtilityTests.testAppURIsc C s� | � d� | j ddd� | j ddd� | j dddd � | j d
ddd � | j d dd d � | j d
ddd � | j ddddd� | j ddddd� | j dddddd� d S )Nzhttp://127.0.0.1/zhttp://127.0.0.1/spamz/spam)r� zhttp://127.0.0.1/sp%E4mu /spämz http://127.0.0.1/spammity/spamz /spammity)r� r, z http://127.0.0.1/spammity/sp%E4mz"http://127.0.0.1/spammity/spam;hamz /spam;hamz/http://127.0.0.1/spammity/spam;cookie=1234,5678z/spam;cookie=1234,5678z%http://127.0.0.1/spammity/spam?say=nizsay=ni)r� r, r+ z'http://127.0.0.1/spammity/spam?s%E4y=nizs%E4y=nir )r� )r r r r � testReqURIs� s"
zUtilityTests.testReqURIsc C s | � ddddg� d S )NZ�xyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyz�x ZxxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzZ xyzxyzxyzxyzxyzxyzxyzxyzxyzxyz)r� )r r r r �testFileWrapper� s z UtilityTests.testFileWrapperc C s� xBd� � D ]6}x0||�� |�� |�� fD ]}| �t�|�� q(W q
W xBd� � D ]6}x0||�� |�� |�� fD ]}| �t�|�� qlW qNW d S )NzbConnection Keep-Alive Proxy-Authenticate Proxy-Authorization TE Trailers Transfer-Encoding Upgradez4Accept Cache-Control Date Pragma Trailer Via Warning)rK �title�upper�lowerrY r Z
is_hop_by_hopr� )r Zhopr� r r r � testHopByHop� s zUtilityTests.testHopByHop)N)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
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� HeaderTestsc C s� dg}| � tt� �d� | � ttg ��d� | � tt|d d � ��d� | � t|d d � ��� dg� | � t|d d � ��� dg� | � t|d d � ��� |� | �t|��� |� t� }|d= d|d< xT|j|j|j |j
fD ]<}| � |d�� | � |d�� | � |d �� | � |d�� q�W | � |d d� d
|d< | � |d d
� | � |� d�d
g� | � |�dd �d
� | � |�d d �d � | � |�
dd �d
� | � |�
d d �d � | � |d d
� | � |d d � d S )
N)r� r� r r; r� r� r� �bar�FooZFOO�bazZwheeZzoo)rL rz r �keys�values�itemsZ assertIsNot� __contains__�get�get_all� __getitem__rY r� �
setdefault)r �test�h�mr r r �testMappingInterface� s4 z HeaderTests.testMappingInterfacec C s | � ttd� d S )Nr� )� assertRaises� TypeErrorr )r r r r �testRequireList� s zHeaderTests.testRequireListc C s� t � }| �t|�d� |jdddd� | �|d d� | �t|�d� |jddd d � | �|�d�dd
g� | �t|�d � d S ) Nz
r� r� Zspam)r� zbar; baz="spam"zfoo: bar; baz="spam"
r� )Zcheesez bar; cheesez*foo: bar; baz="spam"
Foo: bar; cheese
)r rL �strZ
add_headerr� )r r� r r r �
testExtras� s
zHeaderTests.testExtrasN)r r r r� r� r� r r r r r� � s r� c @ s& e Zd ZdZeej�� �Zdd� Z dS )� ErrorHandlerz/Simple handler subclass for testing BaseHandlerc K s* t |� tj| t� t� t� |ddd� d S )NT)� multithread� multiprocess)r r r r r )r r� r r r r s zErrorHandler.__init__N)
r r r r! �dict�osr' r� �
os_environr r r r r r� s r� c @ s e Zd ZdZdd� ZdS )� TestHandlerzASimple handler subclass for testing BaseHandler, w/error passthruc C s � d S )Nr )r r r r � handle_error s zTestHandler.handle_errorN)r r r r! r� r r r r r� 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 S ) � HandlerTestsi� c C s� dddddd�}t �td|�� tdd d
�}|�� W d Q R X x,d D ]$}| �t|d | �|jd
| � qDW i }t|� x$|�� D ]\}}||kr�|||< q�W |� dd dd|�
� |� � dddddt j
d� � | �|j|� d S )Nz/my/homez/my/pathz fr_FR.UTF-8Ztest_script_nameZtest_server_name)�HOME�PATH�LANGr� r� r� �Yz/override/home)�Xr� )rJ r� r� Zrun_onceZ file_wrapperZwsgi_zwsgi.r0 z 127.0.0.1)r; r Fr| T) r� r� r� r� z
wsgi.inputz wsgi.errorsz wsgi.versionz
wsgi.run_oncezwsgi.url_schemezwsgi.multithreadzwsgi.multiprocesszwsgi.file_wrapper)r Z swap_attrr� r rL �getattrr' r r� �update� get_stdin�
get_stderrr r� ZassertDictEqual)r r� Zhandler�attrZexpectedr� r� r r r � testEnviron" s<
zHandlerTests.testEnvironc C s6 t d d d i �}|�� xdD ]}| �||j� q W d S )N)zwsgi.url_schemez
wsgi.inputz wsgi.errors)r r �assertInr' )r r� r� r r r �testCGIEnvironX s
zHandlerTests.testCGIEnvironc C sH t dd�}|�� | �|jd d� t � }|�� | �|jd d� d S )Nr� )r� zwsgi.url_schemer� r| )r� r rL r' )r r� r r r �
testScheme^ s
zHandlerTests.testSchemec C s: t � }x dD ]}| �tt||�� q W | �t|jd� d S )N)Z_flushr� r� Z add_cgi_varsr� )r r� �NotImplementedErrorr� Z_write)r r� �namer r r �testAbstractMethodsd s
z HandlerTests.testAbstractMethodsc C s� dd� }dd� }dd� }dd� }t � }|�|� | �|j�� d �d
�� t � }|�|� | �|j�� d �d
�� t � }|�|� | �|j�� d � t � }|�|� | �|j�� d
� d S )Nc S s |dg � | d � d�gS )Nz200 OKzwsgi.url_schemez
iso-8859-1)r. )ra rb r r r � trivial_app1o s
z4HandlerTests.testContentLength.<locals>.trivial_app1c S s |dg �| d � d�� g S )Nz200 OKzwsgi.url_schemez
iso-8859-1)r. )ra rb r r r � trivial_app2s s z4HandlerTests.testContentLength.<locals>.trivial_app2c S s |dg � d� d�gS )Nz200 OKu тестzutf-8)r. )ra rb r r r � trivial_app3w s
z4HandlerTests.testContentLength.<locals>.trivial_app3c S s |ddg� g S )Nz200 OK)zContent-LengthZ12345r )ra rb r r r � trivial_app4{ s z4HandlerTests.testContentLength.<locals>.trivial_app4z)Status: 200 OK
Content-Length: 4
httpz
iso-8859-1zStatus: 200 OK
https- Status: 200 OK
Content-Length: 8
тестs) Status: 200 OK
Content-Length: 12345
)r� �runrL �stdoutr4 r. )r r� r� r� r� r� r r r �testContentLengthl s,
z HandlerTests.testContentLengthc C s� dd� }dd� }t � }|�|� | �|j�� d�d�� | �|j�� d� t � }|�|� | �|j�� d|jt|j �f �d�|j � | �
d |j�� � d S )
Nc S s |dg � g S )Nz200 OKr )ra rb r r r �
non_error_app� s
z8HandlerTests.testBasicErrorOutput.<locals>.non_error_appc S s t d��d S )Nz This should be caught by handler)r< )ra rb r r r � error_app� s z4HandlerTests.testBasicErrorOutput.<locals>.error_appz%Status: 200 OK
Content-Length: 0
z
iso-8859-1r0 z<Status: %s
Content-Type: text/plain
Content-Length: %d
r< ) r� r� rL r� r4 r. r3 Z error_statusrz Z
error_bodyr� )r r� r� r� r r r �testBasicErrorOutput� s
z!HandlerTests.testBasicErrorOutputc sR d� � fdd�}t � }|�|� | �|j�� d�d�� � | �d|j�� � d S )Ns Some output has been sentc s |dg �� � t d��d S )Nz200 OKz This should be caught by handler)r< )ra rb )�MSGr r r� � s z4HandlerTests.testErrorAfterOutput.<locals>.error_appzStatus: 200 OK
z
iso-8859-1r< )r� r� rL r� r4 r. r� r3 )r r� r� r )r� r �testErrorAfterOutput� s
z!HandlerTests.testErrorAfterOutputc
C s dd� }d}d� d�}x�dD ]�}|r,d| p.d}x�d D ]�}x�d
D ]�}t|d �}d |_||_||_|�|� | �||j�� � t|d �}d
|_||_||_|�|� |dkr�| �|j�� d� q@| � t
� |||f � d�|j�� �|||f � d�|j�� f� q@W q6W q W d S )Nc S s |dg � g S )Nz200 OKr )ra rb r r r r� � s
z5HandlerTests.testHeaderFormats.<locals>.non_error_appzfHTTP/%s 200 OK\r\nDate: \w{3}, [ 0123]\d \w{3} \d{4} \d\d:\d\d:\d\d GMT\r\n%sContent-Length: 0\r\n\r\nz%Status: 200 OK
Content-Length: 0
z
iso-8859-1)z
FooBar/1.0Nz Server: %s
r0 )z1.0z1.1)zHTTP/0.9zHTTP/1.0zHTTP/1.1)r� FTzHTTP/0.9ri ) r. r� Z
origin_serverZ http_versionZserver_softwarer� rL r� r4 rY �rerC ) r r� ZstdpatZshortpatZssw�swrJ �protor� r r r �testHeaderFormats� s4
z HandlerTests.testHeaderFormatsc C s. dd� }t � }|�|� | �d|j�� � d S )Nc S s |ddg� dgS )Nz200 OK)z Content-Typeztext/plain; charset=utf-8s datar )ra rb r r r r5 � s z'HandlerTests.testBytesData.<locals>.appsR Status: 200 OK
Content-Type: text/plain; charset=utf-8
Content-Length: 4
data)r� r� rL r� r4 )r r5 r� r r r �
testBytesData� s
zHandlerTests.testBytesDatac s> ddi�d� � �fdd�}t � }|�|� | ��d d� d S )N� close_calledFs Some output has been sentc s( |dg �� � G �fdd�dt �}|� S )Nz200 OKc s e Zd Zdd� Z� fdd�ZdS )zHHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterablec s s xdV t d��qW d S )Ns blahz This should be caught by handler)r< )r r r r �__iter__ s zQHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterable.__iter__c s d� d< d S )NTr� r )r )� side_effectsr r r{ s zNHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterable.closeN)r r r r� r{ r )r� r r �CrashyIterable s r� )�object)ra rb r� )r� r� r r r� s z0HandlerTests.testCloseOnError.<locals>.error_appT)r� r� rL )r r� r� r )r� r� r �testCloseOnError s
z HandlerTests.testCloseOnErrorc sj t � � G � fdd�d�}ddi}tt� |� tj|�}d}| �t|�� |�t� W d Q R X | � d� � d S )Nc s e Zd Z� fdd�Zdd� ZdS )z4HandlerTests.testPartialWrite.<locals>.PartialWriterc s |d d� }� � |� t|�S )N� )�extendrz )r �b�partial)�writtenr r �write s
z:HandlerTests.testPartialWrite.<locals>.PartialWriter.writec S s d S )Nr )r r r r �flush s z:HandlerTests.testPartialWrite.<locals>.PartialWriter.flushN)r r r r� r r )r� r r �
PartialWriter s r r� zHTTP/1.0z should not do partial writesss HTTP/1.0 200 OK
Content-Type: text/plain
Date: Mon, 05 Jun 2006 18:49:54 GMT
Content-Length: 13
Hello, world!)
� bytearrayr
r r2 r3 ZassertWarnsRegex�DeprecationWarningr� r) rL )r r r' r� �msgr )r� r �testPartialWrite s z HandlerTests.testPartialWritec
sz ddi}xlt ttfD ]^� | j� d��H G � fdd�d�}t� }tt� |� ||�}|�t� | � |�
� � W d Q R X qW d S )Nr� zHTTP/1.0)� exceptionc s e Zd Z� fdd�ZdS )zEHandlerTests.testClientConnectionTerminations.<locals>.AbortingWriterc s � �d S )Nr )r r� )r r r r� 6 s zKHandlerTests.testClientConnectionTerminations.<locals>.AbortingWriter.writeN)r r r r� r )r r r �AbortingWriter5 s r ) �ConnectionAbortedError�BrokenPipeError�ConnectionResetErrorr^ r r
r r� r) r� r4 )r r' r r3 r� r )r r � testClientConnectionTerminations- s
z-HandlerTests.testClientConnectionTerminationsc s� G dd� dt �� G � fdd�d�}t� }ddi}tt� |� ||�}|�t� | �d|�� � | �|j � | �|j
� | �|j � | �|j � d S )Nc @ s e Zd ZdS )zKHandlerTests.testDontResetInternalStateOnException.<locals>.CustomExceptionN)r r r r r r r �CustomException@ s r c s e Zd Z� fdd�ZdS )zJHandlerTests.testDontResetInternalStateOnException.<locals>.AbortingWriterc s � �d S )Nr )r r� )r r r r� H s zPHandlerTests.testDontResetInternalStateOnException.<locals>.AbortingWriter.writeN)r r r r� r )r r r r G s r r� zHTTP/1.0)
�
ValueErrorr r
r r� r) r� r4 ZassertIsNotNone�resultZheadersr\ r' )r r r3 r' r� r )r r �%testDontResetInternalStateOnException? s
z2HandlerTests.testDontResetInternalStateOnExceptionN)r r r ZmaxDiffr� r� r� r� r� r� r� r� r� r� r r r r r r r r� s 62
+r� �__main__)4Zunittestr r� r Ztest.test_httpserversr r Z wsgiref.utilr Zwsgiref.headersr Zwsgiref.handlersr r r
Zwsgirefr Zwsgiref.validater Zwsgiref.simple_serverr
r r Z http.clientr �ior r r Z socketserverr �platformr r� r� rv r2 r� r r" r) r/ r: rG rH r� r� r� r� r� r �mainr r r r �<module> sL
% 9 #7 =