Your IP : 216.73.216.52
���](� � @ sw d d l m Z m Z m Z m Z d d l m Z 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 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/ e0 d( k rse/ � d S))� )�BaseHTTPRequestHandler�
HTTPServer�SimpleHTTPRequestHandler�CGIHTTPRequestHandler)�server�
HTTPStatusN)�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 �5/opt/alt/python35/lib64/python3.5/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 �socket� 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 �shutdown)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
�Eventr! 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 �
connection�request� getresponse)r
Zuri�method�body�headersr r r r6 G s zBaseTestCase.request)r r r r. r0 r6 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' d( � Z d) d* � Z d+ S),�BaseHTTPServerTestCasec @ s� 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 d � Z d d � Z
d S)z&BaseHTTPServerTestCase.request_handlerzHTTP/1.1c C s> | j t j � | j d d � | j d d � | j � d S)Nz Content-Typez text/html�
Connection�close)�
send_responser �
NO_CONTENT� send_header� end_headers)r
r r r �do_TESTR s z.BaseHTTPServerTestCase.request_handler.do_TESTc C s> | j t j � | j d d � | j d d � | j � d S)Nz Content-Typez text/htmlr< z
keep-alive)r> r r? r@ rA )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 t j � d S)N)rD r � NOT_FOUND)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
explanation)rD )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/htmlr< r= )r> r@ rA )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 Mindr< r= zx-special-incomingzutf-8)r> r@ rA r: �encode�wfile�write)r
r9 r r r �do_LATINONEHEADERn s
z8BaseHTTPServerTestCase.request_handler.do_LATINONEHEADERc C s$ | j t | j d d � � � d S)N� )rD �int�path)r
r r r �
do_SEND_ERRORv s z4BaseHTTPServerTestCase.request_handler.do_SEND_ERRORc C s$ | j t | j d d � � � d S)NrN )rD rO rP )r
r r r �do_HEADy s z.BaseHTTPServerTestCase.request_handler.do_HEADN)r r r �protocol_version�default_request_versionrB rC rE rG rH rI rM rQ rR 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. r2 r3 r4 r r �con�connect)r
r r r r. | s
z BaseHTTPServerTestCase.setUpc C s< | j j d d � | j j � } | j | j t j � d S)Nr1 �/)rU r6 r7 � assertEqual�statusr �NOT_IMPLEMENTED)r
�resr r r � test_command� s z#BaseHTTPServerTestCase.test_commandc C sU d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)Nz HTTP/1.1
ZXYZBOGUSrW ) rU �
_http_vsn_str�
putrequest�
endheadersr7 rX rY r rZ )r
r[ r r r �test_request_line_trimming� s
z1BaseHTTPServerTestCase.test_request_line_trimmingc C sU d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)NZFUBARr1 rW ) rU r] r^ r_ r7 rX rY r � BAD_REQUEST)r
r[ r r r �test_version_bogus� s
z)BaseHTTPServerTestCase.test_version_bogusc C sU d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)Nz
HTTP/9.9.9r1 rW ) rU r] r^ r_ r7 rX rY r ra )r
r[ r r r �test_version_digits� s
z*BaseHTTPServerTestCase.test_version_digitsc C sU d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)Nr r1 rW ) rU r] r^ r_ r7 rX rY r rZ )r
r[ r r r �test_version_none_get� s
z,BaseHTTPServerTestCase.test_version_none_getc C sU d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)Nr �CUSTOMrW ) rU r] r^ r_ r7 rX rY r ra )r
r[ r r r �test_version_none� s
z(BaseHTTPServerTestCase.test_version_nonec C sa d | j _ d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)N�c zHTTP/9.9r1 rW )
rU Z _http_vsnr] r^ r_ r7 rX rY r ZHTTP_VERSION_NOT_SUPPORTED)r
r[ r r r �test_version_invalid� s
z+BaseHTTPServerTestCase.test_version_invalidc C sU d | j _ | j j d d � | j j � | j j � } | j | j t j � d S)Nr ) rU r] r^ r_ r7 rX rY r ra )r
r[ r r r �test_send_blank� s
z&BaseHTTPServerTestCase.test_send_blankc C s\ | j j d d � | j j d d � | j j � | j j � } | j | j t j � d S)Nr1 rW r< r= ) rU r^ � putheaderr_ r7 rX rY r rZ )r
r[ r r r �test_header_close� s
z(BaseHTTPServerTestCase.test_header_closec C sh d | j _ | j j d d � | j j d d � | j j � | j j � } | j | j t j � d S)NzHTTP/1.1r1 rW r< z
keep-alive)
rU r] r^ rj r_ r7 rX rY r rZ )r
r[ r r r �test_head_keep_alive� s
z+BaseHTTPServerTestCase.test_head_keep_alivec C s< | j j d d � | j j � } | j | j t j � d S)N�TESTrW )rU r6 r7 rX rY r r? )r
r[ 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)NZKEEPrW r< z
keep-aliverm )rU r6 r7 rX � getheaderZ
addCleanupr= )r
r[ 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)NZKEYERRORrW i� )rU r6 r7 rX rY )r
r[ 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)Nre rW i� )rU r6 r7 rX rY )r
r[ 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 EXPLAINERRORrW i� zContent-Length)rU r6 r7 rX rY �
assertTruerO ro )r
r[ r r r �test_return_explain_error� s z0BaseHTTPServerTestCase.test_return_explain_errorc C sj | j j d d d d d i �| j j � } | j | j d � d � | j | j � d j d � � d S) NZLATINONEHEADERrW r: zX-Special-Incomingu Ärger mit Unicodez X-Specialu Dängerous Mindzutf-8)rU r6 r7 rX ro r rJ )r
r[ r r r �test_latin1_header� s
z)BaseHTTPServerTestCase.test_latin1_headerc C sm | j j d d � | j j � } | j | j t j � | j � } | j t | j d � � t
| � � d S)NZNOTFOUNDrW zContent-Length) rU r6 r7 rX rY r rF r rO ro �len)r
r[ �datar r r �test_error_content_length� s
z0BaseHTTPServerTestCase.test_error_content_lengthc C s� t j t j f } x� t j t j t j t j t j f D]� } | j j d d j | � � | j j � } | j
| | j � | j
d | j d � � | j
d | j d � � | | k r� | j
d | j d � � | j
� } | j
d | � q7 Wd S)NZ
SEND_ERRORz/{}zContent-Lengthz Content-TypezTransfer-Encoding� )r � NOT_MODIFIED�
RESET_CONTENTr? Z
PROCESSING�SWITCHING_PROTOCOLSrU r6 �formatr7 rX rY ro r )r
� allow_transfer_encoding_codes�coder[ rw r r r �test_send_error� s z&BaseHTTPServerTestCase.test_send_errorc C sC t j t j f } x*t j t j t j t j t j f D]} | j j d d j | � � | j j � } | j
| | j � | t j k r� | j t
| j d � � d k � | j d | j d � � n2 | j
d | j d � � | j
d | j d � � | | k r | j
d | j d � � | j � } | j
d | � q7 Wd S) N�HEADz/{}zContent-Lengthr z text/htmlz Content-TypezTransfer-Encodingry )r rz r{ �OKr? r| rU r6 r} r7 rX rY rs rO ro �assertInr )r
r~ r r[ rw r r r �test_head_via_send_error s" " z/BaseHTTPServerTestCase.test_head_via_send_errorN)r r r r r r r. r\ r` rb rc rd rf rh ri rk rl rn rp rq rr rt ru rx r� r� r r r r r; M s* . r; c @ s> e Z d Z Gd d � d e � Z d d � Z d d � Z d S)� RequestHandlerLoggingTestCasec @ s4 e Z d Z d Z d Z d d � Z d d � Z d S)z-RequestHandlerLoggingTestCase.request_handlerzHTTP/1.1c C s | j t j � | j � d S)N)r> r r� rA )r
r r r �do_GET! s z4RequestHandlerLoggingTestCase.request_handler.do_GETc C s | j t j d � d S)NzFile not found)rD r rF )r
r r r �do_ERROR% s z6RequestHandlerLoggingTestCase.request_handler.do_ERRORN)r r r rS rT r� r� r r r r r s r c
C s� t j j | j | j � | _ | j j � t j � �' } | j j d d � | j j
� Wd QRX| j | j � j
d � � d S)Nr1 rW z"GET / HTTP/1.1" 200 -
)r2 r3 r4 r r rU rV r �captured_stderrr6 r7 rs �getvalue�endswith)r
�errr r r �test_get( s
z&RequestHandlerLoggingTestCase.test_getc
C s� t j j | j | j � | _ | j j � t j � �' } | j j d d � | j j
� Wd QRX| j � j d � } | j
| d j d � � | j
| d j d � � d S)N�ERRORrW �
r z code 404, message File not foundrN z"ERROR / HTTP/1.1" 404 -)r2 r3 r4 r r rU rV r r� r6 r7 r� �splitrs r� )r
r� �linesr r r �test_err3 s
z&RequestHandlerLoggingTestCase.test_errN)r r r r r r� r� r r r r r� s r� 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 d � Z d S)�SimpleHTTPServerTestCasec @ s e Z d Z d S)z(SimpleHTTPServerTestCase.request_handlerN)r r r r r r r r A s r c C s� t j | � t j � | _ t j � } t j | � d | _ t j d | � | _
t j j | j
� | _
d | j
| _ t t j j | j
d � d � � } | j | j � Wd QRXd S)Ns We are the knights who say Ni!�dirrW �test�wb)r& r. r( �getcwd�cwd�tempfile�
gettempdir�chdirrw �mkdtemp�tempdirrP �basename� tempdir_name�base_url�open�joinrL )r
Z basetempdirZtempr r r r. D s
$z SimpleHTTPServerTestCase.setUpc C sH z3 t j | j � y t j | j � Wn Yn XWd t j | � Xd S)N)r( r� r� �shutil�rmtreer� r& r0 )r
r r r r0 P s z!SimpleHTTPServerTestCase.tearDownNc s� � � f d d � } d � | � _ � j � } | j � � | j � j | � | j � j � | rv | j | | � | j � j d � | j � j j d d � d � | j � j d � d d � � j
� | S) Nc s � j � d � _ d S)N)�fpr )�reader�responser r �
close_conn[ s zDSimpleHTTPServerTestCase.check_status_and_reason.<locals>.close_conn�
r< r= � ry zConnection should be closed) Z _close_connr rs rX rY �assertIsNotNone�reason�version�msg�getr= )r
r� rY rw r� r9 r )r� r� r �check_status_and_reasonZ s
z0SimpleHTTPServerTestCase.check_status_and_reasonr� � z need support.TESTFN_UNDECODABLEc C sb t j � } t j t j � d } t t j j | j | � d � � } | j
t j � Wd QRX| j | j d � } t j
d k r� x- t j | j � D] } | d k r� | } Pq� W| j | t j � } t j j | d d �} | j d | j | d � | � | j d
t j | � j | d � | � | j | j d | � } | j | t j d t j �d S) Nz.txtr� rW �darwinr� �errors�
surrogatepassz href="%s"�surrogateescapez>%s<rw )�sys�getfilesystemencodingr( �fsdecoder �TESTFN_UNDECODABLEr� rP r� r� rL r6 r� �platform�listdirr� r r� �urllib�parse�quoter� rJ �html�escape)r
�enc�filename�fr� �namer9 Z
quotednamer r r �test_undecodable_filenames s& $ z2SimpleHTTPServerTestCase.test_undecodable_filenamec C sE | j | j d � } | j | t j d | j �| j | j d � } | j | t j � | j | j d � } | j | t j � | j | j � } | j | t j � | j | j d � } | j | t j � | j | j d � } | j | t j � | j | j d � | j d � | j d � } | j | t j � | j d � } | j | t j � d } t
t j j
| j d � d
� � } | j | � Wd QRX| j | j d � } | j | t j | � t j d k rAt j � d k rAt j | j d � z- | j | j d � } | j | t j � Wd t j | j d � Xd S)Nz/testrw z/test/rW z/?hi=2z?hi=1�Locationz/?hi=1z/ThisDoesNotExistZThisDoesNotExists Dummy index file
z
index.htmlr� �posixr i� z/ThisDoesNotExistz/ThisDoesNotExist/)r6 r� r� r r� rw rF �MOVED_PERMANENTLYrX ro r� r( rP r� r� rL r� �geteuid�chmodr� )r
r� rw r� r r r r� � s: $!z!SimpleHTTPServerTestCase.test_getc C st | j | j d d d �} | j | t j � | j | j d � t t | j � � � | j | j d � d � d S)Nz/testr8 r� zcontent-lengthz content-typezapplication/octet-stream)
r6 r� r� r r� rX ro �strrv rw )r
r� r r r � test_head� s z"SimpleHTTPServerTestCase.test_headc C s| | j d d d �} | j | t j � | j d d d �} | j | t j � | j d d d �} | j | t j � d S)NrW r8 ZFOOZcustomZGETs)r6 r� r rZ )r
r� r r r �test_invalid_requests� s z.SimpleHTTPServerTestCase.test_invalid_requestsc C s | j | j d � } | j | t j d | j �| j | j d � } | j | t j � | j | j d � } | j | t j � | j | j � } | j | t j � | j | j d � } | j | t j � | j | j d � } | j | t j � | j | j d � | j d � d S) Nz/testrw z/test/rW z/?hi=2z?hi=1r� z/?hi=1)
r6 r� r� r r� rw rF r� rX ro )r
r� r r r � test_path_without_leading_slash� s z8SimpleHTTPServerTestCase.test_path_without_leading_slash)r r r r r r r. r0 r� r Zrequires_mac_ver�unittestZ
skipUnlessr� r� r� r� r� r� r r r r r� @ s
& r� 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 sB 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 QRXt j | j
d � t j j | j
d � | _ t | j d d
d �� } | j t | j � Wd QRXt j | j d � t j j | j
d � | _ t | j d d
d �� } | j t | j � Wd QRXt j | j d � t j j | j d � | _ t | j d d
d �� } | j t | j � Wd QRXt j | j d � t j j | j
d
� | _ t | j d d
d ��! } | j t! | j d f � Wd QRXt j | j d � t j" | j � d S)Nzcgi-binz child-dir�pythonzutf-8z0Python executable path is not encodable to utf-8znocgi.py�wi� zfile1.py�encodingzfile2.pyzfile3.pyzfile4.pyZ QUERY_STRING)#r& r. r( r� r� r� r� �
parent_dirrP r� �cgi_dir�
cgi_child_dir�mkdir�
nocgi_path�
file1_path�
file2_path�
file3_path�
file4_pathr Z can_symlink� pythonexe�symlinkr� �
executablerJ �UnicodeEncodeErrorr0 ZskipTestr� rL � cgi_file1r� � cgi_file2� cgi_file4r� )r
r� Zfile1Zfile2Zfile3Zfile4r r r r. � sT
!zCGIHTTPServerTestCase.setUpc C s� z� t j | j � | j t j k r5 t j | j � | j rN t j | j � | j rg t j | j � | j r� t j | j � | j
r� t j | j
� | j r� t j | j � t j | j
� t j | j � t j | j � Wd t j | � Xd S)N)r( r� r� r� r� r� �remover� r� r� r� r� �rmdirr� r� r� r& r0 )r
r r r r0 7 s$ z CGIHTTPServerTestCase.tearDownc 8 C s8 d d d t d t d d d d d d d d d d
d
d
d d d
d d d d d d d d d d d d d d d d d d d d d d d d t d d d d d! t d" d d# d i } x� | j � D]u \ } } t | t � r� t | t � r� | j | t j | � q� t j | � } | j | | d$ d% | | | f �q� Wd S)&Nr z//z..z/.//..rW z/\z//\z/.//zcgi-bin/file1.pyz/cgi-bin/file1.py�az//az/az./az./C:/z/C:/z/a/bz/a/b/z/a/b/.z /a/b/c/..z /a/b/c/../dz/a/b/dz/a/b/c/../d/e/../fz/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/../../../../f/..z /a/b/c/../d/e/../../../../f/../.r� z path = %r
Got: %r
Wanted: %r)
�
IndexError�items�
isinstance�type�
issubclass� Exception� assertRaisesr Z_url_collapse_pathrX )r
Z test_vectorsrP �expected�actualr r r �test_url_collapse_pathL sH z,CGIHTTPServerTestCase.test_url_collapse_pathc C sQ | j d � } | j | j � | j d � | j f d | j d t j f � d S)Nz/cgi-bin/file1.pyz Content-types Hello Worldz text/html)r6 rX r ro rY �linesepr r� )r
r[ r r r �test_headers_and_contentv s z.CGIHTTPServerTestCase.test_headers_and_contentc C s) | j d � } | j | j t j � d S)Nz)///////////nocgi.py/../cgi-bin/nothere.sh)r6 rX rY r rF )r
r[ r r r �test_issue19435| s z%CGIHTTPServerTestCase.test_issue19435c C si t j j d d d d d d i � } d d i } | j d d
| | � } | j | j � d | j � d S) NZspamrN Zeggsr� Zbaconi@� z Content-typez!application/x-www-form-urlencodedz/cgi-bin/file2.pyZPOSTs 1, python, 123456)r� r� � urlencoder6 rX r r� )r
�paramsr: r[ r r r � test_post� s
z CGIHTTPServerTestCase.test_postc C s3 | j d � } | j � | j | j t j � d S)Nz/cgi-bin/invalid)r6 r rX rY r rF )r
r[ r r r �test_invaliduri� s
z%CGIHTTPServerTestCase.test_invaliduric C ss d d t j d � i } | j d d d | �} | j d | j d t j f | j � | j d � | j f � d S)
Ns
Authorizations Basic s
username:passz/cgi-bin/file1.pyr1 r: s Hello Worldz text/htmlz Content-type)
�base64Z b64encoder6 rX r� r r� r ro rY )r
r: r[ r r r �test_authorization� s z(CGIHTTPServerTestCase.test_authorizationc C sQ | j d � } | j d | j d t j f | j � | j d � | j f � d S)Nzcgi-bin/file1.pys Hello Worldz text/htmlz Content-type)r6 rX r� r r� r ro rY )r
r[ r r r �test_no_leading_slash� s z+CGIHTTPServerTestCase.test_no_leading_slashc C s{ d } | t j d <| j d � } | j d | j d t j f | j � | j d � | j f � | j t j d | � d S)NzTest CGI ServerZSERVER_SOFTWAREz/cgi-bin/file1.pys Hello Worldz text/htmlz Content-type)
r( r) r6 rX r� r r� r ro rY )r
Z signaturer[ r r r � test_os_environ_is_not_altered� s
"z4CGIHTTPServerTestCase.test_os_environ_is_not_alteredc C sQ | j d � } | j d | j d t j f | j � | j d � | j f � d S)Nz/cgi-bin%2ffile1.pys Hello Worldz text/htmlz Content-type)r6 rX r� r r� r ro rY )r
r[ r r r �#test_urlquote_decoding_in_cgi_check� s z9CGIHTTPServerTestCase.test_urlquote_decoding_in_cgi_checkc C sQ | j d � } | j d | j d t j f | j � | j d � | j f � d S)Nz/cgi-bin/child-dir/file3.pys Hello Worldz text/htmlz Content-type)r6 rX r� r r� r ro rY )r
r[ r r r � test_nested_cgi_path_issue21323� s z5CGIHTTPServerTestCase.test_nested_cgi_path_issue21323c C sQ | j d � } | j d | j d t j f | j � | j d � | j f � d S)Nz/cgi-bin/file4.py?a=b?c=ds a=b?c=dz text/htmlz Content-type)r6 rX r� r r� r ro rY )r
r[ r r r �&test_query_with_multiple_question_mark� s z<CGIHTTPServerTestCase.test_query_with_multiple_question_markc C sQ | j d � } | j d | j d t j 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/htmlz Content-type)r6 rX r� r r� r ro rY )r
r[ r r r �"test_query_with_continuous_slashes� s
z8CGIHTTPServerTestCase.test_query_with_continuous_slashesN)r r r r r r r( r� rJ r. r0 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_calledrS )r
r r r r � s z!SocketlessRequestHandler.__init__c C sG d | _ | j t j � | j d d � | j � | j j d � d S)NTz Content-Typez text/htmls <html><body>Data</body></html>
)r r> r r� r@ rA rK rL )r
r r r r� � s
z SocketlessRequestHandler.do_GETc G s d S)Nr )r
r} r r r r r � s z$SocketlessRequestHandler.log_messageN)r r r r r� r r r r r r � s r c @ s e Z d Z d d � Z d S)�!RejectingSocketlessRequestHandlerc C s | j t j � d S)NF)rD r ZEXPECTATION_FAILED)r
r r r �handle_expect_100� s z3RejectingSocketlessRequestHandler.handle_expect_100N)r r r r r r r r r � 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 � s zAuditableBytesIO.__init__c C s | j j | � d S)N)r �append)r
rw r r r rL � s zAuditableBytesIO.writec C s d j | j � S)Nry )r� r )r
r r r �getData� s zAuditableBytesIO.getDatac C s
t | j � S)N)rv r )r
r r r � numWrites� s zAuditableBytesIO.numWritesN)r r r r rL r �propertyr r r r r r � s r c @ s e 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& d' � Z d( S))� BaseHTTPRequestHandlerTestCases HTTP/1.[0-9]+ 200 OKc C s t � | _ d S)N)r �handler)r
r r r r. � s z$BaseHTTPRequestHandlerTestCase.setUpc C sQ t | � } t � } | | j _ | | j _ | j j � | j d � | j � S)Nr )r r �rfilerK �handle_one_request�seek� readlines)r
�message�input�outputr r r �send_typical_request� s
z3BaseHTTPRequestHandlerTestCase.send_typical_requestc C s | j | j j � d S)N)rs r r )r
r r r �verify_get_called� 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>rN )r r r )rX �sum)r
r: r )r r �verify_expected_headers� s
z6BaseHTTPRequestHandlerTestCase.verify_expected_headersc C s# | j j | � } | j | � d S)N)�HTTPResponseMatch�searchr� )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 rN s <html><body>Data</body></html>
zGET / HTTP/1.1r1 rW zHTTP/1.1���r# )
r r" r r rX r � requestline�commandrP �request_version�assertSequenceEqualr: 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 rN s <html><body>Data</body></html>
zGET / HTTP/1.0r1 rW zHTTP/1.0r# r# )
r r" r r rX r r$ r% rP r& r' r: 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
rN r s <html><body>Data</body></html>
)r rX rv 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 rN s <html><body>Data</body></html>
zGET / HTTP/1.0r1 rW zHTTP/1.0�Expect� 100-continuer# r# �r, r- )r. )
r r" r r rX r r$ r% rP r& r' r: r� )r
r( r: 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
rN s
� s HTTP/1.1 200 OK
s <html><body>Data</body></html>
zGET / HTTP/1.1r1 rW zHTTP/1.1r, � 100-continuer# r# �r, r1 )r2 ) r rX r r r r$ r% rP r& r' r: r� )r
r( r: 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� r0 ) r r r r rK r& r$ r% rD rX 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 rN )
r r r r rK r& Zsend_response_onlyrX r rA )r
r r r r r r �+test_header_buffering_of_send_response_onlyK 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
rN ) r r r r rK r& r@ rX r rA r )r
r r r r r r �$test_header_buffering_of_send_headerY 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� �posrw r r r �_readAndReseekk s
z[BaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continue.<locals>._readAndReseeks( GET / HTTP/1.1
Expect: 100-continue
zHTTP/1.1ry s
r s HTTP/1.1 100 ContinuerN r0 s HTTP/1.1 200 OK) r r r rK r& r
ZassertNotEqualr� rX )r
r9 r r r( r r r �$test_header_unbuffered_when_continuei 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
rN 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 rX 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 rX 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 rX r< r r r$ )r
r( r r r �test_header_length� s
z1BaseHTTPRequestHandlerTestCase.test_header_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
s X-Foo: bar
�e s
r s HTTP/1.1 431 Too many headers
zGET / HTTP/1.1)r rX r< r r r$ )r
r( r r r �test_too_many_headers� s
z4BaseHTTPRequestHandlerTestCase.test_too_many_headersc 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
�iter�handler� �
StopIterationrD )r
r
r )rE r
r �test_close_connection� s
z4BaseHTTPRequestHandlerTestCase.test_close_connectionN)r r r �re�compiler r. r r r r" r) r* r+ r/ r3 r4 r5 r6 r: r= r? rA rC rI r r r r r
� s(
r
c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� SimpleHTTPRequestHandlerTestCasec C s: t j � | _ t j j | j d � | _ t � | _ d S)Nr� )r( r� �
translatedrP r� 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_pathrX rM )r
rP r r r �test_query_arguments� 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 rN rX rM )r
rP 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 QRXd S)NrP zc:c:c:foo/filenamez\c:../filenamezc:\c:..\foo/filenamezc:c:foo\c:c:bar/filename) r Z swap_attrr r( �ntpathr rN �replace�seprX rM )r
rP r r r �test_windows_colon� s z3SimpleHTTPRequestHandlerTestCase.test_windows_colonN)r r r r. rO rP rT r r r r rL � s rL c @ s e Z d Z d d � Z d S)� MiscTestCasec C s� g } d d d h } xf t t � D]X } | j d � s" | | k rF q" t t | � } t | d d � d k r" | j | � q" W| j t j | � d S)Nr� Z
nobody_uidr� �_r z http.server)r� r r �getattrr ZassertCountEqual�__all__)r
r� Z blacklistr� Z
module_objectr r r �test_all� s zMiscTestCase.test_allN)r r r rY r r r r rU � s rU c C sD t j � } z# t j t t t t t t t
� Wd t j | � Xd S)N) r( r� r Z run_unittestr� r
r; r� r� rL rU r� )�verboser� r r r � test_main� s r[ �__main__)1Z http.serverr r r r r2 r r r( r� rJ r� rQ r� Z urllib.parser� r� Z http.clientr� �ior r� r� r �
import_moduler
r r r ZTestCaser& r; r� r� r� r� r� ZskipIf�hasattrr� r� r r r r
rL rU r[ r r r r r �<module> sH " �$�
$ ��(