# 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 problem.models import Problem, ProblemTag class ProblemPageTest(TestCase): def setUp(self): self.client = APIClient() self.user = User.objects.create(username="test", admin_type=SUPER_ADMIN) self.user.set_password("testaa") self.user.save() self.client.login(username="test", password="testaa") 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/1/') self.assertEqual(response.status_code, 200) def test_problem_does_not_exist(self): response = self.client.get('/problem/3/') 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 = User.objects.create(username="test", admin_type=SUPER_ADMIN) self.user.set_password("testaa") self.user.save() self.client.login(username="test", password="testaa") 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=User.objects.get(username="test")) # 以下是发布题目的测试 def test_invalid_format(self): data = {"title": "test1"} response = self.client.post(self.url, data=data) self.assertEqual(response.data["code"], 1) def test_release_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", "tags": [1]} response = self.client.post(self.url, data=json.dumps(data), content_type="application/json") self.assertEqual(response.data["code"], 0) # 以下是编辑题目的测试 def test_invalid_data(self): data = {"title": "test0"} response = self.client.put(self.url, data=data) self.assertEqual(response.data["code"], 1) def test_problem_does_not_exist(self): tags = ProblemTag.objects.filter(id__in=[1]) self.problem.tags.add(*tags) data = self._create_data(2, False, [1]) response = self.client.put(self.url, data=json.dumps(data), content_type="application/json") self.assertEqual(response.data, {"code": 1, "data": u"该题目不存在！"}) def test_edit_problem_successfully(self): tags = ProblemTag.objects.filter(id__in=[1]) self.problem.tags.add(*tags) data = self._create_data(1, True, [1, 2]) problem = Problem.objects.get(id=data["id"]) problem.tags.remove(*problem.tags.all()) problem.tags.add(*ProblemTag.objects.filter(id__in=data["tags"])) 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": 2} response = self.client.get(self.url, data=data) self.assertEqual(response.data, {"code": 1, "data": u"题目不存在"}) def test_query_problem_exists(self): data = {"problem_id": 1} 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 = User.objects.create(username="testx", admin_type=SUPER_ADMIN) self.user.set_password("testxx") self.user.save() self.client.login(username="testx", password="testxx") 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 = User.objects.create(username="test", admin_type=SUPER_ADMIN) self.user.set_password("testaa") self.user.save() self.client.login(username="test", password="testaa") 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=User.objects.get(username="test"))