Your IP : 216.73.216.213
3
� f=� � @ s� d Z ddlmZmZmZmZ ddlmZmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlZddlmZ ddlZddlmZ ejd�ZG dd � d �Z G d
d � d ej �Z G d d
� d
ej �Z G dd� de �Z!G dd� de �Z"G dd� de �Z#dZ$dZ%dZ&ej'e(e d��o6e j)� dkd�G dd� de ��Z*G dd � d e�Z+G d d � d e+�Z,G d d � d �Z-G d!d"� d"ej �Z.G d#d$� d$ej �Z/G d%d&� d&ej �Z0d*d'd(�Z1e2d)k�r�e1� dS )+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 Zd Zdd� Zddd�ZdS )�NoLogRequestHandlerc G s d S )N� )�self�argsr r �:/opt/alt/python36/lib64/python3.6/test/test_httpservers.py� log_message s z NoLogRequestHandler.log_messageNc C s dS )N� r )r
�nr r r �read! s zNoLogRequestHandler.read)N)�__name__�
__module__� __qualname__r r r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�TestServerThreadc C s t jj| � || _|| _d S )N)r
�Thread�__init__�request_handler� test_object)r
r r r r r r & s zTestServerThread.__init__c
C sZ t d| j�| _| jjj� \| j_| j_| jjj � d | _z| jj
d� W d | jj � X d 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� | j� d S )N)r Zshutdown�join)r
r r r �stop5 s
zTestServerThread.stopN)r r r r r! r# r r r r r % s
r c @ s, e Zd Zdd� Zdd� Zddi fdd�ZdS ) � BaseTestCasec C sD t j� | _t j� t_tj� | _t | | j
�| _ | j j � | jj
� d S )N)r Zthreading_setup�_threadsZEnvironmentVarGuard�os�environr
ZEventr r r �thread�start�wait)r
r r r �setUp; s
zBaseTestCase.setUpc C s* | j j� d | _ tjj� tj| j� d S )N)r( r# r& r' �__exit__r Zthreading_cleanupr% )r
r r r �tearDownC s
zBaseTestCase.tearDown�GETNc C s0 t jj| j| j�| _| jj||||� | jj� S )N)�http�client�HTTPConnectionr r Z
connection�request� getresponse)r
Zuri�method�body�headersr r r r2 I s zBaseTestCase.request)r r r r+ r- r2 r r r r r$ : s r$ c @ s� e Zd ZG dd� dee�Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S ),�BaseHTTPServerTestCasec @ s\ e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dS )z&BaseHTTPServerTestCase.request_handlerzHTTP/1.1c C s0 | j tj� | jdd� | jdd� | j� d S )Nz Content-Typez text/html�
Connection�close)�
send_responser �
NO_CONTENT� send_header� end_headers)r
r r r �do_TESTT s z.BaseHTTPServerTestCase.request_handler.do_TESTc C s0 | j tj� | jdd� | jdd� | j� d S )Nz Content-Typez text/htmlr8 z
keep-alive)r: r r; r< r= )r
r r r �do_KEEPZ 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 tj� d S )N)r@ r � NOT_FOUND)r
r r r � do_NOTFOUNDc s z2BaseHTTPServerTestCase.request_handler.do_NOTFOUNDc C s | j ddd� d S )Ni� z
Short Messagez This is a long
explanation)r@ )r
r r r �do_EXPLAINERRORf s z6BaseHTTPServerTestCase.request_handler.do_EXPLAINERRORc C s. | j d� | jdd� | jdd� | j� d S )Ni� z Content-Typez text/htmlr8 r9 )r: r< r= )r
r r r � do_CUSTOMj s
z0BaseHTTPServerTestCase.request_handler.do_CUSTOMc C sJ | j d� | jdd� | jdd� | j� | jd jd�}| jj|� d S )Ni� z X-Specialu Dängerous Mindr8 r9 zx-special-incomingzutf-8)r: r< r= r6 �encode�wfile�write)r
r5 r r r �do_LATINONEHEADERp s
z8BaseHTTPServerTestCase.request_handler.do_LATINONEHEADERc C s | j t| jdd � �� d S )N� )r@ �int�path)r
r r r �
do_SEND_ERRORx s z4BaseHTTPServerTestCase.request_handler.do_SEND_ERRORc C s | j t| jdd � �� d S )NrJ )r@ rK rL )r
r r r �do_HEAD{ s z.BaseHTTPServerTestCase.request_handler.do_HEADN)r r r �protocol_version�default_request_versionr> r? rA rC rD rE rI rM rN r r r r r P s r c C s, t j| � tjj| j| j�| _| jj� d S )N) r$ r+ r/ r0 r1 r r �con�connect)r
r r r r+ ~ s
z BaseHTTPServerTestCase.setUpc C s, | j jdd� | j j� }| j|jtj� d S )Nr. �/)rQ r2 r3 � assertEqual�statusr �NOT_IMPLEMENTED)r
�resr r r � test_command� s
z#BaseHTTPServerTestCase.test_commandc C s> d| j _| j jdd� | j j� | j j� }| j|jtj� d S )Nz HTTP/1.1
ZXYZBOGUSrS ) rQ �
_http_vsn_str�
putrequest�
endheadersr3 rT rU r rV )r
rW r r r �test_request_line_trimming� s
z1BaseHTTPServerTestCase.test_request_line_trimmingc C s> d| j _| j jdd� | j j� | j j� }| j|jtj� d S )NZFUBARr. rS ) rQ rY rZ r[ r3 rT rU r � BAD_REQUEST)r
rW r r r �test_version_bogus� s
z)BaseHTTPServerTestCase.test_version_bogusc C s> d| j _| j jdd� | j j� | j j� }| j|jtj� d S )Nz
HTTP/9.9.9r. rS ) rQ rY rZ r[ r3 rT rU r r] )r
rW r r r �test_version_digits� s
z*BaseHTTPServerTestCase.test_version_digitsc C s> d| j _| j jdd� | j j� | j j� }| j|jtj� d S )Nr r. rS ) rQ rY rZ r[ r3 rT rU r rV )r
rW r r r �test_version_none_get� s
z,BaseHTTPServerTestCase.test_version_none_getc C s> d| j _| j jdd� | j j� | j j� }| j|jtj� d S )Nr �CUSTOMrS ) rQ rY rZ r[ r3 rT rU r r] )r
rW r r r �test_version_none� s
z(BaseHTTPServerTestCase.test_version_nonec C sF d| j _d| j _| j jdd� | j j� | j j� }| j|jtj � d S )N�c zHTTP/9.9r. rS )
rQ Z _http_vsnrY rZ r[ r3 rT rU r ZHTTP_VERSION_NOT_SUPPORTED)r
rW r r r �test_version_invalid� s
z+BaseHTTPServerTestCase.test_version_invalidc C s> d| j _| j jdd� | j j� | j j� }| j|jtj� d S )Nr ) rQ rY rZ r[ r3 rT rU r r] )r
rW r r r �test_send_blank� s
z&BaseHTTPServerTestCase.test_send_blankc C sD | j jdd� | j jdd� | j j� | j j� }| j|jtj� d S )Nr. rS r8 r9 ) rQ rZ � putheaderr[ r3 rT rU r rV )r
rW r r r �test_header_close� s
z(BaseHTTPServerTestCase.test_header_closec C sL d| j _| j jdd� | j jdd� | j j� | j j� }| j|jtj � d S )NzHTTP/1.1r. rS r8 z
keep-alive)
rQ rY rZ rf r[ r3 rT rU r rV )r
rW r r r �test_header_keep_alive� s
z-BaseHTTPServerTestCase.test_header_keep_alivec C s, | j jdd� | j j� }| j|jtj� d S )N�TESTrS )rQ r2 r3 rT rU r r; )r
rW r r r � test_handler� s
z#BaseHTTPServerTestCase.test_handlerc C sJ | j jdd� | j j� }| j|jd�d� | j jdd� | j| j j� d S )NZKEEPrS r8 z
keep-aliveri )rQ r2 r3 rT � getheaderZ
addCleanupr9 )r
rW r r r � test_return_header_keep_alive� s
z4BaseHTTPServerTestCase.test_return_header_keep_alivec C s* | j jdd� | j j� }| j|jd� d S )NZKEYERRORrS i� )rQ r2 r3 rT rU )r
rW r r r �test_internal_key_error� s
z.BaseHTTPServerTestCase.test_internal_key_errorc C s* | j jdd� | j j� }| j|jd� d S )Nra rS i� )rQ r2 r3 rT rU )r
rW r r r �test_return_custom_status� s
z0BaseHTTPServerTestCase.test_return_custom_statusc C s> | j jdd� | j j� }| j|jd� | jt|jd��� d S )NZ EXPLAINERRORrS i� zContent-Length)rQ r2 r3 rT rU �
assertTruerK rk )r
rW r r r �test_return_explain_error� s
z0BaseHTTPServerTestCase.test_return_explain_errorc C sL | j jddddid� | j j� }| j|jd�d� | j|j� djd�� d S ) NZLATINONEHEADERrS zX-Special-Incomingu Ärger mit Unicode)r6 z X-Specialu Dängerous Mindzutf-8)rQ r2 r3 rT rk r rF )r
rW r r r �test_latin1_header� s
z)BaseHTTPServerTestCase.test_latin1_headerc C sN | j jdd� | j j� }| j|jtj� |j� }| jt|j d��t
|�� d S )NZNOTFOUNDrS zContent-Length) rQ r2 r3 rT rU r rB r rK rk �len)r
rW �datar r r �test_error_content_length� s
z0BaseHTTPServerTestCase.test_error_content_lengthc C s� t jt jf}x�t jt jt jt jt jfD ]�}| jjddj|�� | jj � }| j
||j � | j
d |j d�� | j
d |j d�� ||kr�| j
d |j d�� |j
� }| j
d|� q&W d S )NZ
SEND_ERRORz/{}zContent-Lengthz Content-TypezTransfer-Encoding� )r � NOT_MODIFIED�
RESET_CONTENTr; Z
PROCESSING�SWITCHING_PROTOCOLSrQ r2 �formatr3 rT rU rk r )r
� allow_transfer_encoding_codes�coderW rs r r r �test_send_error� s
z&BaseHTTPServerTestCase.test_send_errorc C s� t jt jf}x�t jt jt jt jt jfD ]�}| jjddj|�� | jj � }| j
||j � |t jkr�| j t
|jd��dk� | jd|jd�� n$| j
d |jd�� | j
d |jd�� ||kr�| j
d |jd�� |j� }| j
d|� q&W d S ) N�HEADz/{}zContent-Lengthr z text/htmlz Content-TypezTransfer-Encodingru )r rv rw �OKr; rx rQ r2 ry r3 rT rU ro rK rk �assertInr )r
rz r{ rW rs r r r �test_head_via_send_error s"
z/BaseHTTPServerTestCase.test_head_via_send_errorN)r r r r r r r+ rX r\ r^ r_ r` rb rd re rg rh rj rl rm rn rp rq rt r| r� r r r r r7 O s* . r7 c @ s, e Zd ZG dd� de�Zdd� Zdd� ZdS )� RequestHandlerLoggingTestCasec @ s$ e Zd ZdZdZdd� Zdd� ZdS )z-RequestHandlerLoggingTestCase.request_handlerzHTTP/1.1c C s | j tj� | j� d S )N)r: r r~ r= )r
r r r �do_GET# s z4RequestHandlerLoggingTestCase.request_handler.do_GETc C s | j tjd� d S )NzFile not found)r@ r rB )r
r r r �do_ERROR' s z6RequestHandlerLoggingTestCase.request_handler.do_ERRORN)r r r rO rP r� r� r r r r r s r c
C sb t jj| j| j�| _| jj� tj� � }| jj dd� | jj
� W d Q R X | j |j � j
d�� d S )Nr. rS z"GET / HTTP/1.1" 200 -
)r/ r0 r1 r r rQ rR r �captured_stderrr2 r3 ro �getvalue�endswith)r
�errr r r �test_get* s
z&RequestHandlerLoggingTestCase.test_getc
C s� t jj| j| j�| _| jj� tj� � }| jj dd� | jj
� W d Q R X |j � j d�}| j
|d jd�� | j
|d jd�� d S )NZERRORrS �
r z code 404, message File not foundrJ z"ERROR / HTTP/1.1" 404 -)r/ r0 r1 r r rQ rR r r� r2 r3 r� �splitro r� )r
r� �linesr r r �test_err5 s
z&RequestHandlerLoggingTestCase.test_errN)r r r r r r� r� r r r r r� s r� c @ s� e Zd ZG dd� dee�Zdd� Zdd� Zddd �Ze j
e j d
kd �e j
e j d kd
�e j
ejd�dd� ���Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) �SimpleHTTPServerTestCasec @ s e Zd ZdS )z(SimpleHTTPServerTestCase.request_handlerN)r r r r r r r r C s r c C s� t j| � tj� | _tj� }tj|� d| _tj |d�| _
tj j | j
�| _
d| j
| _ttj j| j
d�d��}|j| j� W d Q R X d S )Ns We are the knights who say Ni!)�dirrS �test�wb)r$ r+ r& �getcwd�cwd�tempfileZ
gettempdir�chdirrs �mkdtemp�tempdirrL �basename� tempdir_name�base_url�openr" rH )r
Z basetempdirZtempr r r r+ F s
z SimpleHTTPServerTestCase.setUpc C s@ z.t j| j� ytj| j� W n Y nX W d tj| � X d S )N)r& r� r� �shutilZrmtreer� r$ r- )r
r r r r- R s
z!SimpleHTTPServerTestCase.tearDownNc s� � �fdd�}d � |�_ �j� }| j�� | j�j|� | j�j� |rT| j||� | j�jd� | j�jj dd�d� | j� jd�dd� � j
� |S ) Nc s �j � d�_ dS )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�
r8 r9 � ru zConnection should be closed) Z _close_connr ro rT rU �assertIsNotNone�reason�version�msg�getr9 )r
r� rU rs r� r5 r )r� r� r �check_status_and_reason\ s
z0SimpleHTTPServerTestCase.check_status_and_reason�darwinz2undecodable name cannot always be decoded on macOSZwin32z+undecodable name cannot be decoded on win32z need support.TESTFN_UNDECODABLEc
C s t j� }tjtj�d }ttjj| j |�d��}|j
tj� W d Q R X | j | j d �}t j
dkr�x"tj| j �D ]}|dkrn|}P qnW | j|tj�}tjj|dd�}| jd| j|d �|� | jd
tj|d d � j|d �|� | j | j d | �}| j|tjtjd
� d S )Nz.txtr� rS r� r� �
surrogatepass)�errorsz href="%s"�surrogateescapez>%s<F)�quote)rs )�sys�getfilesystemencodingr& �fsdecoder �TESTFN_UNDECODABLEr� rL r" r� rH r2 r� �platform�listdirr� r r~ �urllib�parser� r rF �html�escape)r
�enc�filename�fr� �namer5 Z
quotednamer r r �test_undecodable_filenameu s&
z2SimpleHTTPServerTestCase.test_undecodable_filenamec C s� | j | jd �}| j|tj| jd� | j | jd �}| j|tj� | j | jd �}| j|tj� | j | j�}| j|tj� | j | jd �}| j|tj� | j | jd �}| j|tj� | j|j d�| jd � | j d �}| j|tj� | j d�}| j|tj� d }t
t j j
| jd �d
��}|j|� W d Q R X | j | jd �}| j|tj|� t jdk�r�t j� dk�r�t j| jd� z"| j | jd �}| j|tj� W d t j| jd� X d S )Nz/test)rs z/test/rS z/?hi=2z?hi=1�Locationz/?hi=1z/ThisDoesNotExistZThisDoesNotExists Dummy index file
z
index.htmlr� �posixr i� z/ThisDoesNotExistz/ThisDoesNotExist/)r2 r� r� r r~ rs rB �MOVED_PERMANENTLYrT rk r� r& rL r" r� rH r� �geteuid�chmodr� )r
r� rs r� r r r r� � s:
z!SimpleHTTPServerTestCase.test_getc C sT | j | jd dd�}| j|tj� | j|jd�tt| j ��� | j|jd�d� d S )Nz/testr} )r4 zcontent-lengthz content-typezapplication/octet-stream)
r2 r� r� r r~ rT rk �strrr rs )r
r� r r r � test_head� s z"SimpleHTTPServerTestCase.test_headc C sX | j ddd�}| j|tj� | j ddd�}| j|tj� | j ddd�}| j|tj� d S )NrS ZFOO)r4 ZcustomZGETs)r2 r� r rV )r
r� r r r �test_invalid_requests� s z.SimpleHTTPServerTestCase.test_invalid_requestsc C s� | j | jd �}| j|tj| jd� | j | jd �}| j|tj� | j | jd �}| j|tj� | j | j�}| j|tj� | j | jd �}| j|tj� | j | jd �}| j|tj� | j|j d�| jd � d S ) Nz/test)rs z/test/rS z/?hi=2z?hi=1r� z/?hi=1)
r2 r� r� r r~ rs rB r� rT rk )r
r� r r r � test_path_without_leading_slash� s z8SimpleHTTPServerTestCase.test_path_without_leading_slashc C s� d}t jj| j|�}yt|d�j� W n" tk
rH tjd| ��Y nX z,| j | j
d �}| j |t j
�}|jj� }W d t j|� X | j|� dtj|dd� }| j|j|�|� d S )Nz <test&>.txt�wz-Can not create file %s on current file systemrS z>%s<F)r� )r& rL r" r� r� r9 �OSError�unittestZSkipTestr2 r� r� r r~ r6 Zget_content_charset�unlinkr� r� r� r rF )r
r� Zfullpathr� r5 r� Z html_textr r r �test_html_escape_filename� s
z2SimpleHTTPServerTestCase.test_html_escape_filename)N)r r r r r r r+ r- r� r� �skipIfr� r� Z
skipUnlessr r� r� r� r� r� r� r� r r r r r� B 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 Zd ZG dd� dee�Zejjd�Zdd� Z dd� Z
dd � Z d
d � Z d d
� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )!�CGIHTTPServerTestCasec @ s e Zd ZdS )z%CGIHTTPServerTestCase.request_handlerN)r r r r r r r r s r �asciic / C sp t j| � tj� | _tj� | _tjj | jd�| _
tjj | j
d�| _ tj | j
� tj | j � d | _
d | _d | _d | _d | _tj� r�tjj | jd�| _tjtj| j� ntj| _y| jjd� W n& tk
r� | j� | jd� Y nX tjj | jd�| _
t | j
d��}|j t | j � W d Q R X tj | j
d� tjj | j
d �| _t | jddd
��}|j t | j � W d Q R X tj | jd� tjj | j
d �| _t | jddd
��}|j t | j � W d Q R X tj | jd� tjj | j d �| _t | jddd
��}|j t | j � W d Q R X tj | jd� tjj | j
d
�| _t | jddd
��}|j t!| jdf � W d Q R X tj | jd� tj"| 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& r� r� r� r� �
parent_dirrL r" �cgi_dir�
cgi_child_dir�mkdir�
nocgi_path�
file1_path�
file2_path�
file3_path�
file4_pathr Z can_symlink� pythonexe�symlinkr� �
executablerF �UnicodeEncodeErrorr- ZskipTestr� rH � cgi_file1r� � cgi_file2� cgi_file4r� )r
r� Zfile1Zfile2Zfile3Zfile4r r r r+ sT
zCGIHTTPServerTestCase.setUpc C s� z�t j| j� | jtjkr&t j| j� | jr8t j| j� | jrJt j| j� | j r\t j| j � | j
rnt j| j
� | j r�t j| j � t j | j
� t j | j� t j | j� W d tj| � X d S )N)r& r� r� r� r� r� �remover� r� r� r� r� �rmdirr� r� r� r$ r- )r
r r r r- Q s$ z CGIHTTPServerTestCase.tearDownc C s� dt t ddddddddddddddddd d
d
t d d t ddd
� }x^|j� D ]R\}}t|t�rvt|t�rv| j|tj|� qHtj|�}| j ||d|||f d� qHW d S )Nz//z//\z/cgi-bin/file1.pyz//az/C:/z/a/bz/a/b/z/a/b/dz/a/b/d/fz/a/b/fz/a/fz//f) r z..z/.//..rS z//z/\z/.//zcgi-bin/file1.pyz/cgi-bin/file1.py�az/az//az./az./C:/z/a/bz/a/b/z/a/b/.z /a/b/c/..z /a/b/c/../dz/a/b/c/../d/e/../fz/a/b/c/../d/e/../../fz /a/b/c/../d/e/.././././..//fz ../a/b/c/../d/e/.././././..//fz/a/b/c/../d/e/../../../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/../.z path = %r
Got: %r
Wanted: %r)r� )
�
IndexError�items�
isinstance�type�
issubclass� Exception� assertRaisesr Z_url_collapse_pathrT )r
Z test_vectorsrL �expectedZactualr r r �test_url_collapse_pathf sH
z,CGIHTTPServerTestCase.test_url_collapse_pathc C s: | j d�}| j|j� |jd�|jfd| j dtjf� d S )Nz/cgi-bin/file1.pyz Content-types Hello Worldz text/html)r2 rT r rk rU �linesepr r~ )r
rW r r r �test_headers_and_content� s
z.CGIHTTPServerTestCase.test_headers_and_contentc C s | j d�}| j|jtj� d S )Nz)///////////nocgi.py/../cgi-bin/nothere.sh)r2 rT rU r rB )r
rW r r r �test_issue19435� s
z%CGIHTTPServerTestCase.test_issue19435c C sF t jjdddd��}ddi}| jdd||�}| j|j� d | j � d S )
NrJ r� i@� )ZspamZeggsZbaconz Content-typez!application/x-www-form-urlencodedz/cgi-bin/file2.pyZPOSTs 1, python, 123456)r� r� Z urlencoder2 rT r r� )r
Zparamsr6 rW r r r � test_post� s
z CGIHTTPServerTestCase.test_postc C s& | j d�}|j� | j|jtj� d S )Nz/cgi-bin/invalid)r2 r rT rU r rB )r
rW r r r �test_invaliduri� s
z%CGIHTTPServerTestCase.test_invaliduric C sR ddt jd� i}| jdd|d�}| jd| j dtjf|j� |jd �|j f� d S )
Ns
Authorizations Basic s
username:passz/cgi-bin/file1.pyr. )r6 s Hello Worldz text/htmlz Content-type)
�base64Z b64encoder2 rT r� r r~ r rk rU )r
r6 rW r r r �test_authorization� s z(CGIHTTPServerTestCase.test_authorizationc C s: | j d�}| jd| j dtjf|j� |jd�|jf� d S )Nzcgi-bin/file1.pys Hello Worldz text/htmlz Content-type)r2 rT r� r r~ r rk rU )r
rW r r r �test_no_leading_slash� s
z+CGIHTTPServerTestCase.test_no_leading_slashc C sZ d}|t jd< | jd�}| jd| j dtjf|j� |jd�|j f� | jt jd |� d S )NzTest CGI ServerZSERVER_SOFTWAREz/cgi-bin/file1.pys Hello Worldz text/htmlz Content-type)
r& r' r2 rT r� r r~ r rk rU )r
Z signaturerW r r r � test_os_environ_is_not_altered� s
z4CGIHTTPServerTestCase.test_os_environ_is_not_alteredc C s: | j d�}| jd| j dtjf|j� |jd�|jf� d S )Nz/cgi-bin%2ffile1.pys Hello Worldz text/htmlz Content-type)r2 rT r� r r~ r rk rU )r
rW r r r �#test_urlquote_decoding_in_cgi_check� s
z9CGIHTTPServerTestCase.test_urlquote_decoding_in_cgi_checkc C s: | j d�}| jd| j dtjf|j� |jd�|jf� d S )Nz/cgi-bin/child-dir/file3.pys Hello Worldz text/htmlz Content-type)r2 rT r� r r~ r rk rU )r
rW r r r � test_nested_cgi_path_issue21323� s
z5CGIHTTPServerTestCase.test_nested_cgi_path_issue21323c C s: | j d�}| jd| j dtjf|j� |jd�|jf� d S )Nz/cgi-bin/file4.py?a=b?c=ds a=b?c=dz text/htmlz Content-type)r2 rT r� r r~ r rk rU )r
rW r r r �&test_query_with_multiple_question_mark� s
z<CGIHTTPServerTestCase.test_query_with_multiple_question_markc C s: | j d�}| jd| j dtjf|j� |jd�|jf� 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)r2 rT r� r r~ r rk rU )r
rW r r r �"test_query_with_continuous_slashes� s
z8CGIHTTPServerTestCase.test_query_with_continuous_slashesN)r r r r r r r& r� rF r+ r- r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� s :* r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�SocketlessRequestHandlerc C s d| _ d| _d S )NFzHTTP/1.1)�
get_calledrO )r
r r r r � s z!SocketlessRequestHandler.__init__c C s6 d| _ | jtj� | jdd� | j� | jjd� d S )NTz Content-Typez text/htmls <html><body>Data</body></html>
)r� r: r r~ r< r= rG rH )r
r r r r� � s
z SocketlessRequestHandler.do_GETc G s d S )Nr )r
ry 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 Zd Zdd� ZdS )�!RejectingSocketlessRequestHandlerc C s | j tj� dS )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 @ s0 e Zd Zdd� Zdd� Zdd� Zedd� �Zd 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
rs r r r rH � s zAuditableBytesIO.writec C s dj | j�S )Nru )r" r� )r
r r r �getData� s zAuditableBytesIO.getDatac C s
t | j�S )N)rr r� )r
r r r � numWrites� s zAuditableBytesIO.numWritesN)r r r r rH r� �propertyr r r r r r� � s r� c @ s� e Zd ZdZejd�Zdd� Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-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 s: t |�}t � }|| j_|| j_| jj� |jd� |j� S )Nr )r r �rfilerG �handle_one_request�seek� readlines)r
�message�input�outputr r r �send_typical_request s
z3BaseHTTPRequestHandlerTestCase.send_typical_requestc C s | j | jj� d S )N)ro r r� )r
r r r �verify_get_called s z0BaseHTTPRequestHandlerTestCase.verify_get_calledc s0 x*dD ]"� | j t� fdd�|D ��d� qW d S )N� Server: � Date: � Content-Type: c 3 s | ]}|j � �V qd S )N)�
startswith)�.0�h)� fieldNamer r � <genexpr> s zIBaseHTTPRequestHandlerTestCase.verify_expected_headers.<locals>.<genexpr>rJ )r
r r )rT �sum)r
r6 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|dd � � | j� | j|d
d� | j| jjd� | j| jjd� | j| jjd� | j| jj d� | j
| jj j � f � d S ) Ns GET / HTTP/1.1
r rJ s <html><body>Data</body></html>
zGET / HTTP/1.1r. rS zHTTP/1.1���r )
r r r r rT r � requestline�commandrL �request_version�assertSequenceEqualr6 r� )r
�resultr r r �
test_http_1_1 s
z,BaseHTTPRequestHandlerTestCase.test_http_1_1c C s� | j d�}| j|d � | j|dd � � | j� | j|d
d� | j| jjd� | j| jjd� | j| jjd� | j| jj d� | j
| jj j � f � d S ) Ns GET / HTTP/1.0
r rJ s <html><body>Data</body></html>
zGET / HTTP/1.0r. rS zHTTP/1.0r r )
r r r r rT r r r rL r r r6 r� )r
r r r r �
test_http_1_0) s
z,BaseHTTPRequestHandlerTestCase.test_http_1_0c C s6 | j d�}| jt|�d� | j|d d� | j� d S )Ns GET / HTTP/0.9
rJ r s <html><body>Data</body></html>
)r rT rr r )r
r r r r �
test_http_0_95 s
z,BaseHTTPRequestHandlerTestCase.test_http_0_9c C s� | j d�}| j|d � | j|dd � � | j� | j|d d� | j| jjd� | j| jjd� | j| jjd� | j| jj d� d}| j
| jj j � |� d S )Ns( GET / HTTP/1.0
Expect: 100-continue
r rJ s <html><body>Data</body></html>
zGET / HTTP/1.0r. rS zHTTP/1.0�Expect� 100-continuer r �r$ r% )r& )
r r r r rT r r r rL r r r6 r� )r
r r6 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|dd� � | j� | j|d d� | j| jjd � | j| jjd
� | j| jjd � | j| jjd � d}| j | jj
j � |� d S )Ns( GET / HTTP/1.1
Expect: 100-continue
r s HTTP/1.1 100 Continue
rJ s
� s HTTP/1.1 200 OK
s <html><body>Data</body></html>
zGET / HTTP/1.1r. rS zHTTP/1.1r$ � 100-continuer r �r$ r) )r* ) r rT r r r r r rL r r r6 r� )r
r r6 r r r �test_with_continue_1_1H s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_1c C sN t d�}t� }t� }||_||_d|_d|_d |_|jd� | j |j
d� d S )Ns GET / HTTP/1.1
zHTTP/1.1r i� r( ) r r� r� r rG r r r r@ rT r )r
r r
r r r r �#test_header_buffering_of_send_errorW s
zBBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_errorc C sX t d�}t� }t� }||_||_d|_|jd� | j|jd� |j � | j|jd� d S )Ns GET / HTTP/1.1
zHTTP/1.1i� r rJ )
r r� r� r rG r Zsend_response_onlyrT r r= )r
r r
r r r r �+test_header_buffering_of_send_response_onlye s
zJBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_response_onlyc C sv t d�}t� }t� }||_||_d|_|jdd� |jdd� | j|jd� |j � | j|j
� d� | j|jd� d S ) Ns GET / HTTP/1.1
zHTTP/1.1ZFooZfooZbarr s Foo: foo
bar: bar
rJ ) r r� r� r rG r r< rT r r= r� )r
r r
r r r r �$test_header_buffering_of_send_headers s zCBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_headerc C s� dd� }t d�}t � }|| j_|| j_d| j_| jj� | j||�d� ||�jd�}| j|d d� | j|d d� | j|d
d � d S ) Nc S s( | j � }| jd� | j� }| j|� |S )Nr )�tellr r )r� �posrs r r r �_readAndReseek� s
z[BaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continue.<locals>._readAndReseeks( GET / HTTP/1.1
Expect: 100-continue
zHTTP/1.1ru s
r s HTTP/1.1 100 ContinuerJ r( s HTTP/1.1 200 OK) r r r rG r r ZassertNotEqualr� rT )r
r1 r r
r r r r �$test_header_unbuffered_when_continue� s
zCBaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continuec C st | j }t� | _ | jd�}| j|d d� | j|dd� � | j| j j� | jtdd� |dd� D ��d� || _ d S ) Ns( GET / HTTP/1.1
Expect: 100-continue
r s! HTTP/1.1 417 Expectation Failed
rJ c s s | ]}|d kV qdS )s Connection: close
Nr )r �rr r r r � s zMBaseHTTPRequestHandlerTestCase.test_with_continue_rejected.<locals>.<genexpr>r r )r r� r rT r � assertFalser� r )r
Z
usual_handlerr r r r �test_with_continue_rejected� s
"z:BaseHTTPRequestHandlerTestCase.test_with_continue_rejectedc C sD | j ddd �}| j|d d� | j| jj� | j| jjt� d S )Ns GET � xi r s# HTTP/1.1 414 Request-URI Too Long
)r rT r4 r r� ZassertIsInstancer r� )r
r r r r �test_request_length� s z2BaseHTTPRequestHandlerTestCase.test_request_lengthc C sH | j ddd d �}| j|d d� | j| jj� | j| jjd� d S )Ns GET / HTTP/1.1
X-Foo: bar� ri s
r s HTTP/1.1 431 Line too long
zGET / HTTP/1.1)r rT r4 r r� r )r
r r r r �test_header_length� s
z1BaseHTTPRequestHandlerTestCase.test_header_lengthc C sH | j ddd d �}| j|d d� | j| jj� | j| jjd� 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 rT r4 r r� r )r
r r r r �test_too_many_headers� s
z4BaseHTTPRequestHandlerTestCase.test_too_many_headersc C s8 | j d�}dj|�}d}| jtj|dd�jd�|� d S )Ns* <script>alert("hello")</script> / HTTP/1.1ru z <script>alert("hello")</script>F)r� r� )r r" r r� r� rF )r
r �textr r r �test_html_escape_on_error� s
z8BaseHTTPRequestHandlerTestCase.test_html_escape_on_errorc sZ � �fdd�}|�j _td�� �j j� �jtt� � td�� �j j� �jtt� � 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_connectionc C sd t j � }t j|�\ }}}}}}}} }
d| jj| || jj| ||||f } | j| jj|d�| � d S )Nz#%s, %02d %3s %4d %02d:%02d:%02d GMT)Z timestamp)�timeZgmtimer Z weekdaynameZ monthnamerT Zdate_time_string) r
ZnowZyearZmonthZdayZhhZmmZssZwd�y�zr� r r r �test_date_time_string� s
z4BaseHTTPRequestHandlerTestCase.test_date_time_stringN) r r r �__doc__�re�compiler r+ r r r r r! r" r# r' r+ r, r- r. r2 r5 r7 r9 r; r= rB rF r r r r r � s.
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � SimpleHTTPRequestHandlerTestCasez Test url parsing c C s( t j� | _t jj| jd�| _t� | _d S )Nr� )r& r� �
translatedrL r" r� r )r
r r r r+ � s
z&SimpleHTTPRequestHandlerTestCase.setUpc C sR | j jd�}| j|| j� | j jd�}| j|| j� | j jd�}| j|| j� d S )Nz /filenamez/filename?foo=barz/filename?a=b&spam=eggs#zot)r �translate_pathrT rK )r
rL r r r �test_query_arguments� s z5SimpleHTTPRequestHandlerTestCase.test_query_argumentsc C s8 | j jd�}| j|| j� | j jd�}| j|| j� d S )Nz
//filenamez//filename?foo=bar)r rL rT rK )r
rL r r r � test_start_with_double_slash� s z=SimpleHTTPRequestHandlerTestCase.test_start_with_double_slashc
C s� t jtjdt��� | jjd�}|jtjtj�}| j || j
� | jjd�}|jtjtj�}| j || j
� | jjd�}|jtjtj�}| j || j
� | jjd�}|jtjtj�}| j || j
� W d Q R X d S )NrL zc:c:c:foo/filenamez\c:../filenamezc:\c:..\foo/filenamezc:c:foo\c:c:bar/filename) r Z swap_attrr r& �ntpathr rL �replace�seprT rK )r
rL r r r �test_windows_colon� s z3SimpleHTTPRequestHandlerTestCase.test_windows_colonN)r r r rG r+ rM rN rR r r r r rJ � s
rJ c @ s e Zd Zdd� ZdS )� MiscTestCasec C sj g }dddh}xHt t�D ]<}|jd�s||kr0qtt|�}t|dd �dkr|j|� qW | jtj|� 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 rW r r r r rS s rS c C s4 t j� }ztjtttttt t
� W d t j |� X d S )N) r& r� r Z run_unittestr� r r7 r� r� rJ rS r� )�verboser� r r r � test_main s
rY �__main__)N)3rG Z http.serverr r r r r/ r r r& r� rH r� rO r� Z urllib.parser� r� Z http.clientr� rC �ior r� r� r �
import_moduler
r r r ZTestCaser$ r7 r� r� r� r� r� r� �hasattrr� r� r� r� r� r rJ rS rY r r r r r �<module> sT
P$ 5
I b(