Your IP : 216.73.216.213
B
;��]� � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ d d
� ZG d d � d e �Zd
d� Zd
d dgZdS )z0
General helpers for L{twisted.web} unit tests.
� )�division�absolute_import)�succeed)�server)�TestCase)�Failure)�
flattenString)�FlattenerErrorc C sb | � |�}t|t�r.|�|� |�� td �S |tjkrP|jrFtd �S |� � S nt
d|f ��d S )NzUnexpected return value: %r) Zrender�
isinstance�bytes�writeZfinishr r Z NOT_DONE_YET�finishedZ notifyFinish�
ValueError)�resourceZrequest�result� r �I/opt/alt/python37/lib64/python3.7/site-packages/twisted/web/test/_util.py�_render s
r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FlattenTestCasezH
A test case that assists with testing L{twisted.web._flatten}.
c s" t d|�}|�� �fdd�� |S )zS
Assert that a root element, when flattened, is equal to a string.
Nc s � � | ��S )N)Z assertEqual)�s)�self�targetr r �<lambda>- � z2FlattenTestCase.assertFlattensTo.<locals>.<lambda>)r � addCallback)r �rootr �dr )r r r �assertFlattensTo( s
z FlattenTestCase.assertFlattensToc C sX g }| � ||�}|�|j� | �|j� |s6| �d� |d }t|t�rP|�� |d 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}
z'Rendering did not complete immediately.r ) r ZaddBoth�appendZ
addCleanup�cancelZfailr
r ZraiseException)r r r �results�itr r r r �assertFlattensImmediately1 s
z)FlattenTestCase.assertFlattensImmediatelyc s, �� ��|d�t�}|�� �fdd�� |S )zQ
Assert flattening a root element raises a particular exception.
r c s �� | j� �S )N)�assertIsInstance�
_exception)�exc)�exnr r r r P r z8FlattenTestCase.assertFlatteningRaises.<locals>.<lambda>)Z
assertFailurer r r )r r r& r r )r& r r �assertFlatteningRaisesK s z&FlattenTestCase.assertFlatteningRaisesN)�__name__�
__module__� __qualname__�__doc__r r" r' r r r r r $ s r c C s | � |�� t� dS )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# �fileno�int)ZcaseZfileObjr r r �assertIsFilesystemTemporaryU s r. N)r+ �
__future__r r Ztwisted.internet.deferr Z twisted.webr Ztwisted.trial.unittestr Ztwisted.python.failurer Ztwisted.web._flattenr Ztwisted.web.errorr r r r. �__all__r r r r �<module> s 1