update exclude varient format and add variants to tests

2019-01-10 12:41:53 +02:00
parent f054b86e07
commit 9235932f00
2 changed files with 17 additions and 12 deletions
--- a/tests/unit/test_utils.py
+++ b/tests/unit/test_utils.py
@ -69,5 +69,10 @@ def test_exclude_variants_with_pages_querysets():
        page = WagtailPage(path="/" + str(i), depth=0, url_path="/", title="Hoi " + str(i))
        page.save()
    pages = WagtailPage.objects.all().order_by('id')
+    # add variants
+    for page in pages:
+        page.personalisation_metadata = Metadata(is_canonical=False)
+    pages = WagtailPage.objects.all().order_by('id')
    result = exclude_variants(pages)
-    assert result == pages
+    assert type(result) == type(pages)
+    assert result != pages