← Back
Editing: test_resource.cpython-38.pyc
U W[ � @ s� d Z ddlmZ ddlmZ ddlmZmZmZm Z m Z m Z m Z ddl mZ ddlmZ G dd� de�ZG d d � d e�ZG d d � d e�ZG d d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )z$ Tests for L{twisted.web.resource}. � )�TestCase)�UnsupportedMethod)� NOT_FOUND� FORBIDDEN�Resource� ErrorPage� NoResource�ForbiddenResource�getChildForRequest)�Headers)� DummyRequestc @ sD e Zd ZdZeZeZeZ dd� Z dd� Z dd� Z dd � Z d d � Zd S ) �ErrorPageTestszJ Tests for L{ErrorPage}, L{NoResource}, and L{ForbiddenResource}. c C s( | � ddd�}| �|�dt� �|� dS )zj The C{getChild} method of L{ErrorPage} returns the L{ErrorPage} it is called on. �A ZfooZbars nameN)� errorPage�assertIdentical�getChild�object)�self�page� r �@/usr/lib/python3/dist-packages/twisted/web/test/test_resource.py� test_getChild s z ErrorPageTests.test_getChildc C s` t dg�}d}|||||f }| �|�|�|�d�� | �|j|� | �|jtddgi�� d S )N� zg <html> <head><title>%s - %s</title></head> <body> <h1>%s</h1> <p>%s</p> </body> </html> zutf-8s content-types text/html; charset=utf-8)r � assertEqual�render�encodeZ responseCodeZresponseHeadersr )r r �code�brief�detail�request�template�expectedr r r �_pageRenderingTest$ s � � �z!ErrorPageTests._pageRenderingTestc C s. d}d}d}| � |||�}| �||||� dS )a L{ErrorPage.render} returns a C{bytes} describing the error defined by the response code and message passed to L{ErrorPage.__init__}. It also uses that response code to set the response code on the L{Request} passed in. r zbrief description textz much longer text might go hereN)r r" )r r r r r r r r �test_errorPageRendering8 s z&ErrorPageTests.test_errorPageRenderingc C s" d}| � |�}| �|td|� dS )z@ L{NoResource} sets the HTTP I{NOT FOUND} code. z long messagezNo Such ResourceN)� noResourcer" r �r r r r r r �test_noResourceRenderingF s z'ErrorPageTests.test_noResourceRenderingc C s" d}| � |�}| �|td|� dS )zG L{ForbiddenResource} sets the HTTP I{FORBIDDEN} code. zlonger messagezForbidden ResourceN)�forbiddenResourcer" r r% r r r � test_forbiddenResourceRenderingO s z.ErrorPageTests.test_forbiddenResourceRenderingN)�__name__� __module__� __qualname__�__doc__r r r r$ r r' r r" r# r& r( r r r r r s r c @ s e Zd ZdZdd� ZdS )� DynamicChildzG A L{Resource} to be created on the fly by L{DynamicChildren}. c C s t �| � || _|| _d S �N)r �__init__�pathr �r r0 r r r r r/ ] s zDynamicChild.__init__N)r) r* r+ r, r/ r r r r r- Y s r- c @ s e Zd ZdZdd� ZdS )�DynamicChildrenz. A L{Resource} with dynamic children. c C s t ||�S r. )r- r1 r r r r h s zDynamicChildren.getChildN)r) r* r+ r, r r r r r r2 d s r2 c @ s e Zd ZdZdd� Zdd� ZdS )�BytesReturnedRenderablezG A L{Resource} with minimal capabilities to render a response. c C s t �| � || _dS )zo @param response: A C{bytes} object giving the value to return from C{render_GET}. N)r r/ � _response)r Zresponser r r r/ q s z BytesReturnedRenderable.__init__c C s | j S )z} Render a response to a I{GET} request by returning a short byte string to be written by the server. )r4 �r r r r r � render_GETz s z"BytesReturnedRenderable.render_GETN)r) r* r+ r, r/ r6 r r r r r3 m s r3 c @ s e Zd ZdZdd� Zdd� ZdS )�ImplicitAllowedMethodszn A L{Resource} which implicitly defines its allowed methods by defining renderers to handle them. c C s d S r. r r5 r r r r6 � s z!ImplicitAllowedMethods.render_GETc C s d S r. r r5 r r r � render_PUT� s z!ImplicitAllowedMethods.render_PUTN)r) r* r+ r, r6 r8 r r r r r7 � s r7 c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � ResourceTestsz Tests for L{Resource}. c C sF t � }t � }t � }|�d|� |�d|� | �||�dtg ��� dS )z� L{Resource.putChild} adds a I{static} child to the resource. That child is returned from any call to L{Resource.getChildWithDefault} for the child's path. � foo� barN)r �putChildr �getChildWithDefaultr )r �resource�childZsiblingr r r �test_staticChildren� s �z!ResourceTests.test_staticChildrenc C sJ d}t g �}t� }|�||�}| �|t� | �|j|� | �|j|� dS )z� L{Resource.getChildWithDefault} delegates to L{Resource.getChild} when the requested path is not associated with any static child. r: N) r r2 r= ZassertIsInstancer- r r0 r r )r r0 r r> r? r r r �test_dynamicChildren� s z"ResourceTests.test_dynamicChildrenc C s0 d}t g �}d|_t|�}| �||�|�� dS )z� When not otherwise overridden, L{Resource.render} treats a I{HEAD} request as if it were a I{GET} request. s insert response here� HEADN)r �methodr3 r r )r r! r r> r r r �test_defaultHEAD� s z ResourceTests.test_defaultHEADc C sN dddg}t � }||_tg �}d|_| �t|j|�}| �t|�t|j�� dS )z� The L{UnsupportedMethod} raised by L{Resource.render} for an unsupported request method has a C{allowedMethods} attribute set to the value of the C{allowedMethods} attribute of the L{Resource}, if it has one. � GETrB � PUT� FICTIONALN) r �allowedMethodsr rC � assertRaisesr r r �set�r r! r> r �excr r r �test_explicitAllowedMethods� s z)ResourceTests.test_explicitAllowedMethodsc C sH t dddg�}t� }tg �}d|_| �t|j|�}| �|t |j�� dS )al The L{UnsupportedMethod} raised by L{Resource.render} for an unsupported request method has a C{allowedMethods} attribute set to a list of the methods supported by the L{Resource}, as determined by the I{render_}-prefixed methods which it defines, if C{allowedMethods} is not explicitly defined by the L{Resource}. rE rB rF rG N) rJ r7 r rC rI r r r rH rK r r r �test_implicitAllowedMethods� s z)ResourceTests.test_implicitAllowedMethodsN) r) r* r+ r, r@ rA rD rM rN r r r r r9 � s r9 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �GetChildForRequestTestsz* Tests for L{getChildForRequest}. c C s( t g �}t� }t||�}| �||� dS )z� L{getChildForRequest} returns whatever resource has been reached by the time the request's C{postpath} is empty. N)r r r r �r r r> �resultr r r �test_exhaustedPostPath� s z.GetChildForRequestTests.test_exhaustedPostPathc C s2 t ddg�}t� }d|_t||�}| �||� dS )z� L{getChildForRequest} returns the first resource it encounters with a C{isLeaf} attribute set to C{True}. r: r; TN)r r �isLeafr r rP r r r �test_leafResource� s z)GetChildForRequestTests.test_leafResourcec C s` t ddg�}t� }t� }d|_|�d|� | �|t||�� | �|jdg� | �|jdg� dS )z� As path segments from the request are traversed, they are taken from C{postpath} and put into C{prepath}. r: r; TN) r r rS r< r r r ZprepathZpostpath)r r �rootr? r r r �test_postPathToPrePath� s z.GetChildForRequestTests.test_postPathToPrePathN)r) r* r+ r, rR rT rV r r r r rO � s rO N)r, Ztwisted.trial.unittestr Ztwisted.web.errorr Ztwisted.web.resourcer r r r r r r Ztwisted.web.http_headersr Z twisted.web.test.requesthelperr r r- r2 r3 r7 r9 rO r r r r �<module> s $ G N
Save File
Cancel