from __future__ import absolute_import, unicode_literals

import datetime

import pytest

from personalisation.models import Segment, TimeRule
from tests.factories.segment import (
    ReferralRuleFactory, SegmentFactory, TimeRuleFactory)

"""Factory tests"""
@pytest.mark.django_db
def test_create_segment_factory():
    factoried_segment = SegmentFactory()
    segment = Segment(name='TestSegment', status='enabled')
    time_rule = TimeRule(start_time=datetime.time(8,0,0), end_time=datetime.time(23,0,0), segment=segment)

    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():
    segment = SegmentFactory(name='TimeSegment')
    time_rule = TimeRuleFactory(start_time=datetime.time(8, 0, 0), end_time=datetime.time(23, 0, 0), segment=segment)

    assert time_rule.start_time == datetime.time(8,0,0)

"""ReferralRuleFactory tests"""
@pytest.mark.django_db
def test_create_segment_with_referral_rule():
    segment = SegmentFactory(name='Referral')
    referral_rule = ReferralRuleFactory(regex_string='test.test', segment=segment)

    assert referral_rule.regex_string == 'test.test'

@pytest.mark.django_db
def test_create_segment_with_new_referral_rule():
    segment = SegmentFactory()

    segment.referral_rule = ReferralRuleFactory(regex_string='test.notest', segment=segment)

    assert segment.referral_rule.regex_string == 'test.notest'