mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00
Code Issues Releases Wiki Activity

Add assertTagInTemplateScript test helper function

Browse Source 
This asserts that a tag exists within a <script type="text/template"> block in the given HTML,
matching as per assertTagInHTML rules.
This commit is contained in:
Matt Westcott 2017-02-27 18:42:15 +00:00 committed by Mikalai Radchuk
parent 064d06c62c
commit 9c5ab4facf
2 changed files with 54 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
wagtail/tests/utils.py
View File

@ -144,6 +144,22 @@ class WagtailTestUtils(object):
return count
def _tag_is_template_script(self, tag):
if tag.name != 'script':
return False
return any(attr == ('type', 'text/template') for attr in tag.attributes)
def _find_template_script_tags(self, haystack):
if not hasattr(haystack, 'name'):
return
if self._tag_is_template_script(haystack):
yield haystack
else:
for child in haystack.children:
for script_tag in self._find_template_script_tags(child):
yield script_tag
def assertTagInHTML(self, needle, haystack, count=None, msg_prefix=''):
needle = assert_and_parse_html(self, needle, None, 'First argument is not valid HTML:')
haystack = assert_and_parse_html(self, haystack, None, 'Second argument is not valid HTML:')
@ -159,6 +175,27 @@ class WagtailTestUtils(object):
def assertNotInHTML(self, needle, haystack, msg_prefix=''):
self.assertInHTML(needle, haystack, count=0, msg_prefix=msg_prefix)
def assertTagInTemplateScript(self, needle, haystack, count=None, msg_prefix=''):
needle = assert_and_parse_html(self, needle, None, 'First argument is not valid HTML:')
haystack = assert_and_parse_html(self, haystack, None, 'Second argument is not valid HTML:')
real_count = 0
for script_tag in self._find_template_script_tags(haystack):
if script_tag.children:
self.assertEqual(len(script_tag.children), 1)
script_html = assert_and_parse_html(
self, script_tag.children[0], None, 'Script tag content is not valid HTML:'
)
real_count += self._count_tag_occurrences(needle, script_html)
if count is not None:
self.assertEqual(
real_count, count,
msg_prefix + "Found %d instances of '%s' in template script (expected %d)" % (real_count, needle, count)
)
else:
self.assertTrue(real_count != 0, msg_prefix + "Couldn't find '%s' in template script" % needle)
class WagtailPageTests(WagtailTestUtils, TestCase):
"""

17
wagtail/wagtailcore/tests/test_tests.py
View File

@ -29,6 +29,23 @@ class TestAssertTagInHTML(TestCase, WagtailTestUtils):
with self.assertRaises(AssertionError):
self.assertTagInHTML('<li lang="en" class="important really" data-extra="boom">', haystack)
def test_assert_tag_in_template_script(self):
haystack = """<html>
<script type="text/template">
<p class="really important">first template block</p>
</script>
<script type="text/template">
<p class="really important">second template block</p>
</script>
<p class="normal">not in a script tag</p>
</html>"""
self.assertTagInTemplateScript('<p class="important really">', haystack)
self.assertTagInTemplateScript('<p class="important really">', haystack, count=2)
with self.assertRaises(AssertionError):
self.assertTagInTemplateScript('<p class="normal">', haystack)
class TestWagtailPageTests(WagtailPageTests):
def setUp(self):