3
��kS� � @ s| d dl Z d dlZd dlZd dlmZmZmZ d dlmZ G dd� dej �Z
G dd� dej �Z dd � Z e
d
krxejd d � dS ) � N)�Markup�escape�
escape_silent)� text_typec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e j d"krZdd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )#�MarkupTestCasec C s d}t d�}d S )Nz?<script type="application/x-some-script">alert("foo");</script>z<em>username</em>)r )�selfZunsafeZsafe� r � /usr/lib64/python3.6/tests.py� test_adding s zMarkupTestCase.test_addingc C s d S )Nr )r r r r �test_string_interpolation s z(MarkupTestCase.test_string_interpolationc C s t d�}d S )N�foo)r )r �xr r r �test_type_behavior s z!MarkupTestCase.test_type_behaviorc C s G dd� dt �}d S )Nc @ s e Zd Zdd� Zdd� ZeZdS )z-MarkupTestCase.test_html_interop.<locals>.Fooc S s dS )Nz<em>awesome</em>r )r r r r �__html__' s z6MarkupTestCase.test_html_interop.<locals>.Foo.__html__c S s dS )N�awesomer )r r r r � __unicode__) s z9MarkupTestCase.test_html_interop.<locals>.Foo.__unicode__N)�__name__�
__module__� __qualname__r r �__str__r r r r �Foo& s r )�object)r r r r r �test_html_interop$ s z MarkupTestCase.test_html_interopc C s | j td�d td�� d S )Nz<em>%s:%s</em>�<foo>�<bar>z <em><foo>:<bar></em>)r r )� assertEqualr )r r r r �test_tuple_interpol0 s
z"MarkupTestCase.test_tuple_interpolc C s> | j td�ddi td�� | j td�ddd� td�� d S ) Nz<em>%(foo)s</em>r z<foo>z<em><foo></em>z<em>%(foo)s:%(bar)s</em>z<bar>)r �barz <em><foo>:<bar></em>)r r )r r r r �test_dict_interpol6 s
z!MarkupTestCase.test_dict_interpolc C s d S )Nr )r r r r �
test_escaping? s z MarkupTestCase.test_escapingc
C s� x�t d�d dft d�d dft d�d d ft d �jd
d�dft d�jdddig�dft d�jddt d�ig�dffD ]\}}qvW d S )Nz%ig ��Q� @�3z%.2fgn��! @z3.14z%s %s %s�<�{ �>z
< 123 >z<em>{awesome}</em>z <awesome>)r z<em><awesome></em>z {0[1][bar]}r r z<bar/>z <bar/>)r! r" r# )r �format)r �actualZexpectedr r r �test_formattingE s z MarkupTestCase.test_formatting� � c C s t d�jd�}d S )Nz{}r )r r$ )r Z formattedr r r �test_formatting_emptyT s z$MarkupTestCase.test_formatting_emptyc C s$ G dd� dt �}G dd� dt �}d S )Nc @ s e Zd Zdd� ZdS )z:MarkupTestCase.test_custom_formatting.<locals>.HasHTMLOnlyc S s t d�S )Nz<foo>)r )r r r r r Z s zCMarkupTestCase.test_custom_formatting.<locals>.HasHTMLOnly.__html__N)r r r r r r r r � HasHTMLOnlyY s r* c @ s e Zd Zdd� Zdd� ZdS )z?MarkupTestCase.test_custom_formatting.<locals>.HasHTMLAndFormatc S s t d�S )Nz<foo>)r )r r r r r ^ s zHMarkupTestCase.test_custom_formatting.<locals>.HasHTMLAndFormat.__html__c S s t d�S )Nz<FORMAT>)r )r �specr r r �__html_format__` s zOMarkupTestCase.test_custom_formatting.<locals>.HasHTMLAndFormat.__html_format__N)r r r r r, r r r r �HasHTMLAndFormat] s r- )r )r r* r- r r r �test_custom_formattingX s z%MarkupTestCase.test_custom_formattingc C s G dd� dt �}|dd�}d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z;MarkupTestCase.test_complex_custom_formatting.<locals>.Userc S s || _ || _d S )N)�id�username)r r/ r0 r r r �__init__h s zDMarkupTestCase.test_complex_custom_formatting.<locals>.User.__init__c S s2 |dkr t d�j| j| j� �S |r*td��| j� S )N�linkz<a href="/user/{0}">{1}</a>zInvalid format spec)r r$ r/ r �
ValueError)r � format_specr r r r, k s
zKMarkupTestCase.test_complex_custom_formatting.<locals>.User.__html_format__c S s t d�j| j�S )Nz<span class=user>{0}</span>)r r$ r0 )r r r r r t s zDMarkupTestCase.test_complex_custom_formatting.<locals>.User.__html__N)r r r r1 r, r r r r r �Userg s r5 � r )r )r r5 �userr r r � test_complex_custom_formattingf s
z-MarkupTestCase.test_complex_custom_formattingc C s& dd l }x|jD ]}t||� qW d S )Nr )�
markupsafe�__all__�getattr)r Zmarkup�itemr r r � test_all_set{ s zMarkupTestCase.test_all_setc C s d S )Nr )r r r r �test_escape_silent� s z!MarkupTestCase.test_escape_silentc C sd | j td�j� td�td�g� | j td�j� td�td�g� | j td�j� td�td�g� d S )Nza b�a�bza
b)r r �split�rsplit�
splitlines)r r r r �test_splitting� s z MarkupTestCase.test_splittingc C s | j td�d td�� d S )Nr? � Zaaa)r r )r r r r �test_mul� s zMarkupTestCase.test_mulN)r' r( )r r r r
r r r r r r r& �sys� version_infor) r. r8 r= r> rD rF r r r r r s
r c @ s e Zd Zdd� ZdS )�MarkupLeakTestCasec C s` t � }xTtd�D ]H}x0td�D ]$}td� td� td� td� q W |jttj� �� qW d S )N� i� r z<foo>)�set�ranger �add�len�gcZ get_objects)r Zcounts�countr< r r r �test_markup_leaks� s z$MarkupLeakTestCase.test_markup_leaksN)r r r rQ r r r r rI � s rI c C s6 t j� } | jt jt�� ttd�s2| jt jt�� | S )NZ func_code)�unittestZ TestSuiteZaddTestZ makeSuiter �hasattrr rI )�suiter r r rT � s
rT �__main__)Z defaultTest)rO rG rR r9 r r r Zmarkupsafe._compatr ZTestCaser rI rT r �mainr r r r �<module> s