adds related factory
This commit is contained in:
@@ -1,5 +1,5 @@
|
|||||||
import factory
|
import factory
|
||||||
|
from django.db import models
|
||||||
from personalisation.models import Segment, TimeRule
|
from personalisation.models import Segment, TimeRule
|
||||||
|
|
||||||
|
|
||||||
@@ -7,6 +7,8 @@ class SegmentFactory(factory.DjangoModelFactory):
|
|||||||
name = 'TestSegment'
|
name = 'TestSegment'
|
||||||
status = 'enabled'
|
status = 'enabled'
|
||||||
|
|
||||||
|
time_rule = factory.RelatedFactory(TimeRuleFactory, '%(app_label)s_%(class)s_related', action=TimeRule.ACTION_CREATE)
|
||||||
|
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Segment
|
model = Segment
|
||||||
|
|
||||||
@@ -14,3 +16,4 @@ class SegmentFactory(factory.DjangoModelFactory):
|
|||||||
class TimeRuleFactory(factory.DjangoModelFactory):
|
class TimeRuleFactory(factory.DjangoModelFactory):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = TimeRule
|
model = TimeRule
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user