145 lines
4.5 KiB
Python
145 lines
4.5 KiB
Python
from django.test import TestCase, Client
|
|
from django.contrib.auth import get_user_model
|
|
import json
|
|
|
|
from tutorial.models import Tutorial, Exercise
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
def _json(response):
|
|
return json.loads(response.content)
|
|
|
|
|
|
class ExerciseAPITest(TestCase):
|
|
def setUp(self):
|
|
self.client = Client()
|
|
self.admin = User.objects.create(
|
|
username="admin_test", admin_type="Super Admin", is_staff=True
|
|
)
|
|
self.admin.set_password("password")
|
|
self.admin.save()
|
|
|
|
self.tutorial = Tutorial.objects.create(
|
|
title="Test Tutorial",
|
|
content="Hello [[exercise:1]]",
|
|
type="python",
|
|
is_public=True,
|
|
order=0,
|
|
created_by=self.admin,
|
|
)
|
|
self.exercise = Exercise.objects.create(
|
|
tutorial=self.tutorial,
|
|
type="mcq",
|
|
data={
|
|
"question": "Which is valid?",
|
|
"options": ["A", "B"],
|
|
"answer": 1,
|
|
},
|
|
order=0,
|
|
)
|
|
|
|
def test_get_exercises_for_tutorial(self):
|
|
resp = self.client.get(f"/api/exercises?tutorial_id={self.tutorial.id}")
|
|
self.assertEqual(resp.status_code, 200)
|
|
data = _json(resp)["data"]
|
|
self.assertEqual(len(data), 1)
|
|
self.assertEqual(data[0]["type"], "mcq")
|
|
|
|
def test_get_exercises_missing_tutorial_id(self):
|
|
resp = self.client.get("/api/exercises")
|
|
self.assertEqual(_json(resp)["error"], "error")
|
|
|
|
def test_get_exercises_nonexistent_tutorial(self):
|
|
resp = self.client.get("/api/exercises?tutorial_id=99999")
|
|
self.assertEqual(_json(resp)["error"], "error")
|
|
|
|
|
|
class ExerciseAdminAPITest(TestCase):
|
|
def setUp(self):
|
|
self.client = Client()
|
|
self.admin = User.objects.create(
|
|
username="admin2_test", admin_type="Super Admin", is_staff=True
|
|
)
|
|
self.admin.set_password("password")
|
|
self.admin.save()
|
|
self.client.login(username="admin2_test", password="password")
|
|
|
|
self.tutorial = Tutorial.objects.create(
|
|
title="T",
|
|
content="",
|
|
type="python",
|
|
is_public=False,
|
|
order=0,
|
|
created_by=self.admin,
|
|
)
|
|
|
|
def test_create_mcq(self):
|
|
resp = self.client.post(
|
|
"/api/admin/exercise",
|
|
data=json.dumps({
|
|
"tutorial_id": self.tutorial.id,
|
|
"type": "mcq",
|
|
"data": {"question": "Q?", "options": ["A", "B"], "answer": 0},
|
|
"order": 0,
|
|
}),
|
|
content_type="application/json",
|
|
)
|
|
self.assertEqual(resp.status_code, 200)
|
|
self.assertIsNone(_json(resp)["error"])
|
|
self.assertEqual(Exercise.objects.count(), 1)
|
|
|
|
def test_create_invalid_type(self):
|
|
resp = self.client.post(
|
|
"/api/admin/exercise",
|
|
data=json.dumps({
|
|
"tutorial_id": self.tutorial.id,
|
|
"type": "bad_type",
|
|
"data": {},
|
|
"order": 0,
|
|
}),
|
|
content_type="application/json",
|
|
)
|
|
self.assertEqual(_json(resp)["error"], "error")
|
|
|
|
def test_edit_exercise(self):
|
|
ex = Exercise.objects.create(
|
|
tutorial=self.tutorial,
|
|
type="mcq",
|
|
data={"question": "Old", "options": ["A", "B"], "answer": 0},
|
|
order=0,
|
|
)
|
|
resp = self.client.put(
|
|
"/api/admin/exercise",
|
|
data=json.dumps({
|
|
"id": ex.id,
|
|
"type": "mcq",
|
|
"data": {"question": "New", "options": ["A", "B"], "answer": 1},
|
|
"order": 0,
|
|
}),
|
|
content_type="application/json",
|
|
)
|
|
self.assertEqual(resp.status_code, 200)
|
|
ex.refresh_from_db()
|
|
self.assertEqual(ex.data["question"], "New")
|
|
|
|
def test_delete_exercise(self):
|
|
ex = Exercise.objects.create(
|
|
tutorial=self.tutorial,
|
|
type="sort",
|
|
data={"question": "Q", "lines": ["a", "b"]},
|
|
order=0,
|
|
)
|
|
resp = self.client.delete(f"/api/admin/exercise?id={ex.id}")
|
|
self.assertEqual(resp.status_code, 200)
|
|
self.assertEqual(Exercise.objects.count(), 0)
|
|
|
|
def test_requires_admin(self):
|
|
self.client.logout()
|
|
resp = self.client.post(
|
|
"/api/admin/exercise",
|
|
data=json.dumps({"tutorial_id": 1, "type": "mcq", "data": {}, "order": 0}),
|
|
content_type="application/json",
|
|
)
|
|
self.assertEqual(resp.status_code, 403)
|