Your IP : 216.73.216.196
o
0�i{| � @ s� d dl mZ d dlmZ d dlmZmZ d dlmZ d dlm Z d dl m
Z
d dl m Z d dl
mZ d d lmZmZmZ d d
lmZ d d lmZ d d lmZmZ d d
lmZ d dlm Z d dl m Z m Z m 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,dfd d �Z.d d � Z/G d d!� d!e
�Z0G d"d#� d#e
�Z1G d$d%� d%e
�Z2G d&d'� d'e�Z3G d(d)� d)e3�Z4G d*d+� d+e
�Z5e6d,kr�e �7� dS dS )-� )�mock)�support)�
socket_helper�control_characters_c0)�warnings_helper)�NoLogRequestHandler)�TestCase)�setup_testing_defaults)�Headers)� BaseHandler�BaseCGIHandler�
SimpleHandler)�util)� validator)�
WSGIServer�WSGIRequestHandler)� make_server)�HTTPConnection)�StringIO�BytesIO�BufferedReader)�
BaseServer)�python_implementationNc @ � e Zd ZdZdd� Zdd� ZdS )�
MockServerzNon-socket HTTP serverc C s t �| ||� | �� d S �N)r �__init__� server_bind)�self�server_addressZRequestHandlerClass� r �8/opt/alt/python310/lib64/python3.10/test/test_wsgiref.pyr s zMockServer.__init__c C s" | j \}}|| _|| _| �� d S r )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 @ r )� MockHandlerzNon-socket HTTP handlerc C s | j | _| j\| _| _d S r )�requestZ
connectionZrfileZwfile�r r r r! �setup+ s zMockHandler.setupc C � d S r r r) r r r! �finish/ � zMockHandler.finishN)r# r$ r% r& r* r, r r r r! r' ) s r' c C � |dddg� dgS )N�200 OK�� Content-Typez
text/plain��Datez Mon, 05 Jun 2006 18:49:54 GMT�
Hello, world!r ��environZstart_responser r r! � hello_app3 �
�r7 c C s2 |dddg� d� | d | d | d g��d�gS ) Nr/ r0 r2 �;ZHTTP_X_TEST_HEADER� QUERY_STRING� PATH_INFO�
iso-8859-1)�join�encoder5 r r r! �
header_app; s � ��r? s GET / HTTP/1.0
c C sh t dd| tt�}tt|��}t� }tj}t� }t_z
|�||fd� W |t_n|t_w |� � |� � fS )N� �P )� 127.0.0.1i�" )
r r r' r r �sys�stderrr Zfinish_request�getvalue)�app�data�serverZinp�outZolderr�errr r r! � run_amockF s rK c C s� | � }d}|D ]}|| |kst �|d7 }qz|| W n ty% Y nw t d|��zttf W n
ty; Y dS w | � }t|�|urGt �|D ]
}t|�|ksSt �qIzt|� W n
tyd Y dS w t d|��)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_iterT s4
�
� �
rX 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 |rdpd 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
r@ z
Hello, world!r< )r rC �version�split� assertEqualr> )r rI �
has_length�pyverr r r! � check_hello{ s ���
���z IntegrationTests.check_helloc C s t � \}}| �|� d S r )rK r` �r rI rJ 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/)rK r? r] �
splitlines)r r( rI rJ 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
�rG r s! HTTP/1.0 414 Request-URI Too Long)rK r] rd ra r r r! �test_request_length� s �z$IntegrationTests.test_request_lengthc C s" t tt��\}}| j|dd� d S )NF)r^ )rK r r7 r` ra r r r! �test_validated_hello� s z%IntegrationTests.test_validated_helloc C �@ dd� }t t|��\}}| �|�d�� | �|�� d d� d S )Nc S s |dd� dgS )Nr/ r0 z
Hello, world!r r5 r r r! �bad_app� s
z>IntegrationTests.test_simple_validation_error.<locals>.bad_app�; A server error occurred. Please contact the administrator.���z^AssertionError: Headers (('Content-Type', 'text/plain')) must be of type list: <class 'tuple'>�rK r �
assertTrue�endswithr] rd �r rk rI rJ r r r! � test_simple_validation_error� s �
�z-IntegrationTests.test_simple_validation_errorc C s� dd� }g d�}|D ]4\}}| j |d��" t||��\}}| �|�d�� | �|�� d |� W d � n1 s9w Y q
d S )Nc s � fdd�}|S )Nc s |� dg� dgS )N�r1 ztext/plain; charset=utf-8r4 r r5 ��statusr r! rk � s zWIntegrationTests.test_status_validation_errors.<locals>.create_bad_app.<locals>.bad_appr )ru rk r rt 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 codert rl rm )�subTestrK ro rp r] rd )r rv Ztestsru Z exc_messagerI rJ r r r! � test_status_validation_errors� s � ���z.IntegrationTests.test_status_validation_errorsc C rj )Nc S s | d � � |ddg� dgS )N�
wsgi.inputr/ rs � data)�read��e�sr r r! rk � s z1IntegrationTests.test_wsgi_input.<locals>.bad_apprl rm rM rn rq 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 r. )Nr/ rs �r3 z Wed, 24 Dec 2008 13:29:32 GMTrz r r| r r r! rF � r8 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)
rK r ro rp rC r[ r\ r> r r] )r rF rI rJ Zver�pyr_ 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 )Nr/ r0 r� r; �latin1�r> r| r r r! rF � s
�z-IntegrationTests.test_cp1252_url.<locals>.apps GET /�%80 HTTP/1.0rg s HTTP/1.0 200 OKs Content-Type: text/plains# Date: Wed, 24 Dec 2008 13:29:32 GMT� s /��)rK r r] r �ANYrd )r rF rI rJ r r r! �test_cp1252_url� s
���z IntegrationTests.test_cp1252_urlc s� t �td��dd� }G dd� dtt�}ttjd||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 )Nr/ � )r �
SOCK_MAX_SIZEr5 r r r! rF �
z4IntegrationTests.test_interrupted_write.<locals>.appc @ � e Zd ZdS )z<IntegrationTests.test_interrupted_write.<locals>.WsgiHandlerN�r# r$ r% r r r r! � WsgiHandler � r� r )Z
handler_classc s � � � d S r )�set)Zsignum�frame)� interruptedr r! �signal_handler � z?IntegrationTests.test_interrupted_write.<locals>.signal_handlerc s� t �j� } | �dd� | �� �$}|�d� ��tj� � jtd�d�r%nqt |�� ��W d � n1 s6w Y | �
� d S )N�GETrZ �d TrL )Ztimeout) r r r( Z getresponser{ �signal�SIGUSR1�wait�float�len�close)�httpZresponse�r� Z main_threadr� ZreceivedrH r r! �
run_client s
� �
z;IntegrationTests.test_interrupted_write.<locals>.run_client)�targetr� )r Z
get_attributer� r r r r ZHOSTZ
addCleanupZ server_close� threadingZEventr� � get_identZThread�startZhandle_requestr= r] r� )r rF r� r� Zoriginalr� Z
backgroundr r� r! �test_interrupted_write� s" z'IntegrationTests.test_interrupted_writeN)T)r# r$ r% r` rb re rh ri rr rx r r� r� r� r r r r! rY y s
rY c @ s� e Zd Zdd� Zd$dd�Zdd� Zdd � Zd%d d �Zej e
d
�dd� �Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )&� UtilityTestsc C sJ ||d�}t �|� | �t �|�|� | �|d |� | �|d |� |S )N�� SCRIPT_NAMEr; r; r� )r r r] Zshift_path_info)r Zsn_inZpi_in�partZsn_outZpi_out�envr r r! �
checkShift3 s
zUtilityTests.checkShiftNc C s| i }t �|� t|t�r| �|| t� nt|t�r#| �|| t� n| �|| |� ||i}t �|� | �|| |� d S r )r r �
isinstancer ZassertIsInstancer r] ZassertIs)r �key�value�altr� r r r! � checkDefault; s
zUtilityTests.checkDefaultc K s t �|� | �|| |� d S r )r r r] )r r� r� �kwr r r! �checkCrossDefaultK s
z UtilityTests.checkCrossDefaultc K s t �|� | �t �|�|� d S r )r r r] Zapplication_uri)r �urir� r r r! � checkAppURIO s
zUtilityTests.checkAppURIrL c K s" t �|� | �t �||�|� d S r )r r r] Z request_uri)r r� Zqueryr� r r r! � checkReqURIS s
zUtilityTests.checkReqURI)�categoryc C s^ ||fdd�}t ||� |� }| �|jj� |D ]}q| �|jj� |�� | �|jj� d S )Nc S s t �t| �|�S r )r � FileWrapperr )�text�sizer r r! rS Z s z%UtilityTests.checkFW.<locals>.make_it)rX � assertFalseZfilelike�closedr� ro )r r� r� rT rS rU rW r r r! �checkFWW s
zUtilityTests.checkFWc C sR t �td�d�}| �td�� | �|d d� W d � d S 1 s"w Y d S )NZfoobar� z Use iterator protocol insteadrL �foo)r r� r �assertWarnsRegex�DeprecationWarningr] )r �wrapperr r r! �$test_filewrapper_getitem_deprecationj s �"�z1UtilityTests.test_filewrapper_getitem_deprecationc C s^ | � ddddd� | � ddddd� | � ddd dd� | � ddddd� | � dd ddd� d S )
Nr@ rZ z/x�x�/az/x/yz/a/x�/yz/x/�r� r) r r r! �testSimpleShiftsq 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..r� r� r@ z//y�yz/a/b/yz//y/rZ z/./yz/./y/z///./..//y/.//z/y/z///z/a/b/z/.//z/x//r� z/a/b/xz/.r� r) r r r! �testNormalizedShiftsx s z!UtilityTests.testNormalizedShiftsc C sF ddddddddd d
d d t � fd
t� fdfD ]
\}}| �||� qd S )N)� SERVER_NAMErB )� SERVER_PORT�80)�SERVER_PROTOCOL�HTTP/1.0)� HTTP_HOSTrB )ZREQUEST_METHODr� )r� r@ )r; rZ )� wsgi.version�rL r )�
wsgi.run_oncer )�wsgi.multithreadr )�wsgi.multiprocessr ry � wsgi.errors)�wsgi.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� r� �https�on��HTTPS�1�yesr� r� r� 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� )r] r Z guess_schemer) 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 )N�http://127.0.0.1/�http://127.0.0.1/spam�/spam�r� �http://127.0.0.1/sp%E4m� /spämz http://spam.example.com:2071/zspam.example.com:2071Z2071)r� r� zhttp://spam.example.com/zspam.example.comr� rB )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 )Nr� r� r� r� r� r� z http://127.0.0.1/spammity/spamz /spammityr� 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� s0
������
�zUtilityTests.testReqURIsc C s | � ddddg� d S )NZ�xyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyz�x ZxxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzZ xyzxyzxyzxyzxyzxyzxyzxyzxyzxyz)r� r) r r r! �testFileWrapper� s z UtilityTests.testFileWrapperc C s| d� � D ]}||�� |�� |�� fD ]
}| �t�|�� qqd� � D ]}||�� |�� |�� fD ]
}| �t�|�� q0q"d S )NzbConnection Keep-Alive Proxy-Authenticate Proxy-Authorization TE Trailers Transfer-Encoding Upgradez4Accept Cache-Control Date Pragma Trailer Via Warning)r\ �title�upper�lowerro r Z
is_hop_by_hopr� )r Zhopr� r r r! � testHopByHop� s � �� ��zUtilityTests.testHopByHopr )rL )r# r$ r% r� r� r� r� r� r Zignore_warningsr� r� r� r� r� r� r� r� r� r� r� r� r r r r! r� 1 s$
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� 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< |j|j|j |j
fD ] }| � |d�� | � |d�� | � |d �� | � |d�� qn| � |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 rL r� r� r� �bar�FooZFOO�bazZwheeZzoo)r] r� r
�keys�values�itemsZ assertIsNot� __contains__�get�get_all� __getitem__ro 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
r] �str�
add_headerr� �r r� r r r! �
testExtras s �
�zHeaderTests.testExtrasc C s t � D ]z}| �|��k t� }| �t|jd|� �d� | jt|jd|� �ddd� |dkrFd|� �|d< |�dd|� �� |�dd|� �� n(| �t|jdd|� �� | jt|jdd|� �dd� | jt|jddd|� �d� W d � n1 sxw Y qd S )Nr� �val�param)r � )r rw r
r �
ValueError� __setitem__r r� )r Zc0�headersr r r! �testRaisesControlCharacters s
� ���z'HeaderTests.testRaisesControlCharactersN)r# 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�osr6 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 � r r r) r r r! � handle_error4 s zTestHandler.handle_errorN)r# r$ r% r& r r r r r! r 1 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 � n1 s$w Y d D ]}| �t|d | �|jd
| � q+i }t|� |�� D ] \}}||vrT|||< qH|� dd dd|�
� |� � dddddt j
d� � | �|j|� d S )Nz/my/homez/my/pathz fr_FR.UTF-8Ztest_script_nameZtest_server_name)�HOME�PATHZLANGr� r� r �Yz/override/home)�Xr )r[ r r Zrun_onceZ file_wrapperZwsgi_zwsgi.r@ rB r� Fr� T) r r r� r� ry r� r� r� r� r� r� zwsgi.file_wrapper)r Z swap_attrr r" r] �getattrr6 r r� �update� get_stdin�
get_stderrr r� ZassertDictEqual)r r Zhandler�attrZexpectedr� r� r r r! � testEnviron< sF �
� ���zHandlerTests.testEnvironc C s2 t d d d i �}|�� dD ] }| �||j� q
d S )N)r� ry r� )r r" �assertInr6 )r r� r� r r r! �testCGIEnvironr s
�zHandlerTests.testCGIEnvironc C sH t dd�}|�� | �|jd d� t � }|�� | �|jd d� d S )Nr� r� r� r� r� )r r" r] r6 r r r r! �
testSchemex s zHandlerTests.testSchemec C s6 t � }dD ] }| �tt||�� q| �t|jd� d S )N)Z_flushr r Z add_cgi_varsr� )r r �NotImplementedErrorr Z_write)r r� �namer r r! �testAbstractMethods~ 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 �Nr/ r� r< r� r| r r r! � trivial_app1� s
z4HandlerTests.testContentLength.<locals>.trivial_app1c S s |dg �| d � d�� g S r' r� r| r r r! � trivial_app2� s z4HandlerTests.testContentLength.<locals>.trivial_app2c S s |dg � d� d�gS )Nr/ u тестzutf-8r� r| r r r! � trivial_app3� r� z4HandlerTests.testContentLength.<locals>.trivial_app3c S s |ddg� g S )Nr/ )zContent-LengthZ12345r r| r r r! � trivial_app4� s z4HandlerTests.testContentLength.<locals>.trivial_app4z)Status: 200 OK
Content-Length: 4
httpr< zStatus: 200 OK
https- Status: 200 OK
Content-Length: 8
тестs) Status: 200 OK
Content-Length: 12345
)r �runr] �stdoutrE r> )r r( r) r* r+ r� r r r! �testContentLength� s4
�
�
�
�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 � |dg � g S �Nr/ r r| r r r! �
non_error_app� �
z8HandlerTests.testBasicErrorOutput.<locals>.non_error_appc S s t d��)N� This should be caught by handler�rM r| r r r! � error_app� s z4HandlerTests.testBasicErrorOutput.<locals>.error_app�%Status: 200 OK
Content-Length: 0
r< r@ z<Status: %s
Content-Type: text/plain
Content-Length: %d
rM ) r r, r] r- rE r> rD Z error_statusr� Z
error_bodyr! )r r1 r5 r� r r r! �testBasicErrorOutput� s(
�
���z!HandlerTests.testBasicErrorOutputc sR d� � fdd�}t � }|�|� | �|j�� d�d�� � | �d|j�� � d S )N� Some output has been sentc s |dg �� � t d��)Nr/ r3 r4 r| ��MSGr r! r5 � s z4HandlerTests.testErrorAfterOutput.<locals>.error_appzStatus: 200 OK
r< rM )r r, r] r- rE r> r! rD �r r5 r� r r9 r! �testErrorAfterOutput� s
��z!HandlerTests.testErrorAfterOutputc
C s dd� }d}d� d�}dD ]q}|rd| pd}d D ]d}d
D ]_}t|d �}d |_||_||_|�|� | �||j�� � t|d �}d
|_||_||_|�|� |dkr\| �|j�� d� q | � t
� |||f � d�|j�� �|||f � d�|j�� f� q qq
d S )Nc S r/ r0 r r| r r r! r1 � r2 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\nr6 r< )z
FooBar/1.0Nz Server: %s
r@ )z1.0z1.1)�HTTP/0.9r� zHTTP/1.1)r� FTr= r� ) r> r Z
origin_serverZ http_versionZserver_softwarer, r] r- rE ro �rerT ) r r1 ZstdpatZshortpatZssw�swr[ �protor� r r r! �testHeaderFormats� sD ��
������z HandlerTests.testHeaderFormatsc C s. dd� }t � }|�|� | �d|j�� � d S )Nc S s |ddg� dgS )Nr/ rs rz r r| r r r! rF
s �z'HandlerTests.testBytesData.<locals>.appsR Status: 200 OK
Content-Type: text/plain; charset=utf-8
Content-Length: 4
data)r r, r] r- rE )r rF r� r r r! �
testBytesData s
�zHandlerTests.testBytesDatac s> ddi�d� � �fdd�}t � }|�|� | ��d d� d S )N� close_calledFr8 c s( |dg �� � G �fdd�dt �}|� S )Nr/ c s e Zd Zdd� Z� fdd�ZdS )zHHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterablec s s � dV t d��)NTs blahr3 r4 r) r r r! �__iter__" s �zQHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterable.__iter__c s d� d< d S )NTrC r r) �� side_effectsr r! r� & r� zNHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterable.closeN)r# r$ r% rD r� r rE r r! �CrashyIterable! s rG )�object)r} r~ rG �r: rF r r! r5 s z0HandlerTests.testCloseOnError.<locals>.error_appT)r r, r] r; r rI r! �testCloseOnError s
z HandlerTests.testCloseOnErrorc s~ t � � G � fdd�d�}ddi}tt� |� tj|�}d}| �t|��
|�t� W d � n1 s2w Y | � d� � d S )Nc s e Zd Z� fdd�Zdd� ZdS )z4HandlerTests.testPartialWrite.<locals>.PartialWriterc s |d d� }� � |� t|�S )N� )�extendr� )r �b�partial�Zwrittenr r! �write2 s
z:HandlerTests.testPartialWrite.<locals>.PartialWriter.writec S r+ r r r) r r r! �flush7 r- z:HandlerTests.testPartialWrite.<locals>.PartialWriter.flushN)r# r$ r% rP rQ r rO r r! �
PartialWriter1 s rR r� r� z 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 rC rD r� r� r, r7 r] )r rR r6 r� �msgr rO r! �testPartialWrite. s ��z HandlerTests.testPartialWritec s� ddi}t ttfD ]9� | j� d��) G � fdd�d�}t� }tt� |� ||�}|�t� | � |�
� � W d � n1 s=w Y q d S )Nr� r� �Z exceptionc � e Zd Z� fdd�ZdS )zEHandlerTests.testClientConnectionTerminations.<locals>.AbortingWriterc � � �r r �r rM rV r r! rP P r- zKHandlerTests.testClientConnectionTerminations.<locals>.AbortingWriter.writeN�r# r$ r% rP r rV r r! �AbortingWriterO � r[ ) �ConnectionAbortedError�BrokenPipeError�ConnectionResetErrorrw r r
r r, r7 r� rE )r r6 r[ rD r� r rV r! � testClientConnectionTerminationsG s �
���z-HandlerTests.testClientConnectionTerminationsc s� G dd� dt �� G � fdd�d�}t� }ddi}tt� |� ||�}|�t� | �d|�� � | �|j � | �|j
� | �|j � | �|j � d S )Nc @ r� )zKHandlerTests.testDontResetInternalStateOnException.<locals>.CustomExceptionNr� r r r r! �CustomExceptionZ r� ra c rW )zJHandlerTests.testDontResetInternalStateOnException.<locals>.AbortingWriterc rX r r rY �ra r r! rP b r- zPHandlerTests.testDontResetInternalStateOnException.<locals>.AbortingWriter.writeNrZ r rb r r! r[ a r\ r[ r� r� )
r
r r
r r, r7 r! rE ZassertIsNotNone�resultr ru r6 )r r[ rD r6 r� r rb r! �%testDontResetInternalStateOnExceptionY s
z2HandlerTests.testDontResetInternalStateOnExceptionN)r# r$ r% ZmaxDiffr r" r# r& r. r7 r< rA rB rJ rU r` rd r r r r! r 8 s 62
+ r �__main__)8Zunittestr r� r Z test.supportr 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� rC r� r r' r7 r? rK rX rY r� r� r r r r# �mainr r r r! �<module> sT
% 9 +G = �