U
Acj
; @ 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$ >/tmp/pip-unpacked-wheel-nrvc_22y/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 s( 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 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/">sX
click here
Nr+ r, r$ r$ r% test_maliciousRedirectT s
z&RedirectToTests.test_maliciousRedirectN)__name__
__module__ __qualname____doc__r& r) r/ r0 r$ r$ r$ r% r ! s
r c @ s8 e Zd ZdZeedddZdd Zdd Zd d
Zd S ) ParentRedirectTestsz!
Test L{ParentRedirect}.
) requestPathreturnc C sV t t d}d|_|dd|_t }|| |j d}|dk sLt
|\}|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 AssertionError)r" r6 r# r headerslocationr$ r$ r% doLocationTestn s
z"ParentRedirectTests.doLocationTestc C s( | 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/r8 s /?biff=baffNr r? r" r$ r$ r% test_locationRoot s z%ParentRedirectTests.test_locationRootc C s( | d| d | d| d dS )zt
A request for a resource one level down from the URL root produces
a redirect to the root.
r@ s /foos /foo?bar=sproiiingNrA rB 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=bazNrA rB r$ r$ r% test_locationUpOne s z&ParentRedirectTests.test_locationUpOneN) r1 r2 r3 r4 bytesr? rC rD rE r$ r$ r$ r% r5 i s
r5 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 sT dd }|j jd | _z
| W n. tk
rN 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% lineNumberProbeAlsoBroken s z.lineNumberProbeAlsoBroken T)Z captureVarsN)__code__co_firstlinenobase
BaseExceptionr failureframesframe)r" rJ 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}dddg}td|}d}t|D ]Z\}}d }|d
kr||dd g|d
k | j | d | 7 }qL||d| j | d| 7 }qL|
d
}| | j | |S )z
L{_SourceFragmentElement} renders source lines at and around the line
number indicated by a frame object.
rU rV rW Z sourceLinesu# message = "This is a problem"u raise Exception(message) Nz?{}{}
rK Z Highlightu utf8)
r r r rY rZ rS r
enumerateformatrO r\ r[ r ) r" r] sourcer^ ZstringToCheckForrU rW templateZbytesToCheckForr$ r$ r% test_sourceFragmentElement s@
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}.
filenamerV N c ) r r r rZ rS 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}.
rU rV Ns %drK ) r r r rZ rS r
r[ r rO rm 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}.
functionrV Ns& lineNumberProbeAlsoBroken)r r r rZ rS r
r[ r rm 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}.
Nrd ) r rS lookupRenderMethodr rY 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 rR r rK )r rQ rR rr r rY rs listr rt rS ZassertNotEqualru rv r lenrw 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 rQ rr r rY rs r rt Z stackFramesrR r ru rv rw 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.
typerV Ns builtins.Exceptionrh rj r rQ r r rZ 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.
valuerV Ns This is a problemr rm r$ r$ r% test_failureElementValueS s
z,FailureElementTests.test_failureElementValueN)r1 r2 r3 r4 rT r_ rf rn ro rq rz r~ r r r r$ r$ r$ r% rG s *
rG 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 objectrB r$ r$ r% test_flattenerErrord s z&FormatFailureTests.test_flattenerErrorc C s` z t dW n tk
r* tt }Y nX | |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) rH rP r r rs rF
assertTrueallZassertIn)r" ry r$ r$ r% test_returnsBytesk s z$FormatFailureTests.test_returnsBytesN)r1 r2 r3 r4 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)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 )NrK ZfoobarZbaz)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# ry 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^ rQ r errorsr$ r$ r% test_renderNoFailure s
z*DeferredResourceTests.test_renderNoFailureN)r1 r2 r3 r4 r r r r$ r$ r$ r% r s r ),r4 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 r5 rG r r r r r$ r$ r$ r% s& ,
H7 ?