mirror of
https://github.com/zulip/zulip.git
synced 2025-10-26 01:24:02 +00:00
331617efab
This code is not directly related to the template parser, so it can safely live in its own file. The only significant change to the code is to the signature of `html_branches` so that it can be called without requiring a file. Since it's only used in html_grep, that has been updated to reflect this change. Fixes: #1774.
103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
import unittest
|
|
|
|
import tools.lib.template_parser
|
|
|
|
from tools.lib.html_branches import (
|
|
get_tag_info,
|
|
html_branches,
|
|
html_tag_tree,
|
|
)
|
|
|
|
|
|
class TestHtmlBranches(unittest.TestCase):
|
|
|
|
def test_get_tag_info(self):
|
|
# type: () -> None
|
|
html = """
|
|
<p id="test" class="test1 test2">foo</p>
|
|
"""
|
|
|
|
start_tag, end_tag = tools.lib.template_parser.tokenize(html)
|
|
|
|
start_tag_info = get_tag_info(start_tag)
|
|
end_tag_info = get_tag_info(end_tag)
|
|
|
|
self.assertEqual(start_tag_info.text(), 'p.test1.test2#test')
|
|
self.assertEqual(end_tag_info.text(), 'p')
|
|
|
|
def test_html_tag_tree(self):
|
|
# type: () -> None
|
|
html = """
|
|
<!-- test -->
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<!-- test -->
|
|
<head>
|
|
<title>Test</title>
|
|
<meta charset="utf-8" />
|
|
<link rel="stylesheet" href="style.css" />
|
|
</head>
|
|
<body>
|
|
<p>Hello<br />world!</p>
|
|
<p>Goodbye<!-- test -->world!</p>
|
|
</body>
|
|
</html>
|
|
<!-- test -->
|
|
"""
|
|
|
|
tree = html_tag_tree(html)
|
|
|
|
self.assertEqual(tree.children[0].token.kind, 'html_start')
|
|
self.assertEqual(tree.children[0].token.tag, 'html')
|
|
|
|
self.assertEqual(tree.children[0].children[0].token.kind, 'html_start')
|
|
self.assertEqual(tree.children[0].children[0].token.tag, 'head')
|
|
|
|
self.assertEqual(tree.children[0].children[0].children[0].token.kind, 'html_start')
|
|
self.assertEqual(tree.children[0].children[0].children[0].token.tag, 'title')
|
|
|
|
self.assertEqual(tree.children[0].children[1].token.kind, 'html_start')
|
|
self.assertEqual(tree.children[0].children[1].token.tag, 'body')
|
|
|
|
self.assertEqual(tree.children[0].children[1].children[0].token.kind, 'html_start')
|
|
self.assertEqual(tree.children[0].children[1].children[0].token.tag, 'p')
|
|
|
|
self.assertEqual(tree.children[0].children[1].children[0].children[0].token.kind, 'html_singleton')
|
|
self.assertEqual(tree.children[0].children[1].children[0].children[0].token.tag, 'br')
|
|
|
|
self.assertEqual(tree.children[0].children[1].children[1].token.kind, 'html_start')
|
|
self.assertEqual(tree.children[0].children[1].children[1].token.tag, 'p')
|
|
|
|
def test_html_branches(self):
|
|
# type: () -> None
|
|
html = """
|
|
<!-- test -->
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<!-- test -->
|
|
<head>
|
|
<title>Test</title>
|
|
<meta charset="utf-8" />
|
|
<link rel="stylesheet" href="style.css" />
|
|
</head>
|
|
<body>
|
|
<p>Hello<br />world!</p>
|
|
<p>Goodbye<!-- test -->world!</p>
|
|
</body>
|
|
</html>
|
|
<!-- test -->
|
|
"""
|
|
|
|
branches = html_branches(html)
|
|
|
|
self.assertEqual(branches[0].text(), 'html head title')
|
|
self.assertEqual(branches[1].text(), 'html body p br')
|
|
self.assertEqual(branches[2].text(), 'html body p')
|
|
|
|
self.assertEqual(branches[0].staircase_text(), '\n html\n head\n title\n')
|
|
self.assertEqual(branches[1].staircase_text(), '\n html\n body\n p\n br\n')
|
|
self.assertEqual(branches[2].staircase_text(), '\n html\n body\n p\n')
|