Ticket #5775: test_staffmemberrequired.patch

django/contrib/admin/tests/__init__.py

@@ -1 @@
-from django.contrib.admin.tests import widgets
+from django.contrib.admin.tests import widgets, decorators
 
 __test__ = {
     'WIDGET_TESTS': widgets,
+    'DECORATORS_TESTS': decorators,
 }

django/contrib/admin/tests/decorators.py

@@ +1 @@
+"""
+>>> from django.contrib.auth.models import User
+>>> from django.test import Client
+
+Created a regular user and an admin user.
+
+>>> adminuser = User.objects.create_user('staff', 'staff@mysite', 'staffpassword')
+>>> normaluser = User.objects.create_user('user', 'user@mysite', 'userpassword')
+>>> adminuser.is_staff = True
+>>> adminuser.save()
+
+Login as a staff user and access a protected view.
+
+>>> c = Client()
+>>> c.login(username='staff', password='staffpassword')
+True
+>>> response = c.get('/admin/staff_members_only/', {'param':1})
+>>> response.status_code
+200
+>>> response.content
+'param:1'
+
+Login as a normal user and access a protected view.
+
+>>> c = Client()
+>>> c.login(username='user', password='userpassword')
+True
+>>> response = c.get('/admin/staff_members_only/', {'param':1})
+>>> response.status_code
+200
+>>> response.content.find('action="/admin/staff_members_only/?param=1"') > 0
+True
+
+"""
+
+from django.contrib.admin.views.decorators import staff_member_required
+from django.http import HttpResponse
+
+def test_staff_member_required( req ):
+  return HttpResponse( '\n'.join( u'%s:%s' % (item,req.GET[item]) for item in req.GET) )
+test_staff_member_required = staff_member_required( test_staff_member_required )

tests/urls.py

@@ -11 +11 @@
 
     # test urlconf for {% url %} template tag
     (r'^url_tag/', include('regressiontests.templates.urls')),
-    
+
     # django built-in views
     (r'^views/', include('regressiontests.views.urls')),
+
+    # test urlconf for the staff_member_required admin decorator
+    ('^admin/staff_members_only/$', 'django.contrib.admin.tests.decorators.test_staff_member_required'),
 )