Misago/misago/core/tests/test_page.py

from django.test import TestCase

from misago.core.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')