← Back
Editing: 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 %&
Save File
Cancel