61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
|
|
||
|
|
try:
|
||
|
|
import unittest2 as unittest
|
||
|
|
except ImportError:
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
|
||
|
|
from isso.core import Config
|
||
|
|
from isso.utils import html
|
||
|
|
|
||
|
|
|
||
|
|
class TestHTML(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_markdown(self):
|
||
|
|
convert = html.Markdown(extensions=())
|
||
|
|
examples = [
|
||
|
|
("*Ohai!*", "<p><em>Ohai!</em></p>"),
|
||
|
|
("<em>Hi</em>", "<p><em>Hi</em></p>"),
|
||
|
|
("http://example.org/", '<p>http://example.org/</p>')]
|
||
|
|
|
||
|
|
for (input, expected) in examples:
|
||
|
|
self.assertEqual(convert(input), expected)
|
||
|
|
|
||
|
|
def test_markdown_extensions(self):
|
||
|
|
convert = html.Markdown(extensions=("strikethrough", "superscript"))
|
||
|
|
examples = [
|
||
|
|
("~~strike~~ through", "<p><del>strike</del> through</p>"),
|
||
|
|
("sup^(script)", "<p>sup<sup>script</sup></p>")]
|
||
|
|
|
||
|
|
for (input, expected) in examples:
|
||
|
|
self.assertEqual(convert(input), expected)
|
||
|
|
|
||
|
|
@unittest.skipIf(html.html5lib_version == "0.95", "backport")
|
||
|
|
def test_sanitizer(self):
|
||
|
|
sanitizer = html.Sanitizer(elements=[], attributes=[])
|
||
|
|
examples = [
|
||
|
|
('Look: <img src="..." />', 'Look: '),
|
||
|
|
('<a href="http://example.org/">Ha</a>', '<a href="http://example.org/">Ha</a>'),
|
||
|
|
('<a href="sms:+1234567890">Ha</a>', '<a>Ha</a>'),
|
||
|
|
('<p style="visibility: hidden;">Test</p>', '<p>Test</p>'),
|
||
|
|
('<script>alert("Onoe")</script>', 'alert("Onoe")')]
|
||
|
|
|
||
|
|
for (input, expected) in examples:
|
||
|
|
self.assertEqual(html.sanitize(sanitizer, input), expected)
|
||
|
|
|
||
|
|
@unittest.skipIf(html.html5lib_version == "0.95", "backport")
|
||
|
|
def test_sanitizer_extensions(self):
|
||
|
|
sanitizer = html.Sanitizer(elements=["img"], attributes=["src"])
|
||
|
|
examples = [
|
||
|
|
('<img src="cat.gif" />', '<img src="cat.gif">'),
|
||
|
|
('<script src="doge.js"></script>', '')]
|
||
|
|
|
||
|
|
for (input, expected) in examples:
|
||
|
|
self.assertEqual(html.sanitize(sanitizer, input), expected)
|
||
|
|
|
||
|
|
def test_render(self):
|
||
|
|
conf = Config.load(None).section("markup")
|
||
|
|
renderer = html.Markup(conf).render
|
||
|
|
self.assertEqual(renderer("http://example.org/ and sms:+1234567890"),
|
||
|
|
'<p><a href="http://example.org/">http://example.org/</a> and sms:+1234567890</p>')
|