File: //usr/lib/python3/dist-packages/twisted/web/test/__pycache__/test_flatten.cpython-38.pyc
U
W[SG � @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ dd lmZmZmZ dd
lmZ dd lmZmZmZ dd lmZmZm Z m Z m Z m Z dd
lm Z m!Z!m"Z"m#Z# ddl$m%Z% G dd� de%e�Z&dd� j'j(Z)G dd� de �Z*dS )zf
Tests for the flattening portion of L{twisted.web.template}, implemented in
L{twisted.web._flatten}.
� N)�XML)� OrderedDict)� implementer)� _PY35PLUS)�TestCase)�XMLAssertionMixin)�passthru�succeed�
gatherResults)� IRenderable)� UnfilledSlot�UnsupportedType�FlattenerError)�tags�Tag�Comment�CDATA�CharRef�slot)�Element�renderer� TagLoader�
flattenString)�FlattenTestCasec @ s e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Ze s�d2e_ d3d4� Z e s�d2e _ d5d6� Z d7d8� Z d9d:� Z!d;d<� Z"d=S )>�SerializationTestsz.
Tests for flattening various things.
c C s | � tjt�d�dd�d�S )z:
Test that nested tags flatten correctly.
Z42Zthere)�his' <html hi="there"><body>42</body></html>)�assertFlattensTor ZhtmlZbody��self� r �?/usr/lib/python3/dist-packages/twisted/web/test/test_flatten.py�test_nestedTags' s �z"SerializationTests.test_nestedTagsc C s t | �dd�| �dd�g�S )zL
Test that strings will be flattened and escaped correctly.
Zones onez
<abc&&>123s <abc&&>123)r
r r r r r �test_serializeString0 s
�z'SerializationTests.test_serializeStringc C s | � t�� d�S )zP
The serialized form of a self-closing tag is C{'<tagName />'}.
s <img />)r r �imgr r r r � test_serializeSelfClosingTags: s z0SerializationTests.test_serializeSelfClosingTagsc C s | � tjdd�d� dS )zV
The serialized form of attribute I{a} with value I{b} is C{'a="b"'}.
�foo��srcs <img src="foo" />N��assertFlattensImmediatelyr r# r r r r �test_serializeAttributeA s �z*SerializationTests.test_serializeAttributec C s& t �� }tddg�|_| �|d� dS )zg
Multiple attributes are separated by a single space in their serialized
form.
)r' r% )�nameZbars <img src="foo" name="bar" />N)r r# r �
attributesr) �r �tagr r r �!test_serializedMultipleAttributesI s z4SerializationTests.test_serializedMultipleAttributesc C s | � |tj|d�d��d� dS )au
Common implementation of L{test_serializedAttributeWithSanitization}
and L{test_serializedDeferredAttributeWithSanitization},
L{test_serializedAttributeWithTransparentTag}.
@param wrapData: A 1-argument callable that wraps around the
attribute's value so other tests can customize it.
@param wrapData: callable taking L{bytes} and returning something
flattenable
@param wrapTag: A 1-argument callable that wraps around the outer tag
so other tests can customize it.
@type wrapTag: callable taking L{Tag} and returning L{Tag}.
�<>&"r& s! <img src="<>&"" />Nr( )r ZwrapData�wrapTagr r r �checkAttributeSanitizationS s �z-SerializationTests.checkAttributeSanitizationc C s | � tt� dS )z�
Attribute values containing C{"<"}, C{">"}, C{"&"}, or C{'"'} have
C{"<"}, C{">"}, C{"&"}, or C{"""} substituted for those
bytes in the serialized output.
N)r2 r r r r r �(test_serializedAttributeWithSanitizationg s z;SerializationTests.test_serializedAttributeWithSanitizationc C s | � tt� dS )z�
Like L{test_serializedAttributeWithSanitization}, but when the contents
of the attribute are in a L{Deferred
<twisted.internet.defer.Deferred>}.
N)r2 r r r r r r �0test_serializedDeferredAttributeWithSanitizationp s zCSerializationTests.test_serializedDeferredAttributeWithSanitizationc s$ g � | � � fdd�� fdd�� dS )zS
Like L{test_serializedAttributeWithSanitization} but with a slot.
c s � � | �ptd�S )N�stuff)�appendr ��value��tossr r �<lambda> � zUSerializationTests.test_serializedAttributeWithSlotWithSanitization.<locals>.<lambda>c s | j � �� d�S )N)r5 )� fillSlots�pop�r. r9 r r r; � r< N)r2 r r r9 r �0test_serializedAttributeWithSlotWithSanitizationy s
�zCSerializationTests.test_serializedAttributeWithSlotWithSanitizationc C s | � tjt� dS )z�
Attribute values which are supplied via the value of a C{t:transparent}
tag have the same substitution rules to them as values supplied
directly.
N)r2 r � transparentr r r r r �*test_serializedAttributeWithTransparentTag� s z=SerializationTests.test_serializedAttributeWithTransparentTagc s: G � fdd�dt �� g �| ��fdd�� �fdd�� dS )z�
Like L{test_serializedAttributeWithTransparentTag}, but when the
attribute is rendered by a renderer on an element.
c s* e Zd Z�� fdd�Zedd� �Z� ZS )z_SerializationTests.test_serializedAttributeWithTransparentTagWithRenderer.<locals>.WithRendererc s || _ t� | ��|� d S �N)r8 �super�__init__)r r8 �loader)� WithRenderer� __class__r r rE � s zhSerializationTests.test_serializedAttributeWithTransparentTagWithRenderer.<locals>.WithRenderer.__init__c S s | j S rC r7 )r �requestr. r r r r5 � s zeSerializationTests.test_serializedAttributeWithTransparentTagWithRenderer.<locals>.WithRenderer.stuff)�__name__�
__module__� __qualname__rE r r5 �
__classcell__r )rG )rH r rG � s rG c s � � | �ptjdd�S )Nr5 ��render)r6 r rA r7 r9 r r r; � s
z[SerializationTests.test_serializedAttributeWithTransparentTagWithRenderer.<locals>.<lambda>c s � �� � t| ��S rC )r> r r? �rG r: r r r; � r< N)r r2 r r rP r �6test_serializedAttributeWithTransparentTagWithRenderer� s
�zISerializationTests.test_serializedAttributeWithTransparentTagWithRendererc C s( t t�G dd� dt��}| �|t� dS )z�
Like L{test_serializedAttributeWithTransparentTag}, but when the
attribute is a provider of L{IRenderable} rather than a transparent
tag.
c @ s e Zd Zdd� Zdd� ZdS )zLSerializationTests.test_serializedAttributeWithRenderable.<locals>.Arbitraryc S s
|| _ d S rC r7 )r r8 r r r rE � s zUSerializationTests.test_serializedAttributeWithRenderable.<locals>.Arbitrary.__init__c S s | j S rC r7 )r rI r r r rO � s zSSerializationTests.test_serializedAttributeWithRenderable.<locals>.Arbitrary.renderN)rJ rK rL rE rO r r r r � Arbitrary� s rR N)r r �objectr2 r )r rR r r r �&test_serializedAttributeWithRenderable� s z9SerializationTests.test_serializedAttributeWithRenderablec C sL t �d�}t j||�d�}| �|d�}| �|d�}| �t|�jd |� dS )ah
Common implementation of L{test_serializedAttributeWithTag} and
L{test_serializedAttributeWithDeferredTag}.
@param wrapTag: A 1-argument callable that wraps around the attribute's
value so other tests can customize it.
@param wrapTag: callable taking L{Tag} and returning something
flattenable
r0 r& s@ <img src="<a>&lt;&gt;&amp;"</a>" />s <a><>&"</a>r' N)r �ar# r) �assertXMLEqualr �attrib)r r1 ZinnerTagZouterTagZouter�innerr r r � checkTagAttributeSerialization� s
� �z1SerializationTests.checkTagAttributeSerializationc C s | � t� dS )z�
L{Tag} objects which are serialized within the context of an attribute
are serialized such that the text content of the attribute may be
parsed to retrieve the tag.
N)rY r r r r r � test_serializedAttributeWithTag� s z2SerializationTests.test_serializedAttributeWithTagc C s | � t� dS )z�
Like L{test_serializedAttributeWithTag}, but when the L{Tag} is in a
L{Deferred <twisted.internet.defer.Deferred>}.
N)rY r r r r r �'test_serializedAttributeWithDeferredTag� s z:SerializationTests.test_serializedAttributeWithDeferredTagc C s6 | � tjtjdd�d�d�}| �t|�jd d� dS )z�
Similar to L{test_serializedAttributeWithTag}, but for the additional
complexity where the tag which is the attribute value itself has an
attribute value which contains bytes which require substitution.
r0 )Zhrefr& sV <img src="<a href="&lt;&gt;&amp;&quot;"></a>" />r' s" <a href="<>&""></a>N)r) r r# rU rV r rW )r Z flattenedr r r �,test_serializedAttributeWithTagWithAttribute� s ��z?SerializationTests.test_serializedAttributeWithTagWithAttributec C s | � td�d�fS )zI
Test that comments are correctly flattened and escaped.
�foo bars <!--foo bar-->)r r r r r r �test_serializeComment� s z(SerializationTests.test_serializeCommentc sD � fdd�}g }dD ]&}t dt|��}|�|� |�|� qt|�S )a�
The data in a L{Comment} is escaped and mangled in the flattened output
so that the result is a legal SGML and XML comment.
SGML comment syntax is complicated and hard to use. This rule is more
restrictive, and more compatible:
Comments start with <!-- and end with --> and never contain -- or >.
Also by XML syntax, a comment may not end with '-'.
@see: U{http://www.w3.org/TR/REC-xml/#sec-comments}
c s� � � | �d�d| f � � � | �d�d| f � � � t| �dkd| f � | dd� }� �d |� � �d
|� |r�� �|d d � d S )
Ns <!--z)%r does not start with the comment prefixs -->z'%r does not end with the comment suffix� z%%r is too short to be a legal comment� ���s --� >���� -)Z
assertTrue�
startswith�endswith�lenZ assertNotInZassertNotEqual)�cZcontentr r r �
verifyComment s" ��
� z>SerializationTests.test_commentEscaping.<locals>.verifyComment)� z foo---barz
foo---bar-zfoo>barz foo-->barz----------------N)r r Z addCallbackr6 r
)r ri Zresultsrh �dr r r �test_commentEscaping� s
z'SerializationTests.test_commentEscapingc C s$ t | �td�d�| �td�d�g�S )zE
Test that CDATA is correctly flattened and escaped.
r] s <![CDATA[foo bar]]>z foo ]]> bars# <![CDATA[foo ]]]]><![CDATA[> bar]]>)r
r r r r r r �test_serializeCDATA! s ��z&SerializationTests.test_serializeCDATAc C sf d}t | �|d�| �t�|�d�| �t|�d�| �t|�d�| �t|�t�| �td|did�t�g�S ) z�
Test that unicode is encoded correctly in the appropriate places, and
raises an error when it occurs in inappropriate place.
� ☃s ☃s
<p>☃</p>s
<!--☃-->s <![CDATA[☃]]>�prj )r, ) r
r r ro r r �assertFlatteningRaisesr �UnicodeEncodeError)r Zsnowmanr r r �test_serializeUnicode- s
� ��z(SerializationTests.test_serializeUnicodec C s t td��}| �|d�S )ze
A character reference is flattened to a string using the I{&#NNNN;}
syntax.
rn s ☃)r �ordr )r �refr r r �test_serializeCharRef? s z(SerializationTests.test_serializeCharRefc C s | � td�d�S )zz
Test that a deferred is substituted with the current value in the
callback chain when flattened.
Ztwos two)r r r r r r �test_serializeDeferredH s z)SerializationTests.test_serializeDeferredc C s$ t d�}t| �|d�| �|d�g�S )zE
Test that the same deferred can be flattened twice.
Zthrees three)r r
r )r rk r r r � test_serializeSameDeferredTwiceP s
�z2SerializationTests.test_serializeSameDeferredTwicec C s6 ddl m} i }t|d�|� |d }| �|d�d�S )zt
Test that a coroutine returning a value is substituted with the that
value when flattened.
r ��dedentzE
async def coro(x):
return x
�coro�four� four)�textwrapry �execr �r ry � namespacerz r r r �test_serializeCoroutine[ s ��z*SerializationTests.test_serializeCoroutinez*coroutines not available before Python 3.5c C s< ddl m} ttd�}t|d�|� |d }| �|d�d�S )z�
Test that a coroutine returning an awaited deferred value is
substituted with that value when flattened.
r rx )r zT
async def coro(x):
return await succeed(x)
rz r{ r| )r} ry �dictr r~ r r r r r � test_serializeCoroutineWithAwaitr s
��z3SerializationTests.test_serializeCoroutineWithAwaitc s0 t t�G � fdd�dt��}t� �|� d�g�S )zQ
Test that flattening respects all of the IRenderable interface.
c s e Zd Zdd� Z� fdd�ZdS )zASerializationTests.test_serializeIRenderable.<locals>.FakeElementc S s t �dt jdd�dt jdd��S )Nzhello, �testrN z - )r ro rA )�ignZoredr r r rO � s
�zHSerializationTests.test_serializeIRenderable.<locals>.FakeElement.renderc s � � |d� dd� S )Nr� c S s |d�S )NZworldr )r� Znoder r r r; � r< zfSerializationTests.test_serializeIRenderable.<locals>.FakeElement.lookupRenderMethod.<locals>.<lambda>)� assertEqual)r� r+ r r r �lookupRenderMethod� s zTSerializationTests.test_serializeIRenderable.<locals>.FakeElement.lookupRenderMethodN)rJ rK rL rO r� r r r r � FakeElement� s r� s <p>hello, world - world</p>)r r rS r
r )r r� r r r �test_serializeIRenderable� s
�z,SerializationTests.test_serializeIRenderablec C s> t �td��}|�� }|jdd� t| �|t�| �|d�g�S )zS
Test that flattening a slot will use the slot value from the tag.
r� z hello, world�r� s <p>hello, world</p>) r ro r Zcloner= r
rp r r )r �t1�t2r r r �test_serializeSlots� s
�z&SerializationTests.test_serializeSlotsc C s0 t �td��}|jtt �d��d� | �|d�S )z|
Test that a slot with a deferred as its value will be flattened using
the value from the deferred.
r� zfour>r� s <p><em>four></em></p>)r ro r r= r Zemr )r �tr r r �test_serializeDeferredSlots� s z.SerializationTests.test_serializeDeferredSlotsc C s | � dt�S )zT
Test that flattening an unknown type of thing raises an exception.
N)rp r
r r r r �test_unknownTypeRaises� s z)SerializationTests.test_unknownTypeRaisesN)#rJ rK rL �__doc__r! r" r$ r* r/ r2 r3 r4 r@ rB rQ rT rY rZ r[ r\ r^ rl rm rr ru rv rw r� r �skipr� r� r� r� r� r r r r r # sH
/ ��
r c C s d S rC r r r r r r; � r< r; c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �FlattenerErrorTestsz&
Tests for L{FlattenerError}.
c C s< t t�G dd� dt��}| �tttd�|� gg ��d� dS )z�
If a L{FlattenerError} is created with an L{IRenderable} provider root,
the repr of that object is included in the string representation of the
exception.
c @ s e Zd Zdd� ZdS )z7FlattenerErrorTests.test_renderable.<locals>.Renderablec S s dS )Nzrenderable reprr r r r r �__repr__� s z@FlattenerErrorTests.test_renderable.<locals>.Renderable.__repr__N)rJ rK rL r� r r r r �
Renderable� s r� �reasonzCException while flattening:
renderable repr
RuntimeError: reason
N)r r rS r� �strr � RuntimeError)r r� r r r �test_renderable� s ��z#FlattenerErrorTests.test_renderablec C s2 t ddddd�}| �tttd�|gg ��d� dS ) z�
If a L{FlattenerError} is created with a L{Tag} instance with source
location information, the source location is included in the string
representation of the exception.
Zdivz/foo/filename.xhtml� � )�filenameZ
lineNumberZ columnNumberr� zlException while flattening:
File "/foo/filename.xhtml", line 17, column 12, in "div"
RuntimeError: reason
N)r r� r� r r� r- r r r �test_tag� s ��z FlattenerErrorTests.test_tagc C s& | � tttd�td�gg ��d� dS )z�
If a L{FlattenerError} is created with a L{Tag} instance without source
location information, only the tagName is included in the string
representation of the exception.
r� �spanz>Exception while flattening:
Tag <span>
RuntimeError: reason
N)r� r� r r� r r r r r �test_tagWithoutLocation� s �z+FlattenerErrorTests.test_tagWithoutLocationc
s� � fdd�}dd� � z
|� W n@ t k
r^ } z"t�t�� d �dd� }|}W 5 d}~X Y n X | �d� | �tt|g |��d t |j
j d t � j
j d f � dS )
z�
If a L{FlattenerError} is created with traceback frames, they are
included in the string representation of the exception.
c s
� � d S rC r r ��gr r �f s z-FlattenerErrorTests.test_traceback.<locals>.fc S s t d��d S )Nr� )r� r r r r r� s z-FlattenerErrorTests.test_traceback.<locals>.g� � Nzf() must raise RuntimeErrorz�Exception while flattening:
File "%s", line %d, in f
g()
File "%s", line %d, in g
raise RuntimeError("reason")
RuntimeError: reason
) r� � traceback�
extract_tb�sys�exc_infoZfailr� r� r �HERE�__code__�co_firstlineno)r r� �eZtbinfo�excr r� r �test_traceback� s$
���z"FlattenerErrorTests.test_tracebackN)rJ rK rL r� r� r� r� r� r r r r r� � s
r� )+r� r� r� Zxml.etree.cElementTreer � collectionsr Zzope.interfacer Ztwisted.python.compatr Ztwisted.trial.unittestr Ztwisted.test.testutilsr Ztwisted.internet.deferr r r
Ztwisted.web.iwebr Ztwisted.web.errorr r
r Ztwisted.web.templater r r r r r r r r r Ztwisted.web.test._utilr r r� � co_filenamer� r� r r r r �<module> s( #