更新

app/tests/test_ai_consult_ui.py

@@ -1491,6 +1491,55 @@ def test_pending_and_silently_closed_streams_never_show_a_blank_bubble(
     dialog.close()
 
 
+def test_stalled_stream_exits_pending_state_and_ignores_late_events(
+    application: QApplication,
+    monkeypatch: pytest.MonkeyPatch,
+) -> None:
+    dialog = _silent_dialog(monkeypatch)
+    dialog.show()
+    application.processEvents()
+
+    dialog._ask("请总结当前病情")
+    generation, stream_generation = dialog._generation, dialog._stream_generation
+    bubble = dialog._stream_bubble
+    worker = dialog._stream_worker
+    assert bubble is not None and worker is not None
+    assert dialog._stream_timeout_timer.isActive()
+
+    dialog._stream_timed_out()
+    assert worker.is_cancelled()
+    assert not dialog._stream_timeout_timer.isActive()
+    assert not dialog._asking
+    assert dialog.send_button.isEnabled()
+    assert bubble._raw_payload == ai_consult_module.AI_STREAM_TIMEOUT_TEXT
+
+    calls: list[int] = []
+    monkeypatch.setattr(
+        dialog.repository,
+        "stream_diagnosis_ai",
+        lambda *args, **kwargs: calls.append(1),
+        raising=False,
+    )
+    worker.run()
+    assert calls == []
+    dialog._stream_event(generation, stream_generation, {"event": "delta", "text": "迟到回复"})
+    dialog._stream_finished(generation, stream_generation, worker)
+    assert bubble._raw_payload == ai_consult_module.AI_STREAM_TIMEOUT_TEXT
+
+    dialog._ask("请补充说明")
+    second_bubble = dialog._stream_bubble
+    assert second_bubble is not None
+    dialog._stream_event(
+        dialog._generation,
+        dialog._stream_generation,
+        {"event": "delta", "text": "已收到部分分析"},
+    )
+    dialog._stream_timed_out()
+    assert second_bubble._raw_payload.startswith("已收到部分分析\n\n>")
+    assert ai_consult_module.AI_STREAM_TIMEOUT_TEXT in second_bubble._raw_payload
+    dialog.close()
+
+
 def test_answered_stream_replaces_the_pending_placeholder(
     application: QApplication,
     monkeypatch: pytest.MonkeyPatch,
@@ -1501,15 +1550,29 @@ def test_answered_stream_replaces_the_pending_placeholder(
 
     dialog._ask("请总结当前病情")
     generation, stream_generation = dialog._generation, dialog._stream_generation
+    assert dialog._stream_bubble is not None
+    assert dialog._stream_bubble._raw_payload == ai_consult_module.AI_STREAM_PENDING_TEXT
+    dialog._stream_event(
+        generation,
+        stream_generation,
+        {"event": "start", "source_summary": {"patient_name": "敏感姓名"}},
+    )
+    assert dialog._stream_bubble._raw_payload == ai_consult_module.AI_STREAM_WAITING_TEXT
+    assert "敏感姓名" not in dialog._stream_bubble._raw_payload
     dialog._stream_event(generation, stream_generation, {"event": "delta", "text": "证候："})
+    dialog._flush_timer.stop()
+    dialog._flush_stream_chunks()
+    assert dialog._stream_bubble._raw_payload == "证候："
     dialog._stream_event(generation, stream_generation, {"event": "delta", "text": "脾肾两虚"})
     dialog._stream_event(generation, stream_generation, {"event": "done", "model_label": "千问"})
     dialog._stream_finished(generation, stream_generation, dialog._stream_worker)
     application.processEvents()
 
     assert dialog._stream_text == "证候：脾肾两虚"
+    assert not dialog._stream_timeout_timer.isActive()
     texts = _bubble_texts(dialog)
     assert not any(ai_consult_module.AI_STREAM_PENDING_TEXT in text for text in texts)
+    assert not any(ai_consult_module.AI_STREAM_WAITING_TEXT in text for text in texts)
     assert not any(ai_consult_module.AI_STREAM_SILENT_TEXT in text for text in texts)
     dialog.close()