Jacks_Bastards_Website/home/tests.py

43 lines
1.3 KiB
Python
Raw Normal View History

from home.models import HomePage
from wagtail.models import Page, Site
from wagtail.test.utils import WagtailPageTestCase
class HomeSetUpTests(WagtailPageTestCase):
"""
Tests for basic page structure setup and HomePage creation.
"""
def test_root_create(self):
root_page = Page.objects.get(pk=1)
self.assertIsNotNone(root_page)
def test_homepage_create(self):
root_page = Page.objects.get(pk=1)
homepage = HomePage(title="Home")
root_page.add_child(instance=homepage)
self.assertTrue(HomePage.objects.filter(title="Home").exists())
class HomeTests(WagtailPageTestCase):
"""
Tests for homepage functionality and rendering.
"""
def setUp(self):
"""
Create a homepage instance for testing.
"""
root_page = Page.get_first_root_node()
Site.objects.create(hostname="testsite", root_page=root_page, is_default_site=True)
self.homepage = HomePage(title="Home")
root_page.add_child(instance=self.homepage)
def test_homepage_is_renderable(self):
self.assertPageIsRenderable(self.homepage)
def test_homepage_template_used(self):
response = self.client.get(self.homepage.url)
self.assertTemplateUsed(response, "home/home_page.html")