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

tests for querysets in variant_exclude

Browse Source
This commit is contained in:
sewagodimo
2019-01-09 16:54:28 +02:00
parent 875d8302de
commit 94a5c6b289
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/unit/test_utils.py
View File

@ -1,6 +1,8 @@
from wagtail_personalisation.utils import ( from wagtail_personalisation.utils import (
exclude_variants, impersonate_other_page) exclude_variants, impersonate_other_page)
from wagtail.core.models import Page as WagtailPage
class Page(object): class Page(object):
def __init__(self, path, depth, url_path, title): def __init__(self, path, depth, url_path, title):
@ -57,3 +59,15 @@ def test_exclude_variants_excludes_pages_with_metadata_not_canonical():
page.personalisation_metadata.is_canonical = False page.personalisation_metadata.is_canonical = False
result = exclude_variants([page]) result = exclude_variants([page])
assert result == [] assert result == []
def test_exclude_variants_with_pages_querysets():
'''
Test that excludes variant works for querysets too
'''
page1 = WagtailPage(path="/", depth=0, url_path="/", title="Hoi")
page2 = WagtailPage(path="/", depth=0, url_path="/", title="Hoi")
pages = WagtailPage.objects.all().order_by('id')
print(pages)
result = exclude_variants(pages)
assert result == pages