Added a factory for Redirect model

src/wagtail_factories/factories.py

@@ -5,6 +5,7 @@
 from factory import errors, utils
 from factory.declarations import ParameteredAttribute
 from factory.django import DjangoModelFactory
+from wagtail.contrib.redirects.models import Redirect
 from wagtail.documents import get_document_model
 from wagtail.images import get_image_model
 from wagtail.models import Collection, Page, Site
@@ -15,6 +16,7 @@
     "PageFactory",
     "SiteFactory",
     "DocumentFactory",
+    "RedirectFactory",
 ]
 logger = logging.getLogger(__file__)
 
@@ -144,3 +146,26 @@ class Meta:
 
     title = "A document"
     file = factory.django.FileField()
+
+
+class RedirectFactory(DjangoModelFactory):
+    old_path = factory.Faker("slug")
+    site = factory.SubFactory(SiteFactory)
+    redirect_page = factory.SubFactory(PageFactory)
+    redirect_link = factory.Faker("url")
+    is_permanent = factory.Faker("boolean")
+
+    class Meta:
+        model = Redirect
+
+    @classmethod
+    def _create(cls, model_class, *args, **kwargs):
+        """
+        Override _create() to ensure that Redirect.clean() is run
+        in order to normalise `old_path`.
+        @see https://github.com/wagtail/wagtail/blob/main/wagtail/contrib/redirects/models.py#L191
+        """
+        obj = model_class(*args, **kwargs)
+        obj.clean()
+        obj.save()
+        return obj

tests/settings.py

@@ -21,6 +21,7 @@
     "wagtail.contrib.settings",
     "wagtail.contrib.table_block",
     "wagtail.contrib.forms",
+    "wagtail.contrib.redirects",
     "wagtail.search",
     "wagtail.embeds",
     "wagtail.images",

tests/test_factories.py

@@ -2,6 +2,7 @@
 import wagtail_factories
 from wagtail import blocks
 from wagtail.models import Page, Site
+from wagtail.contrib.redirects.models import Redirect
 
 from tests.testapp.factories import MyTestPageFactory, MyTestPageGetOrCreateFactory
 
@@ -171,3 +172,27 @@ def test_document_add_to_collection():
         collection__parent=root_collection, collection__name="new"
     )
     assert document.collection.name == "new"
+
+
+@pytest.mark.django_db()
+def test_redirect_factory():
+    """
+    Test that a `RedirectFactory` generates a `Redirect` object.
+    """
+    redirect = wagtail_factories.RedirectFactory()
+
+    assert type(redirect) is Redirect
+
+
+@pytest.mark.django_db()
+def test_redirect_factory_old_path_normalisation():
+    """
+    Test that `RedirectFactory` uses `Redirect.clean()` to normalise the
+    `old_path` value.
+    """
+
+    redirect = wagtail_factories.RedirectFactory(old_path="test-path/")
+
+    # Verify that `old_path` has been modified according to
+    # @https://github.com/wagtail/wagtail/blob/a09bba67cd58f519f3ae5bff32575e7ce9244031/wagtail/contrib/redirects/models.py#L141
+    assert redirect.old_path == "/test-path"