import pytest

from tests.factories.rule import ReferralRuleFactory
from tests.factories.segment import SegmentFactory


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

    assert referral_rule.regex_string == 'test.test'