Feature/rich text block factory

src/wagtail_factories/blocks.py

@@ -24,6 +24,7 @@
     "PageChooserBlockFactory",
     "ImageChooserBlockFactory",
     "DocumentChooserBlockFactory",
+    "RichTextBlockFactory",
 ]
 
 
@@ -239,3 +240,8 @@ def _build(cls, model_class, document):
     @classmethod
     def _create(cls, model_class, document):
         return document
+
+
+class RichTextBlockFactory(BlockFactory):
+    class Meta:
+        model = blocks.RichTextBlock

tests/test_blocks.py

@@ -6,6 +6,7 @@
 from wagtail.documents.models import Document
 from wagtail.images.models import Image
 from wagtail.models import Page
+from wagtail.rich_text import RichText
 
 from tests.testapp.factories import (
     MyBlockFactory,
@@ -135,6 +136,9 @@ def test_custom_page_streamfield_data_complex():
         body__3__image__image__title="Blub",
         body__4__page__page__title="Bulb",
         body__5__document__document__title="Bubl",
+        body__6__richtext__value=RichText(
+            "<p>This is some <strong>richtext</strong></p>"
+        ),
     )
     assert Image.objects.count() == 1
     image = Image.objects.first()

tests/testapp/factories.py

@@ -52,6 +52,7 @@ class MyTestPageWithStreamFieldFactory(wagtail_factories.PageFactory):
             "document": factory.SubFactory(
                 wagtail_factories.DocumentChooserBlockFactory
             ),
+            "richtext": factory.SubFactory(wagtail_factories.RichTextBlockFactory),
         }
     )
 

tests/testapp/models.py

@@ -41,6 +41,7 @@ class MyTestPage(Page):
             ("page", blocks.PageChooserBlock()),
             ("image", ImageChooserBlock()),
             ("document", DocumentChooserBlock()),
+            ("richtext", blocks.RichTextBlock()),
         ],
         use_json_field=True,
     )