前台比赛、比赛题目与普通题目的一系列界面

submission/tests.py

@@ -1,5 +1,10 @@
+# coding=utf-8
+import json
 from django.test import TestCase
-from account.models import User, REGULAR_USER
+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
 from submission.models import Submission
 from rest_framework.test import APITestCase, APIClient
 
@@ -12,6 +17,7 @@ class SubmissionsListPageTest(TestCase):
         self.user2.set_password("666666")
         self.user.set_password("666666")
         self.user.save()
+        self.user2.save()
         # self.client.login(username="gogoing", password="666666")
         self.submission = Submission.objects.create(user_id=self.user.id,
                                                     language=1,
@@ -42,3 +48,99 @@ class SubmissionsListPageTest(TestCase):
         self.client.login(username="cool", password="666666")
         response = self.client.get('/my_submissions/')
         self.assertEqual(response.status_code, 200)
+
+
+class SubmissionAPITest(APITestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.url = reverse('submission_api')
+        self.user1 = User.objects.create(username="test1", admin_type=REGULAR_USER)
+        self.user1.set_password("testaa")
+        self.user1.save()
+        self.user2 = User.objects.create(username="test2", admin_type=SUPER_ADMIN)
+        self.user2.set_password("testbb")
+        self.user2.save()
+        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="test2"))
+        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=User.objects.get(username="test2"))
+
+        self.submission = Submission.objects.create(user_id=self.user1.id,
+                                                    language=1,
+                                                    code='#include "stdio.h"\nint main(){\n\treturn 0;\n}',
+                                                    problem_id=self.problem.id)
+
+    def test_invalid_format(self):
+        self.client.login(username="test1", password="testaa")
+        data = {"language": 1}
+        response = self.client.post(self.url, data=data)
+        self.assertEqual(response.data["code"], 1)
+
+    def test_problem_does_not_exist(self):
+        self.client.login(username="test1", password="testaa")
+        data = {"language": 1, "code": '#include "stdio.h"\nint main(){\n\treturn 0;\n}',
+                "problem_id": self.problem.id + 10}
+        response = self.client.post(self.url, data=data)
+        self.assertEqual(response.data, {"code": 1, "data": u"题目不存在"})
+
+    def test_submission_successfully(self):
+        self.client.login(username="test1", password="testaa")
+        data = {"language": 1, "code": '#include "stdio.h"\nint main(){\n\treturn 0;\n}',
+                "problem_id": self.problem.id}
+        response = self.client.post(self.url, data=data)
+        self.assertEqual(response.data["code"], 0)
+
+    def test_submission_does_not_exist(self):
+        self.client.login(username="test1", password="testaa")
+        data = {"submission_id": self.submission.id + "111"}
+        response = self.client.get(self.url, data=data)
+        self.assertEqual(response.data, {"code": 1, "data": u"提交不存在"})
+
+    def test_get_submission_successfully(self):
+        self.client.login(username="test1", password="testaa")
+        data = {"submission_id": self.submission.id}
+        response = self.client.get(self.url, data=data)
+        self.assertEqual(response.data["code"], 0)
+
+
+class ContestSubmissionAPITest(APITestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.url = reverse('contest_submission_api')
+        self.user1 = User.objects.create(username="test1", admin_type=REGULAR_USER)
+        self.user1.set_password("testaa")
+        self.user1.save()
+        self.user2 = User.objects.create(username="test2", admin_type=SUPER_ADMIN)
+        self.user2.set_password("testbb")
+        self.user2.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=User.objects.get(username="test2"))
+
+    def test_invalid_format(self):
+        self.client.login(username="test1", password="testaa")
+        data = {"language": 1}
+        response = self.client.post(self.url, data=data)
+        pass
+
+
+
+
+
+
+