Add test for number of queries!

django-cms · Jul 14, 2024 · a3898e1 · a3898e1
1 parent a897942
commit a3898e1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_indicators.py b/tests/test_indicators.py
@@ -219,3 +219,13 @@ def test_mixin_factory(self):
             self.assertContains(response, "cms.pagetree.css"),
             # JS loadeD?
             self.assertContains(response, "indicators.js")
+
+    def test_page_indicator_db_queries(self):
+        """Only one query should be executed to get the indicator"""
+        version = PageVersionFactory(
+            content__language="en",
+        )
+        with self.assertNumQueries(1):
+            from djangocms_versioning.indicators import content_indicator, content_indicator_menu
+
+            content_indicator(version.content)