File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/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