Test

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) assert tree.children[0].token is not None self.assertEqual(tree.children[0].token.kind, "html_start") self.assertEqual(tree.children[0].token.tag, "html") assert tree.children[0].children[0].token is not None self.assertEqual(tree.children[0].children[0].token.kind, "html_start") self.assertEqual(tree.children[0].children[0].token.tag, "head") assert tree.children[0].children[0].children[0].token is not None 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") assert tree.children[0].children[1].token is not None self.assertEqual(tree.children[0].children[1].token.kind, "html_start") self.assertEqual(tree.children[0].children[1].token.tag, "body") assert tree.children[0].children[1].children[0].token is not None 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") assert tree.children[0].children[1].children[0].children[0].token is not None 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") assert tree.children[0].children[1].children[1].token is not None 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"])