Return 404 if variant page is accessed directly (#188)
This commit is contained in:
@ -1,4 +1,7 @@
|
||||
import pytest
|
||||
|
||||
from django.http import Http404
|
||||
|
||||
from wagtail.core.models import Page
|
||||
|
||||
from tests.factories.segment import SegmentFactory
|
||||
@ -16,6 +19,15 @@ def test_serve_variant_no_variant(site, rf):
|
||||
assert result is None
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_variant_accessed_directly_returns_404(segmented_page, rf):
|
||||
request = rf.get('/')
|
||||
args = tuple()
|
||||
kwargs = {}
|
||||
with pytest.raises(Http404):
|
||||
wagtail_hooks.serve_variant(segmented_page, request, args, kwargs)
|
||||
|
||||
|
||||
@pytest.mark.django_db
|
||||
def test_serve_variant_with_variant_no_segment(site, rf, segmented_page):
|
||||
request = rf.get('/')
|
||||
|
Reference in New Issue
Block a user