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