增加了AdminRequired的测试用例

admin/tests.py

@@ -0,0 +1,68 @@
+# coding=utf-8
+import json
+from django.test import TestCase, Client
+from django.core.urlresolvers import reverse
+from django.http import HttpResponse
+
+from account.models import User
+
+
+def middleware_test_func(request):
+    return HttpResponse(json.dumps({"code": 0}))
+
+
+class AdminRequiredMidlewareTest(TestCase):
+    urls = "admin.test_urls"
+
+    def setUp(self):
+        admin_user = User.objects.create(username="test", admin_type=0)
+        admin_user.set_password("test")
+        admin_user.save()
+
+        admin_user = User.objects.create(username="test1", admin_type=1)
+        admin_user.set_password("test")
+        admin_user.save()
+        super_admin_user = User.objects.create(username="test2", admin_type=2)
+        super_admin_user.set_password("test")
+        super_admin_user.save()
+
+        self.client = Client()
+
+    def test_need_admin_login(self):
+        url = "/admin/"
+        response = self.client.get(url)
+        self.assertRedirects(response, "/login/")
+
+        self.client.login(username="test", password="test")
+        response = self.client.get(url)
+        self.assertRedirects(response, "/login/")
+        self.client.logout()
+
+        self.client.login(username="test1", password="test")
+        response = self.client.get(url)
+        self.assertTemplateUsed(response, "admin/admin.html")
+        self.client.logout()
+
+        self.client.login(username="test2", password="test")
+        response = self.client.get(url)
+        self.assertTemplateUsed(response, "admin/admin.html")
+
+    def test_need_admin_login_ajax(self):
+        url = "/api/admin/test/"
+        response = self.client.get(url, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(json.loads(response.content), {"code": 1, "data": u"请先登录"})
+
+        self.client.login(username="test", password="test")
+        rresponse = self.client.get(url, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(json.loads(response.content), {"code": 1, "data": u"请先登录"})
+        self.client.logout()
+
+        self.client.login(username="test1", password="test")
+        response = self.client.get(url, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(json.loads(response.content)["code"], 0)
+        self.client.logout()
+
+        self.client.login(username="test2", password="test")
+        response = self.client.get(url, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(json.loads(response.content)["code"], 0)
+