Fixed #20114 -- support custom project login_url in tests

Thanks to Matias Bordese for the patch
This commit is contained in:
Preston Holmes 2013-04-05 08:43:22 -07:00
parent 975c5afdb5
commit a49e7dd2a3
1 changed files with 4 additions and 1 deletions

View File

@ -1,3 +1,4 @@
from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.contrib.auth.tests.test_views import AuthViewsTestCase
from django.contrib.auth.tests.utils import skipIfCustomUser
@ -27,11 +28,13 @@ class LoginRequiredTestCase(AuthViewsTestCase):
pass
login_required(normal_view)
def testLoginRequired(self, view_url='/login_required/', login_url='/login/'):
def testLoginRequired(self, view_url='/login_required/', login_url=None):
"""
Check that login_required works on a simple view wrapped in a
login_required decorator.
"""
if login_url is None:
login_url = settings.LOGIN_URL
response = self.client.get(view_url)
self.assertEqual(response.status_code, 302)
self.assertTrue(login_url in response.url)