def test_attrs(self):
""" Test allowed attributes are rendered """
list_html = "<ul class=\"our-list\">" \
"<li class=\"list-item\">one</li>" \
"<li>two</li>" \
"</ul>"
test_data = {
'no_strip': "",
'no_tags': list_html,
'bleach_strip': list_html,
'bleach_attrs': list_html,
'bleach_styles': list_html
}
form = BleachForm(data=test_data)
form.is_valid()
self.assertEqual(form.cleaned_data['no_tags'], 'onetwo')
self.assertEqual(
form.cleaned_data['bleach_strip'],
'<ul><li>one</li><li>two</li></ul>'
)
self.assertEqual(
form.cleaned_data['bleach_attrs'],
test_data['bleach_strip']
)
self.assertEqual(
form.cleaned_data['bleach_styles'],
'<ul><li>one</li><li>two</li></ul>'
)
def test_tags(self):
""" Test allowed tags are rendered"""
test_data = {
'no_tags': "<p>No tags here</p>",
'no_strip': "No tags here",
'bleach_strip': "<ul><li>one</li><li>two</li></ul>",
'bleach_attrs': "<a href=\"https://www.google.com\" "
"title=\"Google\">google.com</a>",
'bleach_styles': "<li style=\"color: white;\">item</li>"
}
form = BleachForm(data=test_data)
form.is_valid()
self.assertEqual(form.cleaned_data['no_tags'], "No tags here")
self.assertEqual(form.cleaned_data['no_strip'], "No tags here")
self.assertEqual(
form.cleaned_data['bleach_strip'],
test_data['bleach_strip']
)
self.assertEqual(
form.cleaned_data['bleach_attrs'],
test_data['bleach_attrs']
)
self.assertEqual(
form.cleaned_data['bleach_styles'],
test_data['bleach_styles']
)
def test_bleaching(self):
""" Test values are bleached """
test_data = {
'no_tags': "<h1>Heading</h1>",
'no_strip': "<h1>Heading</h1>",
'bleach_strip': "<!-- script here -->"
"<script>alert(\"Hello World\")</script>",
'bleach_attrs': "<a href=\"https://www.google.com\" "
"target=\"_blank\">google.com</a>",
'bleach_styles': "<li style=\"color: white\">item</li>"
}
form = BleachForm(data=test_data)
form.is_valid()
self.assertEqual(
form.cleaned_data['no_tags'], 'Heading'
)
self.assertEqual(
form.cleaned_data['no_strip'],
'<h1>Heading</h1>'
)
self.assertEqual(
form.cleaned_data['bleach_strip'],
'alert("Hello World")'
)
self.assertEqual(
form.cleaned_data['bleach_attrs'],
'<a href="https://www.google.com">google.com</a>'
)
self.assertNotEqual(
form.cleaned_data['bleach_styles'],
test_data['bleach_styles']
)
def test_custom_widget2(self, settings):
self.assertEqual(
settings.BLEACH_DEFAULT_WIDGET,
'testproject.forms.CustomBleachWidget'
)
test_data = {
'no_tags': "<h1>Heading</h1>",
'no_strip': "<h1>Heading</h1>",
'bleach_strip': "<!-- script here -->"
"<script>alert(\"Hello World\")</script>",
'bleach_attrs': "<a href=\"https://www.google.com\" "
"target=\"_blank\">google.com</a>",
'bleach_styles': "<li style=\"color: white\">item</li>"
}
form = BleachForm(data=test_data)
form.is_valid()
self.assertEqual(
form.cleaned_data['no_tags'], 'Heading'
)
self.assertEqual(
form.cleaned_data['no_strip'],
'<h1>Heading</h1>'
)
self.assertEqual(
form.cleaned_data['bleach_strip'],
'alert("Hello World")'
)
self.assertEqual(
form.cleaned_data['bleach_attrs'],
'<a href="https://www.google.com">google.com</a>'
)
self.assertNotEqual(
form.cleaned_data['bleach_styles'],
test_data['bleach_styles']
)
def test_empty(self):
""" Test no data """
form = BleachForm(data={})
form.is_valid()
self.assertEqual(form.fields['no_tags'].to_python(None), None)
self.assertEqual(form.fields['no_tags'].to_python(''), '')
self.assertEqual(form.fields['no_tags'].to_python([]), [])
self.assertEqual(form.fields['no_tags'].to_python({}), {})
def test_bleaching(self):
""" Test values are bleached """
test_data = {
'no_tags': "<h1>Heading</h1>",
'bleach_strip': """<script>alert("Hello World")</script>""",
'bleach_attrs': "<a href=\"https://www.google.com\" "
"target=\"_blank\">google.com</a>"
}
form = BleachForm(data=test_data)
form.is_valid()
self.assertEqual(
form.cleaned_data['no_tags'], 'Heading'
)
self.assertEqual(
form.cleaned_data['bleach_strip'],
'alert("Hello World")'
)
self.assertEqual(
form.cleaned_data['bleach_attrs'],
'<a href="https://www.google.com">google.com</a>'
)