55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
import datetime
|
|
|
|
import pytest
|
|
from freezegun import freeze_time
|
|
from django.http import HttpRequest
|
|
from wagtail.wagtailcore.models import Site
|
|
|
|
from personalisation import models
|
|
from tests.factories.site import SiteFactory
|
|
|
|
|
|
"""Time Rule tests"""
|
|
@freeze_time("10:00:00")
|
|
def test_create_time_rule():
|
|
time_rule = models.TimeRule(start_time=datetime.time(8, 0, 0), end_time=datetime.time(23, 0, 0))
|
|
|
|
assert time_rule.test_user() is True
|
|
|
|
@freeze_time("10:00:00")
|
|
def test_time_rule_false():
|
|
time_rule = models.TimeRule(start_time=datetime.time(11, 0, 0), end_time=datetime.time(23, 0, 0))
|
|
|
|
assert time_rule.test_user() is False
|
|
|
|
@freeze_time("10:00:00")
|
|
def test_time_rule_reverse():
|
|
time_rule = models.TimeRule(start_time=datetime.time(13, 0, 0), end_time=datetime.time(9, 0, 0))
|
|
|
|
assert time_rule.test_user() is False
|
|
|
|
@freeze_time("10:00:00")
|
|
def test_time_rule_reverse_next_day():
|
|
time_rule = models.TimeRule(start_time=datetime.time(11, 0, 0), end_time=datetime.time(11, 0, 0))
|
|
|
|
assert time_rule.test_user() is False
|
|
|
|
|
|
"""Visit Count Rule tests"""
|
|
def test_visit_count_rule():
|
|
visit_count_rule = models.VisitCountRule()
|
|
|
|
|
|
"""Test test"""
|
|
@pytest.mark.django_db
|
|
def test_test(rf):
|
|
site = SiteFactory()
|
|
|
|
request = HttpRequest()
|
|
request.path = '/'
|
|
request.META['HTTP_HOST'] = 'localhost'
|
|
request.META['SERVER_PORT'] = 8000
|
|
|
|
assert Site.find_for_request(request) == site
|
|
|