Split admin_docs tests into separate files.

1 files changed, 44 insertions, 0 deletions

diff --git a/tests/admin_docs/test_middleware.py b/tests/admin_docs/test_middleware.py
new file mode 100644
index 0000000000..426c78d58f
--- /dev/null
+++ b/tests/admin_docs/test_middleware.py
@@ -0,0 +1,44 @@
+from __future__ import unicode_literals
+
+from django.contrib.auth.models import User
+
+from .tests import AdminDocsTestCase, TestDataMixin
+
+
+class XViewMiddlewareTest(TestDataMixin, AdminDocsTestCase):
+
+    def test_xview_func(self):
+        user = User.objects.get(username='super')
+        response = self.client.head('/xview/func/')
+        self.assertNotIn('X-View', response)
+        self.client.force_login(self.superuser)
+        response = self.client.head('/xview/func/')
+        self.assertIn('X-View', response)
+        self.assertEqual(response['X-View'], 'admin_docs.views.xview')
+        user.is_staff = False
+        user.save()
+        response = self.client.head('/xview/func/')
+        self.assertNotIn('X-View', response)
+        user.is_staff = True
+        user.is_active = False
+        user.save()
+        response = self.client.head('/xview/func/')
+        self.assertNotIn('X-View', response)
+
+    def test_xview_class(self):
+        user = User.objects.get(username='super')
+        response = self.client.head('/xview/class/')
+        self.assertNotIn('X-View', response)
+        self.client.force_login(self.superuser)
+        response = self.client.head('/xview/class/')
+        self.assertIn('X-View', response)
+        self.assertEqual(response['X-View'], 'admin_docs.views.XViewClass')
+        user.is_staff = False
+        user.save()
+        response = self.client.head('/xview/class/')
+        self.assertNotIn('X-View', response)
+        user.is_staff = True
+        user.is_active = False
+        user.save()
+        response = self.client.head('/xview/class/')
+        self.assertNotIn('X-View', response)