/Jf5|dZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$Gdde Z%Gdde Z&Gdde Z'Gdde Z(Gdde j)Z*Gdde 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 redirectToceZdZdZdZdZdS)RedirectToTestsz" Tests for L{redirectTo}. clttd}d|_d}t||||jt ||jd|g||jddgdS)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. TGETshttp://target.example.com/4321slocations content-typestext/html; charset=utf-8N) r rmethodr assertEqualcoder responseHeaders getRawHeadersselfrequest targetURLs ]/home/alex/cs2snipeproduction/venv/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 ( )     cttd}d|_d}|tt ||dS)zW L{redirectTo} will raise TypeError if unicode object is passed in URL Trzhttp://target.example.com/4321N)r rr assertRaises TypeErrorrr$s r(test_redirectToUnicodeURLz)RedirectToTests.test_redirectToUnicodeURL:sC,..$//4  )ZGDDDDDr*N)__name__ __module__ __qualname____doc__r)r.r*r(rr"sA   &EEEEEr*rc4eZdZdZdedefdZdZdZdZdS) ParentRedirectTestsz! Test L{ParentRedirect}. requestPathreturnc&ttd}d|_|dd|_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/sLocation) r rrlstripsplitprepathrrenderr"r#)r%r6r&rheaderslocations r(doLocationTestz"ParentRedirectTests.doLocationTestIs,..$//%,,T2288>>!##   )77 DD""" r*c|d|d|d|ddS)zi At the URL root issue a redirect to the current URL, removing any query string. http://10.0.0.1/r9s /?biff=baffNr r@r%s r(test_locationRootz%ParentRedirectTests.test_locationRoot]sT ,d.A.A$.G.GHHH ,d.A.A..Q.QRRRRRr*c|d|d|d|ddS)zt A request for a resource one level down from the URL root produces a redirect to the root. rBs/foos/foo?bar=sproiiingNrCrDs r(test_locationToRootz'ParentRedirectTests.test_locationToRootes_ ,d.A.A'.J.JKKK  !4!45J!K!K     r*c|d|d|d|d|d|ddS)zp Requests for resources directly under the path C{/foo/} produce redirects to C{/foo/}. shttp://10.0.0.1/foo/s/foo/s/foo/bars/foo/bar?biz=bazNrCrDs r(test_locationUpOnez&ParentRedirectTests.test_locationUpOneos 0$2E2Eh2O2OPPP 0$2E2Ek2R2RSSS  #T%8%89L%M%M     r*N) r/r0r1r2bytesr@rErGrIr3r*r(r5r5Dsp%E(SSS         r*r5cTeZdZdZdZdZdZdZdZdZ dZ 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. cd}|jjdz|_ |dS#t$r0t d|_|jjd|_YdSwxYw)zO Create a L{Failure} which can be used by the rendering tests. c$d}t|)NzThis is a problem) Exception)messages r(lineNumberProbeAlsoBrokenz.lineNumberProbeAlsoBrokens)GG$$ $r*T) captureVarsN)__code__co_firstlinenobase BaseExceptionrfailureframesframe)r%rQs r(setUpzFailureElementTests.setUps  % % % .6EI  1 % % ' ' ' ' ' 1 1 1"t444DL,R0DJJJJ 1s %6AAc 4tttjtjdtjddd}t d|}d}||j|d|S) zN L{_SourceLineElement} renders a source line and line number. lineNumberr= sourceLine2z print 'hello'Nu:
50    print 'hello'
zutf-8) rr rdivspanr addCallbackr encode)r%elementdexpecteds r(test_sourceLineElementz*FailureElementTests.test_sourceLineElements% ,777,9W9W9WXX        $ ( ( O  d&(@(@AAAr*c 4tttjtjdtjdd|j}gd}t d|}d}t|D]d\}}d}|d kr2||dd g|d k|j |zd |zz }?||d|j |zd|zz }e| d }| |j ||S) z L{_SourceFragmentElement} renders source lines at and around the line number indicated by a frame object. r^r_r` sourceLines)u#    message = "This is a problem"u    raise Exception(message)Nrlz?
{}{}
rR Highlightu        utf8) rr rrbrcr[r enumerateformatrWrerdr ) r%rfsourcergstringToCheckForr^r`templatebytesToCheckFors r(test_sourceFragmentElementz.FailureElementTests.test_sourceFragmentElementsT ) I\222I\222(   J       $ ( (&/&7&7   "J XHQ HOO%jAo6I *.;%%  !HOO J.j%%  +11&99 d&888r*c tttjd|j}t d|}||jdtt dzdz|S)z The I{filename} renderer of L{_FrameElement} renders the filename associated with the frame object used to initialize the L{_FrameElement}. filenamer_Nc) rr rrcr[rrdr r__file__rstripr%rfrgs r(test_frameElementFilenamez-FailureElementTests.test_frameElementFilenames  $):*F*F*F G GTT $ ( (     hooc&:&:;; ;j H    r*ctttjd|j}t d|}||jd|jdzfz|S)z The I{lineNumber} renderer of L{_FrameElement} renders the line number associated with the frame object used to initialize the L{_FrameElement}. r^r_Ns%drR) rr rrcr[rrdr rWr}s r(test_frameElementLineNumberz/FailureElementTests.test_frameElementLineNumbersd  $)<*H*H*H I I4:VV $ ( ( d&(:di!m=M(MNNNr*ctttjd|j}t d|}||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}. functionr_Ns&lineNumberProbeAlsoBroken)rr rrcr[rrdr r}s r(test_frameElementFunctionz-FailureElementTests.test_frameElementFunctionsV  $):*F*F*F G GTT $ ( ( d&(QRRRr*cjtd|j}|d}tj}|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}. Nrq) rr[lookupRenderMethodrrbassertIsInstancerassertIdenticalr loaderloadr%rfrenderertagresults r(test_frameElementSourcez+FailureElementTests.test_frameElementSources  dj11--h77hjj$$$ f&<=== V\4:666 # 2 2 4 455555r*ctd|jjdd}|d}t j}|d|}||t||dt| |dj |jjd||dt| |dj |jjd| |dj |dj |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}. NrZrrR)rrYrZrrrbrlistrrr[assertNotEqualrrr lenrs r(test_stackElementz%FailureElementTests.test_stackElementsD  dl&9"1"&=>>--h77hjj$$$ fd+++ fQi777 VAY_dl.A!.DEEE fQi777 VAY_dl.A!.DEEE F1I,1133VAY5E5J5J5L5LMMM CKK(((((r*crt|j}|d}tj}|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) rrYrrrbrrr stackFramesrZr rrrs r(test_failureElementTracebackz0FailureElementTests.test_failureElementTracebacks !..--k::hjj$$$ fm444 V/1DEEE # 2 2 4 455555r*ct|jttjd}t d|}d}||jd|zdz|S)zi The I{type} renderer of L{FailureElement} renders the failure's exception type. typer_Nsbuiltins.ExceptionrxrzrrYr rrcrrdr )r%rfrgexcs r(test_failureElementTypez+FailureElementTests.test_failureElementType#sc !y&9Q9Q9Q/R/RSS $ ( (# d& C*(DEEEr*ct|jttjd}t d|}||jd|S)zi The I{value} renderer of L{FailureElement} renders the value's exception value. valuer_NsThis is a problemrr}s r(test_failureElementValuez,FailureElementTests.test_failureElementValue.sV !y'9R9R9R/S/STT $ ( ( d&(IJJJr*N)r/r0r1r2r\rirur~rrrrrrrr3r*r(rLrL{s 111$&(((T       6 6 6)))$ 6 6 6   r*rLceZdZdZdZdZdS)FormatFailureTestsz Tests for L{twisted.web.util.formatFailure} which returns an HTML string representing the L{Failure} instance passed to it. c`|tttdS)z} If there is an error flattening the L{Failure} instance, L{formatFailure} raises L{FlattenerError}. N)r,r robjectrDs r(test_flattenerErrorz&FormatFailureTests.test_flattenerError?s& .-BBBBBr*c6 td#t$rtt}YnwxYw||t |td|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 bugc3"K|] }|dkV dS)Nr3).0chs r( z7FormatFailureTests.test_returnsBytes..Rs&66BH666666r*s N) rOrXrrrrJ assertTrueallassertIn)r%rs r(test_returnsBytesz$FormatFailureTests.test_returnsBytesFs  .J'' ' . . ."799--FFF . fe,,, 66v66666777 i(((((s %99N)r/r0r1r2rrr3r*r(rr9s? CCC)))))r*rceZdZdZdZdS) SDResourcec||_dSN)default)r%rs r(__init__zSDResource.__init__Xs  r*c~tj|j}tj|}|j||Sr)rsucceedrr rgetChildWithDefault)r%namer&rgrs r(rzSDResource.getChildWithDefault[s8 M$, ' '(+++x+D':::r*N)r/r0r1rrr3r*r(rrWs2;;;;;r*rc$eZdZdZdZdZdZdS)DeferredResourceTestsz( Tests for L{DeferredResource}. ctj}d|_t|}t gd}tj||||jddgdS)NrR)foobarbazrr)rResourceisLeafrrgetChildForRequestr postpath)r%rsrgs r(testDeferredResourcez*DeferredResourceTests.testDeferredResourcefsk     qMM ... / /#Aq))) eU^44444r*cg}tg}|j|_tj}t t j|}|||||gdS)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) rappendr=rrrrrr )r%renderedr&rdeferredResources r( test_renderz!DeferredResourceTests.test_rendernsw r""!"$$+EM&,A,ABB((( F8,,,,,r*ctg}|}tt}t t j|}|||| ||~tj | t}||gdS)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 RuntimeErrorrrfailr=r failureResultOfgccollectflushLoggedErrors)r%r&rgrYrerrorss r(test_renderNoFailurez*DeferredResourceTests.test_renderNoFailure}s r""  " ",..))+EJw,?,?@@((( --a00':::  '' 55 $$$$$r*N)r/r0r1r2rrrr3r*r(rrasK555 - - -%%%%%r*r),r2rtwisted.internetrtwisted.python.compatrtwisted.python.failurertwisted.trial.unittestrr twisted.webrr twisted.web.errorr twisted.web.httpr twisted.web.serverr twisted.web.templater rrtwisted.web.test.requesthelperrrtwisted.web.utilrrrrrrrrrrr5rLrrrrr3r*r(rsx  """"""//////******@@@@@@@@&&&&&&&&,,,,,,""""""&&&&&&??????????EEEEEEEE                      EEEEEhEEED4 4 4 4 4 -4 4 4 n{{{{{({{{|))))))))<;;;;;";;;+%+%+%+%+%/+%+%+%+%+%r*