From 36d472ce74525e225d7948933b60134c23e72f7f Mon Sep 17 00:00:00 2001 From: jordij Date: Mon, 15 Feb 2016 18:24:05 +1300 Subject: [PATCH] Set a valid request.site in userbar tests --- wagtail/wagtailadmin/tests/test_userbar.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/wagtail/wagtailadmin/tests/test_userbar.py b/wagtail/wagtailadmin/tests/test_userbar.py index 812f7a29f1..bb18e8f9b5 100644 --- a/wagtail/wagtailadmin/tests/test_userbar.py +++ b/wagtail/wagtailadmin/tests/test_userbar.py @@ -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):