U s@gL6@sFdZddlmZmZddlZddlmZddlmZm Z ddl m Z ddl m Z mZmZddlmZmZdd lmZdd lmZdd lmZdd lmZmZmZdd lmZmZddl m!Z!ddl m"Z"m#Z#ddl m$Z$m%Z%m&Z&ddl m'Z'm(Z(Gddde Z)Gddde Z*Gddde Z+Gdddej,Z-GdddeZ.dS)z Tests for L{twisted.web.util}. )absolute_importdivisionN)Failure)SynchronousTestCaseTestCase)defer)_PY3 intToBytes networkString)resourceutil)FlattenerError)FOUND)Request) TagLoader flattenStringtags) DummyChannel DummyRequest)DeferredResource)_SourceFragmentElement _FrameElement) _StackElementFailureElement formatFailure) redirectTo_SourceLineElementc@s0eZdZdZddZddZddZdd Zd S) RedirectToTestsz" Tests for L{redirectTo}. cCs^ttd}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) rrmethodr assertEqualcoderZresponseHeadersZ getRawHeadersselfrequestZ targetURLr% click here Nrrr r#r$Zhtmlexpectedr%r%r&test_legitimateRedirect=s   z'RedirectToTests.test_legitimateRedirectcCs(tdg}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_maliciousRedirectOs  z&RedirectToTests.test_maliciousRedirectN)__name__ __module__ __qualname____doc__r'r*r/r0r%r%r%r&rs  rc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)FailureElementTestszn Tests for L{FailureElement} and related helpers which can render a L{Failure} as an HTML string. cCsLdd}|jjd|_z |Wn&tdd|_|jjd|_YnXdS)zO Create a L{Failure} which can be used by the rendering tests. cSsd}t|dS)NzThis is a problem) Exception)messager%r%r&lineNumberProbeAlsoBrokennsz.lineNumberProbeAlsoBrokenT)Z captureVarsN)__code__co_firstlinenobaserfailureframesframe)r#r8r%r%r&setUpjs  zFailureElementTests.setUpcCsNttttjddtjdddd}td|}d}||j|d|S) zN L{_SourceLineElement} renders a source line and line number. lineNumberrender sourceLine2z print 'hello'Nu:
50    print 'hello'
zutf-8) rrrdivspanr addCallbackr encode)r#elementdr.r%r%r&test_sourceLineElement{s    z*FailureElementTests.test_sourceLineElementcstttjtjddtjddddj}dddg}td|}trhd fd d t |D d }nd fd d t |D}| j ||S)z L{_SourceFragmentElement} renders source lines at and around the line number indicated by a frame object. rBrCrEZ sourceLinesu#    message = "This is a problem"u    raise Exception(message)zE# Figure out the line number from which the exception will be raised.Ncs4g|],\}}dddg|dkj|d|fqS)?
%d%s
rN Highlightr9        )r=.0rBrEr#r%r& szBFailureElementTests.test_sourceFragmentElement..utf8cs:g|]2\}}dddg|dkj|d|dfqS)rOrNrPr9rQrV)r=rJrRrTr%r&rUs ) rrrrGrHr@rrjoin enumeraterJrIr )r#rKsourcerLZstringToCheckForr%rTr&test_sourceFragmentElements0     z.FailureElementTests.test_sourceFragmentElementcCsFtttjdd|j}td|}||jdtt dd|S)z The I{filename} renderer of L{_FrameElement} renders the filename associated with the frame object used to initialize the L{_FrameElement}. filenamerCNc) rrrrHr@rrIr r __file__rstripr#rKrLr%r%r&test_frameElementFilenames z-FailureElementTests.test_frameElementFilenamecCsFtttjdd|j}td|}||jdt|j dd|S)z The I{lineNumber} renderer of L{_FrameElement} renders the line number associated with the frame object used to initialize the L{_FrameElement}. rBrCNr\r9r^) rrrrHr@rrIr r r=rar%r%r&test_frameElementLineNumbers z/FailureElementTests.test_frameElementLineNumbercCs4tttjdd|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}. ZfunctionrCNs&lineNumberProbeAlsoBroken)rrrrHr@rrIr rar%r%r&test_frameElementFunctions z-FailureElementTests.test_frameElementFunctioncCs\td|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}. NrY) rr@lookupRenderMethodrrGassertIsInstancerassertIdenticalr loaderloadr#rKZrenderertagresultr%r%r&test_frameElementSources    z+FailureElementTests.test_frameElementSourcecCstd|jjdd}|d}t}|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}. Nr?rr9)rr>r?rerrGrflistrrgr@ZassertNotEqualrhrir lenrjr%r%r&test_stackElements    z%FailureElementTests.test_stackElementcCs\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) rr>rerrGrfrrgZ stackFramesr?r rhrirjr%r%r&test_failureElementTraceback s    z0FailureElementTests.test_failureElementTracebackcCsJt|jttjdd}td|}tr,d}nd}||jd|d|S)zi The I{type} renderer of L{FailureElement} renders the failure's exception type. typerCNsbuiltins.Exceptionsexceptions.Exceptionr\r^) rr>rrrHrrrIr )r#rKrLexcr%r%r&test_failureElementTypes  z+FailureElementTests.test_failureElementTypecCs4t|jttjdd}td|}||jd|S)zi The I{value} renderer of L{FailureElement} renders the value's exception value. valuerCNsThis is a problem)rr>rrrHrrIr rar%r%r&test_failureElementValue+s z,FailureElementTests.test_failureElementValueN)r1r2r3r4rArMrZrbrcrdrmrqrsrvrxr%r%r%r&r5es-r5c@s eZdZdZddZddZdS)FormatFailureTestsz Tests for L{twisted.web.util.formatFailure} which returns an HTML string representing the L{Failure} instance passed to it. cCs|tttdS)z} If there is an error flattening the L{Failure} instance, L{formatFailure} raises L{FlattenerError}. N)r(r robjectrTr%r%r&test_flattenerError>sz&FormatFailureTests.test_flattenerErrorcCsvz tdWntt}YnX||ttrN|tdd|Dn|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 bugcss|]}|dkVqdSNr%rSZchr%r%r& Ssz7FormatFailureTests.test_returnsBytes..css|]}t|dkVqdSr|)ordr~r%r%r&rUss N) r6rrrfbytesrZ assertTrueallZassertIn)r#rlr%r%r&test_returnsBytesFs  z$FormatFailureTests.test_returnsBytesN)r1r2r3r4r{rr%r%r%r&ry9sryc@seZdZddZddZdS) SDResourcecCs ||_dSN)default)r#rr%r%r&__init__\szSDResource.__init__cCs"t|j}t|}|||Sr)rsucceedrr rgetChildWithDefault)r#namer$rLr r%r%r&r`s  zSDResource.getChildWithDefaultN)r1r2r3rrr%r%r%r&r[src@s(eZdZdZddZddZddZdS) DeferredResourceTestsz( Tests for L{DeferredResource}. cCsFt}d|_t|}tdddg}t||||jddgdS)Nr9ZfooZbarZbaz)r ResourceZisLeafrrZgetChildForRequestr Zpostpath)r#rsrLr%r%r&testDeferredResourcels  z*DeferredResourceTests.testDeferredResourcecCsFg}tg}|j|_t}tt|}|||||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) rappendrDr rrrrr )r#Zrenderedr$rldeferredResourcer%r%r& test_renderus z!DeferredResourceTests.test_rendercCshtg}|}tt}tt|}|||| ||~t | 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) rZ notifyFinishr RuntimeErrorrrZfailrDr ZfailureResultOfgcZcollectZflushLoggedErrors)r#r$rLr>rerrorsr%r%r&test_renderNoFailures   z*DeferredResourceTests.test_renderNoFailureN)r1r2r3r4rrrr%r%r%r&rgs r)/r4Z __future__rrrZtwisted.python.failurerZtwisted.trial.unittestrrZtwisted.internetrZtwisted.python.compatrr r Z twisted.webr r Ztwisted.web.errorr Ztwisted.web.httprZtwisted.web.serverrZtwisted.web.templaterrrZtwisted.web.test.requesthelperrrZtwisted.web.utilrrrrrrrrrr5ryrrrr%r%r%r&s,      HU"