fix broken test

This commit is contained in:
virusdefender 2017-01-24 20:53:22 +08:00
parent b9fe8d4669
commit f91970f75a
2 changed files with 5 additions and 5 deletions

View File

@ -30,8 +30,8 @@ class AnnouncementAdminTest(APITestCase):
self.assertEqual(resp_data["content"], "test content")
self.assertEqual(resp_data["visible"], False)
def test_delete_announcemen(self):
def test_delete_announcement(self):
id = self.test_create_announcement().data["data"]["id"]
resp = self.client.delete(self.url, data={"id": id})
resp = self.client.delete(self.url + "?id=" + str(id))
self.assertSuccess(resp)
self.assertFalse(Announcement.objects.filter(id=id).exists())

View File

@ -1,7 +1,7 @@
from django.utils.translation import ugettext as _
from account.decorators import super_admin_required
from utils.api import APIView, IDOnlySerializer, validate_serializer
from utils.api import APIView, validate_serializer
from .models import Announcement
from .serializers import (AnnouncementSerializer, CreateAnnouncementSerializer,
@ -57,8 +57,8 @@ class AnnouncementAdminAPI(APIView):
announcement = announcement.filter(visible=True)
return self.success(self.paginate_data(request, announcement, AnnouncementSerializer))
@validate_serializer(IDOnlySerializer)
@super_admin_required
def delete(self, request):
Announcement.objects.filter(id=request.data["id"]).delete()
if request.GET.get("id"):
Announcement.objects.filter(id=request.GET["id"]).delete()
return self.success()