Your IP : 216.73.216.213
�
g f�� � @ sd d Z d d l m Z m Z m Z m Z d d l m Z d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z d d l m Z e j d � Z Gd d � d � Z Gd
d � d e j � Z Gd d
� d
e j � Z Gd d � d e � Z Gd d � d e � Z d Z! d Z" d Z# e j$ e% e d � o�e j& � d k d � Gd d � d e � � Z' Gd d � d e � Z( Gd d � d e( � Z) Gd d � d � Z* Gd d � d e j � Z+ Gd! d" � d" e j � Z, Gd# d$ � d$ e j � Z- d d% d& � Z. e/ d' k r`e. � n d S)(z�Unittests for the various HTTPServer modules.
Written by Cody A.W. Somerville <cody-somerville@ubuntu.com>,
Josip Dzolonga, and Michael Otteneder for the 2007/08 GHOP contest.
� )�BaseHTTPRequestHandler�
HTTPServer�SimpleHTTPRequestHandler�CGIHTTPRequestHandler)�serverN)�BytesIO)�support� threadingc @ s+ e Z d Z d d � Z d d d � Z d S)�NoLogRequestHandlerc G s d S)N� )�self�argsr r �:/opt/alt/python34/lib64/python3.4/test/test_httpservers.py� log_message s z NoLogRequestHandler.log_messageNc C s d S)N� r )r �nr r r �read s zNoLogRequestHandler.read)�__name__�
__module__� __qualname__r r r r r r r
s r
c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�TestServerThreadc C s& t j j | � | | _ | | _ d S)N)r �Thread�__init__�request_handler� test_object)r r r r r r r % s zTestServerThread.__init__c
C s{ t d | j � | _ | j j j � \ | j _ | j _ | j j j � d | _ z | j j
d � Wd | j j � Xd S)N� localhostr g�������?)r r ) r r r ZsocketZ getsocknamer �HOST�PORT�server_started�setZ
serve_foreverZ server_close)r r r r �run* s $ zTestServerThread.runc C s | j j � d S)N)r Zshutdown)r r r r �stop4 s zTestServerThread.stopN)r r r r r r! r r r r r $ s
r c @ s= e Z d Z d d � Z d d � Z d d i d d � Z d S) � BaseTestCasec C s` t j � | _ t j � t _ t j � | _ t | | j
� | _ | j j � | j j
� d S)N)r Zthreading_setup�_threadsZEnvironmentVarGuard�os�environr ZEventr r r �thread�start�wait)r r r r �setUp9 s
zBaseTestCase.setUpc C s7 | j j � d | _ t j j � t j | j � d S)N)r& r! r$ r% �__exit__r Zthreading_cleanupr# )r r r r �tearDownA s
zBaseTestCase.tearDown�GETNc C sD t j j | j | j � | _ | j j | | | | � | j j � S)N)�http�client�HTTPConnectionr r Z
connection�request� getresponse)r Zuri�method�body�headersr r r r0 G s zBaseTestCase.request)r r r r) r+ r0 r r r r r" 8 s r" c @ s e Z d Z Gd d � d e e � Z d d � Z d d � Z d d � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' S)(�BaseHTTPServerTestCasec @ sp e Z d Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z
d d � Z d S)z&BaseHTTPServerTestCase.request_handlerzHTTP/1.1c C s; | j d � | j d d � | j d d � | j � d S)N�� z Content-Typez text/html�
Connection�close)�
send_response� send_header� end_headers)r r r r �do_TESTR s
z.BaseHTTPServerTestCase.request_handler.do_TESTc C s; | j d � | j d d � | j d d � | j � d S)Nr6 z Content-Typez text/htmlr7 z
keep-alive)r9 r: r; )r r r r �do_KEEPX s
z.BaseHTTPServerTestCase.request_handler.do_KEEPc C s | j d � d S)Ni� )�
send_error)r r r r � do_KEYERROR^ s z2BaseHTTPServerTestCase.request_handler.do_KEYERRORc C s | j d � d S)Ni� )r> )r r r r � do_NOTFOUNDa s z2BaseHTTPServerTestCase.request_handler.do_NOTFOUNDc C s | j d d d � d S)Ni� z
Short Messagez This is a long
explaination)r> )r r r r �do_EXPLAINERRORd s z6BaseHTTPServerTestCase.request_handler.do_EXPLAINERRORc C s; | j d � | j d d � | j d d � | j � d S)Ni� z Content-Typez text/htmlr7 r8 )r9 r: r; )r r r r � do_CUSTOMh s
z0BaseHTTPServerTestCase.request_handler.do_CUSTOMc C sa | j d � | j d d � | j d d � | j � | j d j d � } | j j | � d S)Ni� z X-Specialu Dängerous Mindr7 r8 zx-special-incomingzutf-8)r9 r: r; r4 �encode�wfile�write)r r3 r r r �do_LATINONEHEADERn s
z8BaseHTTPServerTestCase.request_handler.do_LATINONEHEADERN) r r r �protocol_versionZdefault_request_versionr< r= r? r@ rA rB rF r r r r r N s r c C s< t j | � t j j | j | j � | _ | j j � d S)N) r" r) r- r. r/ r r �conZconnect)r r r r r) v s
z BaseHTTPServerTestCase.setUpc C s9 | j j d d � | j j � } | j | j d � d S)Nr, �/i� )rH r0 r1 � assertEqual�status)r �resr r r � test_command{ s z#BaseHTTPServerTestCase.test_commandc C sR d | j _ | j j d d � | j j � | j j � } | j | j d � d S)Nz HTTP/1.1
ZXYZBOGUSrI i� )rH �
_http_vsn_str�
putrequest�
endheadersr1 rJ rK )r rL r r r �test_request_line_trimming� s
z1BaseHTTPServerTestCase.test_request_line_trimmingc C sR d | j _ | j j d d � | j j � | j j � } | j | j d � d S)NZFUBARr, rI i� )rH rN rO rP r1 rJ rK )r rL r r r �test_version_bogus� s
z)BaseHTTPServerTestCase.test_version_bogusc C sR d | j _ | j j d d � | j j � | j j � } | j | j d � d S)Nz
HTTP/9.9.9r, rI i� )rH rN rO rP r1 rJ rK )r rL r r r �test_version_digits� s
z*BaseHTTPServerTestCase.test_version_digitsc C sR d | j _ | j j d d � | j j � | j j � } | j | j d � d S)Nr r, rI i� )rH rN rO rP r1 rJ rK )r rL r r r �test_version_none_get� s
z,BaseHTTPServerTestCase.test_version_none_getc C sR d | j _ | j j d d � | j j � | j j � } | j | j d � d S)Nr �CUSTOMrI i� )rH rN rO rP r1 rJ rK )r rL r r r �test_version_none� s
z(BaseHTTPServerTestCase.test_version_nonec C s^ d | j _ d | j _ | j j d d � | j j � | j j � } | j | j d � d S)N�c zHTTP/9.9r, rI i� )rH Z _http_vsnrN rO rP r1 rJ rK )r rL r r r �test_version_invalid� s
z+BaseHTTPServerTestCase.test_version_invalidc C sR d | j _ | j j d d � | j j � | j j � } | j | j d � d S)Nr i� )rH rN rO rP r1 rJ rK )r rL r r r �test_send_blank� s
z&BaseHTTPServerTestCase.test_send_blankc C sY | j j d d � | j j d d � | j j � | j j � } | j | j d � d S)Nr, rI r7 r8 i� )rH rO � putheaderrP r1 rJ rK )r rL r r r �test_header_close� s
z(BaseHTTPServerTestCase.test_header_closec C se d | j _ | j j d d � | j j d d � | j j � | j j � } | j | j d � d S)NzHTTP/1.1r, rI r7 z
keep-alivei� )rH rN rO rZ rP r1 rJ rK )r rL r r r �test_head_keep_alive� s
z+BaseHTTPServerTestCase.test_head_keep_alivec C s9 | j j d d � | j j � } | j | j d � d S)N�TESTrI r6 )rH r0 r1 rJ rK )r rL r r r � test_handler� s z#BaseHTTPServerTestCase.test_handlerc C se | j j d d � | j j � } | j | j d � d � | j j d d � | j | j j � d S)NZKEEPrI r7 z
keep-aliver] )rH r0 r1 rJ � getheaderZ
addCleanupr8 )r rL r r r � test_return_header_keep_alive� s
z4BaseHTTPServerTestCase.test_return_header_keep_alivec C s9 | j j d d � | j j � } | j | j d � d S)NZKEYERRORrI i� )rH r0 r1 rJ rK )r rL r r r �test_internal_key_error� s z.BaseHTTPServerTestCase.test_internal_key_errorc C s9 | j j d d � | j j � } | j | j d � d S)NrU rI i� )rH r0 r1 rJ rK )r rL r r r �test_return_custom_status� s z0BaseHTTPServerTestCase.test_return_custom_statusc C sU | j j d d � | j j � } | j | j d � | j t | j d � � � d S)NZ EXPLAINERRORrI i� zContent-Length)rH r0 r1 rJ rK �
assertTrue�intr_ )r rL r r r �test_return_explain_error� s z0BaseHTTPServerTestCase.test_return_explain_errorc C sk | j j d d d i d d 6�| j j � } | j | j d � d � | j | j � d j d � � d S) NZLATINONEHEADERrI r4 u Ärger mit UnicodezX-Special-Incomingz X-Specialu Dängerous Mindzutf-8)rH r0 r1 rJ r_ r rC )r rL r r r �test_latin1_header� s
z)BaseHTTPServerTestCase.test_latin1_headerc C sj | j j d d � | j j � } | j | j d � | j � } | j t | j d � � t | � � d S)NZNOTFOUNDrI i� zContent-Length) rH r0 r1 rJ rK r rd r_ �len)r rL �datar r r �test_error_content_length� s
z0BaseHTTPServerTestCase.test_error_content_lengthN)r r r r
r r r) rM rQ rR rS rT rV rX rY r[ r\ r^ r` ra rb re rf ri r r r r r5 M s& ( r5 c @ s� e Z d Z Gd d � d e e � Z d d � Z d d � Z d d d � Z e j
d
d � e j e j
d � d
d � � � Z d d � Z d d � Z d d � Z d S)�SimpleHTTPServerTestCasec @ s e Z d Z d S)z(SimpleHTTPServerTestCase.request_handlerN)r r r r r r r r � s r c C s� t j | � t j � | _ t j � } t j | � d | _ t j d | � | _
t j j | j
� | _
t t j j | j
d � d � � } | j | j � Wd QXd S)Ns We are the knights who say Ni!�dir�test�wb)r" r) r$ �getcwd�cwd�tempfileZ
gettempdir�chdirrh �mkdtemp�tempdir�path�basename� tempdir_name�open�joinrE )r Z basetempdirZtempr r r r) � s
$z SimpleHTTPServerTestCase.setUpc C sH z3 t j | j � y t j | j � Wn Yn XWd t j | � Xd S)N)r$ rq ro �shutilZrmtreers r" r+ )r r r r r+ � s z!SimpleHTTPServerTestCase.tearDownNc C sY | j � } | j | � | j | j | � | j | j � | rU | j | | � n | S)N)r rc rJ rK �assertIsNotNone�reason)r �responserK rh r3 r r r �check_status_and_reason s
z0SimpleHTTPServerTestCase.check_status_and_reason�
� z need support.TESTFN_UNDECODABLEc C sa t j � } t j t j � d } t t j j | j | � d � � } | j
t j � Wd QX| j | j d � } t j
d k r� x3 t j | j � D] } | d k r� | } Pq� q� Wn | j | d � } t j j | d d �} | j d | j | d
� | � | j d t j | � j | d
� | � | j | j d | � } | j | d d t j �d S)
Nz.txtrm rI �darwinrl �� �errors�
surrogatepassz href="%s"�surrogateescapez>%s<rh )�sys�getfilesystemencodingr$ �fsdecoder �TESTFN_UNDECODABLErw rt rx rs rE r0 rv �platform�listdirr} �urllib�parseZquoteZassertInrC �html�escape)r �enc�filename�fr| �namer3 Z
quotednamer r r �test_undecodable_filename s& $ z2SimpleHTTPServerTestCase.test_undecodable_filenamec
C s | j | j d � } | j | d d | j �| j | j d � } | j | d � | j | j d � } | j | d � | j | j � } | j | d � | j | j d � } | j | d � | j | j d � } | j | d � | j | j d
� | j d � | j d � } | j | d � | j d � } | j | d � t t j j | j d � d � �� } | j d | j d � } | j | d � t j
d k rt j � d k rt j | j
d � | j | j d � } | j | d � t j | j
d � n Wd QXd S)Nz/testr� rh z/test/i� rI i- z/?hi=2z?hi=1ZLocationz/?hi=1z/ThisDoesNotExistZThisDoesNotExistz
index.html�w�posixr i� z/ThisDoesNotExistz/ThisDoesNotExist/)r0 rv r} rh rJ r_ rw r$ rt rx r� �geteuid�chmodrs )r r| r� r r r �test_get+ s4 $!z!SimpleHTTPServerTestCase.test_getc C sq | j | j d d d �} | j | d � | j | j d � t t | j � � � | j | j d � d � d S)Nz/testr2 ZHEADr� zcontent-lengthz content-typezapplication/octet-stream)r0 rv r} rJ r_ �strrg rh )r r| r r r � test_headK s z"SimpleHTTPServerTestCase.test_headc C ss | j d d d �} | j | d � | j d d d �} | j | d � | j d d d �} | j | d � d S)NrI r2 ZFOOi� ZcustomZGETs)r0 r} )r r| r r r �test_invalid_requestsT s z.SimpleHTTPServerTestCase.test_invalid_requests)r r r r
r r r) r+ r} r Zrequires_mac_ver�unittestZ
skipUnlessr� r� r� r� r� r r r r rj � s
rj zD#!%s
print("Content-type: text/html")
print()
print("Hello World")
z�#!%s
import cgi
print("Content-type: text/html")
print()
form = cgi.FieldStorage()
print("%%s, %%s, %%s" %% (form.getfirst("spam"), form.getfirst("eggs"),
form.getfirst("bacon")))
zR#!%s
import os
print("Content-type: text/html")
print()
print(os.environ["%s"])
r� z7This test can't be run reliably as root (issue #13308).c @ s� e Z d Z Gd d � d e e � Z e j j d � Z d d � Z d d � Z
d d � Z d
d � Z d d
� Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d S)!�CGIHTTPServerTestCasec @ s e Z d Z d S)z%CGIHTTPServerTestCase.request_handlerN)r r r r r r r r � s r �asciic + C s= t j | � t j � | _ t j � | _ t j j | j d � | _
t j j | j
d � | _ t j | j
� t j | j � d | _
d | _ d | _ d | _ d | _ t j � r� t j j | j d � | _ t j t j | j � n t j | _ y | j j d � Wn) t k
r9| j � | j d � Yn Xt j j | j d � | _
t | j
d � � } | j t | j � Wd QXt j | j
d � t j j | j
d � | _ t | j d d
d �� } | j t | j � Wd QXt j | j d � t j j | j
d � | _ t | j d d
d �� } | j t | j � Wd QXt j | j d � t j j | j d � | _ t | j d d
d �� } | j t | j � Wd QXt j | j d � t j j | j
d
� | _ t | j d d
d ��! } | j t! | j d f � Wd QXt j | j d � t j" | j � d S)Nzcgi-binz child-dir�pythonzutf-8z0Python executable path is not encodable to utf-8znocgi.pyr� i� zfile1.py�encodingzfile2.pyzfile3.pyzfile4.pyZ QUERY_STRING)#r" r) r$ rn ro rp rr �
parent_dirrt rx �cgi_dir�
cgi_child_dir�mkdir�
nocgi_path�
file1_path�
file2_path�
file3_path�
file4_pathr Z can_symlink� pythonexe�symlinkr� �
executablerC �UnicodeEncodeErrorr+ ZskipTestrw rE � cgi_file1r� � cgi_file2� cgi_file4rq )r �fpZfile1Zfile2Zfile3Zfile4r r r r) � sT
zCGIHTTPServerTestCase.setUpc C s
z� t j | j � | j t j k r8 t j | j � n | j rT t j | j � n | j rp t j | j � n | j r� t j | j � n | j
r� t j | j
� n | j r� t j | j � n t j | j
� t j | j � t j | j � Wd t j | � Xd S)N)r$ rq ro r� r� r� �remover� r� r� r� r� �rmdirr� r� r� r" r+ )r r r r r+ � s$ z CGIHTTPServerTestCase.tearDownc C sT i d d 6t d 6t d 6d d 6d d 6d d 6d d 6d d
6d d 6d d 6d d
6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6d d 6t d 6d d 6d d 6t d! 6d d" 6d d# 6} x� | j � D]u \ } } t | t � rt | t � r| j | t j | � q� t j | � } | j | | d$ d% | | | f �q� Wd S)&Nz//r z..z/.//..rI z//\z/\z/.//z/cgi-bin/file1.pyzcgi-bin/file1.pyz//a�az/az./az/C:/z./C:/z/a/bz/a/b/z/a/b/.z /a/b/c/..z/a/b/dz /a/b/c/../dz/a/b/d/fz/a/b/c/../d/e/../fz/a/b/fz/a/b/c/../d/e/../../fz /a/b/c/../d/e/.././././..//fz ../a/b/c/../d/e/.././././..//fz/a/fz/a/b/c/../d/e/../../../fz//fz/a/b/c/../d/e/../../../../fz /a/b/c/../d/e/../../../../../fz /a/b/c/../d/e/../../../../f/..z /a/b/c/../d/e/../../../../f/../.�msgz path = %r
Got: %r
Wanted: %r)
�
IndexError�items�
isinstance�type�
issubclass� Exception� assertRaisesr Z_url_collapse_pathrJ )r Z test_vectorsrt �expectedZactualr r r �test_url_collapse_path� sJ
z,CGIHTTPServerTestCase.test_url_collapse_pathc C sN | j d � } | j d | j d d f | j � | j d � | j f � d S)Nz/cgi-bin/file1.pys Hello Worldz text/htmlr� z Content-type)r0 rJ �linesepr r_ rK )r rL r r r �test_headers_and_content� s z.CGIHTTPServerTestCase.test_headers_and_contentc C s& | j d � } | j | j d � d S)Nz)///////////nocgi.py/../cgi-bin/nothere.shi� )r0 rJ rK )r rL r r r �test_issue19435 s z%CGIHTTPServerTestCase.test_issue19435c C sm t j j i d d 6d d 6d d 6� } i d d 6} | j d d
| | � } | j | j � d | j � d S) N� Zspamr� Zeggsi@� Zbaconz!application/x-www-form-urlencodedz Content-typez/cgi-bin/file2.pyZPOSTs 1, python, 123456)r� r� Z urlencoder0 rJ r r� )r Zparamsr4 rL r r r � test_post s
z CGIHTTPServerTestCase.test_postc C s0 | j d � } | j � | j | j d � d S)Nz/cgi-bin/invalidi� )r0 r rJ rK )r rL r r r �test_invaliduri s
z%CGIHTTPServerTestCase.test_invaliduric C sq i d t j d � d 6} | j d d d | �} | j d | j d d f | j � | j d
� | j f � d S) Ns Basic s
username:passs
Authorizationz/cgi-bin/file1.pyr, r4 s Hello Worldz text/htmlr� z Content-type)�base64Z b64encoder0 rJ r� r r_ rK )r r4 rL r r r �test_authorization s
z(CGIHTTPServerTestCase.test_authorizationc C sN | j d � } | j d | j d d f | j � | j d � | j f � d S)Nzcgi-bin/file1.pys Hello Worldz text/htmlr� z Content-type)r0 rJ r� r r_ rK )r rL r r r �test_no_leading_slash s z+CGIHTTPServerTestCase.test_no_leading_slashc C sx d } | t j d <| j d � } | j d | j d d f | j � | j d � | j f � | j t j d | � d S)NzTest CGI ServerZSERVER_SOFTWAREz/cgi-bin/file1.pys Hello Worldz text/htmlr� z Content-type)r$ r% r0 rJ r� r r_ rK )r Z signaturerL r r r � test_os_environ_is_not_altered! s
"z4CGIHTTPServerTestCase.test_os_environ_is_not_alteredc C sN | j d � } | j d | j d d f | j � | j d � | j f � d S)Nz/cgi-bin%2ffile1.pys Hello Worldz text/htmlr� z Content-type)r0 rJ r� r r_ rK )r rL r r r �#test_urlquote_decoding_in_cgi_check) s z9CGIHTTPServerTestCase.test_urlquote_decoding_in_cgi_checkc C sN | j d � } | j d | j d d f | j � | j d � | j f � d S)Nz/cgi-bin/child-dir/file3.pys Hello Worldz text/htmlr� z Content-type)r0 rJ r� r r_ rK )r rL r r r � test_nested_cgi_path_issue21323. s z5CGIHTTPServerTestCase.test_nested_cgi_path_issue21323c C sN | j d � } | j d | j d d f | j � | j d � | j f � d S)Nz/cgi-bin/file4.py?a=b?c=ds a=b?c=dz text/htmlr� z Content-type)r0 rJ r� r r_ rK )r rL r r r �&test_query_with_multiple_question_mark3 s z<CGIHTTPServerTestCase.test_query_with_multiple_question_markc C sN | j d � } | j d | j d d f | j � | j d � | j f � d S)Nz0/cgi-bin/file4.py?k=aa%2F%2Fbb&//q//p//=//a//b//s k=aa%2F%2Fbb&//q//p//=//a//b//z text/htmlr� z Content-type)r0 rJ r� r r_ rK )r rL r r r �"test_query_with_continuous_slashes9 s
z8CGIHTTPServerTestCase.test_query_with_continuous_slashesN)r r r r
r r r$ r� rC r) r+ r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� } s : * r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�SocketlessRequestHandlerc C s d | _ d | _ d S)NFzHTTP/1.1)�
get_calledrG )r r r r r B s z!SocketlessRequestHandler.__init__c C sD d | _ | j d � | j d d � | j � | j j d � d S)NTr� z Content-Typez text/htmls <html><body>Data</body></html>
)r� r9 r: r; rD rE )r r r r �do_GETF s
z SocketlessRequestHandler.do_GETc G s d S)Nr )r �formatr
r r r r M s z$SocketlessRequestHandler.log_messageN)r r r r r� r r r r r r� A s r� c @ s e Z d Z d d � Z d S)�!RejectingSocketlessRequestHandlerc C s | j d � d S)Ni� F)r> )r r r r �handle_expect_100Q s
z3RejectingSocketlessRequestHandler.handle_expect_100N)r r r r� r r r r r� P s r� c @ sF e Z d Z d d � Z d d � Z d d � Z e d d � � Z d S)
�AuditableBytesIOc C s
g | _ d S)N)�datas)r r r r r X s zAuditableBytesIO.__init__c C s | j j | � d S)N)r� �append)r rh r r r rE [ s zAuditableBytesIO.writec C s d j | j � S)N� )rx r� )r r r r �getData^ s zAuditableBytesIO.getDatac C s
t | j � S)N)rg r� )r r r r � numWritesa s zAuditableBytesIO.numWritesN)r r r r rE r� �propertyr� r r r r r� V s r� c @ s� e Z d Z d Z e j d � Z d d � Z d d � Z d d � Z d d
� Z
d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' S)(� BaseHTTPRequestHandlerTestCasezrTest the functionality of the BaseHTTPServer.
Test the support for the Expect 100-continue header.
s HTTP/1.[0-9]+ 200 OKc C s t � | _ d S)N)r� �handler)r r r r r) n s z$BaseHTTPRequestHandlerTestCase.setUpc C sQ t | � } t � } | | j _ | | j _ | j j � | j d � | j � S)Nr )r r� �rfilerD �handle_one_request�seek� readlines)r �message�input�outputr r r �send_typical_requestq s
z3BaseHTTPRequestHandlerTestCase.send_typical_requestc C s | j | j j � d S)N)rc r� r� )r r r r �verify_get_calledz s z0BaseHTTPRequestHandlerTestCase.verify_get_calledc s> x7 d D]/ � | j t � f d d � | D� � d � q Wd S)N� Server: � Date: � Content-Type: c 3 s | ] } | j � � Vq d S)N)�
startswith)�.0�h)� fieldNamer r � <genexpr> s zIBaseHTTPRequestHandlerTestCase.verify_expected_headers.<locals>.<genexpr>r� )r� r� r� )rJ �sum)r r4 r )r� r �verify_expected_headers} s
z6BaseHTTPRequestHandlerTestCase.verify_expected_headersc C s# | j j | � } | j | � d S)N)�HTTPResponseMatch�searchrz )r r| �matchr r r �verify_http_server_response� s z:BaseHTTPRequestHandlerTestCase.verify_http_server_responsec C s� | j d � } | j | d � | j | d d � � | j � | j | d
d � | j | j j d � | j | j j d � | j | j j d � | j | j j d � | j
| j j j � f � d S) Ns GET / HTTP/1.1
r r� s <html><body>Data</body></html>
zGET / HTTP/1.1r, rI zHTTP/1.1���r� )
r� r� r� r� rJ r� � requestline�commandrt �request_version�assertSequenceEqualr4 r� )r �resultr r r �
test_http_1_1� s
z,BaseHTTPRequestHandlerTestCase.test_http_1_1c C s� | j d � } | j | d � | j | d d � � | j � | j | d
d � | j | j j d � | j | j j d � | j | j j d � | j | j j d � | j
| j j j � f � d S) Ns GET / HTTP/1.0
r r� s <html><body>Data</body></html>
zGET / HTTP/1.0r, rI zHTTP/1.0r� r� )
r� r� r� r� rJ r� r� r� rt r� r� r4 r� )r r� r r r �
test_http_1_0� s
z,BaseHTTPRequestHandlerTestCase.test_http_1_0c C sG | j d � } | j t | � d � | j | d d � | j � d S)Ns GET / HTTP/0.9
r� r s <html><body>Data</body></html>
)r� rJ rg r� )r r� r r r �
test_http_0_9� s z,BaseHTTPRequestHandlerTestCase.test_http_0_9c C s� | j d � } | j | d � | j | d d � � | j � | j | d d � | j | j j d � | j | j j d � | j | j j d � | j | j j d � d } | j
| j j j � | � d S)Ns( GET / HTTP/1.0
Expect: 100-continue
r r� s <html><body>Data</body></html>
zGET / HTTP/1.0r, rI zHTTP/1.0�Expect� 100-continuer� r� �r� r� )r� )
r� r� r� r� rJ r� r� r� rt r� r� r4 r� )r r� r4 r r r �test_with_continue_1_0� s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_0c C s� | j d � } | j | d d � | j | d d � | j | d d � | j | d d � � | j � | j | d d � | j | j j d � | j | j j d
� | j | j j d � | j | j j d � d } | j | j j
j � | � d S)Ns( GET / HTTP/1.1
Expect: 100-continue
r s HTTP/1.1 100 Continue
r� s
� s HTTP/1.1 200 OK
s <html><body>Data</body></html>
zGET / HTTP/1.1r, rI zHTTP/1.1r� � 100-continuer� r� �zExpectr )r ) r� rJ r� r� r� r� r� rt r� r� r4 r� )r r� r4 r r r �test_with_continue_1_1� s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_1c C so t d � } t � } t � } | | _ | | _ d | _ d | _ d | _ | j d � | j | j
d � d S)Ns GET / HTTP/1.1
zHTTP/1.1r i� r ) r r� r� r� rD r� r� r� r> rJ r� )r r� r� r� r r r �#test_header_buffering_of_send_error� s
zBBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_errorc C sz t d � } t � } t � } | | _ | | _ d | _ | j d � | j | j d � | j � | j | j d � d S)Ns GET / HTTP/1.1
zHTTP/1.1i� r r� )
r r� r� r� rD r� Zsend_response_onlyrJ r� r; )r r� r� r� r r r �+test_header_buffering_of_send_response_only� s
zJBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_response_onlyc C s� t d � } t � } t � } | | _ | | _ d | _ | j d d � | j d d � | j | j d � | j � | j | j
� d � | j | j d � d S) Ns GET / HTTP/1.1
zHTTP/1.1ZFooZfooZbarr s Foo: foo
bar: bar
r� ) r r� r� r� rD r� r: rJ r� r; r� )r r� r� r� r r r �$test_header_buffering_of_send_header� s
zCBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_headerc C s� d d � } t d � } t � } | | j _ | | j _ d | j _ | j j � | j | | � d � | | � j d � } | j | d d � | j | d d � | j | d
d � d S) Nc S s6 | j � } | j d � | j � } | j | � | S)Nr )�tellr� r )r� �posrh r r r �_readAndReseek� s
z[BaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continue.<locals>._readAndReseeks( GET / HTTP/1.1
Expect: 100-continue
zHTTP/1.1r� s
r s HTTP/1.1 100 Continuer� r s HTTP/1.1 200 OK) r r� r� rD r� r� ZassertNotEqual�splitrJ )r r r� r� r� r r r �$test_header_unbuffered_when_continue� s
zCBaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continuec C s� | j } t � | _ | j d � } | j | d d � | j | d d � � | j | j j � | j t d d � | d d � D� � d � | | _ d S) Ns( GET / HTTP/1.1
Expect: 100-continue
r s! HTTP/1.1 417 Expectation Failed
r� c s s | ] } | d k Vq d S)s Connection: close
Nr )r� �rr r r r�
s zMBaseHTTPRequestHandlerTestCase.test_with_continue_rejected.<locals>.<genexpr>r� r� )r� r� r� rJ r� � assertFalser� r� )r Z
usual_handlerr� r r r �test_with_continue_rejected s -z:BaseHTTPRequestHandlerTestCase.test_with_continue_rejectedc C sX | j d d d � } | j | d d � | j | j j � | j | j j t � d S)Ns GET � xi r s# HTTP/1.1 414 Request-URI Too Long
)r� rJ r
r� r� ZassertIsInstancer� r� )r r� r r r �test_request_length
s z2BaseHTTPRequestHandlerTestCase.test_request_lengthc C s\ | j d d d d � } | j | d d � | j | j j � | j | j j d � d S)Ns GET / HTTP/1.1
X-Foo: bar� ri s
r s HTTP/1.1 400 Line too long
zGET / HTTP/1.1)r� rJ r
r� r� r� )r r� r r r �test_header_length s
z1BaseHTTPRequestHandlerTestCase.test_header_lengthc s} � � f d d � } | � j _ t d � � � j j � � j t t � � t d � � � j j � � j t t � � d S)Nc s t � � � j _ d S)N)�nextr� Zclose_connectionr )� close_valuesr r r r� s zPBaseHTTPRequestHandlerTestCase.test_close_connection.<locals>.handle_one_requestTF)T)FFT)r� r� �iterZhandler� �
StopIterationr )r r� r )r r r �test_close_connection s
z4BaseHTTPRequestHandlerTestCase.test_close_connectionN)r r r �__doc__�re�compiler� r) r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r� f s(
r� c @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) � SimpleHTTPRequestHandlerTestCasez Test url parsing c C s: t j � | _ t j j | j d � | _ t � | _ d S)Nr� )r$ rn �
translatedrt rx r� r� )r r r r r) . s z&SimpleHTTPRequestHandlerTestCase.setUpc C ss | j j d � } | j | | j � | j j d � } | j | | j � | j j d � } | j | | j � d S)Nz /filenamez/filename?foo=barz/filename?a=b&spam=eggs#zot)r� �translate_pathrJ r )r rt r r r �test_query_arguments3 s z5SimpleHTTPRequestHandlerTestCase.test_query_argumentsc C sN | j j d � } | j | | j � | j j d � } | j | | j � d S)Nz
//filenamez//filename?foo=bar)r� r rJ r )r rt r r r � test_start_with_double_slash; s z=SimpleHTTPRequestHandlerTestCase.test_start_with_double_slashc
C s t j t j d t � �� | j j d � } | j t j t j � } | j | | j
� | j j d � } | j t j t j � } | j | | j
� | j j d � } | j t j t j � } | j | | j
� | j j d � } | j t j t j � } | j | | j
� Wd QXd S)Nrt zc:c:c:foo/filenamez\c:../filenamezc:\c:..\foo/filenamezc:c:foo\c:c:bar/filename) r Z swap_attrr r$ �ntpathr� r �replace�seprJ r )r rt r r r �test_windows_colonA s z3SimpleHTTPRequestHandlerTestCase.test_windows_colonN)r r r r r) r r r# r r r r r , s
r c @ s e Z d Z d d � Z d S)� MiscTestCasec C s� g } d d d h } xl t t � D]^ } | j d � s" | | k rI q" n t t | � } t | d d � d k r" | j | � q" q" W| j t j | � d S)Nr� Z
nobody_uidrl �_r z http.server)rk r r� �getattrr� ZassertCountEqual�__all__)r r� Z blacklistr� Z
module_objectr r r �test_allU s zMiscTestCase.test_allN)r r r r( r r r r r$ T s r$ c
C sA t j � } z t j t t t t t t � Wd t j
| � Xd S)N) r$ rn r Z run_unittestr� r5 rj r� r r$ rq )�verbosero r r r � test_maina s r* �__main__)0r Z http.serverr r r r r- r r$ r� r r� r ry Z urllib.parser� r� Z http.clientrp �ior r� rl r �
import_moduler r
r r ZTestCaser" r5 rj r� r� r� ZskipIf�hasattrr� r� r� r� r� r� r r$ r* r r r r r �<module> sH " �u
$ ��(