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)Markupescape
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 )Nfoo)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 )Nawesomer )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 tdd 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 tdddi td | j tdddd 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 xt dd dft dd dft dd d ft d jd
ddft djdddigdft djddt digdffD ]\}}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 djd}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 dS )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 dS )Nz<foo>)r )r r r r r ^ s zHMarkupTestCase.test_custom_formatting.<locals>.HasHTMLAndFormat.__html__c S s t dS )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)idusername)r r/ r0 r r r __init__h s zDMarkupTestCase.test_complex_custom_formatting.<locals>.User.__init__c S s2 |dkr t dj| j| j S |r*td| j S )Nlinkz<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 dj| jS )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 Zmarkupitemr 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 tdj tdtdg | j tdj tdtdg | j tdj tdtdg d S )Nza babza
b)r r splitrsplit
splitlines)r r r r test_splitting s z MarkupTestCase.test_splittingc C s | j tdd 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 }xTtdD ]H}x0tdD ]$}td td td td q W |jttj qW d S )N i r z<foo>)setranger addlengcZ get_objects)r Zcountscountr< 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 ttds2| 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
All system for education purposes only. For more tools: Telegram @jackleet