website/website/common/tests.py
2022-06-12 15:24:17 +01:00

17 lines
591 B
Python

from django.test import SimpleTestCase
from .models import BasePage
from .utils import get_page_models
class BasePageTestCase(SimpleTestCase):
def test_unique_body_classes(self) -> None:
body_classes = [page.body_class for page in get_page_models()]
self.assertEqual(len(body_classes), len(set(body_classes)))
def test_pages_inherit_base_page(self) -> None:
for page_model in get_page_models():
self.assertTrue(
issubclass(page_model, BasePage),
f"{page_model} does not inherit from {BasePage}.",
)