增加 @admin_required 修饰符代码和对应的测试

account/tests.py

@@ -10,7 +10,7 @@ from rest_framework.views import APIView
 from rest_framework.response import Response
 
 from .models import User
-from .decorators import login_required
+from .decorators import login_required, admin_required
 
 
 class UserLoginTest(TestCase):
@@ -159,6 +159,7 @@ class LoginRequiredCBVTestWithoutArgs(APIView):
     def get(self, request):
         return HttpResponse("class based view login required test1")
 
+
 class LoginRequiredCBVTestWithArgs(APIView):
     @login_required
     def get(self, request, problem_id):
@@ -176,40 +177,113 @@ class LoginRequiredDecoratorTest(TestCase):
 
     def test_fbv_without_args(self):
         # 没登陆
-        response = self.client.get("/test/fbv/1/")
+        response = self.client.get("/login_required_test/fbv/1/")
         self.assertTemplateUsed(response, "utils/error.html")
 
         # 登陆后
         self.client.login(username="test", password="test")
-        response = self.client.get("/test/fbv/1/")
+        response = self.client.get("/login_required_test/fbv/1/")
         self.assertEqual(response.content, "function based view test1")
 
     def test_fbv_with_args(self):
         # 没登陆
-        response = self.client.get("/test/fbv/1024/")
+        response = self.client.get("/login_required_test/fbv/1024/")
         self.assertTemplateUsed(response, "utils/error.html")
 
         # 登陆后
         self.client.login(username="test", password="test")
-        response = self.client.get("/test/fbv/1024/")
+        response = self.client.get("/login_required_test/fbv/1024/")
         self.assertEqual(response.content, "1024")
 
     def test_cbv_without_args(self):
         # 没登陆
-        response = self.client.get("/test/cbv/1/")
+        response = self.client.get("/login_required_test/cbv/1/")
         self.assertTemplateUsed(response, "utils/error.html")
 
         # 登陆后
         self.client.login(username="test", password="test")
-        response = self.client.get("/test/cbv/1/")
+        response = self.client.get("/login_required_test/cbv/1/")
         self.assertEqual(response.content, "class based view login required test1")
 
     def test_cbv_with_args(self):
         # 没登陆
-        response = self.client.get("/test/cbv/1024/", HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        response = self.client.get("/login_required_test/cbv/1024/", HTTP_X_REQUESTED_WITH='XMLHttpRequest')
         self.assertEqual(json.loads(response.content), {"code": 1, "data": u"请先登录"})
 
         # 登陆后
         self.client.login(username="test", password="test")
-        response = self.client.get("/test/cbv/1024/")
+        response = self.client.get("/login_required_test/cbv/1024/")
+        self.assertEqual(response.content, "1024")
+
+
+@admin_required
+def admin_required_FBV_test_without_args(request):
+    return HttpResponse("function based view test1")
+
+
+@admin_required
+def admin_required_FBC_test_with_args(request, problem_id):
+    return HttpResponse(problem_id)
+
+
+class AdminRequiredCBVTestWithoutArgs(APIView):
+    @admin_required
+    def get(self, request):
+        return HttpResponse("class based view login required test1")
+
+
+class AdminRequiredCBVTestWithArgs(APIView):
+    @admin_required
+    def get(self, request, problem_id):
+        return HttpResponse(problem_id)
+
+
+class AdminRequiredDecoratorTest(TestCase):
+    urls = 'account.test_urls'
+
+    def setUp(self):
+        self.client = Client()
+        user = User.objects.create(username="test")
+        user.admin_type = 1
+        user.set_password("test")
+        user.save()
+
+    def test_fbv_without_args(self):
+        # 没登陆
+        response = self.client.get("/admin_required_test/fbv/1/")
+        self.assertTemplateUsed(response, "utils/error.html")
+
+        # 登陆后
+        self.client.login(username="test", password="test")
+        response = self.client.get("/admin_required_test/fbv/1/")
+        self.assertEqual(response.content, "function based view test1")
+
+    def test_fbv_with_args(self):
+        # 没登陆
+        response = self.client.get("/admin_required_test/fbv/1024/")
+        self.assertTemplateUsed(response, "utils/error.html")
+
+        # 登陆后
+        self.client.login(username="test", password="test")
+        response = self.client.get("/admin_required_test/fbv/1024/")
+        self.assertEqual(response.content, "1024")
+
+    def test_cbv_without_args(self):
+        # 没登陆
+        response = self.client.get("/admin_required_test/cbv/1/")
+        self.assertTemplateUsed(response, "utils/error.html")
+
+        # 登陆后
+        self.client.login(username="test", password="test")
+        response = self.client.get("/admin_required_test/cbv/1/")
+        self.assertEqual(response.content, "class based view login required test1")
+
+    def test_cbv_with_args(self):
+        # 没登陆
+        response = self.client.get("/admin_required_test/cbv/1024/", HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(json.loads(response.content), {"code": 1, "data": u"需要管理员权限"})
+
+        # 登陆后
+        self.client.login(username="test", password="test")
+        response = self.client.get("/admin_required_test/cbv/1024/")
         self.assertEqual(response.content, "1024")