Your IP : 216.73.216.52
3
��2aWl � @ 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 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.�Z/G d(d)� d)e�Z0e1d*k�r�e j2� 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 Zdd� Zdd� ZdS )�
MockServerc C s t j| ||� | j� d S )N)r �__init__� server_bind)�self�server_addressZRequestHandlerClass� r �1/opt/alt/python36/lib64/python3.6/test_wsgiref.pyr s zMockServer.__init__c C s" | j \}}|| _|| _| j� d S )N)r Z server_nameZ server_port�
setup_environ)r �hostZportr r r r s
zMockServer.server_bindN)�__name__�
__module__� __qualname__r r r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )� MockHandlerc 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" & s r" c C s |dddg� dgS ) Nz200 OK� Content-Type�
text/plain�Date� Mon, 05 Jun 2006 18:49:54 GMTs
Hello, world!)r& r' )r( r) r )�environ�start_responser r r � hello_app0 s r, c C s2 |dd d g� dj | d | d | d g�jd
�gS )
Nz200 OK� Content-Type�
text/plainr( � Mon, 05 Jun 2006 18:49:54 GMT�;ZHTTP_X_TEST_HEADER� QUERY_STRING� PATH_INFOz
iso-8859-1)r- r. )r( r/ )�join�encode)r* r+ r r r �
header_app8 s r5 s GET / HTTP/1.0
c C sb t dd| tt�}tt|��}t� }tj}t� }t_z|j||fd� W d |t_X |j � |j � fS )N� �P � 127.0.0.1�" )r8 r9 )
r r r" r r �sys�stderrr Zfinish_request�getvalue)�app�data�serverZinp�outZolderr�errr r r � run_amockC s rB 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|��d S )Nr � z Too many items from __getitem__z Too many items from .__next__())�AssertionError�
IndexError�iter�
StopIteration� NameError�next)�make_it�match�it�n�itemr r r �compare_generic_iterQ s4
rO 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 tjj� d }| j|d| d |r0dp2d d jd�� 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
r6 z
Hello, world!z
iso-8859-1)r r: �version�split� assertEqualr4 )r r@ �
has_length�pyverr r r � check_hellox s z IntegrationTests.check_helloc C s t � \}}| j|� d S )N)rB rW )r r@ rA r r r �test_plain_hello� s
z!IntegrationTests.test_plain_helloc C s* d}t t|�\}}| j|j� 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
rC s+ Python test,Python test 2;query=test;/path/���)rB r5 rT �
splitlines)r r# r@ rA r r r � test_environ� s
z IntegrationTests.test_environc C s2 t ddd d d�\}}| j|j� d d� d S )Ns GET � xi s HTTP/1.0
)r> r s! HTTP/1.0 414 Request-URI Too Long)rB rT rZ )r r@ rA r r r �test_request_length� s z$IntegrationTests.test_request_lengthc C s" t tt��\}}| j|dd� d S )NF)rU )rB r r, rW )r r@ rA r r r �test_validated_hello� s z%IntegrationTests.test_validated_helloc C s@ dd� }t t|��\}}| j|jd�� | j|j� d d� d S )Nc S s |dd� dgS )Nz200 OK� Content-Type�
text/plainz
Hello, world!)r_ r` 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'>���)rB r �
assertTrue�endswithrT rZ )r ra r@ rA r r r � test_simple_validation_error� s
z-IntegrationTests.test_simple_validation_errorc C st dd� }d d
dg}x\|D ]T\}}| j |d ��: t||��\}}| j|jd
�� | j|j� d |� W d Q R X qW d S )Nc s � fdd�}|S )Nc s |� dg� dgS )N� Content-Type�text/plain; charset=utf-8s
Hello, world!)rg rh r )r* r+ )�statusr r ra � s zWIntegrationTests.test_status_validation_errors.<locals>.create_bad_app.<locals>.bad_appr )ri ra r )ri r �create_bad_app� s zFIntegrationTests.test_status_validation_errors.<locals>.create_bad_app�200�4AssertionError: Status must be at least 4 characters�20X OK�8AssertionError: Status message must begin w/3-digit code�200OK�;AssertionError: Status message must have a space after code)ri s; A server error occurred. Please contact the administrator.rb )rk rl )rm rn )ro rp rc )ZsubTestrB rd re rT rZ )r rj Ztestsri Z exc_messager@ rA r r r � test_status_validation_errors� s z.IntegrationTests.test_status_validation_errorsc C s@ dd� }t t|��\}}| j|jd�� | j|j� d d� d S )Nc S s | d j � |ddg� dgS )Nz
wsgi.inputz200 OK� Content-Type�text/plain; charset=utf-8s data)rr rs )�read)�e�sr r r ra � s z1IntegrationTests.test_wsgi_input.<locals>.bad_apps; A server error occurred. Please contact the administrator.rb rD rc )rB r rd re rT rZ )r ra r@ rA r r r �test_wsgi_input� s z IntegrationTests.test_wsgi_inputc C sl dd� }t t|��\}}| j|jd�� tjj� d jd�}t� jd�}|d | }| j d| d |� d S ) Nc S s |dddg� dgS ) Nz200 OK� Content-Type�text/plain; charset=utf-8r( � Wed, 24 Dec 2008 13:29:32 GMTs data)rx ry )r( rz r )ru rv r r r r= � 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)
rB r rd re r: rR rS r4 r rT )r r= r@ rA Zver�pyrV r r r �test_bytes_validation� s
z&IntegrationTests.test_bytes_validationc C s> dd� }t t|�dd�\}}| jdtjdddd g|j� � d S )
Nc S s |ddd g� | d j d�gS )
Nz200 OK� Content-Type�
text/plainr( � Wed, 24 Dec 2008 13:29:32 GMTr2 �latin1)r r� )r( r� )r4 )ru rv r r r r= � s z-IntegrationTests.test_cp1252_url.<locals>.apps GET /�%80 HTTP/1.0)r> s HTTP/1.0 200 OKs Content-Type: text/plains# Date: Wed, 24 Dec 2008 13:29:32 GMT� s /��)rB r rT r �ANYrZ )r r= r@ rA r r r �test_cp1252_url� s z IntegrationTests.test_cp1252_urlc s� t jd�}t jtd��dd� }G dd� dtt�}tt jd||d��| j�j � |j
� � � fd d
�}tjtj |�}| jtjtj |� d �|j � �� ����fd d �}|j
|d
�}|j� �j� |j� | j�t jd � d S )N� threading� pthread_killc S s |dg � dt j gS )Nz200 OK� )r �
SOCK_MAX_SIZE)r* r+ r r r r= 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 r� r )Z
handler_classc s � j � d S )N)�set)Zsignum�frame)� interruptedr r �signal_handler s z?IntegrationTests.test_interrupted_write.<locals>.signal_handlerc
sp t �j� } | jdd� | j� �@}|jd� x"��tj� � jtd�d�r,P q,W t |j� ��W d Q R X | j
� d S )N�GETrQ �d rC )Ztimeout) r r r# Z getresponsert �signal�SIGUSR1�wait�float�len�close)�httpZresponse)r� � main_threadr� �receivedr? r r �
run_client s
z;IntegrationTests.test_interrupted_write.<locals>.run_client)�targetr� )r �
import_moduleZ
get_attributer� r r r ZHOSTZ
addCleanupZ server_closeZEventr� � get_identZThread�startZhandle_requestr3 rT r� )r r� r= r� r� Zoriginalr� Z
backgroundr )r� r� r� r� r? r �test_interrupted_write� s$
z'IntegrationTests.test_interrupted_writeN)T)r r r! rW rX r[ r] r^ rf rq rw r~ r� r� r r r r rP v s
rP 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 j|� | jt j|�|� | j|d |� | j|d |� |S )N)� SCRIPT_NAMEr2 r2 r� )r r rT Zshift_path_info)r Zsn_inZpi_in�partZsn_outZpi_out�envr r r �
checkShift1 s
zUtilityTests.checkShiftNc C s| i }t j|� t|t�r*| j|| t� n,t|t�rF| j|| t� n| j|| |� ||i}t j|� | j|| |� d S )N)r r �
isinstancer ZassertIsInstancer rT ZassertIs)r �key�value�altr� r r r � checkDefault9 s
zUtilityTests.checkDefaultc K s t j|� | j|| |� d S )N)r r rT )r r� r� �kwr r r �checkCrossDefaultI s
z UtilityTests.checkCrossDefaultc K s t j|� | jt j|�|� d S )N)r r rT Zapplication_uri)r �urir� r r r � checkAppURIM s
zUtilityTests.checkAppURIrC c K s" t j|� | jt j||�|� d S )N)r r rT Z request_uri)r r� Zqueryr� r r r � checkReqURIQ s
zUtilityTests.checkReqURIc C sb ||fdd�}t ||� |� }| j|jj� x |D ]}q2W | j|jj� |j� | j|jj� d S )Nc S s t jt| �|�S )N)r Z FileWrapperr )�text�sizer r r rJ W s z%UtilityTests.checkFW.<locals>.make_it)rO � assertFalseZfilelike�closedr� rd )r r� r� rK rJ rL rN r r r �checkFWU s
zUtilityTests.checkFWc C s^ | j ddddd� | j ddddd� | j ddd dd� | j ddddd� | j dd ddd� d S )
Nr6 rQ z/x�xz/az/x/yz/a/xz/yz/x/)r� )r r r r �testSimpleShiftsg s
z UtilityTests.testSimpleShiftsc C s� | j ddddd� | j ddddd� | j dddd d� | j dd
dd d � | j dd dd d� | j dd
dd d � | j ddddd� | j ddddd� | j ddddd� | j ddddd � | j ddd dd� d S )Nz/a/bz/../yz..z/az/yr6 z//y�yz/a/b/yz//y/rQ 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 xDddddd d d d d!d"d#dt � fdt� fd$gD ]\}}| j||� q.W d S )%N� SERVER_NAME� 127.0.0.1� SERVER_PORT�80�SERVER_PROTOCOL�HTTP/1.0� HTTP_HOST�REQUEST_METHODr� r� r6 r2 rQ � wsgi.versionrC r �
wsgi.run_once�wsgi.multithread�wsgi.multiprocessz
wsgi.inputz wsgi.errors�wsgi.url_schemer� )r� r� )r� r� )r� r� )r� r� )r� r� )r� r6 )r2 rQ �rC r )r� r� )r� r )r� r )r� r )r� r� )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 | j tji �d� | j tjddi�d� | j tjddi�d� | j tjddi�d� | j tjddi�d� d S )Nr� r� r� r� r� r� r� )rT r Z guess_scheme)r r r r �testGuessScheme� s
z UtilityTests.testGuessSchemec C sv | j 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� | j 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� r2 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� r2 r1 z'http://127.0.0.1/spammity/spam?s%E4y=nizs%E4y=nir )r� )r r r r � testReqURIs� s"
zUtilityTests.testReqURIsc C s" | j dd ddd dd g� d S )NZxyz�2 �x �( �
)r� )r r r r �testFileWrapper� s z UtilityTests.testFileWrapperc C s� xBdj � D ]6}x0||j� |j� |j� fD ]}| jtj|�� q(W q
W xBdj � D ]6}x0||j� |j� |j� fD ]}| jtj|�� qlW qNW d S )NzbConnection Keep-Alive Proxy-Authenticate Proxy-Authorization TE Trailers Transfer-Encoding Upgradez4Accept Cache-Control Date Pragma Trailer Via Warning)rS �title�upper�lowerrd r Z
is_hop_by_hopr� )r Zhopr� r r r � testHopByHop� s zUtilityTests.testHopByHop)N)rC )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� d g}| j tt� �d� | j ttg ��d� | j tt|d d � ��d� | j t|d d � �j� dg� | j t|d d � �j� dg� | j t|d d � �j� |� | jt|�j� |� t� }|d= d|d< xT|j|j|j |j
fD ]<}| j |d�� | j |d�� | j |d�� | j |d�� q�W | j |d d� d |d< | j |d d � | j |j d�d g� | j |jdd
�d � | j |jd d
�d
� | j |j
dd
�d � | j |j
d d
�d
� | j |d d � | j |d d
� d S )
Nr� r� r rC r� �bar�FooZFOO�bazZwheeZzoo)r� r� )rT r� r �keys�values�itemsZ assertIsNot� __contains__�get�get_all� __getitem__rd r� �
setdefault)r �test�h�mr r r �testMappingInterface� s4 z HeaderTests.testMappingInterfacec C s | j ttd� d S )Nr� )� assertRaises� TypeErrorr )r r r r �testRequireList� s zHeaderTests.testRequireListc C s� t � }| jt|�d� |jdddd� | j|d d� | jt|�d� |jddd d � | j|jd�dd
g� | jt|�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 rT �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 Zeejj� �Zdd� ZdS )� ErrorHandlerc 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! �dict�osr* r� Z
os_environr r r r r r� s r� c @ s e Zd Zdd� ZdS )� TestHandlerc C s � d S )Nr )r r r r � handle_error s zTestHandler.handle_errorN)r r r! r� r r r r r� s r� c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) � HandlerTestsc C sJ |j }x>dD ]6}|dkr$|jd kr$q | jt|d| �|d| � q W d S ) NrR r� r� �run_once� file_wrapperZwsgi_zwsgi.)rR r� r� r� r� )r* Zwsgi_file_wrapperrT �getattr)r �handlerr� �attrr r r �checkEnvironAttrs s z HandlerTests.checkEnvironAttrsc C sr i }t |� |j}ddlm} x,|j� D ] \}}||kr(| j|| |� q(W x |j� D ]\}}| j||� qVW d S )Nr )r* )r r* r� r� rT �assertIn)r r� �emptyr� r* �k�vr r r �checkOSEnviron) s zHandlerTests.checkOSEnvironc C s< t dd�}|j� | j|� | j|� | j|jd d� d S )N�Y)�Xr )r� r r� r rT r* )r r� r r r � testEnviron3 s
zHandlerTests.testEnvironc C s6 t d d d i �}|j� xdD ]}| j||j� q W d S )N�wsgi.url_scheme�
wsgi.input� wsgi.errors)r r r )r r r� r* )r r� r� r r r �testCGIEnviron: s
zHandlerTests.testCGIEnvironc C sH t dd�}|j� | j|jd d� t � }|j� | j|jd d� d S )Nr� )r� zwsgi.url_schemer� r� )r� r rT r* )r r� r r r �
testScheme@ s
zHandlerTests.testSchemec C s: t � }x dD ]}| jtt||�� q W | jt|jd� d S )N�_flush� get_stdin�
get_stderr� add_cgi_varsr� )r
r r r
)r r� �NotImplementedErrorr� Z_write)r r� �namer r r �testAbstractMethodsF s
z HandlerTests.testAbstractMethodsc C s� dd� }dd� }dd� }dd� }t � }|j|� | j|jj� d jd
�� t � }|j|� | j|jj� d jd
�� t � }|j|� | j|jj� d � t � }|j|� | j|jj� d
� d S )Nc S s |dg � | d j d�gS )Nz200 OKzwsgi.url_schemez
iso-8859-1)r4 )ru rv r r r � trivial_app1Q s
z4HandlerTests.testContentLength.<locals>.trivial_app1c S s |dg �| d j d�� g S )Nz200 OKzwsgi.url_schemez
iso-8859-1)r4 )ru rv r r r � trivial_app2U s z4HandlerTests.testContentLength.<locals>.trivial_app2c S s |dg � dj d�gS )Nz200 OKu тестzutf-8)r4 )ru rv r r r � trivial_app3Y s
z4HandlerTests.testContentLength.<locals>.trivial_app3c S s |ddg� g S )Nz200 OK�Content-Length�12345)r r r )ru rv 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� �runrT �stdoutr< r4 )r r r r r r� r r r �testContentLengthN s,
z HandlerTests.testContentLengthc C s� dd� }dd� }t � }|j|� | j|jj� djd�� | j|jj� d� t � }|j|� | j|jj� d|jt|j �f jd�|j � | j
d |jj� � d S )
Nc S s |dg � g S )Nz200 OKr )ru rv 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)rD )ru rv r r r � error_app� s z4HandlerTests.testBasicErrorOutput.<locals>.error_appz%Status: 200 OK
Content-Length: 0
z
iso-8859-1r6 z<Status: %s
Content-Type: text/plain
Content-Length: %d
rD ) r� r rT r r< r4 r; Z error_statusr� Z
error_bodyr� )r r r r� r r r �testBasicErrorOutput� s
z!HandlerTests.testBasicErrorOutputc sR d� � fdd�}t � }|j|� | j|jj� djd�� � | jd|jj� � d S )Ns Some output has been sentc s |dg �� � t d��d S )Nz200 OKz This should be caught by handler)rD )ru rv )�MSGr r r � s z4HandlerTests.testErrorAfterOutput.<locals>.error_appzStatus: 200 OK
z
iso-8859-1rD )r� r rT r r< r4 r� r; )r r r� r )r r �testErrorAfterOutput� s
z!HandlerTests.testErrorAfterOutputc C s dd� }d}dj d�}� x�dD ]�}|r.d| p0d}x�dD ]�}x�dD ]�}t|d�}d|_||_||_|j|� | j||jj� � t|d�}d|_||_||_|j|� |d kr�| j|jj� d� qB| j t
j |||f j d�|jj� �|||f j d�|jj� f� qBW q8W q W d S )Nc S s |dg � g S )Nz200 OKr )ru rv 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�
FooBar/1.0z Server: %s
r6 �1.0�1.1�HTTP/0.9�HTTP/1.0�HTTP/1.1)r� FTr� )r N)r r! )r" r# r$ ) r4 r� Z
origin_serverZ http_versionZserver_softwarer rT r r< rd �rerK ) r r ZstdpatZshortpatZssw�swrR �protor� r r r �testHeaderFormats� s4
z HandlerTests.testHeaderFormatsc C s. dd� }t � }|j|� | jd|jj� � d S )Nc S s |ddg� dgS )Nz200 OK� Content-Type�text/plain; charset=utf-8s data)r) r* r )ru rv r r r r= � s z'HandlerTests.testBytesData.<locals>.appsR Status: 200 OK
Content-Type: text/plain; charset=utf-8
Content-Length: 4
data)r� r rT r r< )r r= r� r r r �
testBytesData� s
zHandlerTests.testBytesDatac s> ddi�d� � �fdd�}t � }|j|� | j�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)rD )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)ru rv r/ )r r. r r r � s z0HandlerTests.testCloseOnError.<locals>.error_appT)r� r rT )r r r� r )r r. r �testCloseOnError� s
z HandlerTests.testCloseOnErrorc
sj t � � G � fdd�d�}ddi}tt� |� tj|�}d}| jt|�� |jt� W d Q R X | j d� � d S )Nc s e Zd Z� fdd�Zdd� ZdS )z4HandlerTests.testPartialWrite.<locals>.PartialWriterc s |d d� }� j |� t|�S )N� )�extendr� )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! r7 r8 r )r6 r r �
PartialWriter� s r9 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 r: r; ZassertWarnsRegex�DeprecationWarningr r, rT )r r9 r* r� �msgr )r6 r �testPartialWrite� s z HandlerTests.testPartialWriteN)r r r! r� r r r r r r r r r( r+ r1 r= r r r r r� s
2
+r� �__main__)3Zunittestr 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% r� r: r r" r, r5 rB rO rP r� r� r� r� r� r �mainr r r r �<module> sH
% : #7 s