U  W[ ã@s€dZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZd d „ZGd d „d e ƒZd d gZd S)z0 General helpers for L{twisted.web} unit tests. é)ÚdivisionÚabsolute_import)Úsucceed)Úserver)ÚTestCase)ÚFailure)Ú flattenString)ÚFlattenerErrorcCsb| |¡}t|tƒr.| |¡| ¡tdƒS|tjkrP|jrFtdƒS|  ¡Snt d|fƒ‚dS)NzUnexpected return value: %r) ZrenderÚ isinstanceÚbytesÚwriteZfinishrrZ NOT_DONE_YETZfinishedZ notifyFinishÚ ValueError)ÚresourceZrequestÚresult©rú8/usr/lib/python3/dist-packages/twisted/web/test/_util.pyÚ_renders     rc@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚFlattenTestCasezH A test case that assists with testing L{twisted.web._flatten}. cs"td|ƒ}| ‡‡fdd„¡|S)zS Assert that a root element, when flattened, is equal to a string. Ncs ˆ |ˆ¡S©N)Z assertEqual)Ús©ÚselfÚtargetrrÚ-óz2FlattenTestCase.assertFlattensTo..)rÚ addCallback)rÚrootrÚdrrrÚassertFlattensTo(s z FlattenTestCase.assertFlattensTocCsXg}| ||¡}| |j¡| |j¡|s6| d¡|d}t|tƒrP| ¡|dS)aÖ Assert that a root element, when flattened, is equal to a string, and performs no asynchronus Deferred anything. This version is more convenient in tests which wish to make multiple assertions about flattening, since it can be called multiple times without having to add multiple callbacks. @return: the result of rendering L{root}, which should be equivalent to L{target}. @rtype: L{bytes} z'Rendering did not complete immediately.r) rZaddBothÚappendZ addCleanupZcancelZfailr rZraiseException)rrrZresultsÚitrrrrÚassertFlattensImmediately1s      z)FlattenTestCase.assertFlattensImmediatelycs,ˆ ˆ |d¡t¡}| ‡‡fdd„¡|S)zQ Assert flattening a root element raises a particular exception. rcsˆ |jˆ¡Sr)ZassertIsInstanceZ _exception)Úexc©ÚexnrrrrPrz8FlattenTestCase.assertFlatteningRaises..)Z assertFailurerr r)rrr$rrr#rÚassertFlatteningRaisesKsz&FlattenTestCase.assertFlatteningRaisesN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr!r%rrrrr$s rN)r)Z __future__rrZtwisted.internet.deferrZ twisted.webrZtwisted.trial.unittestrZtwisted.python.failurerZtwisted.web._flattenrZtwisted.web.errorr rrÚ__all__rrrrÚs      0