import pytest

from app.core.knowledge_contracts import KnowledgeHit, KnowledgeQuery
from app.service.knowledge_authority import KnowledgeMysqlAuthority


class FakeRow:
    id = 101
    title = "申购规则"
    content_text = '{"answer":"工作日确认"}'
    version = "v2"
    milvus_collection = "fin_policy_collection"


class FakeSession:
    statement = None

    async def scalars(self, statement):
        self.statement = statement
        return [FakeRow()]


@pytest.mark.asyncio
async def test_authority_filters_to_published_active_effective_knowledge() -> None:
    session = FakeSession()
    authority = KnowledgeMysqlAuthority(session)

    hits = await authority.filter_published((
        KnowledgeHit(
            knowledge_id="101", collection="fin_policy_collection", snippet="摘要", score=0.91,
        ),
    ))

    statement = str(session.statement)
    assert "review_status" in statement
    assert "status" in statement
    assert "effective_date" in statement
    assert "expire_date" in statement
    assert hits[0].answer == "工作日确认"
    assert hits[0].version == "v2"


@pytest.mark.asyncio
async def test_authority_keyword_fallback_only_returns_effective_public_knowledge() -> None:
    session = FakeSession()
    authority = KnowledgeMysqlAuthority(session)

    hits = await authority.search_keyword(
        KnowledgeQuery(query="基金 申购确认", intents=("policy_explain",)),
        ("fin_policy_collection",),
        5,
    )

    statement = str(session.statement)
    assert "milvus_collection" in statement
    assert "content_text" in statement
    assert "review_status" in statement
    assert "effective_date" in statement
    assert hits[0].knowledge_id == "101"
    assert hits[0].collection == "fin_policy_collection"
    assert hits[0].answer == "工作日确认"