zulip/tools/tests/test_html_branches.py

import os
import unittest

import tools.lib.template_parser
from tools.lib.html_branches import build_id_dict, get_tag_info, 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 = """<p id="test" class="test1 test2">foo</p>"""

        start_tag, text, 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")
        self.assertEqual(text.s, "foo")

    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"])