tf
; | 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 d de
Z% G d de Z& G d de
Z' G d de
Z( G d de j) Z* G d 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 * e Zd ZdZd Zd Zd Zd ZdS )RedirectToTestsz"
Tests for L{redirectTo}.
c l t t d }d|_ d}t || | |j t | |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 assertEqualcoder responseHeaders
getRawHeadersselfrequest targetURLs Y/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/web/test/test_util.pytest_headersAndCodez#RedirectToTests.test_headersAndCode& s ,..$// 5 9g&&& u--- # 1 1+ > >
# 1 1/ B B
( )
c t t d }d|_ d}| t t
|| 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$ s r( test_redirectToUnicodeURLz)RedirectToTests.test_redirectToUnicodeURL9 sC ,..$// 4 )ZGDDDDDr* c v t dg }t d| }d}| || dS )zq
Legitimate URLs are fully interpolated in the `redirectTo` response body without transformation
r* s https://twisted.org/s
click here
Nr r r r% r& htmlexpecteds r( test_legitimateRedirectz'RedirectToTests.test_legitimateRedirectB sH u%%17;;
x(((((r* c v t dg }t d| }d}| || dS )zm
Malicious URLs are HTML-escaped before interpolating them in the `redirectTo` response body
r* s? https://twisted.org/">sX
click here
Nr0 r1 s r( test_maliciousRedirectz&RedirectToTests.test_maliciousRedirectT sM u%% NPW
x(((((r* N)__name__
__module__ __qualname____doc__r) r. r4 r6 r* r( r r ! s]
&E E E) ) )$) ) ) ) )r* r c 4 e Zd ZdZdedefdZd Zd Zd ZdS ) ParentRedirectTestsz!
Test L{ParentRedirect}.
requestPathreturnc & t t d }d|_ | d d |_ t
} |j | |j d }|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 Location)
r r r lstripsplitprepathr renderr" r# )r% r> r& r headerslocations r( doLocationTestz"ParentRedirectTests.doLocationTestn s ,..$// %,,T2288>>!## )77 DD"""
r* 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/rA s /?biff=baffNr rH r% s r( test_locationRootz%ParentRedirectTests.test_locationRoot sT
,d.A.A$.G.GHHH ,d.A.A..Q.QRRRRRr* c | d| d | d| d dS )zt
A request for a resource one level down from the URL root produces
a redirect to the root.
rJ s /foos /foo?bar=sproiiingNrK rL s r( test_locationToRootz'ParentRedirectTests.test_locationToRoot s_
,d.A.A'.J.JKKK !4!45J!K!K
r* c | 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=bazNrK rL s r( test_locationUpOnez&ParentRedirectTests.test_locationUpOne s
0$2E2Eh2O2OPPP 0$2E2Ek2R2RSSS #T%8%89L%M%M
r* N) r7 r8 r9 r: bytesrH rM rO rQ r; r* r( r= r= i sp % E (S S S
r* r= c T e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
S )FailureElementTestszn
Tests for L{FailureElement} and related helpers which can render a
L{Failure} as an HTML string.
c d }|j j dz | _ | dS # t $ r0 t d | _ | j j d | _ Y dS w xY w)zO
Create a L{Failure} which can be used by the rendering tests.
c $ d} t | )NzThis is a problem) Exception)messages r( lineNumberProbeAlsoBrokenz.lineNumberProbeAlsoBroken s )GG$$ $r* T) captureVarsN)__code__co_firstlinenobase
BaseExceptionr failureframesframe)r% rY s r( setUpzFailureElementTests.setUp s
% % %
.6EI 1 % % ' ' ' ' ' 1 1 1"t444DL ,R0DJJJJ 1s
% 6A A c
4 t t t j t j d t j d dd }t d| }d}| | j | d |S ) zN
L{_SourceLineElement} renders a source line and line number.
lineNumberrE
sourceLine2 z print 'hello'Nu: 50 print 'hello'
zutf-8) r r
r divspanr addCallbackr encode)r% elementdr3 s r( test_sourceLineElementz*FailureElementTests.test_sourceLineElement s % ,777,9W9W9WXX
$ ( (
O
d&(@(@AAAr* c
4 t t t j t j d t j d d | j }g d}t
d| }d}t | D ]d\ }}d}|d k r2|| dd
g|d k | j |z d |z z
}?|| d| j |z d|z z
}e|
d }| | j | |S )
z
L{_SourceFragmentElement} renders source lines at and around the line
number indicated by a frame object.
rf rg rh sourceLines)u# message = "This is a problem"u raise Exception(message) Nrs z?{}{}
rZ Highlightu utf8)
r r
r rj rk rc r enumerateformatr_ rm rl r ) r% rn sourcero stringToCheckForrf rh templatebytesToCheckFors r( test_sourceFragmentElementz.FailureElementTests.test_sourceFragmentElement sT
) I\222I\222(
J
$ ( ( &/&7&7 "J
XHQ HOO%jAo6I
*.;% % !HOO J.j % % +11&99
d&888r* c t t t j d | j }t d| }| | j dt t
d z dz |S )z
The I{filename} renderer of L{_FrameElement} renders the filename
associated with the frame object used to initialize the
L{_FrameElement}.
filenamerg N c ) r r
r rk rc r rl r r __file__rstripr% rn ro s r( test_frameElementFilenamez-FailureElementTests.test_frameElementFilename s $):*F*F*F G GTT $ ( (
hooc&:&:;; ;j H
r* c t t t j d | j }t d| }| | j d| j dz fz |S )z
The I{lineNumber} renderer of L{_FrameElement} renders the line number
associated with the frame object used to initialize the
L{_FrameElement}.
rf rg Ns %drZ ) r r
r rk rc r rl r r_ r s r( test_frameElementLineNumberz/FailureElementTests.test_frameElementLineNumber sd $)<*H*H*H I I4:VV $ ( (
d&(:di!m=M(MNNNr* c t t t j d | j }t d| }| | j d |S )z
The I{function} renderer of L{_FrameElement} renders the line number
associated with the frame object used to initialize the
L{_FrameElement}.
functionrg Ns& lineNumberProbeAlsoBroken)r r
r rk rc r rl r r s r( test_frameElementFunctionz-FailureElementTests.test_frameElementFunction sV $):*F*F*F G GTT $ ( (
d&(QRRRr* c j t d| j }| d }t j } |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}.
Nrx ) r rc lookupRenderMethodr rj assertIsInstancer assertIdenticalr loaderloadr% rn renderertagresults r( test_frameElementSourcez+FailureElementTests.test_frameElementSource s dj11--h77hjj$$$ f&<=== V\4:666 #
2 2 4 455555r* c t d| j j dd }| d }t j } |d| }| |t | |d t | |d j
| j j d | |d t | |d j
| j j d | |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 rb r rZ )r ra rb r r rj r listr r rc assertNotEqualr r r lenr s r( test_stackElementz%FailureElementTests.test_stackElement) sD
dl&9"1"&=>>--h77hjj$$$ fd+++ fQi 777 VAY_dl.A!.DEEE fQi 777 VAY_dl.A!.DEEE F1I , 1 1 3 3VAY5E5J5J5L5LMMM CKK(((((r* c r t | j }| d }t j } |d| }| |t | |j | j j |
|g|j dS )z
The I{traceback} renderer of L{FailureElement} renders the failure's
stack frames using L{_StackElement}.
tracebackN)
r ra r r rj r r r stackFramesrb r r r r s r( test_failureElementTracebackz0FailureElementTests.test_failureElementTraceback; s
! ..--k::hjj$$$ fm444 V / 1DEEE #
2 2 4 455555r* c t | j t t j d }t d| }d}| | j d|z dz |S )zi
The I{type} renderer of L{FailureElement} renders the failure's
exception type.
typerg Ns builtins.Exceptionr r r ra r
r rk r rl r )r% rn ro excs r( test_failureElementTypez+FailureElementTests.test_failureElementTypeH sc
! y&9Q9Q9Q/R/RSS $ ( (#
d& C*(DEEEr* c t | j t t j d }t d| }| | j d |S )zi
The I{value} renderer of L{FailureElement} renders the value's exception
value.
valuerg Ns This is a problemr r s r( test_failureElementValuez,FailureElementTests.test_failureElementValueS sV
! y'9R9R9R/S/STT $ ( (
d&(IJJJr* N)r7 r8 r9 r: rd rp r| r r r r r r r r r; r* r( rT rT s
1 1 1$ &( ( (T 6 6 6 ) ) )$ 6 6 6 r* rT c e Zd ZdZd Zd ZdS )FormatFailureTestsz
Tests for L{twisted.web.util.formatFailure} which returns an HTML string
representing the L{Failure} instance passed to it.
c ` | t t t dS )z}
If there is an error flattening the L{Failure} instance,
L{formatFailure} raises L{FlattenerError}.
N)r, r
r objectrL s r( test_flattenerErrorz&FormatFailureTests.test_flattenerErrord s&
.-BBBBBr* c 6 t d # t $ r t t }Y nw xY w| |t
| t d |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 3 " K | ]
}|d k V dS ) Nr; ).0chs r( z7FormatFailureTests.test_returnsBytes..w s& 66BH666666r* s N) rW r` r r r rR
assertTrueallassertIn)r% r s r( test_returnsBytesz$FormatFailureTests.test_returnsBytesk s .J'' ' . . ."799--FFF .
fe,,, 66v66666777
i(((((s %99N)r7 r8 r9 r: r r r; r* r( r r ^ s?
C C C) ) ) ) )r* r c e Zd Zd Zd ZdS )
SDResourcec || _ d S N)default)r% r s r( __init__zSDResource.__init__} s
r* c ~ t j | j }t j | } |j || S r )r succeedr r r getChildWithDefault)r% namer& ro r s r( r z SDResource.getChildWithDefault s8 M$, ' '(+++x+D':::r* N)r7 r8 r9 r r r; r* r( r r | s2 ; ; ; ; ;r* r c $ e Zd ZdZd Zd Zd ZdS )DeferredResourceTestsz(
Tests for L{DeferredResource}.
c t j }d|_ t | }t g d }t j || | |j ddg d S )NrZ )foobarbazr r )r ResourceisLeafr r getChildForRequestr postpath)r% rsro s r( testDeferredResourcez*DeferredResourceTests.testDeferredResource sk qMM ... / /#Aq))) eU^44444r* c g }t g }|j |_ t j }t t
j | }| | | ||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 appendrE r r r r r r )r% renderedr& r deferredResources r( test_renderz!DeferredResourceTests.test_render sw r""! "$$+EM&,A,ABB ((( F8,,,,,r* c t g }| }t t }t t j | }| | | | | | ~t j | 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 failrE r failureResultOfgccollectflushLoggedErrors)r% r& ro ra r errorss r( test_renderNoFailurez*DeferredResourceTests.test_renderNoFailure s r"" " ",..))+EJw,?,?@@ ((( --a00':::
'' 55 $$$$$r* N)r7 r8 r9 r: r r r r; r* r( r r sK 5 5 5
-
-
- % % % % %r* r ),r: 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 r= rT r r r r r; r* r( r sx
" " " " " " / / / / / / * * * * * * @ @ @ @ @ @ @ @ & & & & & & & & , , , , , , " " " " " " & & & & & & ? ? ? ? ? ? ? ? ? ? E E E E E E E E
E) E) E) E) E)h E) E) E)P4
4
4
4
4
- 4
4
4
n{ { { { {( { { {|) ) ) ) ) ) ) )<; ; ; ; ;" ; ; ;+% +% +% +% +% / +% +% +% +% +%r*