8

adds related factory

This commit is contained in:
Boris Besemer
2016-11-30 10:45:33 +01:00
parent 8b72bcbe3c
commit 7306863fd8

View File

@@ -1,5 +1,5 @@
import factory
from django.db import models
from personalisation.models import Segment, TimeRule
@@ -7,6 +7,8 @@ class SegmentFactory(factory.DjangoModelFactory):
name = 'TestSegment'
status = 'enabled'
time_rule = factory.RelatedFactory(TimeRuleFactory, '%(app_label)s_%(class)s_related', action=TimeRule.ACTION_CREATE)
class Meta:
model = Segment
@@ -14,3 +16,4 @@ class SegmentFactory(factory.DjangoModelFactory):
class TimeRuleFactory(factory.DjangoModelFactory):
class Meta:
model = TimeRule