OnlineJudge/announcement/tests.py
2016-10-30 02:17:35 +08:00

31 lines
1.1 KiB
Python

# coding=utf-8
from django.core.urlresolvers import reverse
from utils.tests import APITestCase
class AnnouncementAdminTest(APITestCase):
def setUp(self):
self.user = self.create_super_admin(login=True)
self.url = reverse("announcement_admin_api")
def test_announcement_list(self):
response = self.client.get(self.url)
self.assertSuccess(response)
def create_announcement(self):
return self.client.post(self.url, data={"title": "test", "content": "test"})
def test_create_announcement(self):
response = self.create_announcement()
self.assertSuccess(response)
def test_edit_announcement(self):
data = {"id": self.create_announcement().data["data"]["id"], "title": "ahaha", "content": "test content",
"visible": False}
response = self.client.put(self.url, data=data)
self.assertSuccess(response)
resp_data = response.data["data"]
self.assertEqual(resp_data["title"], "ahaha")
self.assertEqual(resp_data["content"], "test content")
self.assertEqual(resp_data["visible"], False)