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 s0 d}t d�}|| tt|��t|� ks,t�d S )Nz?<script type="application/x-some-script">alert("foo");</script>z<em>username</em>)r r r �AssertionError)�selfZunsafeZsafe� r � /usr/lib64/python3.6/tests.py� test_adding s zMarkupTestCase.test_addingc C sX t d�d dkst�t d�ddi dks,t�t d�d dks@t�t d �d d
ksTt�d S ) Nz <em>%s</em>z
<bad user>z<em><bad user></em>z<em>%(username)s</em>�usernamez%ig ��Q� @�3z%.2fz3.14)r r )r r r r
�test_string_interpolation s
z(MarkupTestCase.test_string_interpolationc C s4 t td�d �tkst�td�}|j� |ks0t�d S )N�foo�bar)�typer r �__html__)r �xr r r
�test_type_behavior s z!MarkupTestCase.test_type_behaviorc C s< G dd� dt �}t|� �dks"t�td�|� dks8t�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
r ' 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 z<em>awesome</em>z<strong>%s</strong>z!<strong><em>awesome</em></strong>)�objectr r )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 r z <em><foo>:<bar></em>)r r )r r r r
�test_dict_interpol6 s
z!MarkupTestCase.test_dict_interpolc C s< t d�dkst�td�j� dks$t�td�j� dks8t�d S )Nz"<>&'z"<>&'z<em>Foo & Bar</em>z Foo & Barz <test>z<test>)r r r Z striptagsZunescape)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 ] \}}||ksvtd||f ��qvW d S )Nz%ig ��Q� @r
z%.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/>z%r should be %r!)r$ r% r& )r �formatr )r �actualZexpectedr r r
�test_formattingE s z MarkupTestCase.test_formatting� � c C s" t d�jd�}|t d�ks t�d S )Nz{}r �0)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 �}td�j|� �td�ks<t�td�j|� �td�ksXt�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 r0 r r r r
�HasHTMLAndFormat] s r1 z{0}z<foo>z<FORMAT>)r r r' r )r r. r1 r r r
�test_custom_formattingX s z%MarkupTestCase.test_custom_formattingc C s8 G dd� dt �}|dd�}td�j|�td�ks4t�d S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z;MarkupTestCase.test_complex_custom_formatting.<locals>.Userc S s || _ || _d S )N)�idr )r r3 r 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' r3 r �
ValueError)r � format_specr r r
r0 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' r )r r r r
r t s zDMarkupTestCase.test_complex_custom_formatting.<locals>.User.__html__N)r r r r4 r0 r r r r r
�Userg s r8 � r z<p>User: {0:link}z:<p>User: <a href="/user/1"><span class=user>foo</span></a>)r r r' r )r r8 �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> t d �t� kst�td �td �ks&t�t d�td�ks:t�d S )Nz<foo>z <foo>)r r r r )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 )NrB � 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- r2 r; r@ rA rG rI r r r r
r s
r c @ s e Zd Zdd� ZdS )�MarkupLeakTestCasec C st t � }xTtd�D ]H}x0td�D ]$}td� td� td� td� q W |jttj� �� qW t|�dksptd��d S )N� i� r z<foo>r9 z'ouch, c extension seems to leak objects)�set�ranger �add�len�gcZ get_objectsr )r Zcounts�countr? r r r
�test_markup_leaks� s z$MarkupLeakTestCase.test_markup_leaksN)r r r rT r r r r
rL � s rL c C s6 t j� } | jt jt�� ttd�s2| jt jt�� | S )NZ func_code)�unittestZ TestSuiteZaddTestZ makeSuiter �hasattrr rL )�suiter r r
rW � s
rW �__main__)Z defaultTest)rR rJ rU r< r r r Zmarkupsafe._compatr ZTestCaser rL rW r �mainr r r r
�<module> s