o bf @sdZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZd d ZGd d d eZd dZgdZdS)z0 General helpers for L{twisted.web} unit tests. )Type)Deferredsucceed)SynchronousTestCase)server) flattenString)FlattenerError) FlattenablecCs\||}t|tr|||tdS|tjur'|jr#tdS| St d|)NzUnexpected return value: ) render isinstancebyteswritefinishrr NOT_DONE_YETfinished notifyFinish ValueError)resourcerequestresultr8/usr/lib/python3/dist-packages/twisted/web/test/_util.py_renders    rc@sZeZdZdZdededeefddZdededefddZded e e dd fd d Z d S) FlattenTestCasezH A test case that assists with testing L{twisted.web._flatten}. roottargetreturncs0dtdtffdd }td|}|||S)zS Assert that a root element, when flattened, is equal to a string. rrcs |S)N) assertEqual)rselfrrrcheck,s z/FlattenTestCase.assertFlattensTo..checkN)r r addCallback)rrrr drrrassertFlattensTo's  z FlattenTestCase.assertFlattensTocCs||||S)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} )successResultOfr#)rrrrrrassertFlattensImmediately3s z)FlattenTestCase.assertFlattensImmediatelyexnNcCs(|||dt}||jj|dS)zQ Assert flattening a root element raises a particular exception. N)failureResultOfr#rassertIsInstancevalue _exception)rrr&failurerrrassertFlatteningRaisesBsz&FlattenTestCase.assertFlatteningRaises) __name__ __module__ __qualname____doc__r r rr#r%r Exceptionr-rrrrr"s  rcCs||tdS)z Assert that C{fileObj} is a temporary file on the filesystem. @param case: A C{TestCase} instance to use to make the assertion. @raise: C{case.failureException} if C{fileObj} is not a temporary file on the filesystem. N)r)filenoint)casefileObjrrrassertIsFilesystemTemporaryJsr7)rrr7N)r1typingrtwisted.internet.deferrrtwisted.trial.unittestr twisted.webrtwisted.web._flattenrtwisted.web.errorrtwisted.web.templater rrr7__all__rrrrs      (