✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

B

YI~d�v�@s�
dd
lm
Z

ddlmZ
ddlmZ
ddlmZ
ddlmZ
ddl	m
Z

ddlmZm
Z
mZ
ddlmZ
dd	lmZ
dd
lmZmZ
ddlmZ
ddlmZ
dd
lmZmZmZ
ddlmZ
ddlm Z 
ddl!Z!ddl"Z"ddl#Z#ddl$Z$ddl%Z%ddlZGdd�de�Z&Gdd�de�Z'dd�Z(dd�Z)e(dfdd�
Z*dd�Z+Gdd�de�Z,Gd d!�d!e�Z-Gd"d#�d#e�Z.Gd$d%�d%e
�Z/Gd&d'�d'e/�Z0Gd(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@seZ
dZd
d�Zdd�ZdS)�
MockServercCst�
||
|�
|��
dS)
N)r�__init__�server_bind)�self�server_addressZRequestHandlerClass�r�6/opt/alt/python37/lib64/python3.7/test/test_wsgiref.pyrs

zMockServer.__init__c
Cs"|j\}
}|
|_
||_|��
dS)
N)rZserver_nameZserver_port�
setup_environ)rZhostZportrrrr s




zMockServer.server_bindN)�__name__�
__module__�__qualname__rrrrrrrsrc@seZ
dZd
d�Zdd�ZdS)�MockHandlerc

Cs|j|_
|j
\|_|_dS)
N)�requestZ
connectionZrfileZwfile)
rrrr�setup)s

zMockHandler.setupc


CsdS)
Nr)
rrrr�finish-s
zMockHandler.finishN)rrr r#r$rrrrr!'sr!cCs|
d
ddg�
dg
S)Nz200 OK)zContent-Typez
text/plain)�DatezMon, 05 Jun 2006 18:49:54 GMTs
Hello, world!r)�environ�start_responserrr�	hello_app1s


r(cCs2|
d
ddg�
d�|d|d|dg�
�
d�
g
S)	Nz200 OK)zContent-Typez
text/plain)r%zMon, 05 Jun 2006 18:49:54 GMT�
;ZHTTP_X_TEST_HEADER�QUERY_STRING�	PATH_INFOz
iso-8859-1)�join�encode)r&r'rrr�
header_app9s





r.sGET / HTTP/1.0

cCsbtd
d|t
t�}tt|
�
�
}t�}tj}t�}t_z|�||fd�
Wd|t_X|�	�|�	�fS)N��P)z	127.0.0.1i�")
rrr!rr�sys�stderrrZfinish_request�getvalue)�app�data�serverZinp�outZolderr�errrrr�	run_amockDs





r9cCs�|�}d
}x$|
D]}|||ks$t�
|d7}qWy||
Wnt
k
rP


YnXtd|��
yttf
Wntk
r|


YnbX|�}t|�
|k	r�t�
x|
D]}t|�
|ks�t�
q�Wyt|�

Wntk
r�


YnXtd|��
dS)Nr
�
zToo many items from __getitem__zToo many items from .__next__())�AssertionError�
IndexError�iter�
StopIteration�	NameError�next)�make_it�match�it�
n�itemrrr�compare_generic_iterRs4




















rFc@sfeZ
dZddd�
Zdd�Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dS)�IntegrationTestsTcCsFt�d
t
j��d}|�|
d|d|r0dp2dd�d�
�
dS)	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!z
iso-8859-1)rr1�version�split�assertEqualr-)rr7�
has_length�pyverrrr�check_helloys




zIntegrationTests.check_helloc
Cst�\}
}|�
|
�

dS)
N)r9rN)rr7r8rrr�test_plain_hello�s


z!IntegrationTests.test_plain_helloc
Cs*d
}
tt
|
�\}}|�|��dd�
dS)NsmGET /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/)r9r.rK�
splitlines)rr"r7r8rrr�test_environ�s




zIntegrationTests.test_environc
Cs2td
dddd�
\}
}|�
|
��dd�
dS)NsGET �
xi

s HTTP/1.0

)
r5r
s!HTTP/1.0 414 Request-URI Too Long)r9rKrQ)rr7r8rrr�test_request_length�s


z$IntegrationTests.test_request_lengthc
Cs"tt
t�
�
\}
}|j|
d
d�
dS)NF)
rL)r9rr(rN)rr7r8rrr�test_validated_hello�s
z%IntegrationTests.test_validated_helloc
Cs@d
d�}
tt
|
�
�
\}}|�|�d�
�

|�|��dd�
dS)NcSs|
d
d�
dg
S)Nz200 OK)zContent-Typez
text/plainz
Hello, world!r)r&r'rrr�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'>)r9r�
assertTrue�endswithrKrQ)rrVr7r8rrr�test_simple_validation_error�s





z-IntegrationTests.test_simple_validation_errorc

Cstd
d�}
dddg}x\|D]T\}}|j|d�
�:
t
|
|�
�
\}}|�|�d�
�

|�|��d|�
WdQRXqWdS)	Nc
s�f
d
d�}
|
S)Ncs|
�d
g
�
dg
S)N)zContent-Typeztext/plain; charset=utf-8s
Hello, world!r)r&r')
�statusrrrV�s

zWIntegrationTests.test_status_validation_errors.<locals>.create_bad_app.<locals>.bad_appr)r[rVr)
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.rW)�subTestr9rXrYrKrQ)rr\Ztestsr[Zexc_messager7r8rrr�test_status_validation_errors�s






z.IntegrationTests.test_status_validation_errorsc
Cs@d
d�}
tt
|
�
�
\}}|�|�d�
�

|�|��dd�
dS)NcSs|d
��
|
ddg
�
dg
S)Nz
wsgi.inputz200 OK)zContent-Typeztext/plain; charset=utf-8sdata)
�read)�
e�
srrrrV�s


z1IntegrationTests.test_wsgi_input.<locals>.bad_apps;A server error occurred.  Please contact the administrator.rWr;)r9rrXrYrKrQ)rrVr7r8rrr�test_wsgi_input�s



z IntegrationTests.test_wsgi_inputc
Csld
d�}
tt
|
�
�
\}}|�|�d�
�

tj��d�d�
}t��d�
}|d|}|�	d|d|�
dS)	NcSs|
d
ddg�
dg
S)Nz200 OK)zContent-Typeztext/plain; charset=utf-8)r%zWed, 24 Dec 2008 13:29:32 GMTsdatar)r`rarrrr4�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)
r9rrXrYr1rIrJr-rrK)rr4r7r8Zver�pyrMrrr�test_bytes_validation�s






z&IntegrationTests.test_bytes_validationc
Cs>d
d�}
tt
|
�
dd�\}}|�dtjdddd	g|���
dS)
NcSs|
d
ddg�
|d�d�
g
S)Nz200 OK)zContent-Typez
text/plain)r%zWed, 24 Dec 2008 13:29:32 GMTr+�latin1)
r-)r`rarrrr4�s


z-IntegrationTests.test_cp1252_url.<locals>.appsGET /�%80 HTTP/1.0)
r5sHTTP/1.0 200 OKsContent-Type: text/plains#Date: Wed, 24 Dec 2008 13:29:32 GMT�s/��)r9rrKr�ANYrQ)rr4r7r8rrr�test_cp1252_url�s
	






z IntegrationTests.test_cp1252_urlc
s�t�
td
��dd�}
Gdd�dtt�}ttjd|
|d��|��j�

t	�
���f
dd	�}t�tj|�}|�tjtj|�
d�t	���
��
���fd
d�}t	j
|d�
}|��
���
|��
|��tjd
�
dS)N�pthread_killcSs|
d
g�
dtj
g
S)Nz200 OK�
)r�
SOCK_MAX_SIZE)r&r'rrrr4
s


z4IntegrationTests.test_interrupted_write.<locals>.appc
@seZ
dZd
S)z<IntegrationTests.test_interrupted_write.<locals>.WsgiHandlerN)rrr rrrr�WsgiHandler
s
rnr
)
Z
handler_classcs���
dS)
N)
�set)Zsignum�frame)
�interruptedrr�signal_handler
s
z?IntegrationTests.test_interrupted_write.<locals>.signal_handlerc	spt�j
�}|�d
d�
|���@}
|
�d�

x"��
tj�
�jtd�
d�
r,Pq,Wt	|
���
�WdQRX|�
�
dS)N�GETrH�dr:)
Ztimeout)rrr"Zgetresponser_�signal�SIGUSR1�wait�float�len�close)�httpZresponse)rq�main_threadrk�receivedr6rr�
run_client
s










z;IntegrationTests.test_interrupted_write.<locals>.run_client)
�targetrt)rZ
get_attributerurrrZHOSTZ
addCleanupZserver_close�	threadingZEventrv�	get_identZThread�startZhandle_requestr,rKrm)rr4rnrrZoriginalr~Z
backgroundr)rqr|rkr}r6r�test_interrupted_write�s"








z'IntegrationTests.test_interrupted_writeN)
T)rrr rNrOrRrTrUrZr^rbrfrjr�rrrrrGws



rGc@s�eZ
dZd
d�Zd!dd�
Zdd�Zdd	�Zd"dd�
Zd
d�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd �ZdS)#�UtilityTestscCsJ|
|d
�}t�
|�

|�t�|�
|�
|�|d|�
|�|d|�
|S)N)�SCRIPT_NAMEr+r+r�)rrrKZshift_path_info)rZsn_inZpi_in�partZsn_outZpi_out�envrrr�
checkShift1
s







zUtilityTests.checkShiftNcCs|i}t�
|�

t|t�r*|�||
t�
n,t|t�rF|�||
t�
n|�||
|�
|
|i
}t�
|�

|�||
|�
dS)
N)rr�
isinstancerZassertIsInstancerrKZassertIs)r�key�value�altr�rrr�checkDefault9
s










zUtilityTests.checkDefaultcKst�
|�

|�||
|�
dS)
N)rrrK)rr�r��kwrrr�checkCrossDefaultI
s


zUtilityTests.checkCrossDefaultcKs t�
|�

|�t�|�
|
�
dS)
N)rrrKZapplication_uri)r�urir�rrr�checkAppURIM
s


zUtilityTests.checkAppURIr:cKs"t�
|�

|�t�||�|
�
dS)
N)rrrKZrequest_uri)rr�Zqueryr�rrr�checkReqURIQ
s


zUtilityTests.checkReqURIcCsb|
|fd
d�
}t||�
|�}|�
|jj�

x|D]}q2W|�
|jj�

|��
|�|jj�

dS)NcSst�
t|�
|
�S)
N)r�FileWrapperr)�text�sizerrrrAW
s
z%UtilityTests.checkFW.<locals>.make_it)rF�assertFalseZfilelike�closedrzrX)rr�r�rBrArCrErrr�checkFWU
s




zUtilityTests.checkFWc

Cs^|�d
dd
dd
�
|�d
dddd
�
|�dd
ddd
�
|�ddddd�
|�dd	ddd�
dS)
Nr/rHz/x�
xz/az/x/yz/a/xz/yz/x/)
r�)
rrrr�testSimpleShiftsg
s





zUtilityTests.testSimpleShiftsc

Cs�|�d
dddd�
|�ddddd�
|�d
ddd	d�
|�d
d
dd	d�
|�d
ddd	d�
|�d
d
dd	d�
|�d
dddd�
|�d
dddd�
|�d
dddd�
|�d
dddd�
|�d
ddd
d�
dS)Nz/a/bz/../yz..z/az/yr/z//y�
yz/a/b/yz//y/rHz/./yz/./y/z///./..//y/.//z/y/z///z/a/b/z/.//z/x//r�z/a/b/xz/.)
r�)
rrrr�testNormalizedShiftsn
s










z!UtilityTests.testNormalizedShiftsc
CsJxDd
dddddddd	d
ddt�fd
t
�fdgD]\}
}|�|
|�
q.WdS)N)�SERVER_NAMEz	127.0.0.1)�SERVER_PORT�80)�SERVER_PROTOCOLzHTTP/1.0)�	HTTP_HOSTz	127.0.0.1)ZREQUEST_METHODrs)r�r/)r+rH)zwsgi.version)r:r
)z
wsgi.run_oncer
)zwsgi.multithreadr
)zwsgi.multiprocessr
z
wsgi.inputzwsgi.errors)zwsgi.url_schemer{)rrr�)rr�r�rrr�testDefaults{
s 














zUtilityTests.testDefaultsc

Cst|jd
ddd�
|jdddd�
|jdddd�
|jddd	d�
|jdd
dd�
|jdd
dd�
|jdddd�
dS)Nr�zfoo.bar)
r�zwsgi.url_scheme�https�on)
�HTTPS�
1�yesr{�foor�r�Z443)
r�)
rrrr�testCrossDefaults�
s






zUtilityTests.testCrossDefaultsc

Csn|�t
�i�
d
�
|�t
�ddi
�
d
�
|�t
�ddi
�
d�
|�t
�ddi
�
d�
|�t
�ddi
�
d�
dS)Nr{r�r�r�r�r�r�)rKrZguess_scheme)
rrrr�testGuessScheme�
s





zUtilityTests.testGuessSchemec

Csv|�d
�

|jddd�
|jddd�
|jddd	d
�
|jddd
�
|jd
ddd�
|jddd�
|jdddd�
dS)Nzhttp://127.0.0.1/zhttp://127.0.0.1/spamz/spam)
r�zhttp://127.0.0.1/sp%E4mu/spämzhttp://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�)
rrrr�testAppURIs�
s














zUtilityTests.testAppURIsc

Cs�|�d
�

|jddd�
|jddd�
|jdddd	�
|jd
ddd	�
|jdddd	�
|jd
ddd	�
|jddddd�
|jddddd�
|jdddddd�
dS)Nzhttp://127.0.0.1/zhttp://127.0.0.1/spamz/spam)
r�zhttp://127.0.0.1/sp%E4mu/spämzhttp://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�)
rrrr�testReqURIs�
s"





















zUtilityTests.testReqURIsc

Cs|�d
dddg�
dS)NZ�xyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyz�xZxxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzZxyzxyzxyzxyzxyzxyzxyzxyzxyzxyz)
r�)
rrrr�testFileWrapper�
s
zUtilityTests.testFileWrapperc
Cs�xBd
��D]6}
x0|
|
�
�|
��|
��fD]}|�t�|�
�

q(Wq
WxBd��D]6}
x0|
|
�
�|
��|
��fD]}|�t�|�
�

qlWqNWdS)NzbConnection Keep-Alive Proxy-Authenticate Proxy-Authorization TE Trailers Transfer-Encoding Upgradez4Accept Cache-Control Date Pragma Trailer Via Warning)rJ�title�upper�lowerrXrZ
is_hop_by_hopr�)rZhopr�rrr�testHopByHop�
s




zUtilityTests.testHopByHop)
N)
r:)rrr r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrr�/
s


	r�c@s$eZ
dZd
d�Zdd�Zdd�ZdS)�HeaderTestsc
Cs�
d
g
}
|�t
t��
d�
|�t
tg�
�
d�
|�t
t|
dd��
�
d�
|�t|
dd��
��dg
�
|�t|
dd��
��dg
�
|�t|
dd��
��|
�
|�t|
�
��|
�
t�}|d=d|d<xT|j|j|j	|j
fD]<}|�|d�
�

|�|d�
�

|�|d	�
�

|�|d�
�

q�W|�|dd�
d
|d<|�|d	d
�
|�|�	d�
d
g
�
|�|�dd�d
�
|�|�dd�d�
|�|�
dd�d
�
|�|�
dd�d�
|�|dd
�
|�|dd�
dS)
N)r�r�r
r:r�r�r��bar�FooZFOO�bazZwheeZzoo)rKryr�keys�values�itemsZassertIsNot�__contains__�get�get_all�__getitem__rXr��
setdefault)r�test�
h�
mrrr�testMappingInterface�
s4





















z HeaderTests.testMappingInterfacec

Cs|�t
td
�
dS)Nr�)�assertRaises�	TypeErrorr)
rrrr�testRequireList�
s
zHeaderTests.testRequireListc
Cs�t�}
|�
t|
�
d
�
|
jdddd�
|�
|
dd�
|�
t|
�
d�
|
jdddd	�
|�
|
�d�
dd
g�
|�
t|
�
d�
dS)Nz
r�r�Zspam)
r�zbar; baz="spam"zfoo: bar; baz="spam"

r�)
Zcheesezbar; cheesez*foo: bar; baz="spam"
Foo: bar; cheese

)rrK�strZ
add_headerr�)rr�rrr�
testExtras�
s








zHeaderTests.testExtrasN)rrr r�r�r�rrrrr��
s r�c@s"eZ
dZeej���
Zd
d�ZdS)�ErrorHandlerc
	Ks*t|
�

t
j|t�t�t�|
d
d
d�
dS)NT)�multithread�multiprocess)rr	rrr)rr�rrrrs



zErrorHandler.__init__N)	rrr �dict�osr&r��
os_environrrrrrr�sr�c@seZ
dZd
d�ZdS)�TestHandlerc

Cs�dS)
Nr)
rrrr�handle_errors
zTestHandler.handle_errorN)rrr r�rrrrr�sr�c@sxeZ
dZd
Zdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�ZdS)�HandlerTestsi�c
Cs�d
ddddd�}
t�
td|
��
tdd	d
�}|��
WdQRXx,dD]$}|�t|d|�|jd
|�
qDWi}t|�

x$|
��D]\}}||kr�|||<q�W|�	dd	dd|�
�|��dddddtj
d��

|�|j|�
dS)Nz/my/homez/my/pathzfr_FR.UTF-8Ztest_script_nameZtest_server_name)�HOME�PATH�LANGr�r�r��
Yz/override/home)�
Xr�)rIr�r�Zrun_onceZfile_wrapperZwsgi_zwsgi.r/z	127.0.0.1)r:r
Fr{T)r�r�r�r�z
wsgi.inputzwsgi.errorszwsgi.versionz
wsgi.run_oncezwsgi.url_schemezwsgi.multithreadzwsgi.multiprocesszwsgi.file_wrapper)rZ	swap_attrr�rrK�getattrr&rr��update�	get_stdin�
get_stderrrr�ZassertDictEqual)rr�Zhandler�attrZexpectedr�r�rrr�testEnviron"s<

















zHandlerTests.testEnvironc
Cs6tdddi�}
|
�
�
xd
D]}|�||
j�
qWdS)N)zwsgi.url_schemez
wsgi.inputzwsgi.errors)r	r�assertInr&)rr�r�rrr�testCGIEnvironXs




zHandlerTests.testCGIEnvironc
CsHtd
d�
}
|
�
�
|�|
jdd�
t�}
|
�
�
|�|
jdd�
dS)Nr�)
r�zwsgi.url_schemer�r{)r�rrKr&)rr�rrr�
testScheme^s




zHandlerTests.testSchemec
Cs:t�}
xd
D]}|�
tt|
|��
qW|�
t|
jd�
dS)N)Z_flushr�r�Zadd_cgi_varsr�)rr��NotImplementedErrorr�Z_write)rr��namerrr�testAbstractMethodsds



z HandlerTests.testAbstractMethodsc
Cs�d
d�}
dd�}dd�}dd�}t�}|�
|
�

|�|j��d	�d
�
�
t�}|�
|�

|�|j��d�d
�
�
t�}|�
|�

|�|j��d�
t�}|�
|�

|�|j��d
�
dS)NcSs|
d
g�
|d�d�
g
S)Nz200 OKzwsgi.url_schemez
iso-8859-1)
r-)r`rarrr�trivial_app1os


z4HandlerTests.testContentLength.<locals>.trivial_app1cSs|
d
g�|d�d�
�

gS)Nz200 OKzwsgi.url_schemez
iso-8859-1)
r-)r`rarrr�trivial_app2ss

z4HandlerTests.testContentLength.<locals>.trivial_app2cSs|
d
g�
d�d�
g
S)Nz200 OKuтестzutf-8)
r-)r`rarrr�trivial_app3ws


z4HandlerTests.testContentLength.<locals>.trivial_app3cSs|
d
dg
�
gS)Nz200 OK)zContent-LengthZ12345r)r`rarrr�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��runrK�stdoutr3r-)rr�r�r�r�r�rrr�testContentLengthls,















zHandlerTests.testContentLengthc
Cs�d
d�}
dd�}t�}|�
|
�

|�|j��d�d�
�
|�|j��d�
t�}|�
|�

|�|j��d|jt|j	�
f�d�
|j	�
|�
d	|j���
dS)
NcSs|
d
g�
gS)Nz200 OKr)r`rarrr�
non_error_app�s


z8HandlerTests.testBasicErrorOutput.<locals>.non_error_appcSstd
�
�
dS)Nz This should be caught by handler)
r;)r`rarrr�	error_app�s
z4HandlerTests.testBasicErrorOutput.<locals>.error_appz%Status: 200 OK
Content-Length: 0

z
iso-8859-1r/z<Status: %s
Content-Type: text/plain
Content-Length: %d

r;)r�r�rKr�r3r-r2Zerror_statusryZ
error_bodyr�)rr�r�r�rrr�testBasicErrorOutput�s










z!HandlerTests.testBasicErrorOutputc
sRd
��f
dd�}
t�}|�
|
�

|�|j��d�d�
��
|�d|j���
dS)NsSome output has been sentcs|
d
g���

td�
�
dS)Nz200 OKz This should be caught by handler)
r;)r`ra)
�MSGrrr��s

z4HandlerTests.testErrorAfterOutput.<locals>.error_appzStatus: 200 OK

z
iso-8859-1r;)r�r�rKr�r3r-r�r2)rr�r�r)
r�r�testErrorAfterOutput�s







z!HandlerTests.testErrorAfterOutputc
	
Cs
d
d�}
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@Wq6WqWdS)NcSs|
d
g�
gS)Nz200 OKr)r`rarrrr��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
r/)z1.0z1.1)zHTTP/0.9zHTTP/1.0zHTTP/1.1)
r�FTzHTTP/0.9rh)r-r�Z
origin_serverZhttp_versionZserver_softwarer�rKr�r3rX�rerB)	rr�ZstdpatZshortpatZssw�swrI�protor�rrr�testHeaderFormats�s4

























zHandlerTests.testHeaderFormatsc
Cs.d
d�}
t�}|�
|
�

|�d|j���
dS)NcSs|
d
dg
�
dg
S)Nz200 OK)zContent-Typeztext/plain; charset=utf-8sdatar)r`rarrrr4�s

z'HandlerTests.testBytesData.<locals>.appsRStatus: 200 OK
Content-Type: text/plain; charset=utf-8
Content-Length: 4

data)r�r�rKr�r3)rr4r�rrr�
testBytesData�s




zHandlerTests.testBytesDatac
s>d
di
�
d���
fdd�}
t�}|�
|
�

|��
d
d�
dS)N�close_calledFsSome output has been sentcs(|
d
g���

G�
f
dd�dt�}|�S)Nz200 OKcs eZ
dZd
d�Z�f
dd�ZdS)zHHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterablec

ssxd
V
td�
�
qWdS)Nsblahz This should be caught by handler)
r;)
rrrr�__iter__s


zQHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterable.__iter__c

sd
�d<dS)NTr�r)
r)
�side_effectsrrrzs
zNHandlerTests.testCloseOnError.<locals>.error_app.<locals>.CrashyIterable.closeN)rrr r�rzr)
r�rr�CrashyIterables
r�)
�object)r`rar�)r�r�rrr�s

z0HandlerTests.testCloseOnError.<locals>.error_appT)r�r�rK)rr�r�r)r�r�r�testCloseOnErrors





zHandlerTests.testCloseOnErrorc
	sjt��G�f
d
d�d�}
ddi
}t
t�|
�tj|�}d}|�t|��
|�t�

WdQRX|�	d��
dS)Ncs eZ
dZ�f
d
d�Zdd�ZdS)z4HandlerTests.testPartialWrite.<locals>.PartialWritercs|
dd
�}��|�

t
|�
S)N�)�extendry)r�
b�partial)
�writtenrr�writes



z:HandlerTests.testPartialWrite.<locals>.PartialWriter.writec


SsdS)
Nr)
rrrr�flushs
z:HandlerTests.testPartialWrite.<locals>.PartialWriter.flushN)rrr r�r�r)
r�rr�
PartialWriters
r
r�zHTTP/1.0zshould not do partial writesssHTTP/1.0 200 OK
Content-Type: text/plain
Date: Mon, 05 Jun 2006 18:49:54 GMT
Content-Length: 13

Hello, world!)
�	bytearrayr
rr1r2ZassertWarnsRegex�DeprecationWarningr�r(rK)rr
r&r��msgr)
r�r�testPartialWrites
	




zHandlerTests.testPartialWritec

szd
di
}
xltt
tfD]^�|j�d�
�H
G�f
dd�d�}t�}tt�|�||
�}|�t�

|�	|�
��

WdQRXqWdS)Nr�zHTTP/1.0)
�	exceptioncseZ
dZ�f
d
d�ZdS)zEHandlerTests.testClientConnectionTerminations.<locals>.AbortingWriterc
s��
dS)
Nr)rr�)
r
rrr�6s
zKHandlerTests.testClientConnectionTerminations.<locals>.AbortingWriter.writeN)rrr r�r)
r
rr�AbortingWriter5s
r
)�ConnectionAbortedError�BrokenPipeError�ConnectionResetErrorr]rr
rr�r(r�r3)rr&r
r2r�r)
r
r� testClientConnectionTerminations-s









z-HandlerTests.testClientConnectionTerminationsc
s�Gd
d�dt��G�f
dd�d�}
t
�}ddi
}tt�|
�||�}|�t�

|�d|���
|�|j	�

|�|j
�

|�|j�

|�|j�

dS)Nc
@seZ
dZd
S)zKHandlerTests.testDontResetInternalStateOnException.<locals>.CustomExceptionN)rrr rrrr�CustomException@s
r
cseZ
dZ�f
d
d�ZdS)zJHandlerTests.testDontResetInternalStateOnException.<locals>.AbortingWriterc
s��
dS)
Nr)rr�)
r
rrr�Hs
zPHandlerTests.testDontResetInternalStateOnException.<locals>.AbortingWriter.writeN)rrr r�r)
r
rrr
Gs
r
r�zHTTP/1.0)
�
ValueErrorrr
rr�r(r�r3ZassertIsNotNone�resultZheadersr[r&)rr
r2r&r�r)
r
r�%testDontResetInternalStateOnException?s







z2HandlerTests.testDontResetInternalStateOnExceptionN)rrr ZmaxDiffr�r�r�r�r�r�r�r�r�r�r
r

r
rrrrr�s62
+r��__main__)4Zunittestrr�rZtest.test_httpserversrrZwsgiref.utilrZwsgiref.headersrZwsgiref.handlersrr	r
ZwsgirefrZwsgiref.validaterZwsgiref.simple_serverr
rrZhttp.clientr�iorrrZsocketserverr�platformrr�r�rur1r�rr!r(r.r9rFrGr�r�r�r�r�r�mainrrrr�<module>
sL



















%9#7=