✅ [#1821] Tests for category list visibility

Author: stevenbal

src/open_inwoner/components/tests/test_header.py

@@ -23,10 +23,32 @@ def setUpTestData(cls):
         # PrimaryNavigation.html requires apphook + categories
         create_apphook_page(ProductsApphook)
         cls.published1 = CategoryFactory(
-            path="0001", name="First one", slug="first-one"
+            path="0001",
+            name="First one",
+            slug="first-one",
+            visible_for_anonymous=True,
+            visible_for_authenticated=True,
         )
         cls.published2 = CategoryFactory(
-            path="0002", name="Second one", slug="second-one"
+            path="0002",
+            name="Second one",
+            slug="second-one",
+            visible_for_anonymous=True,
+            visible_for_authenticated=False,
+        )
+        cls.published3 = CategoryFactory(
+            path="0003",
+            name="Third one",
+            slug="third-one",
+            visible_for_anonymous=False,
+            visible_for_authenticated=True,
+        )
+        cls.published4 = CategoryFactory(
+            path="0004",
+            name="Fourth one",
+            slug="fourth-one",
+            visible_for_anonymous=False,
+            visible_for_authenticated=False,
         )
 
     def test_categories_hidden_from_anonymous_users(self):
@@ -55,6 +77,37 @@ def test_categories_not_hidden_from_anonymous_users(self):
         self.assertEqual(categories[0].tag, "a")
         self.assertEqual(categories[1].tag, "button")
 
+        links = [x for x in doc.find("[title='Onderwerpen'] + ul li a").items()]
+        self.assertEqual(len(links), 4)
+        self.assertEqual(links[0].attr("href"), self.published1.get_absolute_url())
+        self.assertEqual(links[1].attr("href"), self.published2.get_absolute_url())
+        self.assertEqual(links[2].attr("href"), self.published1.get_absolute_url())
+        self.assertEqual(links[3].attr("href"), self.published2.get_absolute_url())
+
+    def test_categories_visibility_for_authenticated_users(self):
+        config = SiteConfiguration.get_solo()
+        config.hide_categories_from_anonymous_users = False
+        config.save()
+
+        self.client.force_login(self.user)
+
+        response = self.client.get("/", user=self.user)
+
+        doc = PyQuery(response.content)
+
+        categories = doc.find("[title='Onderwerpen']")
+
+        self.assertEqual(len(categories), 2)
+        self.assertEqual(categories[0].tag, "a")
+        self.assertEqual(categories[1].tag, "button")
+
+        links = [x for x in doc.find("[title='Onderwerpen'] + ul li a").items()]
+        self.assertEqual(len(links), 4)
+        self.assertEqual(links[0].attr("href"), self.published1.get_absolute_url())
+        self.assertEqual(links[1].attr("href"), self.published3.get_absolute_url())
+        self.assertEqual(links[2].attr("href"), self.published1.get_absolute_url())
+        self.assertEqual(links[3].attr("href"), self.published3.get_absolute_url())
+
     def test_search_bar_hidden_from_anonymous_users(self):
         config = SiteConfiguration.get_solo()
         config.hide_search_from_anonymous_users = True

src/open_inwoner/pdc/tests/test_category.py

@@ -16,16 +16,38 @@ class TestPublishedCategories(WebTest):
     def setUp(self):
         self.user = UserFactory()
         self.published1 = CategoryFactory(
-            path="0001", name="First one", slug="first-one"
+            path="0001",
+            name="First one",
+            slug="first-one",
+            visible_for_anonymous=True,
+            visible_for_authenticated=True,
         )
         self.published2 = CategoryFactory(
-            path="0002", name="Second one", slug="second-one"
+            path="0002",
+            name="Second one",
+            slug="second-one",
+            visible_for_anonymous=True,
+            visible_for_authenticated=False,
+        )
+        self.published3 = CategoryFactory(
+            path="0003",
+            name="Third one",
+            slug="third-one",
+            visible_for_anonymous=False,
+            visible_for_authenticated=True,
+        )
+        self.published4 = CategoryFactory(
+            path="0004",
+            name="Fourth one",
+            slug="fourth-one",
+            visible_for_anonymous=False,
+            visible_for_authenticated=False,
         )
         self.draft1 = CategoryFactory(
-            path="0003", name="Third one", slug="third-one", published=False
+            path="0005", name="Fifth one", slug="fifth-one", published=False
         )
         self.draft2 = CategoryFactory(
-            path="0004", name="Wourth one", slug="wourth-one", published=False
+            path="0006", name="Sixth one", slug="sixth-one", published=False
         )
         cms_tools.create_homepage()
 
@@ -40,7 +62,7 @@ def test_only_published_categories_exist_in_breadcrumbs_when_logged_in(self):
         response = self.app.get("/", user=self.user)
         self.assertEqual(
             list(response.context["menu_categories"]),
-            [self.published1, self.published2],
+            [self.published1, self.published3],
         )
 
     def test_only_published_categories_exist_in_list_page_when_anonymous(self):
@@ -52,7 +74,7 @@ def test_only_published_categories_exist_in_list_page_when_anonymous(self):
     def test_only_published_categories_exist_in_list_page_when_logged_in(self):
         response = self.app.get(reverse("products:category_list"), user=self.user)
         self.assertEqual(
-            list(response.context["categories"]), [self.published1, self.published2]
+            list(response.context["categories"]), [self.published1, self.published3]
         )
 
     def test_only_published_subcategories_exist_in_detail_page_when_anonymous(self):
@@ -87,7 +109,7 @@ def test_only_published_categories_exist_in_my_categories_page(self):
         response = self.app.get(reverse("profile:categories"), user=self.user)
         self.assertEqual(
             list(response.context["form"].fields["selected_categories"].queryset.all()),
-            [self.published1, self.published2],
+            [self.published1, self.published2, self.published3, self.published4],
         )