Cavemanon/cavemanon-wagtail-personalisation
Archived
7
Fork 0
Code Issues Pull Requests Activity

adds a few more middleware tests

Browse Source
This commit is contained in:
Boris Besemer
2016-12-01 12:00:51 +01:00
parent 38a6f8f32a
commit 510ed6d7c7
1 changed files with 51 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
tests/unit/test_middleware.py
View File

@ -45,3 +45,54 @@ class TestUserSegmenting(object):
client.get('/', **{ 'HTTP_REFERER': 'test.test'})
assert client.session['segments'][0]['encoded_name'] == 'referral'
@freeze_time("10:00:00")
def test_time_and_referral_segment(self, client):
segment = SegmentFactory(name='Both')
time_rule = TimeRuleFactory(
start_time=datetime.time(8, 0, 0),
end_time=datetime.time(23, 0, 0),
segment=segment
)
referral_rule = ReferralRuleFactory(
regex_string="test.test",
segment=segment
)
client.get('/', **{ 'HTTP_REFERER': 'test.test'})
assert client.session['segments'][0]['encoded_name'] == 'both'
@freeze_time("7:00:00")
def test_no_time_but_referral_segment(self, client):
segment = SegmentFactory(name='Not both')
time_rule = TimeRuleFactory(
start_time=datetime.time(8, 0, 0),
end_time=datetime.time(23, 0, 0),
segment=segment
)
referral_rule = ReferralRuleFactory(
regex_string="test.test",
segment=segment
)
client.get('/', **{ 'HTTP_REFERER': 'test.test'})
assert len(client.session['segments']) == 0
@freeze_time("9:00:00")
def test_time_but_no_referral_segment(self, client):
segment = SegmentFactory(name='Not both')
time_rule = TimeRuleFactory(
start_time=datetime.time(8, 0, 0),
end_time=datetime.time(23, 0, 0),
segment=segment
)
referral_rule = ReferralRuleFactory(
regex_string="test.test",
segment=segment
)
client.get('/')
assert len(client.session['segments']) == 0