7

Day rule and accompanying tests

Signed-off-by: Jasper Berghoef <jasper.berghoef@gmail.com>
This commit is contained in:
Jasper Berghoef
2017-01-10 15:36:25 +01:00
parent 9b90551e3b
commit a7b477d71f
6 changed files with 122 additions and 5 deletions

View File

@ -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()