70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
|
"""对话落库返回人/助 message id，供 segment lineage 配对。"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from unittest.mock import AsyncMock, MagicMock
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from app.features.conversation.history_store import HumanAiTurnIds
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.mark.asyncio
|
|||
|
|
async def test_record_human_ai_turn_returns_both_message_ids(monkeypatch) -> None:
|
|||
|
|
conv_id = "conv-1"
|
|||
|
|
captured: list[object] = []
|
|||
|
|
|
|||
|
|
class FakeMsg:
|
|||
|
|
def __init__(self, **kwargs) -> None:
|
|||
|
|
for k, v in kwargs.items():
|
|||
|
|
setattr(self, k, v)
|
|||
|
|
|
|||
|
|
class _FakeRepo:
|
|||
|
|
@staticmethod
|
|||
|
|
def add_conversation_message(msg: object, db) -> None:
|
|||
|
|
captured.append(msg)
|
|||
|
|
|
|||
|
|
monkeypatch.setattr(
|
|||
|
|
"app.features.conversation.history_store.ConversationMessage",
|
|||
|
|
FakeMsg,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
db = MagicMock(spec=AsyncSession)
|
|||
|
|
db.commit = AsyncMock()
|
|||
|
|
db.refresh = AsyncMock()
|
|||
|
|
|
|||
|
|
import app.features.conversation.history_store as hs_mod
|
|||
|
|
|
|||
|
|
orig_repo = hs_mod.repo
|
|||
|
|
hs_mod.repo = _FakeRepo # type: ignore[misc]
|
|||
|
|
try:
|
|||
|
|
from app.features.conversation import history_store as hs
|
|||
|
|
|
|||
|
|
store = hs.ConversationHistoryStore(db)
|
|||
|
|
store._sync_redis_best_effort = AsyncMock() # type: ignore[method-assign]
|
|||
|
|
store._touch_conversation = AsyncMock() # type: ignore[method-assign]
|
|||
|
|
|
|||
|
|
out = await store.record_human_ai_turn(
|
|||
|
|
conv_id,
|
|||
|
|
"hello",
|
|||
|
|
["reply a", "reply b"],
|
|||
|
|
user_message_timestamp=None,
|
|||
|
|
is_from_voice=False,
|
|||
|
|
voice_session_id=None,
|
|||
|
|
audio_duration_seconds=None,
|
|||
|
|
tts_audio_urls=None,
|
|||
|
|
segment_id="seg-1",
|
|||
|
|
memory_retrieval_trace=None,
|
|||
|
|
)
|
|||
|
|
finally:
|
|||
|
|
hs_mod.repo = orig_repo # type: ignore[misc]
|
|||
|
|
|
|||
|
|
assert isinstance(out, HumanAiTurnIds)
|
|||
|
|
assert len(captured) == 2
|
|||
|
|
assert captured[0].role == "human"
|
|||
|
|
assert captured[1].role == "ai"
|
|||
|
|
assert captured[0].segment_id == "seg-1"
|
|||
|
|
assert out.human_message_id == captured[0].id
|
|||
|
|
assert out.assistant_message_id == captured[1].id
|