56 lines
1.0 KiB
Python
56 lines
1.0 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
import datetime
|
|
|
|
import factory
|
|
|
|
from personalisation import models
|
|
|
|
|
|
class SegmentFactory(factory.DjangoModelFactory):
|
|
name = 'TestSegment'
|
|
status = 'enabled'
|
|
|
|
class Meta:
|
|
model = models.Segment
|
|
|
|
|
|
class TimeRuleFactory(factory.DjangoModelFactory):
|
|
start_time = datetime.time(8, 0, 0)
|
|
end_time = datetime.time(23, 0, 0)
|
|
|
|
class Meta:
|
|
model = models.TimeRule
|
|
|
|
|
|
class DayRuleFactory(factory.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = models.DayRule
|
|
|
|
|
|
class ReferralRuleFactory(factory.DjangoModelFactory):
|
|
regex_string = "test.test"
|
|
|
|
class Meta:
|
|
model = models.ReferralRule
|
|
|
|
|
|
class VisitCountRuleFactory(factory.DjangoModelFactory):
|
|
operator = "more_than"
|
|
count = 0
|
|
|
|
class Meta:
|
|
model = models.VisitCountRule
|
|
|
|
|
|
class QueryRuleFactory(factory.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = models.QueryRule
|
|
|
|
class DeviceRuleFactory(factory.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = models.DeviceRule
|