import os import unittest import tools.lib.template_parser from tools.lib.html_branches import ( build_id_dict, get_tag_info, html_branches, html_tag_tree, split_for_id_and_class, ) ZULIP_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) TEST_TEMPLATES_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "test_template_data") class TestHtmlBranches(unittest.TestCase): def test_get_tag_info(self) -> None: html = """

foo

""" 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) -> None: html = """ Test

Hello
world!

Goodbyeworld!

""" 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) -> None: html = """ Test

Hello
world!

Goodbyeworld!

""" 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') def test_build_id_dict(self) -> None: templates = ["test_template1.html", "test_template2.html"] templates = [os.path.join(TEST_TEMPLATES_DIR, fn) for fn in templates] template_id_dict = build_id_dict(templates) self.assertEqual(set(template_id_dict.keys()), {'below_navbar', 'hello_{{ message }}', 'intro'}) self.assertEqual(template_id_dict['hello_{{ message }}'], [ f'Line 12:{ZULIP_PATH}/tools/tests/test_template_data/test_template1.html', f'Line 12:{ZULIP_PATH}/tools/tests/test_template_data/test_template2.html']) self.assertEqual(template_id_dict['intro'], [ f'Line 10:{ZULIP_PATH}/tools/tests/test_template_data/test_template1.html', f'Line 11:{ZULIP_PATH}/tools/tests/test_template_data/test_template1.html', f'Line 11:{ZULIP_PATH}/tools/tests/test_template_data/test_template2.html']) self.assertEqual(template_id_dict['below_navbar'], [ f'Line 10:{ZULIP_PATH}/tools/tests/test_template_data/test_template2.html']) def test_split_for_id_and_class(self) -> None: id1 = "{{ red|blue }}" id2 = "search_box_{{ page }}" class1 = "chat_box message" class2 = "stream_{{ topic }}" class3 = "foo {{ a|b|c }} bar" self.assertEqual(split_for_id_and_class(id1), ['{{ red|blue }}']) self.assertEqual(split_for_id_and_class(id2), ['search_box_{{ page }}']) self.assertEqual(split_for_id_and_class(class1), ['chat_box', 'message']) self.assertEqual(split_for_id_and_class(class2), ['stream_{{ topic }}']) self.assertEqual(split_for_id_and_class(class3), ['foo', '{{ a|b|c }}', 'bar'])