Add tests for the wagtail_hooks.serve_variaton

tests/unit/test_wagtail_hooks.py

@@ -0,0 +1,41 @@
+import pytest
+
+from wagtail_personalisation import adapters, wagtail_hooks
+
+
+@pytest.mark.django_db
+def test_serve_variation_no_variant(site, rf):
+    page = site.root_page
+    request = rf.get('/')
+    args = tuple()
+    kwargs = {}
+
+    result = wagtail_hooks.serve_variation(page, request, args, kwargs)
+    assert result is None
+
+
+@pytest.mark.django_db
+def test_serve_variation_with_variant_no_segment(site, rf, segmented_page):
+    request = rf.get('/')
+    args = tuple()
+    kwargs = {}
+
+    page = segmented_page.personalisable_metadata.canonical_page
+    result = wagtail_hooks.serve_variation(page, request, args, kwargs)
+    assert result is None
+
+
+@pytest.mark.django_db
+def test_serve_variation_with_variant_segmented(site, rf, segmented_page):
+    request = rf.get('/')
+    args = tuple()
+    kwargs = {}
+
+    page = segmented_page.personalisable_metadata.canonical_page
+    segment = segmented_page.personalisable_metadata.segment
+
+    adapter = adapters.get_segment_adapter(request)
+    adapter.set_segments([segment])
+
+    result = wagtail_hooks.serve_variation(page, request, args, kwargs)
+    assert result.status_code == 200