more factories

tests/factories/page.py

@@ -0,0 +1,16 @@
+import factory
+from wagtail.wagtailcore.models import Page
+from personalisation.models import PersonalisablePage
+
+from tests.sandbox.pages.models import HomePage
+
+class PersonalisablePageFactory(factory.DjangoModelFactory):
+    class Meta:
+        model = PersonalisablePage
+
+    @classmethod
+    def _create(cls, *args, **kwargs):
+        obj = super(PersonalisablePageFactory, cls)._build(*args, **kwargs)
+        if not obj.title:
+            obj.title = "Page-Test"
+        return obj

tests/factories/segment.py

@@ -0,0 +1,16 @@
+import factory
+
+from personalisation.models import Segment, TimeRule
+
+
+class SegmentFactory(factory.DjangoModelFactory):
+    name = 'TestSegment'
+    status = 'enabled'
+
+    class Meta:
+        model = Segment
+
+
+class TimeRuleFactory(factory.DjangoModelFactory):
+    class Meta:
+        model = TimeRule

tests/sandbox/pages/models.py

@@ -0,0 +1,14 @@
+from django.db import models
+from wagtail.wagtailadmin.edit_handlers import FieldPanel
+from wagtail.wagtailcore.fields import RichTextField
+from personalisation.models import PersonalisablePage
+
+
+class HomePage(PersonalisablePage):
+    subtitle = models.CharField(max_length=255)
+    body = RichTextField(blank=True, default='')
+
+    content_panels = PersonalisablePage.content_panels + [
+        FieldPanel('subtitle'),
+        FieldPanel('body'),
+    ]