o
i
; @ s* d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z m
Z
ddl m Z m
Z
ddlmZ dd lmZ dd
lmZ dd lmZmZmZ dd lmZmZ dd
lm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$ G dd de
Z%G dd de Z&G dd de
Z'G dd de
Z(G dd de j)Z*G dd de Z+dS )z
Tests for L{twisted.web.util}.
N)defer)
networkString)Failure)SynchronousTestCaseTestCase)resourceutil)FlattenerError)FOUND)Request) TagLoader
flattenStringtags) DummyChannel DummyRequest) DeferredResourceFailureElementParentRedirect
_FrameElement_SourceFragmentElement_SourceLineElement
_StackElement
formatFailure
redirectToc @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) RedirectToTestsz"
Tests for L{redirectTo}.
c C s^ t t d}d|_d}t|| | |jt | |jd|g | |jddg 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 assertEqualcoder
responseHeaders
getRawHeadersselfrequestZ targetURL r$ `C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/web/test/test_util.pytest_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_redirectToUnicodeURL9 s z)RedirectToTests.test_redirectToUnicodeURLc C ( t dg}td|}d}| || dS )zq
Legitimate URLs are fully interpolated in the `redirectTo` response body without transformation
s https://twisted.org/s
click here
Nr r r r" r# htmlexpectedr$ r$ r% test_legitimateRedirectB s
z'RedirectToTests.test_legitimateRedirectc C r* )zm
Malicious URLs are HTML-escaped before interpolating them in the `redirectTo` response body
r+ s? https://twisted.org/">sX
click here
Nr, r- r$ r$ r% test_maliciousRedirectT s
z&RedirectToTests.test_maliciousRedirectN)__name__
__module__ __qualname____doc__r& r) r0 r1 r$ r$ r$ r% r ! s r c @ s: e Zd ZdZdedefddZdd Zdd Zd
d Zd S )
ParentRedirectTestsz!
Test L{ParentRedirect}.
requestPathreturnc C sV t t d}d|_|dd|_t }|| |j d}|dus&J |\}|S )z
Render a response to a request with path *requestPath*
@param requestPath: A slash-separated path like C{b'/foo/bar'}.
@returns: The value of the I{Location} header.
Tr /s LocationN)
r r r lstripsplitZprepathr renderr r )r" r7 r# r headerslocationr$ r$ r% doLocationTestn s
z"ParentRedirectTests.doLocationTestc C ( | d| d | d| d dS )zi
At the URL root issue a redirect to the current URL, removing any query
string.
http://10.0.0.1/r9 s /?biff=baffNr r? r" r$ r$ r% test_locationRoot s z%ParentRedirectTests.test_locationRootc C r@ )zt
A request for a resource one level down from the URL root produces
a redirect to the root.
rA s /foos /foo?bar=sproiiingNrB rC r$ r$ r% test_locationToRoot s
z'ParentRedirectTests.test_locationToRootc C s: | d| d | d| d | d| d dS )zp
Requests for resources directly under the path C{/foo/} produce
redirects to C{/foo/}.
s http://10.0.0.1/foo/s /foo/s /foo/bars /foo/bar?biz=bazNrB rC r$ r$ r% test_locationUpOne s
z&ParentRedirectTests.test_locationUpOneN) r2 r3 r4 r5 bytesr? rD rE rF r$ r$ r$ r% r6 i s
r6 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 sR dd }|j jd | _z| W dS ty( tdd| _| jjd | _Y dS w )zO
Create a L{Failure} which can be used by the rendering tests.
c S s d} t | )NzThis is a problem) Exception)messager$ r$ r% lineNumberProbeAlsoBroken s z.lineNumberProbeAlsoBroken T)Z captureVarsN)__code__co_firstlinenobase
BaseExceptionr failureframesframe)r" rK r$ r$ r% setUp s zFailureElementTests.setUpc C sN t tttjddtjdddd}td|}d}|| j|d |S ) zN
L{_SourceLineElement} renders a source line and line number.
lineNumberr<
sourceLine2 z print 'hello'Nu: 50 print 'hello'
zutf-8) r r r divspanr
addCallbackr encode)r" elementdr/ r$ r$ r% test_sourceLineElement s
z*FailureElementTests.test_sourceLineElementc C s t ttjtjddtjdddd| j}g d}td|}d}t|D ]-\}}d}|d krD||dd
g|d k | j | d | 7 }q%||d| j | d| 7 }q%|
d }| | j | |S )
z
L{_SourceFragmentElement} renders source lines at and around the line
number indicated by a frame object.
rV rW rX Z sourceLines)u# message = "This is a problem"u raise Exception(message) Nra z?{}{}
rL Z Highlightu utf8)
r r r rZ r[ rT r
enumerateformatrP r] r\ r ) r" r^ sourcer_ ZstringToCheckForrV rX templateZbytesToCheckForr$ r$ r% test_sourceFragmentElement s6
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}.
filenamerW N c ) r r r r[ rT r
r\ r r __file__rstripr" r^ r_ r$ r$ r% test_frameElementFilename s
z-FailureElementTests.test_frameElementFilenamec C s@ t ttjdd| j}td|}|| jd| jd f |S )z
The I{lineNumber} renderer of L{_FrameElement} renders the line number
associated with the frame object used to initialize the
L{_FrameElement}.
rV rW Ns %drL ) r r r r[ rT r
r\ r rP rn 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}.
functionrW Ns& lineNumberProbeAlsoBroken)r r r r[ rT r
r\ r rn 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}.
Nre ) r rT lookupRenderMethodr rZ assertIsInstancer assertIdenticalr loaderloadr" r^ Zrenderertagresultr$ 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 rS r rL )r rR rS rs r rZ rt listr ru rT ZassertNotEqualrv rw r lenrx 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 rR rs r rZ rt r ru Z stackFramesrS r rv rw rx r$ r$ r% test_failureElementTraceback; s
z0FailureElementTests.test_failureElementTracebackc C s@ t | jttjdd}td|}d}|| jd| d |S )zi
The I{type} renderer of L{FailureElement} renders the failure's
exception type.
typerW Ns builtins.Exceptionri rk r rR r r r[ r
r\ r )r" r^ r_ excr$ r$ r% test_failureElementTypeH 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.
valuerW Ns This is a problemr rn r$ r$ r% test_failureElementValueS s
z,FailureElementTests.test_failureElementValueN)r2 r3 r4 r5 rU r` rg ro rp rr r{ r r r r r$ r$ r$ r% rH s *
rH 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 objectrC r$ r$ r% test_flattenerErrord s z&FormatFailureTests.test_flattenerErrorc C sZ zt d ty tt }Y nw | |t | 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$ ).0chr$ r$ r% w s z7FormatFailureTests.test_returnsBytes..s N) rI rQ r r rt rG
assertTrueallZassertIn)r" rz r$ r$ r% test_returnsBytesk s z$FormatFailureTests.test_returnsBytesN)r2 r3 r4 r5 r r r$ r$ r$ r% r ^ s r 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# r_ r r$ r$ r% r s
z SDResource.getChildWithDefaultN)r2 r3 r4 r r r$ r$ r$ r% r | s r c @ s( e Zd ZdZdd Zdd Zdd ZdS ) DeferredResourceTestsz(
Tests for L{DeferredResource}.
c C sD t }d|_t|}tg d}t || | |jddg d S )NrL )Zfoobarbazr r )r ResourceZisLeafr r ZgetChildForRequestr Zpostpath)r" rsr_ r$ r$ r% testDeferredResource 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 appendr< r r r r r r )r" renderedr# rz deferredResourcer$ r$ r% test_render 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 Zfailr< r ZfailureResultOfgcZcollectZflushLoggedErrors)r" r# r_ rR r errorsr$ r$ r% test_renderNoFailure s
z*DeferredResourceTests.test_renderNoFailureN)r2 r3 r4 r5 r r r r$ r$ r$ r% r s
r ),r5 r Ztwisted.internetr Ztwisted.python.compatr Ztwisted.python.failurer Ztwisted.trial.unittestr 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 r r6 rH r r r r r$ r$ r$ r% s( ,
H7 ?