Add page factories for the services module

tbx/core/tests/test_models.py

@@ -18,7 +18,7 @@ class TestPageFactory(TestCase):
 
     # Exclude these modules from the check.
     # (They currently don't have factories. Un-exclude once they have factories.)
-    EXCLUDE = ["tbx.events", "tbx.impact_reports", "tbx.services"]
+    EXCLUDE = ["tbx.events", "tbx.impact_reports"]
 
     def test_pages(self):
         app_configs = apps.get_app_configs()

tbx/services/factories.py

@@ -0,0 +1,30 @@
+import factory
+import wagtail_factories
+from tbx.core.factories import StoryBlockFactory
+
+from .models import ServiceAreaPage, ServicePage
+
+
+class ServicePageFactory(wagtail_factories.PageFactory):
+    class Meta:
+        model = ServicePage
+
+    title = factory.Faker("text", max_nb_chars=100)
+
+    @factory.post_generation
+    def body(obj, create, extracted, **kwargs):
+        blocks = kwargs or {"0": "paragraph"}
+        obj.body = StoryBlockFactory(**blocks)
+
+
+class ServiceAreaPageFactory(wagtail_factories.PageFactory):
+    class Meta:
+        model = ServiceAreaPage
+
+    title = factory.Faker("text", max_nb_chars=100)
+    subtitle = factory.Faker("text", max_nb_chars=100)
+
+    @factory.post_generation
+    def body(obj, create, extracted, **kwargs):
+        blocks = kwargs or {"0": "paragraph"}
+        obj.body = StoryBlockFactory(**blocks)