medical_training/test/test_case_happy.py

"""病例域 2 条 happy-path 流程测试。"""

from unittest.mock import patch, MagicMock

from django.core.cache import cache

from apps.case.models import CaseBase, TraditionalCase, ScoringRule, CaseExamItem
from apps.user.throttling import PdfParseUserThrottle, ScoringRuleGenerateUserThrottle
from .conftest import (
    CacheTestCase,
    CASE_PARSE_URL, CASE_GENERATE_RULES_URL, CASE_FULL_CREATE_URL,
    case_full_url, create_test_user, get_auth_client, ensure_department,
    build_traditional_payload, make_fake_pdf,
    MOCK_C1_PARSE_RESULT, MOCK_C2_SCORING_RULES,
)


class CaseFormHappyPathTest(CacheTestCase):
    """HP-5: 表单录入 → full-create → GET → PATCH → GET 验证"""

    def setUp(self):
        super().setUp()
        self.user = create_test_user(phone='13900100001', password='CaseTest1')
        self.client = get_auth_client(self.user)
        ensure_department('儿科')

    def test_flow_form_create_read_update(self):
        """HP-5: C3 full-create → C4 GET full → C5 PATCH → C4 GET verify"""
        # C3: full-create（2 条评分规则）
        payload = build_traditional_payload(
            department_name='儿科', scoring_rules_count=2, with_exam_items=True,
        )
        resp = self.client.post(CASE_FULL_CREATE_URL, payload, format='json')
        self.assertEqual(resp.status_code, 201, resp.content)

        created = resp.json()
        case_id = created['case']['id']
        self.assertEqual(created['case']['case_type'], 'traditional')
        self.assertEqual(created['case']['publish_status'], 0)  # 草稿
        self.assertIn('traditional', created)
        self.assertIsNotNone(created['traditional'])
        self.assertEqual(len(created['scoring_rules']), 2)
        self.assertEqual(len(created['exam_items']), 1)
        self.assertEqual(created['exam_items'][0]['item_code'], 'blood_routine')

        # C4: GET full
        resp = self.client.get(case_full_url(case_id))
        self.assertEqual(resp.status_code, 200, resp.content)
        full = resp.json()
        self.assertEqual(full['case']['title'], payload['title'])
        self.assertEqual(len(full['scoring_rules']), 2)
        self.assertEqual(len(full['exam_items']), 1)

        # C5: PATCH（改标题 + 改子表 + 替换为 1 条评分规则）
        patch_data = {
            'title': '更新后的标题',
            'traditional': {
                'standard_diagnosis': '更新后的诊断',
            },
            'scoring_rules': [
                {
                    'dimension': '更新后的维度',
                    'score_weight': 1.0,
                    'ai_auto_score': False,
                    'scoring_standard': '更新后的标准',
                },
            ],
        }
        resp = self.client.patch(case_full_url(case_id), patch_data, format='json')
        self.assertEqual(resp.status_code, 200, resp.content)

        # C4: GET 验证更新
        resp = self.client.get(case_full_url(case_id))
        self.assertEqual(resp.status_code, 200, resp.content)
        full = resp.json()
        self.assertEqual(full['case']['title'], '更新后的标题')
        self.assertEqual(full['traditional']['standard_diagnosis'], '更新后的诊断')
        self.assertEqual(len(full['scoring_rules']), 1)
        self.assertEqual(full['scoring_rules'][0]['dimension'], '更新后的维度')

        # 验证 DB
        case = CaseBase.objects.get(id=case_id)
        self.assertEqual(case.title, '更新后的标题')
        self.assertEqual(ScoringRule.objects.filter(case_id=case_id).count(), 1)
        self.assertEqual(CaseExamItem.objects.filter(case_id=case_id).count(), 1)
        tc = TraditionalCase.objects.get(case_id=case_id)
        self.assertEqual(tc.standard_diagnosis, '更新后的诊断')


class CasePdfMockHappyPathTest(CacheTestCase):
    """HP-6: PDF 解析（mock AI）→ 生成评分规则 → full-create → GET → PATCH"""

    def setUp(self):
        super().setUp()
        self.user = create_test_user(phone='13900100002', password='CaseTest2')
        self.client = get_auth_client(self.user)
        ensure_department('儿科')

    @patch('apps.case.services.case_importer.extract_text_from_pdfs',
           return_value='患儿，男，4岁，因发热3天就诊。体温38.5°C...')
    def test_flow_pdf_mock_full_pipeline(self, mock_pdf):
        """HP-6: C1 parse-pdf → C2 generate-scoring-rules → C3 full-create → C4 GET → C5 PATCH"""

        # mock call_deepseek: 第 1 次返回 C1 解析结果，第 2 次返回 C2 评分规则
        call_count = {'n': 0}
        def mock_deepseek(system_prompt, user_content):
            call_count['n'] += 1
            if call_count['n'] == 1:
                return MOCK_C1_PARSE_RESULT
            return MOCK_C2_SCORING_RULES

        with (
            patch('apps.case.services.deepseek_client.call_deepseek', side_effect=mock_deepseek),
            patch.object(PdfParseUserThrottle, 'allow_request', return_value=True),
            patch.object(ScoringRuleGenerateUserThrottle, 'allow_request', return_value=True),
        ):
            # C1: parse-pdf
            fake_pdf = make_fake_pdf()
            resp = self.client.post(
                CASE_PARSE_URL,
                {'files': fake_pdf, 'case_type': 'traditional'},
                format='multipart',
            )
            self.assertEqual(resp.status_code, 200, resp.content)
            parse_result = resp.json()
            self.assertIn('parse_id', parse_result)
            self.assertEqual(parse_result['case_type'], 'traditional')
            data = parse_result['data']
            self.assertEqual(data['case_type'], 'traditional')
            self.assertIn('traditional', data)
            self.assertNotIn('scoring_rules', data)

            # C2: generate-scoring-rules
            resp = self.client.post(
                CASE_GENERATE_RULES_URL,
                data,
                format='json',
            )
            self.assertEqual(resp.status_code, 200, resp.content)
            gen_result = resp.json()
            self.assertGreaterEqual(gen_result['generated'], 1)
            scoring_rules = gen_result['scoring_rules']

        # C3: full-create（组装 C1 data + C2 scoring_rules）
        create_payload = {**data}
        create_payload['scoring_rules'] = scoring_rules
        create_payload['parse_id'] = parse_result['parse_id']
        resp = self.client.post(CASE_FULL_CREATE_URL, create_payload, format='json')
        self.assertEqual(resp.status_code, 201, resp.content)
        created = resp.json()
        case_id = created['case']['id']
        self.assertEqual(len(created['scoring_rules']), len(scoring_rules))
        self.assertEqual(len(created['exam_items']), 1)

        # C4: GET full
        resp = self.client.get(case_full_url(case_id))
        self.assertEqual(resp.status_code, 200, resp.content)
        full = resp.json()
        self.assertEqual(full['case']['id'], case_id)
        self.assertEqual(full['case']['case_type'], 'traditional')

        # C5: PATCH
        resp = self.client.patch(case_full_url(case_id), {
            'title': 'AI-更新标题',
        }, format='json')
        self.assertEqual(resp.status_code, 200, resp.content)

        # 验证 PATCH 生效
        resp = self.client.get(case_full_url(case_id))
        self.assertEqual(resp.status_code, 200, resp.content)
        self.assertEqual(resp.json()['case']['title'], 'AI-更新标题')