Michael Yick
b8d7dd53ae
Co-authored-by: nick.moreton <nick.moreton@torchbox.com> Co-authored-by: Nick Moreton <nick.moreton@torchbox.com> Reviewed-on: #1
39 lines
949 B
Python
39 lines
949 B
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import factory
|
|
from django.utils.text import slugify
|
|
from wagtail.models import Locale
|
|
from wagtail_factories.factories import PageFactory
|
|
|
|
from tests.site.pages import models
|
|
from wagtail_personalisation.models import PersonalisablePageMetadata
|
|
|
|
|
|
class LocaleFactory(factory.django.DjangoModelFactory):
|
|
language_code = "en"
|
|
|
|
class Meta:
|
|
model = Locale
|
|
|
|
|
|
class ContentPageFactory(PageFactory):
|
|
parent = None
|
|
title = "Test page"
|
|
slug = factory.LazyAttribute(lambda obj: slugify(obj.title))
|
|
|
|
class Meta:
|
|
model = models.ContentPage
|
|
|
|
|
|
class RegularPageFactory(PageFactory):
|
|
title = "Regular page"
|
|
slug = factory.LazyAttribute(lambda obj: slugify(obj.title))
|
|
|
|
class Meta:
|
|
model = models.RegularPage
|
|
|
|
|
|
class PersonalisablePageMetadataFactory(factory.django.DjangoModelFactory):
|
|
class Meta:
|
|
model = PersonalisablePageMetadata
|