增加了公告发布成功的测试，以及公告分页的APIview

announcement/serializers.py

@@ -1,8 +1,24 @@
 # coding=utf-8
 from rest_framework import serializers
 
+from account.models import User
+from .models import Announcement
 
-class AnnouncementSerializer(serializers.Serializer):
+
+class CreateAnnouncementSerializer(serializers.Serializer):
     title = serializers.CharField(max_length=50)
     content = serializers.CharField(max_length=10000)
 
+
+class AnnouncementSerializer(serializers.ModelSerializer):
+
+    class UserSerializer(serializers.ModelSerializer):
+        class Meta:
+            model = User
+            fields = ["username"]
+
+    created_by = UserSerializer()
+
+    class Meta:
+        model = Announcement
+

announcement/tests.py

@@ -3,14 +3,25 @@ from django.core.urlresolvers import reverse
 
 from rest_framework.test import APITestCase, APIClient
 
+from account.models import User
+
 
 class AnnouncementAPITest(APITestCase):
         def setUp(self):
             self.client = APIClient()
-            self.url = reverse("announcement_api")
+            self.url = reverse("announcement_admin_api")
+            user = User.objects.create(username="test")
+            user.set_password("test")
+            user.save()
 
         def test_invalid_format(self):
-            # todo 判断用户是否登录
+            self.client.login(username="test", password="test")
             data = {"title": "test1"}
             response = self.client.post(self.url, data=data)
             self.assertEqual(response.data["code"], 1)
+
+        def test_success_announcement(self):
+            self.client.login(username="test", password="test")
+            data = {"title": "title0", "content": "content0"}
+            response = self.client.post(self.url, data=data)
+            self.assertEqual(response.data, {"code": 0, "data": u"公告发布成功！"})

announcement/views.py

@@ -4,20 +4,19 @@ from rest_framework.views import APIView
 from utils.shortcuts import serializer_invalid_response, error_response, success_response
 
 from account.models import User
-
+from utils.shortcuts import paginate
 from .models import Announcement
-from .serializers import AnnouncementSerializer
+from .serializers import CreateAnnouncementSerializer, AnnouncementSerializer
 
 
-class AnnouncementAPIView(APIView):
-    # todo 判断用户是否需要登录
+class AnnouncementAdminAPIView(APIView):
     def post(self, request):
         """
         公告发布json api接口
         ---
-        request_serializer: AnnouncementSerializer
+        request_serializer: CreateAnnouncementSerializer
         """
-        serializer = AnnouncementSerializer(data=request.DATA)
+        serializer = CreateAnnouncementSerializer(data=request.DATA)
         if serializer.is_valid():
             data = serializer.data
             Announcement.objects.create(title=data["title"],
@@ -25,4 +24,15 @@ class AnnouncementAPIView(APIView):
                                         created_by=request.user)
             return success_response(u"公告发布成功！")
         else:
-            return serializer_invalid_response(serializer)
+            return serializer_invalid_response(serializer)
+
+
+class AnnouncementAPIView(APIView):
+    def get(self, request):
+        """
+        公告分页json api接口
+        ---
+        request_serializer: AnnouncementSerializer
+        """
+        announcement = Announcement.objects.all().order_by("last_update_time")
+        return paginate(request, announcement, AnnouncementSerializer)