diff --git a/announcement/serializers.py b/announcement/serializers.py index f1a5f494..3aabb08b 100644 --- a/announcement/serializers.py +++ b/announcement/serializers.py @@ -8,6 +8,7 @@ from .models import Announcement class CreateAnnouncementSerializer(serializers.Serializer): title = serializers.CharField(max_length=50) content = serializers.CharField(max_length=10000) + visible = serializers.BooleanField() class AnnouncementSerializer(serializers.ModelSerializer): diff --git a/announcement/tests.py b/announcement/tests.py index a97b753c..dd702bad 100644 --- a/announcement/tests.py +++ b/announcement/tests.py @@ -13,7 +13,7 @@ class AnnouncementAdminTest(APITestCase): self.assertSuccess(response) def create_announcement(self): - return self.client.post(self.url, data={"title": "test", "content": "test"}) + return self.client.post(self.url, data={"title": "test", "content": "test", "visible": True}) def test_create_announcement(self): resp = self.create_announcement() diff --git a/announcement/views.py b/announcement/views.py index 4d1d4496..26fa7f5f 100644 --- a/announcement/views.py +++ b/announcement/views.py @@ -18,7 +18,8 @@ class AnnouncementAdminAPI(APIView): data = request.data announcement = Announcement.objects.create(title=data["title"], content=data["content"], - created_by=request.user) + created_by=request.user, + visible=data["visible"]) return self.success(AnnouncementSerializer(announcement).data) @validate_serializer(EditAnnouncementSerializer)