o b5@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@ eZdZdZddZddZdS)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&r?r&r&r&r'r0Ds  r0c@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#rDr&r&r'setUps   zFailureElementTests.setUpcCsNttttjddtjdddd}td|}d}||j|d|S) zN L{_SourceLineElement} renders a source line and line number. lineNumberr6 sourceLine2z print 'hello'Nu:
50    print 'hello'
zutf-8) rr rdivspanr addCallbackrencode)r#elementdexpectedr&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. rPrQrR sourceLines)u#    message = "This is a problem"u    raise Exception(message)Nr]z?
{}{}
rE Highlightu        utf8) rr rrTrUrNr enumerateformatrJrWrVr) r#rXsourcerYstringToCheckForrPrRtemplatebytesToCheckForr&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}. filenamerQNc) rr rrUrNr rVrr__file__rstripr#rXrYr&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}. rPrQNs%drE) rr rrUrNr rVrrJrmr&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}. functionrQNs&lineNumberProbeAlsoBroken)rr rrUrNr rVrrmr&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}. Nrb) rrNlookupRenderMethodrrTassertIsInstancerassertIdenticalrloaderloadr#rXrenderertagresultr&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}. NrMrrE)rrLrMrrrrTrslistrrtrNassertNotEqualrurvrlenrwr&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) rrLrrrrTrsrrt stackFramesrMrrurvrwr&r&r'test_failureElementTracebacks    z0FailureElementTests.test_failureElementTracebackcCs@t|jttjdd}td|}d}||jd|d|S)zi The I{type} renderer of L{FailureElement} renders the failure's exception type. typerQNsbuiltins.ExceptionrhrjrrLr rrUr rVr)r#rXrYexcr&r&r'test_failureElementType!s  z+FailureElementTests.test_failureElementTypecCs4t|jttjdd}td|}||jd|S)zi The I{value} renderer of L{FailureElement} renders the value's exception value. valuerQNsThis is a problemrrmr&r&r'test_failureElementValue,s z,FailureElementTests.test_failureElementValueN)r,r-r.r/rOr[rfrnrorqr{rrrrr&r&r&r'rAys*   rAc@r)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 robjectr<r&r&r'test_flattenerError=sz&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' Psz7FormatFailureTests.test_returnsBytes..s N) rBrKrrrsr@ assertTrueallassertIn)r#rzr&r&r'test_returnsBytesDs  z$FormatFailureTests.test_returnsBytesN)r,r-r.r/rrr&r&r&r'r7s rc@seZdZddZddZdS) SDResourcecCs ||_dSN)default)r#rr&r&r'__init__Vs zSDResource.__init__cCs"t|j}t|}|||Sr)rsucceedrrrgetChildWithDefault)r#namer$rYrr&r&r'rYs   zSDResource.getChildWithDefaultN)r,r-r.rrr&r&r&r'rUs rc@s(eZdZdZddZddZddZdS) DeferredResourceTestsz( Tests for L{DeferredResource}. cCsDt}d|_t|}tgd}t||||jddgdS)NrE)foobarbazrr)rResourceisLeafrrgetChildForRequestrpostpath)r#rsrYr&r&r'testDeferredResourceds   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) rappendr6rrrrrr)r#renderedr$rzdeferredResourcer&r&r' test_renderls 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 RuntimeErrorrrfailr6rfailureResultOfgccollectflushLoggedErrors)r#r$rYrLrerrorsr&r&r'test_renderNoFailure{s   z*DeferredResourceTests.test_renderNoFailureN)r,r-r.r/rrrr&r&r&r'r_s  r),r/rtwisted.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.utilrrrrrrrrrrr0rArrrrr&r&r&r's(      , "5?