OnlineJudge/problem/tests.py

# coding=utf-8
import json

from django.test import TestCase, Client

from django.core.urlresolvers import reverse
from rest_framework.test import APITestCase, APIClient

from account.models import User, SUPER_ADMIN
from account.tests import create_user
from problem.models import Problem, ProblemTag


class ProblemPageTest(TestCase):
    def setUp(self):
        self.client = APIClient()
        self.user = create_user()
        self.client.login(username="test", password="111111")
        self.problem = Problem.objects.create(title="title1",
                                              description="description1",
                                              input_description="input1_description",
                                              output_description="output1_description",
                                              test_case_id="1",
                                              source="source1",
                                              samples=json.dumps([{"input": "1 1", "output": "2"}]),
                                              time_limit=100,
                                              memory_limit=1000,
                                              difficulty=1,
                                              hint="hint1",
                                              created_by=User.objects.get(username="test"))

    def test_visit_problem_successfully(self):
        response = self.client.get('/problem/' + str(self.problem.id) + "/")
        self.assertTemplateUsed(response, "oj/problem/problem.html")

    def test_problem_does_not_exist(self):
        response = self.client.get('/problem/3000/')
        self.assertTemplateUsed(response, "utils/error.html")


class ProblemAdminTest(APITestCase):
    def _create_data(self, problem_id, visible, tags):
        data = {"id": problem_id,
                "title": "title0",
                "description": "description0",
                "input_description": "input_description0",
                "output_description": "output_description0",
                "test_case_id": "1",
                "source": "source1",
                "samples": [{"input": "1 1", "output": "2"}],
                "time_limit": "100",
                "memory_limit": "1000",
                "difficulty": "1",
                "hint": "hint1",
                "visible": visible,
                "tags": tags}
        return data

    def setUp(self):
        self.client = APIClient()
        self.url = reverse("problem_admin_api")
        self.user = create_user(admin_type=SUPER_ADMIN)
        self.client.login(username="test", password="111111")
        ProblemTag.objects.create(name="tag1")
        ProblemTag.objects.create(name="tag2")
        self.problem = Problem.objects.create(title="title1",
                                              description="description1",
                                              input_description="input1_description",
                                              output_description="output1_description",
                                              test_case_id="1",
                                              source="source1",
                                              samples=json.dumps([{"input": "1 1", "output": "2"}]),
                                              time_limit=100,
                                              memory_limit=1000,
                                              difficulty=1,
                                              hint="hint1",
                                              created_by=self.user)

    def test_create_problem_successfully(self):
        data = {"title": "title2",
                "description": "description2",
                "input_description": "input_description2",
                "output_description": "output_description2",
                "test_case_id": "1",
                "source": "source1",
                "samples": [{"input": "1 1", "output": "2"}],
                "time_limit": "100",
                "memory_limit": "1000",
                "difficulty": "1",
                "hint": "hint1",
                "visible": True,
                "tags": [1]}
        response = self.client.post(self.url, data=json.dumps(data), content_type="application/json")
        self.assertEqual(response.data["code"], 0)

    def test_edit_problem_successfully(self):
        tags = ProblemTag.objects.filter(id__in=[1])
        self.problem.tags.add(*tags)
        data = self._create_data(self.problem.id, True, [1, 2])
        response = self.client.put(self.url, data=json.dumps(data), content_type="application/json")
        self.assertEqual(response.data["code"], 0)

    # 以下是题目分页的测试
    def test_success_get_data(self):
        self.assertEqual(self.client.get(self.url).data["code"], 0)

    def test_query_by_keyword(self):
        response = self.client.get(self.url + "?keyword=title1")
        self.assertEqual(response.data["code"], 0)

    def test_query_by_visible(self):
        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_query_problem_does_not_exist(self):
        data = {"problem_id": 200}
        response = self.client.get(self.url, data=data)
        self.assertEqual(response.data, {"code": 1, "data": u"题目不存在"})

    def test_query_existed_problem(self):
        data = {"problem_id": self.problem.id}
        response = self.client.get(self.url, data=data)
        self.assertEqual(response.data["code"], 0)


class ProblemTagAdminAPITest(APITestCase):
    def setUp(self):
        self.client = APIClient()
        self.url = reverse('problem_tag_admin_api')
        self.user = create_user(admin_type=SUPER_ADMIN)
        self.client.login(username="test", password="111111")
        ProblemTag.objects.create(name="tag1")

    # 以下是返回所有的问题的标签
    def test_get_all_problem_tag_successfully(self):
        self.assertEqual(self.client.get(self.url).data["code"], 0)


class ProblemListPageTest(TestCase):
    def setUp(self):
        self.client = Client()
        self.url = reverse('problem_list_page', kwargs={"page": 1})
        self.user = create_user(admin_type=SUPER_ADMIN)
        self.client.login(username="test", password="111111")
        ProblemTag.objects.create(name="tag1")
        ProblemTag.objects.create(name="tag2")
        self.problem = Problem.objects.create(title="title1",
                                              description="description1",
                                              input_description="input1_description",
                                              output_description="output1_description",
                                              test_case_id="1",
                                              source="source1",
                                              samples=json.dumps([{"input": "1 1", "output": "2"}]),
                                              time_limit=100,
                                              memory_limit=1000,
                                              difficulty=1,
                                              hint="hint1",
                                              created_by=self.user)

    def test_problemListPage_not_exist(self):
        response = self.client.get('/problems/999/')
        self.assertTemplateUsed(response, "utils/error.html")

    def test_query_by_keyword(self):
        response = self.client.get(self.url + "?keyword=title1")
        self.assertEqual(response.status_code, 200)

    def test_query_by_tag_successfully(self):
        response = self.client.get(self.url + "?tag=")
        self.assertEqual(response.status_code, 200)

    def test_tag_does_not_exists(self):
        response = self.client.get(self.url + "?tag=xxxxxx")
        self.assertTemplateUsed(response, "utils/error.html")