OnlineJudge/announcement/tests.py

# coding=utf-8
from django.core.urlresolvers import reverse
from django.test import TestCase

from rest_framework.test import APITestCase, APIClient

from account.models import User
from group.models import Group
from announcement.models import Announcement
from account.models import REGULAR_USER, ADMIN, SUPER_ADMIN


class AnnouncementAdminAPITest(APITestCase):
        def setUp(self):
            self.client = APIClient()
            self.url = reverse("announcement_admin_api")
            user1 = User.objects.create(username="test1", admin_type=SUPER_ADMIN)
            user1.set_password("testaa")
            user1.save()
            user2 = User.objects.create(username="test2", admin_type=ADMIN)
            user2.set_password("testbb")
            user2.save()
            self.group = Group.objects.create(name="group1", description="des0",
                                              join_group_setting=0, visible=True,
                                              admin=user2)
            self.announcement = Announcement.objects.create(title="bb",
                                                            content="BB",
                                                            created_by=User.objects.get(username="test2"),
                                                            is_global=False)

        # 以下是发布公告的测试
        def test_invalid_format(self):
            self.client.login(username="test1", password="testaa")
            data = {"title": "test1"}
            response = self.client.post(self.url, data=data)
            self.assertEqual(response.data["code"], 1)

        def test_group_at_least_one(self):
            self.client.login(username="test1", password="testaa")
            data = {"title": "title0", "content": "content0", "is_global": False}
            response = self.client.post(self.url, data=data)
            self.assertEqual(response.data, {"code": 1, "data": u"至少选择一个小组"})

        def test_global_announcement_successfully(self):
            self.client.login(username="test1", password="testaa")
            data = {"title": "title0", "content": "content0", "is_global": True}
            response = self.client.post(self.url, data=data)
            self.assertEqual(response.data, {"code": 0, "data": u"公告发布成功！"})

        def test_group_announcement_successfully(self):
            self.client.login(username="test2", password="testbb")
            data = {"title": "title0", "content": "content0", "is_global": False, "groups": [self.group.id]}
            response = self.client.post(self.url, data=data)
            self.assertEqual(response.data, {"code": 0, "data": u"公告发布成功！"})

        def test_global_announcement_does_not_has_privileges(self):
            self.client.login(username="test2", password="testbb")
            data = {"title": "title0", "content": "content0", "is_global": True}
            response = self.client.post(self.url, data=data)
            self.assertEqual(response.data, {"code": 1, "data": u"只有超级管理员可以创建全局公告"})

        # 以下是编辑公告的测试
        def test_put_invalid_data(self):
            self.client.login(username="test1", password="testaa")
            data = {"title": "test0", "content": "test0", "visible": "True"}
            response = self.client.put(self.url, data=data)
            self.assertEqual(response.data["code"], 1)

        def test_announcement_does_not_exist(self):
            self.client.login(username="test1", password="testaa")
            announcement = Announcement.objects.create(title="aa",
                                                       content="AA",
                                                       created_by=User.objects.get(username="test1"),
                                                       is_global=True)
            data = {"id": announcement.id + 1, "title": "11", "content": "22",
                    "visible": True, "is_global": True}
            response = self.client.put(self.url, data=data)
            self.assertEqual(response.data, {"code": 1, "data": u"公告不存在"})

        def test_edit_global_announcement_successfully(self):
            self.client.login(username="test1", password="testaa")
            data = {"id": self.announcement.id, "title": "11", "content": "22",
                    "visible": True, "is_global": True}
            response = self.client.put(self.url, data=data)
            self.assertEqual(response.data["code"], 0)

        def test_edit_group_announcement_successfully(self):
            self.client.login(username="test2", password="testbb")
            data = {"id": self.announcement.id, "title": "11", "content": "22",
                    "visible": True, "is_global": False, "groups": [self.group.id]}
            response = self.client.put(self.url, data=data)
            self.assertEqual(response.data["code"], 0)

        def test_edit_group_at_least_one(self):
            self.client.login(username="test1", password="testaa")
            data = {"id": self.announcement.id, "title": "title0", "content": "content0",
                    "visible": True, "is_global": False}
            response = self.client.put(self.url, data=data)
            self.assertEqual(response.data, {"code": 1, "data": u"至少选择一个小组"})

        # 以下是公告分页的测试
        def test_get_data_successfully(self):
            self.client.login(username="test1", password="testaa")
            self.assertEqual(self.client.get(self.url).data["code"], 0)

        def test_keyword_global_announcement(self):
            self.client.login(username="test1", password="testaa")
            Announcement.objects.create(title="aa",
                                        content="AA",
                                        created_by=User.objects.get(username="test1"),
                                        visible=True,
                                        is_global=True)

            Announcement.objects.create(title="bb",
                                        content="BB",
                                        created_by=User.objects.get(username="test1"),
                                        visible=False,
                                        is_global=True)

            response = self.client.get(self.url + "?visible=true")
            self.assertEqual(response.data["code"], 0)
            for item in response.data["data"]:
                self.assertEqual(item["visible"], True)

        def test_keyword_group_announcement(self):
            self.client.login(username="test2", password="testbb")
            Announcement.objects.create(title="aa",
                                        content="AA",
                                        created_by=User.objects.get(username="test2"),
                                        visible=True,
                                        is_global=False)

            Announcement.objects.create(title="cc",
                                        content="CC",
                                        created_by=User.objects.get(username="test2"),
                                        visible=False,
                                        is_global=False)

            response = self.client.get(self.url + "?visible=true")
            self.assertEqual(response.data["code"], 0)
            for item in response.data["data"]:
                self.assertEqual(item["visible"], True)


class AnnouncementPageTest(TestCase):
    def setUp(self):
        user = User.objects.create(username="test")
        user.set_password("testaa")
        user.save()
        Announcement.objects.create(title="aa",
                                    content="AA",
                                    created_by=User.objects.get(username="test"),
                                    visible=True,
                                    is_global=True)

        Announcement.objects.create(title="bb",
                                    content="BB",
                                    created_by=User.objects.get(username="test"),
                                    visible=False,
                                    is_global=True)

    def test_visit_announcement_successfully(self):
        response = self.client.get('/announcement/1/')
        self.assertEqual(response.status_code, 200)

    def test_announcement_does_not_exist(self):
        response = self.client.get('/announcement/3/')
        self.assertTemplateUsed(response, "utils/error.html")