Misago/misago/core/tests/test_page.py

from django.test import TestCase

from ..page import Page


class SiteTests(TestCase):
    def setUp(self):
        self.page = Page("test")

    def test_pages(self):
        """add_section adds section to page"""
        self.page.add_section(
            link="misago:user-posts", name="Posts", after="misago:user-threads"
        )

        self.page.add_section(link="misago:user-threads", name="Threads")

        self.page.add_section(
            link="misago:user-follows", name="Follows", before="misago:user-posts"
        )

        self.page.assert_is_finalized()

        sorted_sections = self.page._sorted_list
        self.assertEqual(sorted_sections[0]["name"], "Threads")
        self.assertEqual(sorted_sections[1]["name"], "Follows")
        self.assertEqual(sorted_sections[2]["name"], "Posts")

        self.assertEqual(self.page.get_default_link(), "misago:user-threads")