website/website/common/factories.py

import factory
import wagtail_factories

from . import models


class BaseContentFactory(wagtail_factories.PageFactory):
    title = factory.Faker("catch_phrase")
    subtitle = factory.Faker("bs")


class BaseListingFactory(wagtail_factories.PageFactory):
    title = factory.Faker("catch_phrase")


class ContentPageFactory(BaseContentFactory):
    class Meta:
        model = models.ContentPage


class ListingPageFactory(BaseListingFactory):
    class Meta:
        model = models.ListingPage