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

Wagtail 3 changes

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

93
tests/unit/test_utils.py
View File

@ -2,39 +2,43 @@ import pytest
from django.test import override_settings
from wagtail.core.models import Page as WagtailPage
from tests.factories.page import (ContentPageFactory, PersonalisablePageMetadataFactory)
from tests.factories.page import ContentPageFactory, PersonalisablePageMetadataFactory
from wagtail_personalisation.utils import (
can_delete_pages, exclude_variants, get_client_ip, impersonate_other_page)
can_delete_pages,
exclude_variants,
get_client_ip,
impersonate_other_page,
)
locale_factory = False
try:
from tests.factories.page import LocaleFactory
locale_factory = True
from tests.factories.page import LocaleFactory # noqa
locale_factory = True # noqa
except ImportError:
pass
@pytest.fixture
def rootpage():
return ContentPageFactory(parent=None, path='/', depth=0, title='root')
return ContentPageFactory(parent=None, path="/", depth=0, title="root")
@pytest.fixture
def page(rootpage):
return ContentPageFactory(parent=rootpage, path='/hi', title='Hi')
return ContentPageFactory(parent=rootpage, path="/hi", title="Hi")
@pytest.fixture
def otherpage(rootpage):
return ContentPageFactory(parent=rootpage, path='/bye', title='Bye')
return ContentPageFactory(parent=rootpage, path="/bye", title="Bye")
@pytest.mark.django_db
def test_impersonate_other_page(page, otherpage):
impersonate_other_page(page, otherpage)
assert page.title == otherpage.title == 'Bye'
assert page.title == otherpage.title == "Bye"
assert page.path == otherpage.path
@ -50,58 +54,63 @@ def test_cannot_delete_pages_with_standard_user(user, segmented_page):
def test_get_client_ip_with_remote_addr(rf):
request = rf.get('/', REMOTE_ADDR='173.231.235.87')
assert get_client_ip(request) == '173.231.235.87'
request = rf.get("/", REMOTE_ADDR="173.231.235.87")
assert get_client_ip(request) == "173.231.235.87"
def test_get_client_ip_with_x_forwarded_for(rf):
request = rf.get('/', HTTP_X_FORWARDED_FOR='173.231.235.87',
REMOTE_ADDR='10.0.23.24')
assert get_client_ip(request) == '173.231.235.87'
request = rf.get(
"/", HTTP_X_FORWARDED_FOR="173.231.235.87", REMOTE_ADDR="10.0.23.24"
)
assert get_client_ip(request) == "173.231.235.87"
@override_settings(
WAGTAIL_PERSONALISATION_IP_FUNCTION='some.non.existent.path'
)
@override_settings(WAGTAIL_PERSONALISATION_IP_FUNCTION="some.non.existent.path")
def test_get_client_ip_custom_get_client_ip_function_does_not_exist(rf):
with pytest.raises(ImportError):
get_client_ip(rf.get('/'))
get_client_ip(rf.get("/"))
@override_settings(
WAGTAIL_PERSONALISATION_IP_FUNCTION='tests.utils.get_custom_ip'
)
@override_settings(WAGTAIL_PERSONALISATION_IP_FUNCTION="tests.utils.get_custom_ip")
def test_get_client_ip_custom_get_client_ip_used(rf):
assert get_client_ip(rf.get('/')) == '123.123.123.123'
assert get_client_ip(rf.get("/")) == "123.123.123.123"
def test_exclude_variants_with_pages_querysets():
'''
"""
Test that excludes variant works for querysets
'''
"""
for i in range(5):
page = ContentPageFactory(path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i))
page = ContentPageFactory(
path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i)
)
page.save()
pages = WagtailPage.objects.all().specific().order_by('id')
pages = WagtailPage.objects.all().specific().order_by("id")
result = exclude_variants(pages)
assert type(result) == type(pages)
assert set(result.values_list('pk', flat=True)) == set(pages.values_list('pk', flat=True))
assert set(result.values_list("pk", flat=True)) == set(
pages.values_list("pk", flat=True)
)
def test_exclude_variants_with_pages_querysets_not_canonical():
'''
"""
Test that excludes variant works for querysets with
personalisation_metadata canonical False
'''
"""
for i in range(5):
page = ContentPageFactory(path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i))
page = ContentPageFactory(
path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i)
)
page.save()
pages = WagtailPage.objects.all().specific().order_by('id')
pages = WagtailPage.objects.all().specific().order_by("id")
# add variants
for page in pages:
variant = ContentPageFactory(title='variant %d' % page.pk)
page.personalisation_metadata = PersonalisablePageMetadataFactory(canonical_page=page, variant=variant)
variant = ContentPageFactory(title="variant %d" % page.pk)
page.personalisation_metadata = PersonalisablePageMetadataFactory(
canonical_page=page, variant=variant
)
page.save()
pages = WagtailPage.objects.all().specific()
@ -111,19 +120,25 @@ def test_exclude_variants_with_pages_querysets_not_canonical():
def test_exclude_variants_with_pages_querysets_meta_none():
'''
"""
Test that excludes variant works for querysets with meta as none
'''
"""
for i in range(5):
page = ContentPageFactory(path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i))
page = ContentPageFactory(
path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i)
)
page.save()
pages = WagtailPage.objects.all().specific().order_by('id')
pages = WagtailPage.objects.all().specific().order_by("id")
# add variants
for page in pages:
page.personalisation_metadata = PersonalisablePageMetadataFactory(canonical_page=page, variant=page)
page.personalisation_metadata = PersonalisablePageMetadataFactory(
canonical_page=page, variant=page
)
page.save()
pages = WagtailPage.objects.all().specific()
result = exclude_variants(pages)
assert type(result) == type(pages)
assert set(result.values_list('pk', flat=True)) == set(pages.values_list('pk', flat=True))
assert set(result.values_list("pk", flat=True)) == set(
pages.values_list("pk", flat=True)
)