Your IP : 216.73.216.52
��Yf(� � @ s} d Z 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( � Z0 e1 d) k rye0 � 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)�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 �:/opt/alt/python35/lib64/python3.5/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 r1 G s zBaseTestCase.request)r r r r* r, r1 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/htmlr7 z
keep-alive)r9 r r: 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 t j � d S)N)r? 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)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< r5 �encode�wfile�write)r
r4 r r r �do_LATINONEHEADERn s
z8BaseHTTPServerTestCase.request_handler.do_LATINONEHEADERc C s$ | j t | j d d � � � d S)N� )r? �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)NrI )r? rJ rK )r
r r r �do_HEADy s z.BaseHTTPServerTestCase.request_handler.do_HEADN)r r r �protocol_version�default_request_versionr= r> r@ rB rC rD rH rL rM 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/ r0 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)Nr- �/)rP r1 r2 � 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
ZXYZBOGUSrR ) rP �
_http_vsn_str�
putrequest�
endheadersr2 rS rT r rU )r
rV 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)NZFUBARr- rR ) rP rX rY rZ r2 rS rT r � BAD_REQUEST)r
rV 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.9r- rR ) rP rX rY rZ r2 rS rT r r\ )r
rV 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 r- rR ) rP rX rY rZ r2 rS rT r rU )r
rV 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 �CUSTOMrR ) rP rX rY rZ r2 rS rT r r\ )r
rV 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.9r- rR )
rP Z _http_vsnrX rY rZ r2 rS rT r ZHTTP_VERSION_NOT_SUPPORTED)r
rV 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 ) rP rX rY rZ r2 rS rT r r\ )r
rV 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)Nr- rR r7 r8 ) rP rY � putheaderrZ r2 rS rT r rU )r
rV 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.1r- rR r7 z
keep-alive)
rP rX rY re rZ r2 rS rT r rU )r
rV 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�TESTrR )rP r1 r2 rS rT r r: )r
rV 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)NZKEEPrR r7 z
keep-aliverh )rP r1 r2 rS � getheaderZ
addCleanupr8 )r
rV 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)NZKEYERRORrR i� )rP r1 r2 rS rT )r
rV 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)Nr` rR i� )rP r1 r2 rS rT )r
rV 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 EXPLAINERRORrR i� zContent-Length)rP r1 r2 rS rT �
assertTruerJ rj )r
rV 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) NZLATINONEHEADERrR r5 zX-Special-Incomingu Ärger mit Unicodez X-Specialu Dängerous Mindzutf-8)rP r1 r2 rS rj r rE )r
rV 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)NZNOTFOUNDrR zContent-Length) rP r1 r2 rS rT r rA r rJ rj �len)r
rV �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_PROTOCOLSrP r1 �formatr2 rS rT rj r )r
� allow_transfer_encoding_codes�coderV rr 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-Encodingrt )r ru rv �OKr: rw rP r1 rx r2 rS rT rn rJ rj �assertInr )r
ry rz rV rr r r r �test_head_via_send_error s" " z/BaseHTTPServerTestCase.test_head_via_send_errorN)r r r r r r r* rW r[ r] r^ r_ ra rc rd rf rg ri rk rl rm ro rp rs r{ r r r r r r6 M s* . r6 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)r9 r r} r< )r
r r r �do_GET! s z4RequestHandlerLoggingTestCase.request_handler.do_GETc C s | j t j d � d S)NzFile not found)r? r rA )r
r r r �do_ERROR% s z6RequestHandlerLoggingTestCase.request_handler.do_ERRORN)r r r rN rO 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)Nr- rR z"GET / HTTP/1.1" 200 -
)r. r/ r0 r r rP rQ r �captured_stderrr1 r2 rn �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)NZERRORrR �
r z code 404, message File not foundrI z"ERROR / HTTP/1.1" 404 -)r. r/ r0 r r rP rQ r r� r1 r2 r� �splitrn 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!�dirrR �test�wb)r# r* r% �getcwd�cwd�tempfileZ
gettempdir�chdirrr �mkdtemp�tempdirrK �basename� tempdir_name�base_url�open�joinrG )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� �shutilZrmtreer� r# r, )r
r r r r, 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)zWDon't close reader yet so we can check if there was leftover
buffered inputN)�fpr )�reader�responser r �
close_conn[ s zDSimpleHTTPServerTestCase.check_status_and_reason.<locals>.close_conn�
r7 r8 � rt zConnection should be closed) Z _close_connr rn rS rT �assertIsNotNone�reason�version�msg�getr8 )r
r� rT rr r� r4 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� rR �darwinr� �errors�
surrogatepassz href="%s"�surrogateescapez>%s<rr )�sys�getfilesystemencodingr% �fsdecoder �TESTFN_UNDECODABLEr� rK r� r� rG r1 r� �platform�listdirr� r r} �urllib�parseZquoter~ rE �html�escape)r
�enc�filename�fr� �namer4 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/testrr z/test/rR z/?hi=2z?hi=1�Locationz/?hi=1z/ThisDoesNotExistZThisDoesNotExists Dummy index file
z
index.htmlr� �posixr i� z/ThisDoesNotExistz/ThisDoesNotExist/)r1 r� r� r r} rr rA �MOVED_PERMANENTLYrS rj r� r% rK r� r� rG r� �geteuid�chmodr� )r
r� rr 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/testr3 r| zcontent-lengthz content-typezapplication/octet-stream)
r1 r� r� r r} rS rj �strrq rr )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)NrR r3 ZFOOZcustomZGETs)r1 r� r rU )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/testrr z/test/rR z/?hi=2z?hi=1r� z/?hi=1)
r1 r� r� r r} rr rA r� rS rj )r
r� r r r � test_path_without_leading_slash� s z8SimpleHTTPServerTestCase.test_path_without_leading_slash)r r r r r r r* r, 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_dirrK r� �cgi_dir�
cgi_child_dir�mkdir�
nocgi_path�
file1_path�
file2_path�
file3_path�
file4_pathr Z can_symlink� pythonexe�symlinkr� �
executablerE �UnicodeEncodeErrorr, ZskipTestr� rG � 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# r, )r
r r r r, 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/.//..rR 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_pathrS )r
Z test_vectorsrK �expectedZactualr 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)r1 rS r rj rT �linesepr r} )r
rV 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)r1 rS rT r rA )r
rV 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) NZspamrI Zeggsr� Zbaconi@� z Content-typez!application/x-www-form-urlencodedz/cgi-bin/file2.pyZPOSTs 1, python, 123456)r� r� Z urlencoder1 rS r r� )r
Zparamsr5 rV 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)r1 r rS rT r rA )r
rV 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.pyr- r5 s Hello Worldz text/htmlz Content-type)
�base64Z b64encoder1 rS r� r r} r rj rT )r
r5 rV 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)r1 rS r� r r} r rj rT )r
rV 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& r1 rS r� r r} r rj rT )r
Z signaturerV 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)r1 rS r� r r} r rj rT )r
rV 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)r1 rS r� r r} r rj rT )r
rV 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)r1 rS r� r r} r rj rT )r
rV 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)r1 rS r� r r} r rj rT )r
rV r r r �"test_query_with_continuous_slashes� s
z8CGIHTTPServerTestCase.test_query_with_continuous_slashesN)r r r r r r r% r� rE 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_calledrN )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� r9 r r} r; r< rF rG )r
r r r r� � s
z SocketlessRequestHandler.do_GETc G s d S)Nr )r
rx 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)r? 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
rr r r r rG � s zAuditableBytesIO.writec C s d j | j � S)Nrt )r� r� )r
r r r �getData� s zAuditableBytesIO.getDatac C s
t | j � S)N)rq r� )r
r r r � numWrites� s zAuditableBytesIO.numWritesN)r r r r rG r� �propertyr� r r r r r� � 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' 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* � s z$BaseHTTPRequestHandlerTestCase.setUpc C sQ t | � } t � } | | j _ | | j _ | j j � | j d � | j � S)Nr )r r� �rfilerF �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)rn 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>rI )r r
r )rS �sum)r
r5 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 rI s <html><body>Data</body></html>
zGET / HTTP/1.1r- rR zHTTP/1.1���r )
r r r r rS r� � requestline�commandrK �request_version�assertSequenceEqualr5 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 rI s <html><body>Data</body></html>
zGET / HTTP/1.0r- rR zHTTP/1.0r r )
r r r r rS r� r r rK r r r5 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
rI r s <html><body>Data</body></html>
)r rS rq 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 rI s <html><body>Data</body></html>
zGET / HTTP/1.0r- rR zHTTP/1.0�Expect� 100-continuer r �r r! )r" )
r r r r rS r� r r rK r r r5 r� )r
r r5 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
rI s
� s HTTP/1.1 200 OK
s <html><body>Data</body></html>
zGET / HTTP/1.1r- rR zHTTP/1.1r � 100-continuer r �r r% )r& ) r rS r r r� r r rK r r r5 r� )r
r r5 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 rF r r r r? rS 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 rI )
r r� r� r rF r Zsend_response_onlyrS r� r< )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
rI ) r r� r� r rF r r; rS r� r< 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� �posrr r r r �_readAndReseekk s
z[BaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continue.<locals>._readAndReseeks( GET / HTTP/1.1
Expect: 100-continue
zHTTP/1.1rt s
r s HTTP/1.1 100 ContinuerI r$ s HTTP/1.1 200 OK) r r� r rF r r ZassertNotEqualr� rS )r
r- 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
rI 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 rS 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 rS r0 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 rS r0 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 rS r0 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 �iterZhandler� �
StopIterationr8 )r
r r )r9 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. r1 r3 r5 r7 r< r r r r r� � 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% r� �
translatedrK 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_pathrS rA )r
rK 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� rB rS rA )r
rK 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)NrK zc:c:c:foo/filenamez\c:../filenamezc:\c:..\foo/filenamezc:c:foo\c:c:bar/filename) r Z swap_attrr r% �ntpathr� rB �replace�seprS rA )r
rK r r r �test_windows_colon� s z3SimpleHTTPRequestHandlerTestCase.test_windows_colonN)r r r r= r* rC rD rH 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 } 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 rM r r r r rI � s rI 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� r6 r� r� r@ rI r� )�verboser� r r r � test_main� s rO �__main__)2r= Z http.serverr r r r r. r r r% r� r> r� rE r� Z urllib.parser� r� Z http.clientr� �ior r� r� r �
import_moduler
r r r ZTestCaser# r6 r� r� r� r� r� ZskipIf�hasattrr� r� r� r� r� r� r@ rI rO r r r r r �<module> sJ " �$�
$ ��(