Python get_page_model Examples

Example #1

 def get_test_site_instance(self):
     Site = utils.get_site_model()
     Page = utils.get_page_model()
     return Site.objects.all().first() or Site.objects.create(
         id=1,
         hostname='wagtailmenus.co.uk',
         root_page=Page.objects.first(),
     )

Example #2

from django.test import TestCase

from wagtailmenus.models import SectionMenu
from wagtailmenus.tests import base, utils

Page = utils.get_page_model()


class SectionMenuTestCase(TestCase):
    def get_test_menu_instance(self):
        return SectionMenu(root_page=Page(), max_levels=3, use_specific=1)


class TestSectionMenuGeneralMethods(SectionMenuTestCase):
    def test_get_from_collected_values_is_not_implemented(self):
        # Non model-based menus use create_from_collected_values() instead of
        # get_from_collected_values(), because there's no 'getting' involved.
        menu = self.get_test_menu_instance()
        with self.assertRaises(NotImplementedError):
            menu.get_from_collected_values(None, None)


class TestGetSubMenuTemplateNames(SectionMenuTestCase,
                                  base.GetSubMenuTemplateNamesMethodTestCase):
    """
    Tests SectionMenu.get_sub_menu_template_names() using common test cases
    from base.GetTemplateNamesMethodTestCase
    """
    expected_default_result_length = 4

Example #3

File: test_sectionmenu_class.py Project: ababic/wagtailmenus

from django.test import TestCase

from wagtailmenus.models import SectionMenu
from wagtailmenus.tests import base, utils

Page = utils.get_page_model()


class SectionMenuTestCase(TestCase):

    def get_test_menu_instance(self):
        return SectionMenu(root_page=Page(), max_levels=3, use_specific=1)


class TestSectionMenuGeneralMethods(SectionMenuTestCase):

    def test_get_from_collected_values_is_not_implemented(self):
        # Non model-based menus use create_from_collected_values() instead of
        # get_from_collected_values(), because there's no 'getting' involved.
        menu = self.get_test_menu_instance()
        with self.assertRaises(NotImplementedError):
            menu.get_from_collected_values(None, None)


class TestGetSubMenuTemplateNames(
    SectionMenuTestCase, base.GetSubMenuTemplateNamesMethodTestCase
):
    """
    Tests SectionMenu.get_sub_menu_template_names() using common test cases
    from base.GetTemplateNamesMethodTestCase
    """