Set a valid request.site in userbar tests

This commit is contained in:
jordij 2016-02-15 18:24:05 +13:00 committed by Josh Barr
parent 632b0c1821
commit 36d472ce74

View file

@ -6,7 +6,7 @@ from django.contrib.auth import get_user_model
from django.contrib.auth.models import AnonymousUser
from wagtail.tests.utils import WagtailTestUtils
from wagtail.wagtailcore.models import Page, PAGE_TEMPLATE_VAR
from wagtail.wagtailcore.models import Page, Site, PAGE_TEMPLATE_VAR
from wagtail.tests.testapp.models import BusinessIndex, BusinessChild
@ -22,6 +22,7 @@ class TestUserbarTag(TestCase):
def dummy_request(self, user=None):
request = RequestFactory().get('/')
request.user = user or AnonymousUser()
request.site = Site.objects.first()
return request
def test_userbar_tag(self):