Filter listings by whether the page is public

tbx/blog/feeds.py

@@ -14,7 +14,7 @@ class BlogFeed(Feed):
     description = "The latest news and views from Torchbox on the work we do, the web and the wider world"
 
     def items(self):
-        return BlogPage.objects.live().order_by("-date")[:10]
+        return BlogPage.objects.live().public().order_by("-date")[:10]
 
     def item_title(self, item):
         return item.title

tbx/blog/models.py

@@ -54,6 +54,7 @@ def blog_posts(self):
         # Get list of blog pages that are descendants of this page
         blog_posts = (
             BlogPage.objects.live()
+            .public()
             .descendant_of(self)
             .distinct()
             .prefetch_related(

tbx/blog/tests/test_models.py

@@ -1,14 +1,48 @@
+from django.core.paginator import Page as PaginatorPage
+
 from faker import Faker
 from tbx.blog.factories import BlogIndexPageFactory, BlogPageFactory
+from tbx.blog.models import BlogPage
 from tbx.taxonomy.factories import SectorFactory, ServiceFactory
+from wagtail.coreutils import get_dummy_request
+from wagtail.models import PageViewRestriction
 from wagtail.test.utils import WagtailPageTestCase
 
 fake = Faker(["en_GB"])
 
 
 class TestBlogIndexPageFactory(WagtailPageTestCase):
-    def test_create(self):
-        BlogIndexPageFactory()
+    @classmethod
+    def setUpTestData(cls):
+        cls.blog_index = BlogIndexPageFactory(title="The Torchbox Blog")
+        cls.blog_post = BlogPageFactory(title="The blog post", parent=cls.blog_index)
+        cls.private_blog_post = BlogPageFactory(
+            title="Private blog post", parent=cls.blog_index
+        )
+        cls.draft_blog_post = BlogPageFactory(
+            title="Draft blog post", live=False, parent=cls.blog_index
+        )
+        PageViewRestriction.objects.create(
+            page=cls.private_blog_post,
+            restriction_type="password",
+            password="password123",
+        )
+
+    def test_get_context(self):
+        context = self.blog_index.get_context(get_dummy_request())
+        blog_posts = context["blog_posts"]
+        print("blog_posts")
+        print(blog_posts)
+        self.assertIsInstance(blog_posts, PaginatorPage)
+        self.assertEqual(blog_posts.object_list.first(), self.blog_post)
+
+    def test_blog_posts_property(self):
+        """Checks that the blog_posts property returns public and published blog posts under the given blog index."""
+        another_blog = BlogIndexPageFactory(title="Tech blog")
+        BlogPageFactory(title="Tech blog", parent=another_blog)
+        self.assertQuerysetEqual(
+            self.blog_index.blog_posts, BlogPage.objects.filter(pk=self.blog_post.pk)
+        )
 
 
 class TestBlogPageFactory(WagtailPageTestCase):

tbx/core/templatetags/torchbox_tags.py

@@ -20,15 +20,15 @@ def get_googe_maps_key():
 
 @register.simple_tag
 def get_next_sibling_by_order(page):
-    sibling = page.get_next_siblings().live().first()
+    sibling = page.get_next_siblings().live().public().first()
 
     if sibling:
         return sibling.specific
 
 
 @register.simple_tag
 def get_prev_sibling_by_order(page):
-    sibling = page.get_prev_siblings().live().first()
+    sibling = page.get_prev_siblings().live().public().first()
 
     if sibling:
         return sibling.specific
@@ -37,7 +37,11 @@ def get_prev_sibling_by_order(page):
 @register.simple_tag
 def get_next_sibling_blog(page):
     sibling = (
-        BlogPage.objects.filter(date__lt=page.date).order_by("-date").live().first()
+        BlogPage.objects.filter(date__lt=page.date)
+        .order_by("-date")
+        .live()
+        .public()
+        .first()
     )
     if sibling:
         return sibling.specific
@@ -46,7 +50,11 @@ def get_next_sibling_blog(page):
 @register.simple_tag
 def get_prev_sibling_blog(page):
     sibling = (
-        BlogPage.objects.filter(date__gt=page.date).order_by("-date").live().last()
+        BlogPage.objects.filter(date__gt=page.date)
+        .order_by("-date")
+        .live()
+        .public()
+        .last()
     )
     if sibling:
         return sibling.specific

tbx/work/models.py

@@ -120,6 +120,7 @@ def related_works(self):
                 Q(related_sectors__in=sectors) | Q(related_services__in=services)
             )
             .live()
+            .public()
             .distinct()
             .order_by(F("date").desc(nulls_last=True))
             .exclude(pk=self.pk)[:4]
@@ -379,6 +380,7 @@ def works(self):
         pages = (
             self.get_children()
             .live()
+            .public()
             .type(HistoricalWorkPage, WorkPage)
             .specific()
             .prefetch_related(

tbx/work/tests/test_models.py

@@ -1,15 +1,51 @@
+from django.core.paginator import Page as PaginatorPage
+
 from tbx.taxonomy.factories import ServiceFactory
 from tbx.work.factories import (
     HistoricalWorkPageFactory,
     WorkIndexPageFactory,
     WorkPageFactory,
 )
+from tbx.work.models import WorkPage
+from wagtail.coreutils import get_dummy_request
+from wagtail.models import PageViewRestriction
 from wagtail.test.utils import WagtailPageTestCase
 
 
 class TestWorkIndexPageFactory(WagtailPageTestCase):
-    def test_create(self):
-        WorkIndexPageFactory()
+    # def test_create(self):
+    #     WorkIndexPageFactory()
+    @classmethod
+    def setUpTestData(cls):
+        cls.work_index = WorkIndexPageFactory(title="Our work")
+        cls.work_page = WorkPageFactory(title="Work page", parent=cls.work_index)
+        cls.private_work_page = WorkPageFactory(
+            title="Private work page", parent=cls.work_index
+        )
+        cls.draft_work_page = WorkPageFactory(
+            title="Draft work page", live=False, parent=cls.work_index
+        )
+        PageViewRestriction.objects.create(
+            page=cls.private_work_page,
+            restriction_type="password",
+            password="password123",
+        )
+
+    def test_get_context(self):
+        context = self.work_index.get_context(get_dummy_request())
+        work_pages = context["works"]
+        print("work_pages")
+        print(work_pages)
+        self.assertIsInstance(work_pages, PaginatorPage)
+        self.assertEqual(work_pages[0]["title"], self.work_page.title)
+
+    def test_works_property(self):
+        """Checks that the works property returns public and published work pages under the given work index."""
+        another_work_index = WorkIndexPageFactory(title="Another work index")
+        WorkPageFactory(title="Another work page", parent=another_work_index)
+        self.assertQuerysetEqual(
+            self.work_index.works, WorkPage.objects.filter(pk=self.work_page.pk)
+        )
 
 
 class TestHistoricalWorkPageFactory(WagtailPageTestCase):