from pathlib import Path

import pytest

from haiku.rag.agents.research.dependencies import ResearchContext
from haiku.rag.agents.research.graph import build_research_graph
from haiku.rag.agents.research.models import ResearchReport
from haiku.rag.agents.research.state import ResearchDeps, ResearchState
from haiku.rag.client import HaikuRAG


@pytest.fixture(scope="module")
def vcr_cassette_dir():
    return str(
        Path(__file__).parent.parent.parent / "cassettes" / "test_research_graph"
    )


@pytest.mark.vcr()
async def test_graph_end_to_end(allow_model_requests, temp_db_path, qa_corpus):
    """Test research graph with real LLM calls recorded via VCR."""
    graph = build_research_graph()

    client = HaikuRAG(temp_db_path, create=True)
    doc = qa_corpus[0]
    await client.create_document(
        content=doc["document_extracted"], uri=doc["document_id"]
    )

    state = ResearchState(
        context=ResearchContext(original_question=doc["question"]),
        max_iterations=1,
        confidence_threshold=0.5,
        max_concurrency=1,
    )

    deps = ResearchDeps(client=client)

    result = await graph.run(state=state, deps=deps)

    assert result is not None
    assert isinstance(result, ResearchReport)
    assert result.title
    assert result.executive_summary

    client.close()


def test_research_context_initial_context():
    """Test ResearchContext accepts initial_context."""
    context = ResearchContext(
        original_question="What is X?",
        initial_context="Background: X is a concept in domain Y.",
    )
    assert context.initial_context == "Background: X is a concept in domain Y."


def test_research_context_initial_context_defaults_to_none():
    """Test ResearchContext initial_context defaults to None."""
    context = ResearchContext(original_question="What is X?")
    assert context.initial_context is None


def test_format_context_for_prompt_includes_initial_context():
    """Test format_context_for_prompt includes initial_context in output."""
    from haiku.rag.agents.research.graph import format_context_for_prompt

    context = ResearchContext(
        original_question="What is X?",
        initial_context="Background: X is a concept in domain Y.",
    )
    result = format_context_for_prompt(context)
    assert "Background: X is a concept in domain Y." in result
    assert "initial_context" in result


def test_format_context_for_prompt_excludes_initial_context_when_none():
    """Test format_context_for_prompt excludes initial_context when None."""
    from haiku.rag.agents.research.graph import format_context_for_prompt

    context = ResearchContext(original_question="What is X?")
    result = format_context_for_prompt(context)
    assert "initial_context" not in result


def test_format_conversational_context_for_prompt_includes_initial_context():
    """Test format_conversational_context_for_prompt includes initial_context."""
    from haiku.rag.agents.research.graph import format_conversational_context_for_prompt

    context = ResearchContext(
        original_question="What is X?",
        initial_context="Background: X is a concept in domain Y.",
    )
    result = format_conversational_context_for_prompt(context)
    assert "Background: X is a concept in domain Y." in result
    assert "initial_context" in result


def test_format_conversational_context_for_prompt_excludes_initial_context_when_none():
    """Test format_conversational_context_for_prompt excludes initial_context when None."""
    from haiku.rag.agents.research.graph import format_conversational_context_for_prompt

    context = ResearchContext(original_question="What is X?")
    result = format_conversational_context_for_prompt(context)
    assert "initial_context" not in result