def test_sanitize(self):
text_before = u"<żarcik>To wcale nie jest śmieszne!!</żarcik>"
allowed_tags = "b i em strong ol li ul blockquote a:href img:src h2 spoiler"
text_after = sanitize(text_before, allowed_tags)
self.failUnlessEqual(text_after, u"<żarcik>To wcale nie jest śmieszne!!")
text_before = u"<h1>Lorem ipsum</h1>"
text_after = sanitize(text_before, allowed_tags)
self.failUnlessEqual(text_after, u"Lorem ipsum")
def test_allowed_tags(self):
"""
ALLOWED TAGS set h5, img with src attribute
"""
text_before = u"Lorem <h3>ipsum</h3> <h1>lorem</h1> <h5>ipsum</h5> <img src=\"#\" />"
allowed_tags = "h5 img:src"
text_after = sanitize(text_before, allowed_tags)
self.failUnlessEqual(text_after, u"Lorem ipsum lorem <h5>ipsum</h5> <img src=\"#\" />")
