"""对话落库返回人/助 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