OnlineJudge/contest_submission/tests.py

# coding=utf-8
import json
from django.test import TestCase, Client
from django.core.urlresolvers import reverse
from account.models import User, REGULAR_USER, ADMIN, SUPER_ADMIN
from problem.models import Problem
from contest.models import Contest, ContestProblem
from submission.models import Submission
from rest_framework.test import APITestCase, APIClient


class ContestSubmissionAPITest(APITestCase):
    def setUp(self):
        self.client = APIClient()
        self.url = reverse('contest_submission_api')
        self.user1 = User.objects.create(username="test1", admin_type=SUPER_ADMIN)
        self.user1.set_password("testaa")
        self.user1.save()
        self.user2 = User.objects.create(username="test2", admin_type=REGULAR_USER)
        self.user2.set_password("testbb")
        self.user2.save()
        self.global_contest = Contest.objects.create(title="titlex", description="descriptionx", mode=1,
                                                     contest_type=1, show_rank=True, show_user_submission=True,
                                                     start_time="2015-08-15T10:00:00.000Z",
                                                     end_time="2015-08-30T12:00:00.000Z",
                                                     created_by=User.objects.get(username="test1"))
        self.contest_problem = ContestProblem.objects.create(title="titlex",
                                                             description="descriptionx",
                                                             input_description="input1_description",
                                                             output_description="output1_description",
                                                             test_case_id="1",
                                                             samples=json.dumps([{"input": "1 1", "output": "2"}]),
                                                             time_limit=100,
                                                             memory_limit=1000,
                                                             hint="hint1",
                                                             created_by=User.objects.get(username="test1"),
                                                             contest=Contest.objects.get(title="titlex"),
                                                             sort_index="a")

    # 以下是创建比赛的提交
    def test_invalid_format(self):
        self.client.login(username="test1", password="testaa")
        data = {"contest_id": self.global_contest.id, "language": 1}
        response = self.client.post(self.url, data=data)
        self.assertEqual(response.data["code"], 1)

    def test_contest_submission_successfully(self):
        self.client.login(username="test1", password="testaa")
        data = {"contest_id": self.global_contest.id, "problem_id": self.contest_problem.id,
                "language": 1, "code": '#include "stdio.h"\nint main(){\n\treturn 0;\n}'}
        response = self.client.post(self.url, data=data)
        self.assertEqual(response.data["code"], 0)

    def test_contest_problem_does_not_exist(self):
        self.client.login(username="test1", password="testaa")
        data = {"contest_id": self.global_contest.id, "problem_id": self.contest_problem.id + 10,
                "language": 1, "code": '#include "stdio.h"\nint main(){\n\treturn 0;\n}'}
        response = self.client.post(self.url, data=data)
        self.assertEqual(response.data, {"code": 1, "data": u"题目不存在"})


class ContestProblemMySubmissionListTest(TestCase):
    # 以下是我比赛单个题目的提交列表的测试
    def setUp(self):
        self.client = Client()
        self.user1 = User.objects.create(username="test1", admin_type=SUPER_ADMIN)
        self.user1.set_password("testaa")
        self.user1.save()
        self.user2 = User.objects.create(username="test2", admin_type=REGULAR_USER)
        self.user2.set_password("testbb")
        self.user2.save()
        self.global_contest = Contest.objects.create(title="titlex", description="descriptionx", mode=1,
                                                     contest_type=1, show_rank=True, show_user_submission=True,
                                                     start_time="2015-08-15T10:00:00.000Z",
                                                     end_time="2015-08-30T12:00:00.000Z",
                                                     created_by=User.objects.get(username="test1"))
        self.contest_problem = ContestProblem.objects.create(title="titlex",
                                                             description="descriptionx",
                                                             input_description="input1_description",
                                                             output_description="output1_description",
                                                             test_case_id="1",
                                                             samples=json.dumps([{"input": "1 1", "output": "2"}]),
                                                             time_limit=100,
                                                             memory_limit=1000,
                                                             hint="hint1",
                                                             created_by=self.user1,
                                                             contest=self.global_contest,
                                                             sort_index="a")

    def test_contestsList_page_not_exist(self):
        self.client.login(username="test1", password="testaa")
        response = self.client.get('/contest/1/submissions/999/')
        self.assertTemplateUsed(response, "utils/error.html")


class SubmissionAPITest(APITestCase):
    def setUp(self):
        self.client = APIClient()
        self.url = reverse('contest_submission_admin_api_view')
        self.userA = User.objects.create(username="test1", admin_type=ADMIN)
        self.userA.set_password("testaa")
        self.userA.save()
        self.userS = User.objects.create(username="test2", admin_type=SUPER_ADMIN)
        self.userS.set_password("testbb")
        self.userS.save()
        self.global_contest = Contest.objects.create(title="titlex", description="descriptionx", mode=1,
                                                     contest_type=2, show_rank=True, show_user_submission=True,
                                                     start_time="2015-08-15T10:00:00.000Z",
                                                     end_time="2015-08-15T12:00:00.000Z",
                                                     password="aacc", created_by=self.userS
                                                     )
        self.problem = ContestProblem.objects.create(title="title1",
                                                     description="description1",
                                                     input_description="input1_description",
                                                     output_description="output1_description",
                                                     test_case_id="1",
                                                     sort_index="1",
                                                     samples=json.dumps([{"input": "1 1", "output": "2"}]),
                                                     time_limit=100,
                                                     memory_limit=1000,
                                                     hint="hint1",
                                                     contest=self.global_contest,
                                                     created_by=self.userS)
        self.submission = Submission.objects.create(user_id=self.userA.id,
                                                    language=1,
                                                    code='#include "stdio.h"\nint main(){\n\treturn 0;\n}',
                                                    problem_id=self.problem.id)
        self.submissionS = Submission.objects.create(user_id=self.userS.id,
                                                     language=2,
                                                     code='#include "stdio.h"\nint main(){\n\treturn 0;\n}',
                                                     problem_id=self.problem.id)

    def test_submission_contest_does_not_exist(self):
        self.client.login(username="test2", password="testbb")
        response = self.client.get(self.url + "?contest_id=99")
        self.assertEqual(response.data["code"], 1)

    def test_submission_contest_parameter_error(self):
        self.client.login(username="test2", password="testbb")
        response = self.client.get(self.url)
        self.assertEqual(response.data["code"], 1)

    def test_submission_access_denied(self):
        self.client.login(username="test1", password="testaa")
        response = self.client.get(self.url + "?problem_id=" + str(self.problem.id))
        self.assertEqual(response.data["code"], 1)

    def test_submission_access_denied_with_contest_id(self):
        self.client.login(username="test1", password="testaa")
        response = self.client.get(self.url + "?contest_id=" + str(self.global_contest.id))
        self.assertEqual(response.data["code"], 1)

    def test_get_submission_successfully(self):
        self.client.login(username="test2", password="testbb")
        response = self.client.get(
            self.url + "?contest_id=" + str(self.global_contest.id) + "&problem_id=" + str(self.problem.id))
        self.assertEqual(response.data["code"], 0)

    def test_get_submission_successfully_problem(self):
        self.client.login(username="test2", password="testbb")
        response = self.client.get(self.url + "?problem_id=" + str(self.problem.id))
        self.assertEqual(response.data["code"], 0)

    def test_get_submission_problem_do_not_exist(self):
        self.client.login(username="test2", password="testbb")
        response = self.client.get(self.url + "?problem_id=9999")
        self.assertEqual(response.data["code"], 1)