merge problem and contest_problem

2017-09-24 09:48:17 +08:00
parent e9c7344815
commit 51c229a2c5
13 changed files with 263 additions and 97 deletions
--- a/submission/tests.py
+++ b/submission/tests.py
@@ -0,0 +1,57 @@
+from unittest import mock
+from copy import deepcopy
+
+from .models import Submission
+from problem.models import Problem, ProblemTag
+from utils.api.tests import APITestCase
+
+DEFAULT_PROBLEM_DATA = {"_id": "110", "title": "test", "description": "<p>test</p>", "input_description": "test",
+                        "output_description": "test", "time_limit": 1000, "memory_limit": 256, "difficulty": "Low",
+                        "visible": True, "tags": ["test"], "languages": ["C", "C++", "Java", "Python2"], "template": {},
+                        "samples": [{"input": "test", "output": "test"}], "spj": False, "spj_language": "C",
+                        "spj_code": "", "test_case_id": "499b26290cc7994e0b497212e842ea85",
+                        "test_case_score": [{"output_name": "1.out", "input_name": "1.in", "output_size": 0,
+                                             "stripped_output_md5": "d41d8cd98f00b204e9800998ecf8427e",
+                                             "input_size": 0, "score": 0}],
+                        "rule_type": "ACM", "hint": "<p>test</p>", "source": "test"}
+
+DEFAULT_SUBMISSION_DATA = {
+    "problem_id": "110",
+    "user_id": 1,
+    "username": "test",
+    "code": "xxxxxxxxxxxxxx",
+    "result": -2,
+    "info": {},
+    "language": "C",
+    "statistic_info": {}
+}
+
+
+class SubmissionListTest(APITestCase):
+    def setUp(self):
+        self.create_user("123", "345")
+        self.url = self.reverse("submission_list_api")
+        Submission.objects.create(**DEFAULT_SUBMISSION_DATA)
+
+    def test_get_submission_list(self):
+        resp = self.client.get(self.url, data={"limit": "10"})
+        self.assertSuccess(resp)
+
+
+@mock.patch("submission.views.oj.judge_task.delay")
+class SubmissionAPITest(APITestCase):
+    def setUp(self):
+        self.user = self.create_user("test", "test123")
+        tag = ProblemTag.objects.create(name="test")
+        problem_data = deepcopy(DEFAULT_PROBLEM_DATA)
+        problem_data.pop("tags")
+        problem_data["created_by"] = self.user
+        self.problem = Problem.objects.create(**problem_data)
+        self.problem.tags.add(tag)
+        self.problem.save()
+        self.url = self.reverse("submission_api")
+
+    def test_create_submission(self, judge_task):
+        resp = self.client.post(self.url, DEFAULT_SUBMISSION_DATA)
+        self.assertSuccess(resp)
+        judge_task.assert_called()