7

Wagtail 3 changes

This commit is contained in:
Alex Bridge
2022-01-28 11:53:57 +00:00
committed by nick.moreton
parent dd4530203f
commit c7eaec1315
89 changed files with 3003 additions and 1456 deletions

View File

@ -11,80 +11,68 @@ from tests.factories.segment import SegmentFactory
@pytest.mark.django_db
def test_no_segments(client, site):
response = client.get('/')
response = client.get("/")
assert response.status_code == 200
assert client.session['segments'] == []
assert client.session["segments"] == []
@pytest.mark.django_db
def test_referral_segment(client, site):
referral_segment = SegmentFactory(name='Referral')
ReferralRuleFactory(
regex_string="test.test",
segment=referral_segment
)
referral_segment = SegmentFactory(name="Referral")
ReferralRuleFactory(regex_string="test.test", segment=referral_segment)
response = client.get('/', **{'HTTP_REFERER': 'test.test'})
response = client.get("/", **{"HTTP_REFERER": "test.test"})
assert response.status_code == 200
assert client.session['segments'][0]['encoded_name'] == 'referral'
assert client.session["segments"][0]["encoded_name"] == "referral"
@pytest.mark.django_db
@freeze_time("10:00:00")
def test_time_and_referral_segment(client, site):
segment = SegmentFactory(name='Both')
segment = SegmentFactory(name="Both")
TimeRuleFactory(
start_time=datetime.time(8, 0, 0),
end_time=datetime.time(23, 0, 0),
segment=segment
)
ReferralRuleFactory(
regex_string="test.test",
segment=segment
segment=segment,
)
ReferralRuleFactory(regex_string="test.test", segment=segment)
response = client.get('/', **{'HTTP_REFERER': 'test.test'})
response = client.get("/", **{"HTTP_REFERER": "test.test"})
assert response.status_code == 200
assert client.session['segments'][0]['encoded_name'] == 'both'
assert client.session["segments"][0]["encoded_name"] == "both"
@pytest.mark.django_db
@freeze_time("7:00:00")
def test_no_time_but_referral_segment(client, site):
segment = SegmentFactory(name='Not both')
segment = SegmentFactory(name="Not both")
TimeRuleFactory(
start_time=datetime.time(8, 0, 0),
end_time=datetime.time(23, 0, 0),
segment=segment
)
ReferralRuleFactory(
regex_string="test.test",
segment=segment
segment=segment,
)
ReferralRuleFactory(regex_string="test.test", segment=segment)
response = client.get('/', **{'HTTP_REFERER': 'test.test'})
response = client.get("/", **{"HTTP_REFERER": "test.test"})
assert response.status_code == 200
assert len(client.session['segments']) == 0
assert len(client.session["segments"]) == 0
@pytest.mark.django_db
@freeze_time("9:00:00")
def test_time_but_no_referral_segment(client, site):
segment = SegmentFactory(name='Not both')
segment = SegmentFactory(name="Not both")
TimeRuleFactory(
start_time=datetime.time(8, 0, 0),
end_time=datetime.time(23, 0, 0),
segment=segment
)
ReferralRuleFactory(
regex_string="test.test",
segment=segment
segment=segment,
)
ReferralRuleFactory(regex_string="test.test", segment=segment)
response = client.get('/')
response = client.get("/")
assert response.status_code == 200
assert len(client.session['segments']) == 0
assert len(client.session["segments"]) == 0