"""CMS 训练记录 + 带教能力相关接口测试。 - CMS-TRN-1 超管训练记录列表 GET /api/cms/training-records/ - CMS-TEA-3 教学工具-训练记录 GET /api/cms/students/training-records/ - CMS-TEA-4 学生能力画像 GET /api/cms/students/{id}/competency/ - CMS-TEA-5 学生排行榜 GET /api/cms/students/ranking/ training_record 是 managed=False 只读表，迁移占位结构缺新列，故在 setUpClass 里按真实列 重建该表（仅测试库），用 TransactionTestCase 允许 DDL（与 test_mobile_training_stats 一致）。 """ import json from django.core.cache import cache from django.db import connection from django.test import TransactionTestCase from django.utils import timezone from rest_framework.test import APIClient from apps.user.models import Department, User from apps.case.models import CaseBase from .conftest import ( create_test_user, get_auth_client, ensure_institution, create_teacher_student_relation, ) CREATE_TR = """ CREATE TABLE training_record ( id BIGINT NOT NULL AUTO_INCREMENT PRIMARY KEY, user_id BIGINT NULL, case_id BIGINT NULL, teacher_id BIGINT NULL, training_mode VARCHAR(50) NULL, case_type VARCHAR(30) NULL, start_time DATETIME NULL, end_time DATETIME NULL, duration_seconds INT NULL, total_score DECIMAL(5,2) NULL, ai_score DECIMAL(5,2) NULL, teacher_score DECIMAL(5,2) NULL, evaluation_level VARCHAR(20) NULL, status VARCHAR(30) NULL, feedback TEXT NULL, thinking_chain TEXT NULL, diagnosis_path TEXT NULL, wrong_points JSON NULL, missed_questions JSON NULL, recommendation_result JSON NULL, ai_feedback_structured JSON NULL, osce_station_score JSON NULL, interruption_count INT NULL, emotion_analysis JSON NULL, prompt_version VARCHAR(50) NULL, rag_context_version VARCHAR(50) NULL, external_user_id VARCHAR(128) NULL, session_id BIGINT NULL, evaluation_record_id BIGINT NULL, score_type VARCHAR(20) NULL, pdf_file_path VARCHAR(512) NULL, created_at DATETIME NULL, updated_at DATETIME NULL ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 """ SUP_URL = '/api/cms/training-records/' TEA_TR_URL = '/api/cms/students/training-records/' RANK_URL = '/api/cms/students/ranking/' def _dim(name, score, mx): return {'dimension': name, 'score': score, 'max_score': mx} # 各学生维度评分（得分率见注释） DIMS_A = [_dim('信息获取', 20, 25), _dim('分析推理', 18, 20), _dim('处置决策', 9, 10), _dim('沟通人文', 7, 10), _dim('临床整合', 8, 10)] # 80/90/90/70/80 DIMS_B = [_dim('信息获取', 20, 25), _dim('分析推理', 16, 20), _dim('处置决策', 8, 10), _dim('沟通人文', 8, 10), _dim('临床整合', 8, 10)] # 全 80 DIMS_C = [_dim('信息获取', 15, 25), _dim('分析推理', 12, 20), _dim('处置决策', 6, 10), _dim('沟通人文', 6, 10), _dim('临床整合', 6, 10)] # 全 60 class CmsTrainingTest(TransactionTestCase): @classmethod def setUpClass(cls): super().setUpClass() with connection.cursor() as c: c.execute('SET FOREIGN_KEY_CHECKS=0') c.execute('DROP TABLE IF EXISTS training_record') c.execute(CREATE_TR) c.execute('SET FOREIGN_KEY_CHECKS=1') def _insert(self, user_id, total_score, dims, status='completed', score_type='percentage', case_type='diagnosis_treatment'): now = timezone.now() with connection.cursor() as c: c.execute( "INSERT INTO training_record (user_id, case_id, training_mode, case_type, status, " "total_score, duration_seconds, end_time, start_time, score_type, evaluation_level, " "ai_feedback_structured, created_at, updated_at) " "VALUES (%s,%s,'practice',%s,%s,%s,%s,%s,%s,%s,'good',%s,%s,%s)", [user_id, self.case.id, case_type, status, total_score, 1800, now, now, score_type, json.dumps({'dimension_scores': dims}, ensure_ascii=False), now, now], ) return c.lastrowid def setUp(self): cache.clear() with connection.cursor() as c: c.execute('DELETE FROM training_record') self.inst = ensure_institution(name='测试医院', code='TR-H1') self.dept = Department.objects.create(name='心内科', category='临床') self.case = CaseBase.objects.create(title='急性心梗', case_type='traditional', department=self.dept, institution=self.inst) self.sup = create_test_user(phone='13970000001', role_type='super_admin', institution=self.inst) self.doc = create_test_user(phone='13970000002', real_name='张医生', role_type='doctor', institution=self.inst) self.other_doc = create_test_user(phone='13970000003', role_type='doctor', institution=self.inst) self.stu1 = create_test_user(phone='13970000011', real_name='学生甲', role_type='student', institution=self.inst) self.stu1.department = self.dept self.stu1.save(update_fields=['department']) self.stu2 = create_test_user(phone='13970000012', real_name='学生乙', role_type='student', institution=self.inst) self.stu_other = create_test_user(phone='13970000013', real_name='别家学生', role_type='student', institution=self.inst) create_teacher_student_relation(self.doc, self.stu1, status=1) create_teacher_student_relation(self.doc, self.stu2, status=1) create_teacher_student_relation(self.other_doc, self.stu_other, status=1) # 训练记录：stu1 两条完成 + 一条进行中；stu2 一条完成；stu_other 一条完成 self._insert(self.stu1.id, 90, DIMS_A) self._insert(self.stu1.id, 80, DIMS_B) self._insert(self.stu1.id, None, [], status='in_progress') self._insert(self.stu2.id, 60, DIMS_C) self._insert(self.stu_other.id, 70, DIMS_B) self.sup_client = get_auth_client(self.sup) self.doc_client = get_auth_client(self.doc) # ── CMS-TRN-1 超管训练记录 ────────────────────────────────────────────────── def test_trn_unauthenticated_401(self): self.assertEqual(APIClient().get(SUP_URL).status_code, 401) def test_trn_non_super_403(self): resp = self.doc_client.get(SUP_URL) self.assertEqual(resp.status_code, 403, resp.content) self.assertEqual(resp.json()['code'], 'CMS_PERMISSION_DENIED') def test_trn_super_list_all(self): resp = self.sup_client.get(SUP_URL) self.assertEqual(resp.status_code, 200, resp.content) data = resp.json() self.assertEqual(data['count'], 5) # 全平台所有记录 self.assertEqual(data['summary']['total'], 5) self.assertEqual(data['summary']['completed'], 4) self.assertEqual(data['summary']['avg_score'], 75.0) # avg(90,80,60,70) item = data['results'][0] for k in ('record_id', 'user_id', 'user_name', 'case_title', 'score', 'score_normalized', 'status', 'trained_at'): self.assertIn(k, item) def test_trn_filter_by_user_and_status(self): resp = self.sup_client.get(SUP_URL, {'user_id': self.stu1.id, 'status': 'completed'}) data = resp.json() self.assertEqual(data['count'], 2) self.assertTrue(all(r['user_id'] == self.stu1.id for r in data['results'])) def test_trn_filter_score_range(self): resp = self.sup_client.get(SUP_URL, {'min_score': 75}) data = resp.json() # 归一分≥75 的完成记录：90、80（stu1）、未含 60/70 self.assertEqual(data['count'], 2) def test_trn_search_by_student_name(self): resp = self.sup_client.get(SUP_URL, {'search': '学生甲'}) data = resp.json() self.assertEqual(data['count'], 3) # stu1 全部记录（2 完成 + 1 进行中） def test_trn_five_point_normalized(self): self._insert(self.stu2.id, 4, DIMS_B, score_type='five_point') # 4×20=80 resp = self.sup_client.get(SUP_URL, {'user_id': self.stu2.id, 'min_score': 80}) data = resp.json() self.assertEqual(data['count'], 1) self.assertEqual(data['results'][0]['score'], 4.0) self.assertEqual(data['results'][0]['score_normalized'], 80.0) # ── CMS-TEA-3 教学工具-训练记录（名下学生）────────────────────────────────── def test_tea_tr_scope_own_students(self): resp = self.doc_client.get(TEA_TR_URL) self.assertEqual(resp.status_code, 200, resp.content) data = resp.json() self.assertEqual(data['count'], 4) # stu1(3) + stu2(1)，不含 stu_other uids = {r['user_id'] for r in data['results']} self.assertNotIn(self.stu_other.id, uids) self.assertEqual(data['summary']['completed'], 3) def test_tea_tr_non_teacher_403(self): resp = self.sup_client.get(TEA_TR_URL) self.assertEqual(resp.status_code, 403, resp.content) def test_tea_tr_filter_by_student(self): resp = self.doc_client.get(TEA_TR_URL, {'user_id': self.stu2.id}) self.assertEqual(resp.json()['count'], 1) # ── CMS-TEA-4 能力画像 ────────────────────────────────────────────────────── def test_competency_own_student(self): resp = self.doc_client.get(f'/api/cms/students/{self.stu1.id}/competency/') self.assertEqual(resp.status_code, 200, resp.content) d = resp.json() self.assertEqual(d['student_id'], self.stu1.id) self.assertEqual(d['completed_cases'], 2) self.assertEqual(d['avg_score'], 85.0) # avg(90,80) self.assertEqual(d['diagnosis_accuracy'], 85) # 分析推理 avg(90,80) radar = {x['dimension']: x['score'] for x in d['radar']} self.assertEqual(set(radar), {'信息获取', '分析推理', '处置决策', '沟通人文', '临床整合'}) self.assertEqual(radar['信息获取'], 80) self.assertEqual(radar['沟通人文'], 75) self.assertEqual(d['weak_dimensions'], ['沟通人文']) def test_competency_other_student_404(self): resp = self.doc_client.get(f'/api/cms/students/{self.stu_other.id}/competency/') self.assertEqual(resp.status_code, 404, resp.content) # ── CMS-TEA-5 排行榜 ──────────────────────────────────────────────────────── def test_ranking_by_score_default(self): resp = self.doc_client.get(RANK_URL) self.assertEqual(resp.status_code, 200, resp.content) d = resp.json() self.assertEqual(d['dimension'], 'score') self.assertEqual(d['count'], 2) self.assertEqual(d['ranking'][0]['student_id'], self.stu1.id) # 85 > 60 self.assertEqual(d['ranking'][0]['rank'], 1) self.assertEqual(d['ranking'][1]['student_id'], self.stu2.id) def test_ranking_by_train_count(self): resp = self.doc_client.get(RANK_URL, {'dimension': 'train_count'}) d = resp.json() self.assertEqual(d['dimension'], 'train_count') self.assertEqual(d['ranking'][0]['student_id'], self.stu1.id) # 3 > 1 self.assertEqual(d['ranking'][0]['value'], 3) def test_ranking_by_competency_dimension(self): resp = self.doc_client.get(RANK_URL, {'dimension': '信息获取'}) d = resp.json() self.assertEqual(d['dimension'], '信息获取') self.assertEqual(d['ranking'][0]['student_id'], self.stu1.id) # 80 > 60 def test_ranking_invalid_dimension_falls_back_to_score(self): resp = self.doc_client.get(RANK_URL, {'dimension': '不存在'}) self.assertEqual(resp.json()['dimension'], 'score') def test_ranking_non_teacher_403(self): self.assertEqual(self.sup_client.get(RANK_URL).status_code, 403)