o pj:@s*dZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$Gddde Z%Gddde Z&Gddde Z'Gddde Z(Gddde j)Z*Gddde 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@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) r rmethodr assertEqualcoder responseHeaders getRawHeadersselfrequest targetURLr% click here Nrrrr"r#htmlexpectedr%r%r&test_legitimateRedirectBs   z'RedirectToTests.test_legitimateRedirectcCr+)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_maliciousRedirectTs  z&RedirectToTests.test_maliciousRedirectN)__name__ __module__ __qualname____doc__r'r*r1r2r%r%r%r&r!s  rc@s6eZdZdZdefddZddZddZd d Zd S) ParentRedirectTestsz! Test L{ParentRedirect}. requestPathcCsFttd}d|_|dd|_t}|||j d\}|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 rrlstripsplitprepathrrenderrr )r"r8r#rlocationr%r%r&doLocationTestns  z"ParentRedirectTests.doLocationTestcC(|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=baffNrr?r"r%r%r&test_locationRootsz%ParentRedirectTests.test_locationRootcCr@)zt A request for a resource one level down from the URL root produces a redirect to the root. rAs/foos/foo?bar=sproiiingNrBrCr%r%r&test_locationToRoots z'ParentRedirectTests.test_locationToRootcCs:|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=bazNrBrCr%r%r&test_locationUpOnes  z&ParentRedirectTests.test_locationUpOneN) r3r4r5r6bytesr?rDrErFr%r%r%r&r7is  r7c@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. cCsRdd}|jjd|_z|WdSty(tdd|_|jjd|_YdSw)zO Create a L{Failure} which can be used by the rendering tests. cSs d}t|)NzThis is a problem) Exception)messager%r%r&lineNumberProbeAlsoBrokensz.lineNumberProbeAlsoBrokenT) captureVarsN)__code__co_firstlinenobase BaseExceptionrfailureframesframe)r"rKr%r%r&setUps   zFailureElementTests.setUpcCsNttttjddtjdddd}td|}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 addCallbackrencode)r"elementdr0r%r%r&test_sourceLineElements z*FailureElementTests.test_sourceLineElementc Cstttjtjddtjdddd|j}gd}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. rWrXrY sourceLines)u#    message = "This is a problem"u    raise Exception(message)Nrcz?
{}{}
rL Highlightu        utf8) rr rr[r\rUr enumerateformatrQr^r]r) r"r_sourcer`stringToCheckForrWrYtemplatebytesToCheckForr%r%r&test_sourceFragmentElements6    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}. filenamerXNc) rr rr\rUr r]rr__file__rstripr"r_r`r%r%r&test_frameElementFilenames z-FailureElementTests.test_frameElementFilenamecCs@tttjdd|j}td|}||jd|jdf|S)z The I{lineNumber} renderer of L{_FrameElement} renders the line number associated with the frame object used to initialize the L{_FrameElement}. rWrXNs%drL) rr rr\rUr r]rrQrsr%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}. functionrXNs&lineNumberProbeAlsoBroken)rr rr\rUr r]rrsr%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}. Nrh) rrUlookupRenderMethodrr[assertIsInstancerassertIdenticalrloaderloadr"r_renderertagresultr%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}. NrTrrL)rrSrTrxrr[rylistrrzrUassertNotEqualr{r|rlenr}r%r%r&test_stackElement's    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) rrSrxrr[ryrrz stackFramesrTrr{r|r}r%r%r&test_failureElementTraceback9s    z0FailureElementTests.test_failureElementTracebackcCs@t|jttjdd}td|}d}||jd|d|S)zi The I{type} renderer of L{FailureElement} renders the failure's exception type. typerXNsbuiltins.ExceptionrnrprrSr rr\r r]r)r"r_r`excr%r%r&test_failureElementTypeFs  z+FailureElementTests.test_failureElementTypecCs4t|jttjdd}td|}||jd|S)zi The I{value} renderer of L{FailureElement} renders the value's exception value. valuerXNsThis is a problemrrsr%r%r&test_failureElementValueQs z,FailureElementTests.test_failureElementValueN)r3r4r5r6rVrarlrtrurwrrrrrr%r%r%r&rHs*   rHc@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 robjectrCr%r%r&test_flattenerErrorbsz&FormatFailureTests.test_flattenerErrorcCsZztdtytt}Ynw||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 bugcss|]}|dkVqdS)Nr%).0chr%r%r& usz7FormatFailureTests.test_returnsBytes..s N) rIrRrrryrG assertTrueallassertIn)r"rr%r%r&test_returnsBytesis  z$FormatFailureTests.test_returnsBytesN)r3r4r5r6rrr%r%r%r&r\s rc@seZdZddZddZdS) SDResourcecCs ||_dSN)default)r"rr%r%r&__init__{s zSDResource.__init__cCs"t|j}t|}|||Sr)rsucceedrrrgetChildWithDefault)r"namer#r`rr%r%r&r~s   zSDResource.getChildWithDefaultN)r3r4r5rrr%r%r%r&rzs rc@s(eZdZdZddZddZddZdS) DeferredResourceTestsz( Tests for L{DeferredResource}. cCsDt}d|_t|}tgd}t||||jddgdS)NrL)foobarbazrr)rResourceisLeafrrgetChildForRequestrpostpath)r"rsr`r%r%r&testDeferredResources   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) rappendr=rrrrrr)r"renderedr#rdeferredResourcer%r%r& test_renders 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) r notifyFinishr RuntimeErrorrrfailr=rfailureResultOfgccollectflushLoggedErrors)r"r#r`rSrerrorsr%r%r&test_renderNoFailures   z*DeferredResourceTests.test_renderNoFailureN)r3r4r5r6rrrr%r%r%r&rs  r),r6rtwisted.internetrtwisted.python.compatrtwisted.python.failurertwisted.trial.unittestrr twisted.webrrtwisted.web.errorr twisted.web.httpr twisted.web.serverr twisted.web.templater r rtwisted.web.test.requesthelperrrtwisted.web.utilrrrrrrrrrrr7rHrrrrr%r%r%r&s(      , H5?