o
>h
; @ 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
getRawHeadersselfrequest targetURL r% z/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/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* r1 r2 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 lstripsplitprepathr renderr r )r" r8 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/r: s /?biff=baffNr rA r" r% r% r& test_locationRoot s z%ParentRedirectTests.test_locationRootc C rB )zt
A request for a resource one level down from the URL root produces
a redirect to the root.
rC s /foos /foo?bar=sproiiingNrD rE 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=bazNrD rE r% r% r& test_locationUpOne s
z&ParentRedirectTests.test_locationUpOneN) r3 r4 r5 r6 bytesrA rF rG rH r% r% r% r& r7 i s
r7 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) captureVarsN)__code__co_firstlinenobase
BaseExceptionr failureframesframe)r" rM 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" elementdr0 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.
rY rZ r[ sourceLines)u# message = "This is a problem"u raise Exception(message) Nre z?{}{}
rN Highlightu utf8)
r r r r] r^ rW r
enumerateformatrS r` r_ r ) r" ra sourcerb stringToCheckForrY r[ templatebytesToCheckForr% 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}.
filenamerZ N c ) r r r r^ rW r
r_ r r __file__rstripr" ra rb 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}.
rY rZ Ns %drN ) r r r r^ rW r
r_ r rS ru 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}.
functionrZ Ns& lineNumberProbeAlsoBroken)r r r r^ rW r
r_ r ru 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}.
Nrj ) r rW lookupRenderMethodr r] assertIsInstancer assertIdenticalr loaderloadr" ra renderertagresultr% 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 rV r rN )r rU rV rz r r] r{ listr r| rW assertNotEqualr} r~ r lenr 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 rU rz r r] r{ r r| stackFramesrV r r} r~ r 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.
typerZ Ns builtins.Exceptionrp rr r rU r r r^ r
r_ r )r" ra rb 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.
valuerZ Ns This is a problemr ru r% r% r& test_failureElementValueS s
z,FailureElementTests.test_failureElementValueN)r3 r4 r5 r6 rX rc rn rv rw ry r r r r r r% r% r% r& rJ s *
rJ 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 objectrE 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) rK rT r r r{ rI
assertTrueallassertIn)r" r r% r% r& test_returnsBytesk s z$FormatFailureTests.test_returnsBytesN)r3 r4 r5 r6 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# rb r r% r% r& r s
z SDResource.getChildWithDefaultN)r3 r4 r5 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 )NrN )foobarbazr r )r ResourceisLeafr r getChildForRequestr postpath)r" rsrb 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# r 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 notifyFinishr RuntimeErrorr r failr> r failureResultOfgccollectflushLoggedErrors)r" r# rb rU r errorsr% r% r& test_renderNoFailure s
z*DeferredResourceTests.test_renderNoFailureN)r3 r4 r5 r6 r r r r% r% r% r& r s
r ),r6 r twisted.internetr twisted.python.compatr twisted.python.failurer twisted.trial.unittestr r twisted.webr r twisted.web.errorr twisted.web.httpr
twisted.web.serverr twisted.web.templater r
r twisted.web.test.requesthelperr r twisted.web.utilr r r r r r r r r r r7 rJ r r r r r% r% r% r& s( ,
H7 ?