AnonymousFox

//usr/lib64/python2.6/site-packages/markupsafe/tests.py

import gc
import unittest
from markupsafe import Markup, escape

class MarkupTestCase(unittest.TestCase):

def test_markup_operations(self):
 # adding two strings should escape the unsafe one
 unsafe = '<script type="application/x-some-script">alert("foo");</script>'
 safe = Markup('username')
 assert unsafe + safe == unicode(escape(unsafe)) + unicode(safe)

# string interpolations are safe to use too
 assert Markup('%s') % '<bad user>' == \
 '&lt;bad user&gt;'
 assert Markup('%(username)s') % {
 'username': '<bad user>'
 } == '&lt;bad user&gt;'

# an escaped object is markup too
        assert type(Markup('foo') + 'bar') is Markup

# and it implements __html__ by returning itself
        x = Markup("foo")
        assert x.__html__() is x

# it also knows how to treat __html__ objects
 class Foo(object):
 def __html__(self):
 return 'awesome'
 def __unicode__(self):
 return 'awesome'
 assert Markup(Foo()) == 'awesome'
 assert Markup('%s') % Foo() == \
 'awesome'

# escaping and unescaping
 assert escape('"<>&\'') == '&#34;&lt;&gt;&amp;&#39;'
 assert Markup("Foo &amp; Bar").striptags() == "Foo & Bar"
 assert Markup("&lt;test&gt;").unescape() == "<test>"

class MarkupLeakTestCase(unittest.TestCase):

def test_markup_leaks(self):
 counts = set()
 for count in xrange(20):
 for item in xrange(1000):
 escape("foo")
 escape("<foo>")
 escape(u"foo")
 escape(u"<foo>")
 counts.add(len(gc.get_objects()))
 assert len(counts) == 1, 'ouch, c extension seems to leak objects'

def suite():
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(MarkupTestCase))

# this test only tests the c extension
    if not hasattr(escape, 'func_code'):
        suite.addTest(unittest.makeSuite(MarkupLeakTestCase))

return suite

if __name__ == '__main__':
    unittest.main(defaultTest='suite')