更新

app/tests/test_mock_repository.py

@@ -232,6 +232,69 @@ def test_demo_call_lifecycle_mutates_record(repository: DemoDoctorRepository) ->
     assert ended["room_id"] == "room-501"
 
 
+def test_demo_transcript_upsert_and_finish_round_trip_in_call_records(
+    repository: DemoDoctorRepository,
+) -> None:
+    """Demo replay reads expose one finalized segment for a repeated segment ID."""
+
+    started = repository.start_call(501, 301)
+    call_record_id = started["id"]
+    repository.start_call_transcription(501, call_record_id, "session-1")
+    repository.upsert_call_transcript_segments(
+        501,
+        call_record_id,
+        "session-1",
+        [
+            {
+                "segment_id": "seg-1",
+                "speaker_user_id": "patient_301",
+                "speaker_role": "patient",
+                "timestamp": 1200,
+                "text": "draft words",
+            }
+        ],
+    )
+    repository.upsert_call_transcript_segments(
+        501,
+        call_record_id,
+        "session-1",
+        [
+            {
+                "segment_id": "seg-1",
+                "speaker_user_id": "patient_301",
+                "speaker_role": "patient",
+                "timestamp": 1200,
+                "text": "final words",
+            }
+        ],
+    )
+    repository.finish_call_transcription(
+        501,
+        call_record_id,
+        "session-1",
+        expected_segment_count=1,
+        status="completed",
+    )
+    repository.end_call(501)
+
+    record = next(
+        row for row in repository.list_call_records(501) if row["id"] == call_record_id
+    )
+    assert record["status"] == 2
+    assert record["transcription_status"] == "completed"
+    assert record["transcription_segment_count"] == 1
+    assert record["transcript_segments"] == [
+        {
+            "segment_id": "seg-1",
+            "speaker_role": "patient",
+            "speaker_user_id": "patient_301",
+            "timestamp": 1200,
+            "text": "final words",
+        }
+    ]
+    assert "final words" in record["transcript_text"]
+
+
 def test_tolerant_page_parsing_accepts_aliases_and_bad_rows() -> None:
     """List parsing handles nullable fields, aliases and non-object rows safely."""
 
@@ -315,6 +378,8 @@ class _StubApiClient:
                 "userSig": "short-lived",
                 "patientUserId": "patient_2",
             }
+        if endpoint == "tcm.diagnosis/startCall":
+            return {"call_record_id": 901}
         return {"ok": True}