from django.test import TestCase from django.contrib.auth import get_user_model from task.models import Task from submission.models import Submission from prompt.models import Conversation, Message User = get_user_model() def _make_user(username="student1"): return User.objects.create_user(username=username, password="pw") def _make_task(): return Task.objects.create( title="Test Task", task_type="challenge", display=1, content="" ) class SubmissionMessageLinkTest(TestCase): def setUp(self): self.user = _make_user("student1") self.task = _make_task() self.conv = Conversation.objects.create(user=self.user, task=self.task) self.user_msg = Message.objects.create( conversation=self.conv, role="user", content="帮我做个按钮" ) self.asst_msg = Message.objects.create( conversation=self.conv, role="assistant", content="好的", code_html="", code_css="", code_js="" ) def test_create_submission_links_message(self): """POST /api/submission/ with message_id links the assistant message""" self.client.force_login(self.user) resp = self.client.post( "/api/submission/", data={ "task_id": self.task.id, "html": "", "css": "", "js": "", "message_id": self.asst_msg.id, }, content_type="application/json", ) self.assertEqual(resp.status_code, 200) self.asst_msg.refresh_from_db() sub = Submission.objects.filter(user=self.user, task=self.task).first() self.assertIsNotNone(sub) self.assertEqual(self.asst_msg.submission, sub) def test_create_submission_without_message_id(self): """POST /api/submission/ without message_id still works""" self.client.force_login(self.user) resp = self.client.post( "/api/submission/", data={"task_id": self.task.id, "html": "

hi

", "css": "", "js": ""}, content_type="application/json", ) self.assertEqual(resp.status_code, 200) def test_cannot_link_message_from_different_task(self): """message_id from a different task is silently ignored""" other_task = Task.objects.create( title="Other Task", task_type="challenge", display=2, content="" ) other_conv = Conversation.objects.create(user=self.user, task=other_task) other_msg = Message.objects.create( conversation=other_conv, role="assistant", content="other" ) self.client.force_login(self.user) resp = self.client.post( "/api/submission/", data={ "task_id": self.task.id, "html": "

x

", "css": "", "js": "", "message_id": other_msg.id, }, content_type="application/json", ) self.assertEqual(resp.status_code, 200) # Submission created, but message NOT linked (wrong task) other_msg.refresh_from_db() self.assertIsNone(other_msg.submission) class DeleteSubmissionCascadeTest(TestCase): def setUp(self): self.user = _make_user("student2") self.task = _make_task() self.conv = Conversation.objects.create(user=self.user, task=self.task) self.user_msg = Message.objects.create( conversation=self.conv, role="user", content="问题" ) self.asst_msg = Message.objects.create( conversation=self.conv, role="assistant", content="回答", code_html="

hi

", code_css="", code_js="" ) self.sub = Submission.objects.create( user=self.user, task=self.task, html="

hi

", css="", js="" ) self.asst_msg.submission = self.sub self.asst_msg.save(update_fields=["submission"]) def test_delete_submission_also_deletes_message_pair(self): """DELETE /api/submission/{id} deletes linked user+assistant messages""" self.client.force_login(self.user) resp = self.client.delete(f"/api/submission/{self.sub.id}") self.assertEqual(resp.status_code, 200) self.assertFalse(Submission.objects.filter(id=self.sub.id).exists()) self.assertFalse(Message.objects.filter(id=self.asst_msg.id).exists()) self.assertFalse(Message.objects.filter(id=self.user_msg.id).exists()) def test_delete_submission_without_linked_message(self): """DELETE /api/submission/{id} works even with no linked message""" sub2 = Submission.objects.create( user=self.user, task=self.task, html="", css="", js="" ) self.client.force_login(self.user) resp = self.client.delete(f"/api/submission/{sub2.id}") self.assertEqual(resp.status_code, 200) self.assertFalse(Submission.objects.filter(id=sub2.id).exists()) class DeleteMessagePairTest(TestCase): def setUp(self): self.user = _make_user("student3") self.task = _make_task() self.conv = Conversation.objects.create(user=self.user, task=self.task) self.user_msg = Message.objects.create( conversation=self.conv, role="user", content="问题" ) self.asst_msg = Message.objects.create( conversation=self.conv, role="assistant", content="回答", code_html="

ok

", code_css="", code_js="" ) self.sub = Submission.objects.create( user=self.user, task=self.task, html="

ok

", css="", js="" ) self.asst_msg.submission = self.sub self.asst_msg.save(update_fields=["submission"]) def test_delete_message_pair_also_deletes_submission(self): self.client.force_login(self.user) resp = self.client.delete(f"/api/prompt/messages/{self.asst_msg.id}/pair") self.assertEqual(resp.status_code, 200) self.assertFalse(Message.objects.filter(id=self.asst_msg.id).exists()) self.assertFalse(Message.objects.filter(id=self.user_msg.id).exists()) self.assertFalse(Submission.objects.filter(id=self.sub.id).exists()) data = resp.json() self.assertTrue(data["deleted"]) self.assertTrue(data["submission_deleted"]) def test_delete_message_pair_without_submission(self): # Create a second pair without a submission link user2 = Message.objects.create( conversation=self.conv, role="user", content="另一问" ) asst2 = Message.objects.create( conversation=self.conv, role="assistant", content="另一条" ) self.client.force_login(self.user) resp = self.client.delete(f"/api/prompt/messages/{asst2.id}/pair") self.assertEqual(resp.status_code, 200) data = resp.json() self.assertFalse(data["submission_deleted"]) self.assertFalse(Message.objects.filter(id=asst2.id).exists()) self.assertFalse(Message.objects.filter(id=user2.id).exists()) def test_delete_message_pair_forbidden_for_other_user(self): other = _make_user("other") self.client.force_login(other) resp = self.client.delete(f"/api/prompt/messages/{self.asst_msg.id}/pair") self.assertEqual(resp.status_code, 403)