File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_util.cpython-38.pyc
U
s�@gL6 � @ sF d Z ddlmZmZ ddlZddlmZ ddlmZm Z ddl
m Z ddl m
Z
mZmZ ddlmZmZ dd lmZ dd
lmZ dd lmZ dd lmZmZm Z dd
l m Z m Z ddl m!Z! ddl m"Z"m#Z# ddl m$Z$m%Z%m&Z& ddl m'Z'm(Z( G dd� de �Z)G dd� de �Z*G dd� de �Z+G dd� dej,�Z-G dd� de�Z.dS ) z
Tests for L{twisted.web.util}.
� )�absolute_import�divisionN)�Failure)�SynchronousTestCase�TestCase)�defer)�_PY3�
intToBytes�
networkString)�resource�util)�FlattenerError)�FOUND)�Request)� TagLoader�
flattenString�tags)� DummyChannel� DummyRequest)�DeferredResource)�_SourceFragmentElement�
_FrameElement)�
_StackElement�FailureElement�
formatFailure)�
redirectTo�_SourceLineElementc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �RedirectToTestsz"
Tests for L{redirectTo}.
c C s^ t t� d�}d|_d}t||� | �|jt� | �|j�d�|g� | �|j�d�dg� dS )z�
L{redirectTo} will set the C{Location} and C{Content-Type} headers on
its request, and set the response code to C{FOUND}, so the browser will
be redirected.
T� GETs http://target.example.com/4321s locations content-types text/html; charset=utf-8N) r r �methodr � assertEqual�coder ZresponseHeadersZ
getRawHeaders��self�requestZ targetURL� r% �</usr/lib/python3/dist-packages/twisted/web/test/test_util.py�test_headersAndCode" s
�
�z#RedirectToTests.test_headersAndCodec C s* t t� d�}d|_d}| �tt||� dS )zW
L{redirectTo} will raise TypeError if unicode object is passed in URL
Tr z http://target.example.com/4321N)r r r � assertRaises� TypeErrorr r" r% r% r& �test_redirectToUnicodeURL4 s z)RedirectToTests.test_redirectToUnicodeURLc C s( t dg�}td|�}d}| �||� dS )zq
Legitimate URLs are fully interpolated in the `redirectTo` response body without transformation
� s https://twisted.org/s�
<html>
<head>
<meta http-equiv="refresh" content="0;URL=https://twisted.org/">
</head>
<body bgcolor="#FFFFFF" text="#000000">
<a href="https://twisted.org/">click here</a>
</body>
</html>
N�r r r �r# r$ Zhtml�expectedr% r% r& �test_legitimateRedirect= s
z'RedirectToTests.test_legitimateRedirectc C s( t dg�}td|�}d}| �||� dS )zm
Malicious URLs are HTML-escaped before interpolating them in the `redirectTo` response body
r+ s? https://twisted.org/"><script>alert(document.location)</script>sX
<html>
<head>
<meta http-equiv="refresh" content="0;URL=https://twisted.org/"><script>alert(document.location)</script>">
</head>
<body bgcolor="#FFFFFF" text="#000000">
<a href="https://twisted.org/"><script>alert(document.location)</script>">click here</a>
</body>
</html>
Nr, r- r% r% r& �test_maliciousRedirectO s
�
z&RedirectToTests.test_maliciousRedirectN)�__name__�
__module__� __qualname__�__doc__r' r* r/ r0 r% r% r% r& r s
r c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�FailureElementTestszn
Tests for L{FailureElement} and related helpers which can render a
L{Failure} as an HTML string.
c C sL dd� }|j jd | _z
|� W n& tdd�| _| jjd | _Y nX dS )zO
Create a L{Failure} which can be used by the rendering tests.
c S s d} t | ��d S )NzThis is a problem)� Exception)�messager% r% r& �lineNumberProbeAlsoBrokenn s z<FailureElementTests.setUp.<locals>.lineNumberProbeAlsoBroken� T)Z captureVars���N)�__code__�co_firstlineno�baser �failure�frames�frame)r# r8 r% r% r& �setUpj s
zFailureElementTests.setUpc C sN t tt�tjdd�tjdd���dd�}td|�}d}|�| j|�d�� |S ) zN
L{_SourceLineElement} renders a source line and line number.
�
lineNumber��render�
sourceLine�2 z print 'hello'Nu: <div><span>50</span><span> print 'hello'</span></div>zutf-8) r r r �div�spanr � addCallbackr �encode)r# �element�dr. r% r% r& �test_sourceLineElement{ s
� �
� �z*FailureElementTests.test_sourceLineElementc s� t ttjtjdd�tjdd�dd��� j�}dddg}td|�}trhd �� fd
d �t |�D ���
d �}n d �� fd
d �t |�D ��}|� � j |� |S )z�
L{_SourceFragmentElement} renders source lines at and around the line
number indicated by a frame object.
rB rC rE Z sourceLinesu# message = "This is a problem"u raise Exception(message)zE# Figure out the line number from which the exception will be raised.N� c s4 g | ],\}}d ddg|dk � j | d| f �qS )�?<div class="snippet%sLine"><span>%d</span><span>%s</span></div>rN � Highlightr9 � )r= ��.0rB rE �r# r% r& �
<listcomp>� s ���zBFailureElementTests.test_sourceFragmentElement.<locals>.<listcomp>�utf8c s: g | ]2\}}d ddg|dk � j | d| �d�f �qS )rO rN rP r9 rQ rV )r= rJ rR rT r% r& rU � s � ��)
r r r rG rH r@ r r �join� enumeraterJ rI r )r# rK �sourcerL ZstringToCheckForr% rT r& �test_sourceFragmentElement� s0
���
��
� z.FailureElementTests.test_sourceFragmentElementc C sF t ttjdd��| j�}td|�}|�| jdtt �
d�� d � |S )z�
The I{filename} renderer of L{_FrameElement} renders the filename
associated with the frame object used to initialize the
L{_FrameElement}.
�filenamerC N� <span>�c� </span>) r r r rH r@ r rI r r
�__file__�rstrip�r# rK rL r% r% r& �test_frameElementFilename� s �
�z-FailureElementTests.test_frameElementFilenamec C sF t ttjdd��| j�}td|�}|�| jdt| j d � d � |S )z�
The I{lineNumber} renderer of L{_FrameElement} renders the line number
associated with the frame object used to initialize the
L{_FrameElement}.
rB rC Nr\ r9 r^ )
r r r rH r@ r rI r r r= ra r% r% r& �test_frameElementLineNumber� s �
�z/FailureElementTests.test_frameElementLineNumberc C s4 t ttjdd��| j�}td|�}|�| jd� |S )z�
The I{function} renderer of L{_FrameElement} renders the line number
associated with the frame object used to initialize the
L{_FrameElement}.
ZfunctionrC Ns& <span>lineNumberProbeAlsoBroken</span>)r r r rH r@ r rI r ra r% r% r& �test_frameElementFunction� s �
�z-FailureElementTests.test_frameElementFunctionc C s\ t d| j�}|�d�}t�� }|d|�}| �|t� | �|j| j� | �|g|j �
� � dS )z�
The I{source} renderer of L{_FrameElement} renders the source code near
the source filename/line number associated with the frame object used to
initialize the L{_FrameElement}.
NrY ) r r@ �lookupRenderMethodr rG �assertIsInstancer �assertIdenticalr �loader�load�r# rK Zrenderer�tag�resultr% r% r& �test_frameElementSource� s
z+FailureElementTests.test_frameElementSourcec C s� t d| jjdd� �}|�d�}t�� }|d|�}| �|t� | �|d t� | � |d j
| jjd � | �|d t� | � |d j
| jjd � | � |d j �
� |d j �
� � | �dt|�� dS )z�
The I{frames} renderer of L{_StackElement} renders each stack frame in
the list of frames used to initialize the L{_StackElement}.
N� r? r r9 )r r> r? re r rG rf �listr rg r@ ZassertNotEqualrh ri r �lenrj r% r% r& �test_stackElement� s
z%FailureElementTests.test_stackElementc C s\ t | j�}|�d�}t�� }|d|�}| �|t� | �|j| jj � | �
|g|j � � � dS )z�
The I{traceback} renderer of L{FailureElement} renders the failure's
stack frames using L{_StackElement}.
� tracebackN)
r r> re r rG rf r rg Z stackFramesr? r rh ri rj r% r% r& � test_failureElementTraceback s
z0FailureElementTests.test_failureElementTracebackc C sJ t | jttjdd���}td|�}tr,d}nd}|�| jd| d � |S )zi
The I{type} renderer of L{FailureElement} renders the failure's
exception type.
�typerC Ns builtins.Exceptions exceptions.Exceptionr\ r^ ) r r> r r rH r r rI r )r# rK rL �excr% r% r& �test_failureElementType s �
�z+FailureElementTests.test_failureElementTypec C s4 t | jttjdd���}td|�}|�| jd� |S )zi
The I{value} renderer of L{FailureElement} renders the value's exception
value.
�valuerC Ns <span>This is a problem</span>)r r> r r rH r rI r ra r% r% r& �test_failureElementValue+ s �
�z,FailureElementTests.test_failureElementValueN)r1 r2 r3 r4 rA rM rZ rb rc rd rm rq rs rv rx r% r% r% r& r5 e s -r5 c @ s e Zd ZdZdd� Zdd� ZdS )�FormatFailureTestsz�
Tests for L{twisted.web.util.formatFailure} which returns an HTML string
representing the L{Failure} instance passed to it.
c C s | � ttt� � dS )z}
If there is an error flattening the L{Failure} instance,
L{formatFailure} raises L{FlattenerError}.
N)r( r
r �objectrT r% r% r& �test_flattenerError> s z&FormatFailureTests.test_flattenerErrorc C sv z t d��W n tt� �}Y nX | �|t� trN| �tdd� |D ��� n| �tdd� |D ��� | �d|� dS )z�
The return value of L{formatFailure} is a C{str} instance (not a
C{unicode} instance) with numeric character references for any non-ASCII
characters meant to appear in the output.
zFake bugc s s | ]}|d k V qdS �� Nr% �rS Zchr% r% r& � <genexpr>S s z7FormatFailureTests.test_returnsBytes.<locals>.<genexpr>c s s | ]}t |�d k V qdS r| )�ordr~ r% r% r& r U s s  N) r6 r r rf �bytesr Z
assertTrue�allZassertIn)r# rl r% r% r& �test_returnsBytesF s z$FormatFailureTests.test_returnsBytesN)r1 r2 r3 r4 r{ r� r% r% r% r& ry 9 s ry c @ s e Zd Zdd� Zdd� ZdS )�
SDResourcec C s
|| _ d S �N)�default)r# r� r% r% r& �__init__\ s zSDResource.__init__c C s" t �| j�}t�|�}|�||�S r� )r �succeedr� r r �getChildWithDefault)r# �namer$ rL r r% r% r& r� ` s
z SDResource.getChildWithDefaultN)r1 r2 r3 r� r� r% r% r% r& r� [ s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DeferredResourceTestsz(
Tests for L{DeferredResource}.
c C sF t �� }d|_t|�}tdddg�}t �||� | �|jddg� d S )Nr9 ZfooZbarZbaz)r �ResourceZisLeafr� r ZgetChildForRequestr Zpostpath)r# �r�srL r% r% r& �testDeferredResourcel s z*DeferredResourceTests.testDeferredResourcec C sF g }t g �}|j|_t�� }tt�|��}|�|� | �||g� dS )z�
L{DeferredResource} uses the request object's C{render} method to
render the resource which is the result of the L{Deferred} being
handled.
N) r �appendrD r r� r r r� r )r# Zrenderedr$ rl �deferredResourcer% r% r& � test_renderu s
z!DeferredResourceTests.test_renderc C sh t g �}|�� }tt� �}tt�|��}|�|� | �| � |�|� ~t
� � | � t�}| �|g � dS )z�
If the L{Deferred} fails, L{DeferredResource} reports the failure via
C{processingFailed}, and does not cause an unhandled error to be
logged.
N)
r Z notifyFinishr � RuntimeErrorr r ZfailrD r ZfailureResultOf�gcZcollectZflushLoggedErrors)r# r$ rL r> r� �errorsr% r% r& �test_renderNoFailure� s
z*DeferredResourceTests.test_renderNoFailureN)r1 r2 r3 r4 r� r� r� r% r% r% r& r� g s r� )/r4 Z
__future__r r r� Ztwisted.python.failurer Ztwisted.trial.unittestr r Ztwisted.internetr Ztwisted.python.compatr r r
Z twisted.webr r Ztwisted.web.errorr
Ztwisted.web.httpr Ztwisted.web.serverr Ztwisted.web.templater r r Z twisted.web.test.requesthelperr r Ztwisted.web.utilr r r r r r r r r r5 ry r� r� r� r% r% r% r& �<module> s, H U"