File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_cgi.cpython-38.pyc
U
W[�5 � @ s* d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z m
Z
m Z m Z ddl
mZmZmZ ddlmZmZ ddlmZmZmZmZmZ dd lmZ dd
l m Z d Z d Z d
Z dZ!dZ"dZ#dZ$dZ%G dd� dej&�Z'G dd� dej(�Z)G dd� dej(�Z*G dd� dej(�Z+G dd � d ej(�Z,d d � Z-dS ) z!
Tests for L{twisted.web.twcgi}.
� N)�BytesIO)�unittest)�address�reactor�
interfaces�error)�util�failure�log)� NOT_FOUND�INTERNAL_SERVER_ERROR)�client�twcgi�server�resource� http_headers)�_render)� DummyRequestz2print("Header: OK")
print("")
print("cgi output")
zJprint("Header: spam")
print("Header: eggs")
print("")
print("cgi output")
z+print("XYZ")
print("")
print("cgi output")
zPprint("Server: monkeys")
print("Date: last year")
print("")
print("cgi output")
a& # This is an example of a correctly-written CGI script which reads a body
# from stdin, which only reads env['CONTENT_LENGTH'] bytes.
import os, sys
body_length = int(os.environ.get('CONTENT_LENGTH',0))
indata = sys.stdin.read(body_length)
print("Header: OK")
print("")
print("readinput ok")
a+ # This is an example of the typical (incorrect) CGI script which expects
# the server to close stdin when the body of the request is complete.
# A correct CGI should only read env['CONTENT_LENGTH'] bytes.
import sys
indata = sys.stdin.read()
print("Header: OK")
print("")
print("readallinput ok")
zMprint("content-type: text/cgi-duplicate-test")
print("")
print("cgi output")
z�import json
import os
print("")
print("")
vals = {x:y for x,y in os.environ.items() if x.startswith("HTTP_")}
print(json.dumps(vals))
c @ s e Zd ZejZdS )� PythonScriptN)�__name__�
__module__� __qualname__�sys�
executable�filter� r r �;/usr/lib/python3/dist-packages/twisted/web/test/test_cgi.pyr X s r c @ s� e Zd ZdZej�e�s dZdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zde_dd� Zd d � Zde_d d � Zd d!� Zde_d"d#� Zd$d%� Zd&S )'�CGITestsz,
Tests for L{twcgi.FilteredScript}.
z5CGI tests require a functional reactor.spawnProcess()c C sH t �� }t�t|�}|�dt|�� t�|�}t �
d|�| _ | j � � j
S )Ns cgir )r ZResourcer Zsibpath�__file__ZputChildr r ZSiter Z listenTCP�pZgetHostZport)�selfZcgi�rootZcgipathZsiter r r � startServerf s
zCGITests.startServerc C s t | dd �r| j�� S d S )Nr )�getattrr Z
stopListening�r r r r �tearDowno s zCGITests.tearDownc C s4 t j�| �� �}t|d��}|�|� W 5 Q R X |S )N�wt)�os�path�abspath�mktemp�open�write)r �source� cgiFilename�cgiFiler r r �writeCGIt s zCGITests.writeCGIc C sV | � t�}| �|�}d|f }|�d�}t�t��d|�}|�tj � |�| j
� |S )N�http://localhost:%d/cgi�ascii� GET) r0 � DUMMY_CGIr" �encoder
�Agentr �request� addCallback�readBody�
_testCGI_1)r r. �portnum�url�dr r r �test_CGI{ s
zCGITests.test_CGIc C s | � |dtj�d� � d S )Ns
cgi outputr2 )� assertEqualr' �linesepr5 )r �resr r r r: � s zCGITests._testCGI_1c sb � � t�}� �|�}d|f }|�d�}t�t�}|�d|�}|�t � � fdd�}|�|� |S )zc
If the CGI script emits a I{Server} or I{Date} header, these are
ignored.
r1 r2 r3 c s, � � d| j�d�� � � d| j�d�� d S )NZmonkeysr z last year�date)Z assertNotIn�headers�
getRawHeaders��responser$ r r �
checkResponse� s
�
�z;CGITests.test_protectedServerAndDate.<locals>.checkResponse)
r0 �SPECIAL_HEADER_CGIr" r5 r
r6 r r7 r8 � discardBody�r r. r; r< �agentr= rG r r$ r �test_protectedServerAndDate� s
z$CGITests.test_protectedServerAndDatec sb � � t�}� �|�}d|f }|�d�}t�t�}|�d|�}|�t � � fdd�}|�|� |S )z�
If the CGI script emits a I{content-type} header, make sure that the
server doesn't add an additional (duplicate) one, as per ticket 4786.
r1 r2 r3 c s � � | j�d�dg� | S )Nz content-typeztext/cgi-duplicate-test�r? rC rD rE r$ r r rG � s
�zBCGITests.test_noDuplicateContentTypeHeaders.<locals>.checkResponse)
r0 �$NO_DUPLICATE_CONTENT_TYPE_HEADER_CGIr" r5 r
r6 r r7 r8 rI rJ r r$ r �"test_noDuplicateContentTypeHeaders� s
z+CGITests.test_noDuplicateContentTypeHeadersc s| � � t�}� �|�}d|f }|�d�}t�t�}t�dgdgd��}|j d||d�}� fdd �}|�
tj � |�
|� |S )
zV
The CGI script is never called with the Proxy header passed through.
r1 r2 s foos bar)s Proxys X-Innocent-Headerr3 )rC c s. t �| �d��}� �t|�� �dddh� d S )Nr2 Z HTTP_HOSTZHTTP_CONNECTIONZHTTP_X_INNOCENT_HEADER)�json�loads�decoder? �set�keys)rF rC r$ r r rG � s
�z7CGITests.test_noProxyPassthrough.<locals>.checkResponse) r0 �HEADER_OUTPUT_CGIr" r5 r
r6 r r ZHeadersr7 r8 r9 )r r. r; r< rK rC r= rG r r$ r �test_noProxyPassthrough� s
�
z CGITests.test_noProxyPassthroughc sb � � t�}� �|�}d|f }|�d�}t�t�}|�d|�}|�t � � fdd�}|�|� |S )zp
If a CGI script emits two instances of the same header, both are sent
in the response.
r1 r2 r3 c s � � | j�d�ddg� d S )N�headerZspamZeggsrM rE r$ r r rG � s
�z7CGITests.test_duplicateHeaderCGI.<locals>.checkResponse)
r0 �DUAL_HEADER_CGIr" r5 r
r6 r r7 r8 rI rJ r r$ r �test_duplicateHeaderCGI� s
z CGITests.test_duplicateHeaderCGIc s� �� t�}��|�}d|f }|�d�}t�t�}|�d|�}|�t � g � � fdd�}t
� |� �� t
j
|� � �fdd�}|�|� |S )zF
Check for the error message in the duplicated header
r1 r2 r3 c s � � t�| �� d S �N)�appendr
ZtextFromEventDict)Z eventDict)�loggedMessagesr r �
addMessage� s z4CGITests.test_malformedHeaderCGI.<locals>.addMessagec s �� dtd� � � d S )Nz ignoring malformed CGI header: s XYZ)ZassertIn�repr�Zignored�r\ r r r rG � s �z7CGITests.test_malformedHeaderCGI.<locals>.checkResponse)r0 �BROKEN_HEADER_CGIr" r5 r
r6 r r7 r8 rI r
Z addObserverZ
addCleanupZremoveObserver)r r. r; r< rK r= r] rG r r` r �test_malformedHeaderCGI� s
z CGITests.test_malformedHeaderCGIc C s� t j�| �� �}t|d��}|�t� W 5 Q R X | �|�}t� t
�}d|f }|� d�}|� d|�}|�
tj� |�
| j� |S )Nr& r1 r2 r3 )r' r( r) r* r+ r, �
READINPUT_CGIr" r
r6 r r5 r7 r8 r9 �_test_ReadEmptyInput_1�r r. r/ r; rK r< r= r r r �test_ReadEmptyInput s
z CGITests.test_ReadEmptyInput� c C s& d� tj�}|�d�}| �||� d S �Nzreadinput ok{}r2 ��formatr' r@ r5 r? �r rA Zexpectedr r r rd s
z CGITests._test_ReadEmptyInput_1c C s� t j�| �� �}t|d��}|�t� W 5 Q R X | �|�}t� t
�}d|f }|� d�}|j |dt�
td��d�}|�tj� |�| j� |S �Nr& r1 r2 s POSTs Here is your stdin)Zuri�methodZ bodyProducer)r' r( r) r* r+ r, rc r" r
r6 r r5 r7 �FileBodyProducerr r8 r9 �_test_ReadInput_1re r r r �test_ReadInput s"
�� zCGITests.test_ReadInputc C s& d� tj�}|�d�}| �||� d S rh ri rk r r r ro - s
zCGITests._test_ReadInput_1c C s� t j�| �� �}t|d��}|�t� W 5 Q R X | �|�}d|f }|�d�}t �
t �j |dt �
td��d�}|�t j� |�| j� |S rl )r' r( r) r* r+ r, �READALLINPUT_CGIr" r5 r
r6 r r7 rn r r8 r9 �_test_ReadAllInput_1)r r. r/ r; r< r= r r r �test_ReadAllInput3 s
�� zCGITests.test_ReadAllInputc C s& d� tj�}|�d�}| �||� d S )Nzreadallinput ok{}r2 ri rk r r r rr G s
z CGITests._test_ReadAllInput_1c C sX G dd� d�}|� }t ddg�}t�ddd�|_tjd|d �}t||� | �|j� d
S ) zw
L{twcgi.FilteredScript.runProcess} uses the reactor passed as an
argument to the constructor.
c @ s e Zd ZdZdZdd� ZdS )z5CGITests.test_useReactorArgument.<locals>.FakeReactorzR
A fake reactor recording whether spawnProcess is called.
Fc _ s
d| _ dS )z�
Set the C{called} flag to C{True} if C{spawnProcess} is called.
@param args: Positional arguments.
@param kwargs: Keyword arguments.
TN)�called)r �args�kwargsr r r � spawnProcessW s zBCGITests.test_useReactorArgument.<locals>.FakeReactor.spawnProcessN)r r r �__doc__rt rw r r r r � FakeReactorR s ry �a�b�TCP� 127.0.0.1�90 z
dummy-file�r N) r r � IPv4Addressr
r �FilteredScriptr Z
assertTruert )r ry Z fakeReactorr7 r r r r �test_useReactorArgumentM s
z CGITests.test_useReactorArgumentN)r r r rx r ZIReactorProcessZ
providedByr �skipr" r% r0 r> r: rL rO rV rY rb rf Ztimeoutrd rp ro rs rr r� r r r r r ] s.
r c @ s e Zd ZdZdd� ZdS )�CGIScriptTestsz'
Tests for L{twcgi.CGIScript}.
c C sb G dd� d�}|� }t j| �� |d�}tddg�}t�ddd�|_t||� | �|j d d
� d S ) zt
L{twcgi.CGIScript.render} sets the process environment
I{PATH_INFO} from the request path.
c @ s e Zd ZdZdd� ZdS )z1CGIScriptTests.test_pathInfo.<locals>.FakeReactorzZ
A fake reactor recording the environment passed to spawnProcess.
c S s
|| _ dS )a9
Store the C{env} L{dict} to an instance attribute.
@param process: Ignored
@param filename: Ignored
@param args: Ignored
@param env: The environment L{dict} which will be stored
@param wdir: Ignored
N)� process_env)r Zprocess�filenameru �envZwdirr r r rw x s
z>CGIScriptTests.test_pathInfo.<locals>.FakeReactor.spawnProcessN)r r r rx rw r r r r ry t s ry r rz r{ r| r} r~ Z PATH_INFOz/a/bN)
r Z CGIScriptr* r r r� r
r r? r� )r ry Z_reactorr r7 r r r �
test_pathInfoo s
�z CGIScriptTests.test_pathInfoN)r r r rx r� r r r r r� j s r� c @ s e Zd ZdZdd� Zdd� ZdS )�CGIDirectoryTestsz*
Tests for L{twcgi.CGIDirectory}.
c s> t ���� �}tdg�� t|� �}� �fdd�}|�|� |S )zc
L{twcgi.CGIDirectory.render} sets the HTTP response code to I{NOT
FOUND}.
� c s �� � jt� d S rZ �r? � responseCoder r_ �r7 r r r �
cbRendered� s z1CGIDirectoryTests.test_render.<locals>.cbRendered)r � CGIDirectoryr* r r r8 )r r r= r� r r� r � test_render� s
z CGIDirectoryTests.test_renderc sX �� � }t�|� t�|�}tdg�� |�d� �}t|� �}� �fdd�}|�|� |S )a
L{twcgi.CGIDirectory.getChild} returns a resource which renders an
response with the HTTP I{NOT FOUND} status code if the indicated child
does not exist as an entry in the directory used to initialized the
L{twcgi.CGIDirectory}.
Zfooc s �� � jt� d S rZ r� r_ r� r r r� � s z8CGIDirectoryTests.test_notFoundChild.<locals>.cbRendered) r* r' �makedirsr r� r ZgetChildr r8 )r r( r Zchildr= r� r r� r �test_notFoundChild� s
z$CGIDirectoryTests.test_notFoundChildN)r r r rx r� r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�CGIProcessProtocolTestsz0
Tests for L{twcgi.CGIProcessProtocol}.
c C s: t dg�}t�|�}|�t�t�� �� | �|j t
� dS )z�
If the process communicating with L{CGIProcessProtocol} ends before
finishing writing out headers, the response has I{INTERNAL SERVER
ERROR} as its status code.
r� N) r r ZCGIProcessProtocolZ processEndedr ZFailurer ZProcessTerminatedr? r� r )r r7 Zprotocolr r r �test_prematureEndOfHeaders� s
z2CGIProcessProtocolTests.test_prematureEndOfHeadersN)r r r rx r� r r r r r� � s r� c s t �� ��� fdd��S )zn
Discard the body of a HTTP response.
@param response: The response.
@return: The response.
c s � S rZ r )�_rE r r �<lambda>� � z discardBody.<locals>.<lambda>)r
r9 r8 rE r rE r rI � s rI ).rx r r' rP �ior Z
twisted.trialr Ztwisted.internetr r r r Ztwisted.pythonr r r
Ztwisted.web.httpr r Z twisted.webr
r r r r Ztwisted.web.test._utilr Ztwisted.web.test.test_webr r4 rX ra rH rc rq rN rU r� r ZTestCaser r� r� r� rI r r r r �<module> s6
%&