← Back
Editing: test_domhelpers.cpython-38.pyc
U W[_+ � @ sn d Z ddlmZ ddlmZ ddlmZ ddlmZm Z G dd� d�Z G dd � d e e�Z G d d � d e e�Z d S ) zH Specific tests for (some of) the methods in L{twisted.web.domhelpers}. � )�minidom)�unicode)�TestCase)� domhelpers�microdomc @ s| e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � ZdS ) �DOMHelpersTestsMixinz� A mixin for L{TestCase} subclasses which defines test methods for domhelpers functionality based on a DOM creation function provided by a subclass. Nc C s� | j �d�}t�|d�d j}d}| �||� |j}t�|d�d j}| �||� d}| j �|�}t�|d�}d�dd� |D ��}d}| �||� |j}t�|d�}d�d d� |D ��}| �||� d } | j �| �} t�| d�}d�d d� |D ��}d }| �||� | j} t�| d�}d�d d� |D ��}| �||� d} | j �| �} t�| d�}| j}||jd jd g}| �||� t�|d�}| �||� d S )Nz<foo/>�foor z<a><foo in="a"/><b><foo in="b"/></b><c><foo in="c"/></c><foo in="d"/><foo in="ef"/><g><foo in="g"/><h><foo in="h"/></h></g></a>� c S s g | ]}|� d ��qS ��in�Z getAttribute��.0�node� r �B/usr/lib/python3/dist-packages/twisted/web/test/test_domhelpers.py� <listcomp>$ s zBDOMHelpersTestsMixin.test_getElementsByTagName.<locals>.<listcomp>�abcdefghc S s g | ]}|� d ��qS r r r r r r r ) s aH <a><foo in="a"/> <b><foo in="b"/> <d><foo in="d"/> <g><foo in="g"/></g> <h><foo in="h"/></h> </d> <e><foo in="e"/> <i><foo in="i"/></i> </e> </b> <c><foo in="c"/> <f><foo in="f"/> <j><foo in="j"/></j> </f> </c> </a>c S s g | ]}|� d ��qS r r r r r r r ? s � abdgheicfjc S s g | ]}|� d ��qS r r r r r r r D s z'<foo><bar></bar><baz><foo/></baz></foo>���) �dom� parseStringr ZgetElementsByTagNameZnodeName� assertEqual�documentElement�join� childNodes)�self�doc1�actual�expectedZel1�doc2_xml�doc2Ztag_listZel2�doc3_xml�doc3Zel3�doc4_xml�doc4�rootr r r �test_getElementsByTagName sF z.DOMHelpersTestsMixin.test_getElementsByTagNamec C s� | j �d�}t�|�}d}| �||� t�|j�}| �||� d}| j �|�}t�|�}d}| �||� t�|j�}| �||� d}| j �|�}t�|�}d}| �||� t�|j�}| �||� d S )Nz <a>foo</a>r z+<a>a<b>b</b><c>c</c>def<g>g<h>h</h></g></a>r zP<a>a<b>b<d>d<g>g</g><h>h</h></d><e>e<i>i</i></e></b><c>c<f>f<j>j</j></f></c></a>r �r r r ZgatherTextNodesr r )r r r r r r! r" r# r r r �test_gatherTextNodesQ s( z)DOMHelpersTestsMixin.test_gatherTextNodesc C s� | j �d�}|j}t�|� | �|�� | j �d��� � | j �d�}|jjd }t�|� |j�� }| j �d�}|� | j �d�� | �||�� � d S )Nz<a><b><c><d/></c></b></a>�ar �b) r r r r Z clearNoder �toxmlZElementr � appendChild)r r Za_noder! Zb_noder r r r r �test_clearNodek s � z#DOMHelpersTestsMixin.test_clearNodec C s� | j �d�}| j �� }t�|d�}|�� }|�d�}|�dd� | �||�� � t�|d�}|�� }|�d�}|�dd� | �||�� � | � tj tj|d� d S ) N�$<a><b id="bar"/><c class="foo"/></a>r �c�class�barr+ �id�pzork) r r �Documentr �getr, � createElement� setAttributer Z assertRaisesZNodeLookupError�r r �docr r r r r r �test_get| s" �z DOMHelpersTestsMixin.test_getc C sl | j �d�}| j �� }t�|d�}|�� }|�d�}|�dd� | �||�� � t�|d�}| � |d � d S )Nr/ r r0 r1 r4 ) r r r5 r Z getIfExistsr, r7 r8 r ZassertIdenticalr9 r r r �test_getIfExists� s z%DOMHelpersTestsMixin.test_getIfExistsc C sT | j �d�}| j �� }t�|d�}|�� }|�d�}|�dd� | �||�� � d S )Nz <a><b id="foo"><c></c></b></a>r r+ r3 ) r r r5 r Z getAndClearr, r7 r8 r r9 r r r �test_getAndClear� s z%DOMHelpersTestsMixin.test_getAndClearc C s� | j �d�}| j �� }tj|jdddd�}d�dd� |D ��}|�d �}|�dd� |�d �}|�dd� |� |� | � ||� � � tj|jddd d�}d�d d� |D ��}| � ||� � |� � � d S ) Nz:<a><b foo="olive"><c foo="olive"/></b><d foo="poopy"/></a>r Zolive� )Z noNestingr c S s g | ] }|� � �qS r �r, r r r r r � s z9DOMHelpersTestsMixin.test_locateNodes.<locals>.<listcomp>r+ r0 r c S s g | ] }|� � �qS r r? r r r r r � s ) r r r5 r Z locateNodesr r r7 r8 r- r r, )r r r: � node_listr r r0 r r r �test_locateNodes� s. � �z%DOMHelpersTestsMixin.test_locateNodesc C sL | j �d�}t�|jd jd jd �}d�dd� |D ��}| �|d� d S )Nz!<a><b><c><d/></c><e/></b><f/></a>r r c S s g | ]}t |d �r|j�qS ��tagName)�hasattrrC r r r r r � s �z8DOMHelpersTestsMixin.test_getParents.<locals>.<listcomp>Zcba)r r r Z getParentsr r r �r r r@ r r r r �test_getParents� s �z$DOMHelpersTestsMixin.test_getParentsc C sj | j �d�}t�|d�}d�dd� |D ��}| �|d� t�|dd�}d�dd� |D ��}| �|d � d S ) Nz+<a foo="1"><b foo="2"/><c foo="1"/><d/></a>r r c S s g | ] }|j �qS r rB r r r r r � s zGDOMHelpersTestsMixin.test_findElementsWithAttribute.<locals>.<listcomp>�abc�1c S s g | ] }|j �qS r rB r r r r r � s Zac)r r r ZfindElementsWithAttributer r rE r r r � test_findElementsWithAttribute� s z3DOMHelpersTestsMixin.test_findElementsWithAttributec C s0 | j �d�}t�|d�}t|�}| �|d� d S )Nz#<doc><foo/><bar/><foo>a</foo></doc>r � )r r r ZfindNodesNamed�lenr rE r r r �test_findNodesNamed� s z(DOMHelpersTestsMixin.test_findNodesNamedc C s d}d}| � t�|�|� d S )Nz9this string " contains many & characters> xml< won't likezHthis string " contains many & characters> xml< won't like)r r �escape�r �jr r r r � test_escape� s z DOMHelpersTestsMixin.test_escapec C s d}d}| � t�|�|� d S )NzTthis string " has && entities > < and some characters xml won't like<zEthis string " has && entities > < and some characters xml won't like<)r r ZunescaperN r r r � test_unescape� s z"DOMHelpersTestsMixin.test_unescapec C s" | j �d�}| �t�|�d� dS )z| L{getNodeText} returns the concatenation of all the text data at or beneath the node passed to it. z(<foo><bar>baz</bar><bar>quux</bar></foo>ZbazquuxN)r r r r � getNodeText)r r r r r �test_getNodeText� s z%DOMHelpersTestsMixin.test_getNodeText)�__name__� __module__� __qualname__�__doc__r r' r) r. r; r< r= rA rF rI rL rP rQ rS r r r r r s 9 r c @ s e Zd ZeZdd� Zdd� ZdS )�MicroDOMHelpersTestsc C sF d}| j �|�}t�|�}d}| �||� t�|j�}| �||� dS )z� Microdom discards whitespace-only text nodes, so L{gatherTextNodes} returns only the text from nodes which had non-whitespace characters. z?<html> <head> </head> <body> stuff </body> </html> z stuff Nr( )r r$ r% r r r r r �#test_gatherTextNodesDropsWhitespace� s z8MicroDOMHelpersTests.test_gatherTextNodesDropsWhitespacec C sF d}| j �|�}t�|�}d}| �||� t�|j�}| �||� dS )zB Microdom does not decode entities in text nodes. �<x>Souffl&</x>z Souffl&Nr( �r Zdoc5_xmlZdoc5r r r r r �test_textEntitiesNotDecoded s z0MicroDOMHelpersTests.test_textEntitiesNotDecodedN)rT rU rV r r rY r\ r r r r rX � s rX c @ s e Zd ZeZdd� Zdd� ZdS )�MiniDOMHelpersTestsc C sF d}| j �|�}t�|�}d}| �||� t�|j�}| �||� dS )z= Minidom does decode entities in text nodes. rZ zSouffl&Nr( r[ r r r �test_textEntitiesDecoded s z,MiniDOMHelpersTests.test_textEntitiesDecodedc C sf | j �d�}t�|�}| �|d� | �|t� | j �d�d��}t�|�}| �|d� | �|t� dS )z� L{domhelpers.getNodeText} returns a C{unicode} string when text nodes are represented in the DOM with unicode, whether or not there are non-ASCII characters present. z<foo>bar</foo>r2 u <foo>☃</foo>zutf-8u ☃N)r r r rR r ZassertIsInstancer �encode)r r �textr r r �test_getNodeUnicodeText# s z+MiniDOMHelpersTests.test_getNodeUnicodeTextN)rT rU rV r r r^ ra r r r r r] s r] N) rW Zxml.domr Ztwisted.python.compatr Ztwisted.trial.unittestr Z twisted.webr r r rX r] r r r r �<module> s ^&
Save File
Cancel