website/website/common/factories.py

23 lines
497 B
Python

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