"""Memory 检索：事实 fallback 开关；compaction 后事实 stale。"""

from __future__ import annotations

from unittest.mock import MagicMock

import pytest

from app.core.config import settings
from app.features.memory import repo as memory_repo


def test_search_facts_sync_skips_recent_fallback_when_disabled(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setattr(
        settings, "memory_fact_search_use_recent_fallback", False, raising=False
    )
    calls: list[bool] = []

    def boom(*_a, **_k):
        calls.append(True)
        raise AssertionError("get_facts_for_user_sync should not run")

    monkeypatch.setattr(memory_repo, "get_facts_for_user_sync", boom)

    mock_session = MagicMock()
    mock_session.execute.return_value.unique.return_value.scalars.return_value.all.return_value = []

    out = memory_repo.search_facts_for_user_sync(
        mock_session, "user-1", "no_such_subject_xyz", 5
    )
    assert out == []
    assert calls == []


def test_search_facts_sync_uses_recent_fallback_when_enabled(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    monkeypatch.setattr(
        settings, "memory_fact_search_use_recent_fallback", True, raising=False
    )
    stub_fact = object()

    def fake_get_facts(sess, uid, lim):
        assert sess is mock_session
        assert uid == "user-1"
        return [stub_fact]  # type: ignore[list-item]

    monkeypatch.setattr(memory_repo, "get_facts_for_user_sync", fake_get_facts)

    mock_session = MagicMock()
    mock_session.execute.return_value.unique.return_value.scalars.return_value.all.return_value = []

    out = memory_repo.search_facts_for_user_sync(
        mock_session, "user-1", "no_match_query", 5
    )
    assert out == [stub_fact]


def test_mark_facts_stale_for_excluded_chunk_returns_rowcount() -> None:
    session = MagicMock()
    session.execute.return_value.rowcount = 3
    n = memory_repo.mark_facts_stale_for_excluded_chunk_sync(
        session, user_id="u1", chunk_id="chunk-9"
    )
    assert n == 3