重构 problem 和 contest;修改测试和部分模板的结构
This commit is contained in:
@@ -7,16 +7,15 @@ 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 = 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.user = create_user()
|
||||
self.client.login(username="test", password="111111")
|
||||
self.problem = Problem.objects.create(title="title1",
|
||||
description="description1",
|
||||
input_description="input1_description",
|
||||
@@ -31,11 +30,11 @@ class ProblemPageTest(TestCase):
|
||||
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)
|
||||
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/3/')
|
||||
response = self.client.get('/problem/3000/')
|
||||
self.assertTemplateUsed(response, "utils/error.html")
|
||||
|
||||
|
||||
@@ -60,10 +59,8 @@ class ProblemAdminTest(APITestCase):
|
||||
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")
|
||||
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",
|
||||
@@ -77,15 +74,9 @@ class ProblemAdminTest(APITestCase):
|
||||
memory_limit=1000,
|
||||
difficulty=1,
|
||||
hint="hint1",
|
||||
created_by=User.objects.get(username="test"))
|
||||
created_by=self.user)
|
||||
|
||||
# 以下是发布题目的测试
|
||||
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):
|
||||
def test_create_problem_successfully(self):
|
||||
data = {"title": "title2",
|
||||
"description": "description2",
|
||||
"input_description": "input_description2",
|
||||
@@ -97,30 +88,15 @@ class ProblemAdminTest(APITestCase):
|
||||
"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_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"]))
|
||||
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)
|
||||
|
||||
@@ -139,11 +115,11 @@ class ProblemAdminTest(APITestCase):
|
||||
self.assertEqual(item["visible"], True)
|
||||
|
||||
def test_query_problem_does_not_exist(self):
|
||||
data = {"problem_id": 2}
|
||||
data = {"problem_id": 200}
|
||||
response = self.client.get(self.url, data=data)
|
||||
self.assertEqual(response.data, {"code": 1, "data": u"题目不存在"})
|
||||
|
||||
def test_query_problem_exists(self):
|
||||
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)
|
||||
@@ -153,10 +129,8 @@ 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")
|
||||
self.user = create_user(admin_type=SUPER_ADMIN)
|
||||
self.client.login(username="test", password="111111")
|
||||
ProblemTag.objects.create(name="tag1")
|
||||
|
||||
# 以下是返回所有的问题的标签
|
||||
@@ -168,10 +142,8 @@ 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")
|
||||
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",
|
||||
@@ -185,7 +157,7 @@ class ProblemListPageTest(TestCase):
|
||||
memory_limit=1000,
|
||||
difficulty=1,
|
||||
hint="hint1",
|
||||
created_by=User.objects.get(username="test"))
|
||||
created_by=self.user)
|
||||
|
||||
def test_problemListPage_not_exist(self):
|
||||
response = self.client.get('/problems/999/')
|
||||
|
||||
Reference in New Issue
Block a user