File: //usr/local/lib/python3.7/test/__pycache__/test_httpservers.cpython-37.pyc
B
��gM� � @ s� d Z ddlmZmZmZmZ ddlmZmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m Z ddlZddl m Z G 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*e�+e,e d��oXe �-� dkd�G dd� de$��Z.G dd � d e�Z/G d d � d e/�Z0G d d � d �Z1G d!d"� d"ej#�Z2G d#d$� d$ej#�Z3G d%d&� d&ej#�Z4d*d'd(�Z5e6d)k�r�e5� 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)�mock)�BytesIO)�supportc @ s e Zd Zdd� Zddd�ZdS )�NoLogRequestHandlerc G s d S )N� )�self�argsr r �1/usr/local/lib/python3.7/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 j�| � || _|| _d S )N)� threading�Thread�__init__�request_handler� test_object)r
r r r r r r + s zTestServerThread.__init__c C sZ t d| j�| _| jj�� \| j_| j_| jj� � d | _z| j�
d� W d | j� � X d S )N)Z localhostr g�������?) r r r ZsocketZ getsocknamer �HOST�PORT�server_started�setZ
serve_foreverZ server_close)r
r r r �run0 s zTestServerThread.runc C s | j �� | �� d S )N)r Zshutdown�join)r
r r r �stop: 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 �� | _t �� t_t�� | _t | | j
�| _ | j � � | j�
� 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 �� d | _ tj�� tj| j� d S )N)r( r# r& r' �__exit__r
Zthreading_cleanupr% )r
r r r �tearDownH s
zBaseTestCase.tearDown�GETNc C s0 t j�| j| j�| _| j�||||� | j�� S )N)�http�client�HTTPConnectionr r Z
connection�request� getresponse)r
Zuri�method�body�headersr r r r2 N 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 | � tj� | �dd� | �dd� | �� d S )Nz Content-Typez text/html�
Connection�close)�
send_responser �
NO_CONTENT� send_header� end_headers)r
r r r �do_TESTY s z.BaseHTTPServerTestCase.request_handler.do_TESTc C s0 | � tj� | �dd� | �dd� | �� d S )Nz Content-Typez text/htmlr8 z
keep-alive)r: r r; r< r= )r
r r r �do_KEEP_ s z.BaseHTTPServerTestCase.request_handler.do_KEEPc C s | � d� d S )Ni� )�
send_error)r
r r r � do_KEYERRORe s z2BaseHTTPServerTestCase.request_handler.do_KEYERRORc C s | � tj� d S )N)r@ r � NOT_FOUND)r
r r r � do_NOTFOUNDh s z2BaseHTTPServerTestCase.request_handler.do_NOTFOUNDc C s | � ddd� d S )Ni� z
Short Messagez This is a long
explanation)r@ )r
r r r �do_EXPLAINERRORk s z6BaseHTTPServerTestCase.request_handler.do_EXPLAINERRORc C s. | � d� | �dd� | �dd� | �� d S )Ni� z Content-Typez text/htmlr8 r9 )r: r< r= )r
r r r � do_CUSTOMo s
z0BaseHTTPServerTestCase.request_handler.do_CUSTOMc C sJ | � d� | �dd� | �dd� | �� | jd �d�}| j�|� 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_LATINONEHEADERu s
z8BaseHTTPServerTestCase.request_handler.do_LATINONEHEADERc C s | � t| jdd � �� d S )N� )r@ �int�path)r
r r r �
do_SEND_ERROR} s z4BaseHTTPServerTestCase.request_handler.do_SEND_ERRORc C s | � 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 U s r c C s, t �| � tj�| j| j�| _| j�� d S )N) r$ r+ r/ r0 r1 r r �con�connect)r
r r r r+ � s
z BaseHTTPServerTestCase.setUpc C s, | j �dd� | 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 �dd� | 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 �dd� | 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 �dd� | 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 �dd� | 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 �dd� | 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 �dd� | 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 �dd� | 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 �dd� | j �dd� | 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 �dd� | j �dd� | 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 �dd� | 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 �dd� | j �� }| �|�d�d� | j �dd� | �| 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 �dd� | 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 �dd� | 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 �dd� | j �� }| �|jd� | �t|�d��� 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 �� }| �|�d�d� | �|�� d�d�� 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 �dd� | j �� }| �|jtj� |�� }| �t|� 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 ]�}| j�dd�|�� | j� � }| �
||j � | �
d |� d�� | �
d |� d�� ||kr�| �
d |� d�� |�
� }| �
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 ]�}| j�dd�|�� | j� � }| �
||j � |t jkr�| � t
|�d��dk� | �d|�d�� n$| �
d |�d�� | �
d |�d�� ||kr�| �
d |�d�� |�� }| �
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 T 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 | � tj� | �� d S )N)r: r r~ r= )r
r r r �do_GET( s z4RequestHandlerLoggingTestCase.request_handler.do_GETc C s | � 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 j�| j| j�| _| j�� t�� � }| j� dd� | j�
� W d Q R X | � |� � �
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 j�| j| j�| _| j�� t�� � }| j� dd� | j�
� W d Q R X |� � � d�}| �
|d �d�� | �
|d �d�� 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_err: 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� Zd#dd �Ze �
e j d
kd �e �
e j d kd
�e �
ejd�dd� ���Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZdS )$�SimpleHTTPServerTestCasec @ s e Zd ZdS )z(SimpleHTTPServerTestCase.request_handlerN)r r r r r r r r H s r c C s� t �| � t�� | _t�� }t�|� d| _tj |d�| _
tj � | j
�| _
d| j
| _tj �| j
d�}t|d��}|�| j� |�� W d Q R X t�|�j}tj�|tjj�}|jdd�| _tj j |� � dd �| _ d S )
Ns We are the knights who say Ni!)�dirrS �test�wbr )Z microsecondT)�usegmt) r$ r+ r& �getcwd�cwd�tempfileZ
gettempdir�chdirrs �mkdtemp�tempdirrL �basename� tempdir_name�base_urlr" �openrH �flush�stat�st_mtime�datetimeZ
fromtimestampZtimezoneZutc�replace�last_modif_datetime�email�utilsZ
formatdate� timestamp�last_modif_header)r
Z basetempdirZtempnameZtemp�mtimeZ
last_modifr r r r+ K s$
z SimpleHTTPServerTestCase.setUpc C s@ z.t �| j� yt�| j� W n Y nX W d t�| � X d S )N)r& r� r� �shutilZrmtreer� r$ r- )r
r r r r- ` s
z!SimpleHTTPServerTestCase.tearDownNc s� � �fdd�}d � |�_ ��� }| ��� | ��j|� | ��j� |rT| �||� | ��jd� | ��j� dd�d� | �� �d�dd� � �
� |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_connk 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_reasonj s
z0SimpleHTTPServerTestCase.check_status_and_reason�darwinz2undecodable name cannot always be decoded on macOS�win32z+undecodable name cannot be decoded on win32z need support.TESTFN_UNDECODABLEc C s t �� }t�tj�d }ttj�| j |�d��}|�
tj� W d Q R X | � | j d �}t j
dkr�x"t�| j �D ]}|dkrn|}P qnW | �|tj�}tjj|dd�}| �d| �|d �|� | �d
tj|d d � �|d �|� | � | 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_filename� s&
z2SimpleHTTPServerTestCase.test_undecodable_filenamec
C s� | � | jd �}| j|tj| jd� | � | jd �}| �|tj� | � | jd �}| �|tj� | � | j�}| �|tj� | � | jd �}| �|tj� | � | jd �}| �|tj� | �|� d�| jd � | � d �}| �|tj� | � d
�}| �|tj� d }t
t j �
| jd �d
��}|�|� W d Q R X | � | jd �}| �|tj|� t jdk�r�t �� dk�r�t �| jd� z"| � | jd �}| �|tj� W d t �| jd� X d S )Nz/test)rs z/test/rS z/?hi=2z?hi=1�Locationz/?hi=1z/ThisDoesNotExistz/ThisDoesNotExist/s Dummy index file
z
index.htmlr� �posixr i� )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�}| �|tj� | �|�d�tt| j ��� | �|�d�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 s� t j�� }| j|d< | j| jd |d�}| �|tj� | j t
j dd� }t j�� }t j j
|dd�|d< | j| jd |d�}| �|tj� dS ) z�Check that when a request to /test is sent with the request header
If-Modified-Since set to date of last modification, the server returns
status code 304, not 200
zIf-Modified-Sincez/test)r6 rJ )ZhoursT)r� N)r� �message�Messager� r2 r� r� r rv r� r� � timedeltar� �format_datetime)r
r6 r� Znew_dtr r r �test_browser_cache� s
z+SimpleHTTPServerTestCase.test_browser_cachec C sZ | j }|tjdd� }tj�� }tjj|dd�|d< | j| j d |d�}| �
|t j � d S )Nim )ZdaysT)r� zIf-Modified-Sincez/test)r6 )
r� r� r� r� r� r� r� r� r2 r� r� r r~ )r
ZdtZold_dtr6 r� r r r � test_browser_cache_file_changed� s
z8SimpleHTTPServerTestCase.test_browser_cache_file_changedc C sB t j�� }| j|d< d|d< | j| jd |d�}| �|tj� d S )NzIf-Modified-Since�*z
If-None-Matchz/test)r6 ) r� r� r� r� r2 r� r� r r~ )r
r6 r� r r r �,test_browser_cache_with_If_None_Match_header� s
zESimpleHTTPServerTestCase.test_browser_cache_with_If_None_Match_headerc C sX | j ddd�}| �|tj� | j ddd�}| �|tj� | j ddd�}| �|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@ | � | jd �}| j|tj| jd� |jd }| �|| j� dS )z�Checks that the datetime returned in Last-Modified response header
is the actual datetime of last modification, rounded to the second
z/test)rs z
Last-modifiedN) r2 r� r� r r~ rs r6 rT r� )r
r� r� r r r �test_last_modified� s
z+SimpleHTTPServerTestCase.test_last_modifiedc C s� | � | jd �}| j|tj| jd� | � | jd �}| �|tj� | � | jd �}| �|tj� | � | j�}| �|tj� | � | jd �}| �|tj� | � | jd �}| �|tj� | �|� 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 j�| j|�}yt|d��� W n" tk
rH t�d| ��Y nX z,| � | j
d �}| � |t j
�}|j�� }W d t �|� X | �|� dtj|dd� }| �|�|�|� 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 r r r r� G 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�Zej�d�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 O s r �asciic C sp t �| � t�� | _t�� | _tj� | jd�| _
tj� | j
d�| _ t� | j
� t� | j � d | _
d | _d | _d | _d | _t�� r�tj� | jd�| _t�tj| j� ntj| _y| j�d� W n& tk
r� | �� | �d� Y nX tj� | jd�| _
t | j
d��}|� t | j � W d Q R X t� | j
d� tj� | j
d �| _t | jddd
��}|� t | j � W d Q R X t� | jd� tj� | j
d �| _t | jddd
��}|� t | j � W d Q R X t� | jd� tj� | j d �| _t | jddd
��}|� t | j � W d Q R X t� | jd� tj� | j
d
�| _t | jddd
��}|� t!| jdf � W d Q R X t� | jd� t�"| 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+ T sT
zCGIHTTPServerTestCase.setUpc C s� z�t �| j� | jtjkr&t �| j� | jr8t �| j� | jrJt �| j� | j r\t �| j � | j
rnt �| j
� | j r�t �| j � t � | j
� t � | j� t � | j� W d t�| � 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- � s$ z CGIHTTPServerTestCase.tearDownc C s� dt t ddddddddddddddddd d
d
t d d t ddd
� }x^|�� D ]R\}}t|t�rvt|t�rv| �|tj|� qHt�|�}| 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 �expected�actualr r r �test_url_collapse_path� sH
z,CGIHTTPServerTestCase.test_url_collapse_pathc C s: | � d�}| �|�� |�d�|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 | � d�}| �|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 j�dddd��}ddi}| �dd||�}| �|�� 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
�paramsr6 rW r r r � test_post� s
z CGIHTTPServerTestCase.test_postc C s& | � d�}|�� | �|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 �d� i}| jdd|d�}| �d| j dtjf|�� |�d �|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: | � d�}| �d| j dtjf|�� |�d�|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< | �d�}| �d| j dtjf|�� |�d�|j f� | �t 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: | � d�}| �d| j dtjf|�� |�d�|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: | � d�}| �d| j dtjf|�� |�d�|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: | � d�}| �d| j dtjf|�� |�d�|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: | � d�}| �d| j dtjf|�� |�d�|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� L s :* r� c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�SocketlessRequestHandlerc s2 t �� }t� |j_t� �|d d � d| _d| _d S )NFzHTTP/1.1) r ZMockr ZmakefileZ return_value�superr �
get_calledrO )r
r �kwargsr2 )� __class__r r r s
z!SocketlessRequestHandler.__init__c C s6 d| _ | �tj� | �dd� | �� | j�d� 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_message)r r r r r� r �
__classcell__r r )r r r s r c @ s e Zd Zdd� ZdS )�!RejectingSocketlessRequestHandlerc C s | � 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 1 s zAuditableBytesIO.__init__c C s | j �|� d S )N)r �append)r
rs r r r rH 4 s zAuditableBytesIO.writec C s d� | j�S )Nru )r" r )r
r r r �getData7 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 ZdZe�d�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-d.� Z d/S )0� 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+ G s z$BaseHTTPRequestHandlerTestCase.setUpc C s: t |�}t � }|| j_|| j_| j�� |�d� |�� S )Nr )r r �rfilerG �handle_one_request�seek� readlines)r
r� �input�outputr r r �send_typical_requestJ s
z3BaseHTTPRequestHandlerTestCase.send_typical_requestc C s | � | jj� d S )N)ro r r )r
r r r �verify_get_calledS s z0BaseHTTPRequestHandlerTestCase.verify_get_calledc s0 x*dD ]"� | � t� fdd�|D ��d� qW d S )N)s Server: s Date: s Content-Type: c 3 s | ]}|� � �V qd S )N)�
startswith)�.0�h)� fieldNamer r � <genexpr>X s zIBaseHTTPRequestHandlerTestCase.verify_expected_headers.<locals>.<genexpr>rJ )rT �sum)r
r6 r )r+ r �verify_expected_headersV s
z6BaseHTTPRequestHandlerTestCase.verify_expected_headersc C s | j �|�}| �|� d S )N)�HTTPResponseMatch�searchr� )r
r� �matchr r r �verify_http_server_responseZ s z:BaseHTTPRequestHandlerTestCase.verify_http_server_responsec C s� | � d�}| �|d � | �|dd� � | �� | �|d d� | �| jjd� | �| jjd� | �| jjd� | �| jj d � | �
| jj � � d
� d S ) Ns GET / HTTP/1.1
r rJ ���s <html><body>Data</body></html>
zGET / HTTP/1.1r. rS zHTTP/1.1r )
r& r2 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� | � d�}| �|d � | �|dd� � | �� | �|d d� | �| jjd� | �| jjd� | �| jjd� | �| jj d � | �
| jj � � d
� d S ) Ns GET / HTTP/1.0
r rJ r3 s <html><body>Data</body></html>
zGET / HTTP/1.0r. rS zHTTP/1.0r )
r& r2 r. r' rT r r4 r5 rL r6 r7 r6 r� )r
r8 r r r �
test_http_1_0j s
z,BaseHTTPRequestHandlerTestCase.test_http_1_0c C s6 | � d�}| �t|�d� | �|d d� | �� d S )Ns GET / HTTP/0.9
rJ r s <html><body>Data</body></html>
)r& rT rr r' )r
r8 r r r �
test_http_0_9v s
z,BaseHTTPRequestHandlerTestCase.test_http_0_9c C sH | � d�}| �|d �d�� | �|d|�d�� � | �| jj� d S )Ns) GET /spaced out HTTP/1.1
Host: dummy
r s
HTTP/1.1 400 rJ s
)r& ro r( r. �index� assertFalser r )r
r8 r r r �test_extra_space| s
z/BaseHTTPRequestHandlerTestCase.test_extra_spacec C s� | � d�}| �|d � | �|dd� � | �� | �|d d� | �| jjd� | �| jjd� | �| jjd� | �| jj d � d
}| �
| jj � � |� d S ) Ns( GET / HTTP/1.0
Expect: 100-continue
r rJ r3 s <html><body>Data</body></html>
zGET / HTTP/1.0r. rS zHTTP/1.0))�Expectz 100-continue)
r& r2 r. r' rT r r4 r5 rL r6 r7 r6 r� )r
r8 r6 r r r �test_with_continue_1_0� s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_0c C s� | � d�}| �|d d� | �|d d� | �|d d� | �|dd� � | �� | �|d d � | �| jjd
� | �| jjd � | �| jjd � | �| jjd
� d}| � | jj
� � |� 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
r3 s <html><body>Data</body></html>
zGET / HTTP/1.1r. rS zHTTP/1.1))r? z 100-continue) r& rT r. r' r r4 r5 rL r6 r7 r6 r� )r
r8 r6 r r r �test_with_continue_1_1� s
z5BaseHTTPRequestHandlerTestCase.test_with_continue_1_1c C sN t d�}t� }t� }||_||_d|_d|_d |_|�d� | � |j
d� d S )Ns GET / HTTP/1.1
zHTTP/1.1r i� rA ) r r r r rG r6 r4 r5 r@ rT r )r
r$ r% r r r r �#test_header_buffering_of_send_error� s
zBBaseHTTPRequestHandlerTestCase.test_header_buffering_of_send_errorc C sX t d�}t� }t� }||_||_d|_|�d� | �|jd� |� � | �|jd� d S )Ns GET / HTTP/1.1
zHTTP/1.1i� r rJ )
r r r r rG r6 Zsend_response_onlyrT 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 sv t d�}t� }t� }||_||_d|_|�dd� |�dd� | �|jd� |� � | �|�
� d� | �|jd� d S ) Ns GET / HTTP/1.1
zHTTP/1.1ZFooZfooZbarr s Foo: foo
bar: bar
rJ ) r r r r rG r6 r< rT 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� dd� }t d�}t � }|| j_|| j_d| j_| j�� | �||�d� ||��d�}| �|d d� | �|d d� | �|d
d � d S ) Nc S s( | � � }| �d� | �� }| �|� |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 rA s HTTP/1.1 200 OK) r r r rG r6 r! ZassertNotEqualr� rT )r
rH r$ r% r8 r r r �$test_header_unbuffered_when_continue� s
zCBaseHTTPRequestHandlerTestCase.test_header_unbuffered_when_continuec C st | j }t� | _ | �d�}| �|d d� | �|dd� � | �| j j� | �tdd� |dd� D ��d� || _ d S )Ns( GET / HTTP/1.1
Expect: 100-continue
r s! HTTP/1.1 417 Expectation Failed
rJ r3 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& rT r. r= r r- )r
Z
usual_handlerr8 r r r �test_with_continue_rejected� s
"z:BaseHTTPRequestHandlerTestCase.test_with_continue_rejectedc C sD | � ddd �}| �|d d� | �| jj� | �| jjt� d S )Ns GET � xi r s# HTTP/1.1 414 Request-URI Too Long
)r& rT r= r r ZassertIsInstancer4 r� )r
r8 r r r �test_request_length� s z2BaseHTTPRequestHandlerTestCase.test_request_lengthc C sH | � ddd d �}| �|d d� | �| jj� | �| 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 r= r r r4 )r
r8 r r r �test_header_length� s
z1BaseHTTPRequestHandlerTestCase.test_header_lengthc C s< | � d�}| �|d d� | �| jj� | �| jjd� d S )Ns� GET / HTTP/1.1
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
X-Foo: bar
r s HTTP/1.1 431 Too many headers
zGET / HTTP/1.1)r& rT r= r r r4 )r
r8 r r r �test_too_many_headers s
z4BaseHTTPRequestHandlerTestCase.test_too_many_headersc C s8 | � d�}d�|�}d}| �tj|dd��d�|� 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
r8 �textr r r �test_html_escape_on_error s
z8BaseHTTPRequestHandlerTestCase.test_html_escape_on_errorc sZ � �fdd�}|�j _td�� �j �� ��tt� � td�� �j �� ��tt� � 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_request)T)FFT)r r! �iterZhandler� �
StopIterationrS )r
r! r )rT r
r �test_close_connection s
z4BaseHTTPRequestHandlerTestCase.test_close_connectionc C sd t � � }t �|�\ }}}}}}}} }
d| jj| || jj| ||||f } | �| jj|d�| � d S )Nz#%s, %02d %3s %4d %02d:%02d:%02d GMT)r� )�timeZgmtimer Z weekdaynameZ monthnamerT Zdate_time_string) r
ZnowZyearZmonthZdayZhhZmm�ssZwd�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. r2 r9 r: r; r> r@ rB rC rD rE rI rK rM rO rP rR rW r\ r r r r r ? s0
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � SimpleHTTPRequestHandlerTestCasez Test url parsing c C s( t �� | _t j�| jd�| _t� | _d S )Nr� )r& r� �
translatedrL r" r r )r
r r r r+ , s
z&SimpleHTTPRequestHandlerTestCase.setUpc C sR | j �d�}| �|| j� | j �d�}| �|| j� | j �d�}| �|| j� d S )Nz /filenamez/filename?foo=barz/filename?a=b&spam=eggs#zot)r �translate_pathrT ra )r
rL r r r �test_query_arguments1 s z5SimpleHTTPRequestHandlerTestCase.test_query_argumentsc C s8 | j �d�}| �|| j� | j �d�}| �|| j� d S )Nz
//filenamez//filename?foo=bar)r rb rT ra )r
rL r r r � test_start_with_double_slash9 s z=SimpleHTTPRequestHandlerTestCase.test_start_with_double_slashc C s� t �tjdt��� | j�d�}|�tjtj�}| � || j
� | j�d�}|�tjtj�}| � || j
� | j�d�}|�tjtj�}| � || j
� | j�d�}|�tjtj�}| � || 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 rb r� �seprT ra )r
rL r r r �test_windows_colon? s z3SimpleHTTPRequestHandlerTestCase.test_windows_colonN)r r r r] r+ rc rd rg r r r r r` * s
r` c @ s e Zd Zdd� ZdS )� MiscTestCasec C sj g }dddh}xHt t�D ]<}|�d�s||kr0qtt|�}t|dd �dkr|�|� qW | �tj|� 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_allS s
zMiscTestCase.test_allN)r r r rl r r r r rh R s rh c C s4 t �� }zt�tttttt t
� W d t � |� X d S )N) r& r� r
Z run_unittestr� r r7 r� r� r` rh r� )�verboser� r r r � test_main_ s
rn �__main__)N)7r] Z http.serverr r r r r/ r r r& r� r^ r re r� Z
email.messager� Z email.utilsr� Z http.clientZ urllib.parser� r� rX r� r r� r �ior r� r
r r r ZTestCaser$ r7 r� r� r� r� r� r� �hasattrr� r� r r r r r` rh rn r r r r r �<module> s\ P$ m
I l(