Day rule and accompanying tests
Signed-off-by: Jasper Berghoef <jasper.berghoef@gmail.com>
This commit is contained in:
@ -6,7 +6,7 @@ import pytest
|
||||
|
||||
from personalisation.models import Segment, TimeRule
|
||||
from tests.factories.segment import (
|
||||
ReferralRuleFactory, SegmentFactory, TimeRuleFactory)
|
||||
ReferralRuleFactory, SegmentFactory, TimeRuleFactory, DayRuleFactory)
|
||||
|
||||
"""Factory tests"""
|
||||
@pytest.mark.django_db
|
||||
@ -18,6 +18,7 @@ def test_create_segment_factory():
|
||||
assert factoried_segment.name == segment.name
|
||||
assert factoried_segment.status == segment.status
|
||||
|
||||
|
||||
"""TimeRuleFactory tests"""
|
||||
@pytest.mark.django_db
|
||||
def test_create_segment_with_time_rule():
|
||||
@ -26,6 +27,18 @@ def test_create_segment_with_time_rule():
|
||||
|
||||
assert time_rule.start_time == datetime.time(8,0,0)
|
||||
|
||||
|
||||
"""TimeRuleFactory tests"""
|
||||
@pytest.mark.django_db
|
||||
def test_create_segment_with_day_rule():
|
||||
segment = SegmentFactory(name='DaySegment')
|
||||
day_rule = DayRuleFactory(mon=True, thu=True, segment=segment)
|
||||
|
||||
assert day_rule.mon is True
|
||||
assert day_rule.thu is True
|
||||
assert day_rule.sun is False
|
||||
|
||||
|
||||
"""ReferralRuleFactory tests"""
|
||||
@pytest.mark.django_db
|
||||
def test_create_segment_with_referral_rule():
|
||||
@ -34,6 +47,7 @@ def test_create_segment_with_referral_rule():
|
||||
|
||||
assert referral_rule.regex_string == 'test.test'
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_create_segment_with_new_referral_rule():
|
||||
segment = SegmentFactory()
|
||||
|
Reference in New Issue
Block a user