import pytest

from review_agent.models import KnowledgeBaseDocument
from review_agent.services import build_knowledge_context


pytestmark = pytest.mark.django_db


def test_build_knowledge_context_ignores_irrelevant_rag_chunks(monkeypatch):
    monkeypatch.setattr(
        "review_agent.services.search_knowledge_base",
        lambda query, n_results=5: {
            "query": query,
            "results": [
                {
                    "source": "附件 4 体外诊断试剂注册申报资料要求及说明.doc",
                    "text": "预期用途应明确产品用于检测的分析物和功能。",
                    "score": 7.636,
                    "metadata": {"source_type": "regulatory_document"},
                }
            ],
            "error_message": "",
        },
    )

    context = build_knowledge_context("孙之烨是谁")

    assert context == ""


def test_build_knowledge_context_uses_full_document_when_name_matches(settings, tmp_path, monkeypatch, django_user_model):
    settings.MEDIA_ROOT = tmp_path
    user = django_user_model.objects.create_user(username="owner", password="pass")
    document_path = tmp_path / "resume.txt"
    document_path.write_text(
        "孙之烨，负责审核智能体项目。\n完整经历：曾组织技术分享并带队参加竞赛。",
        encoding="utf-8",
    )
    KnowledgeBaseDocument.objects.create(
        user=user,
        display_name="孙之烨简历",
        original_name="孙之烨-260510.txt",
        storage_path=str(document_path),
        file_size=document_path.stat().st_size,
        status=KnowledgeBaseDocument.Status.ACTIVE,
        is_active=True,
        indexed_chunk_count=2,
    )
    monkeypatch.setattr(
        "review_agent.services.search_knowledge_base",
        lambda query, n_results=5: {"query": query, "results": [], "error_message": ""},
    )

    context = build_knowledge_context("孙之烨是谁")

    assert "全文材料" in context
    assert "来源：用户知识库/孙之烨-260510.txt" in context
    assert "完整经历：曾组织技术分享并带队参加竞赛" in context