"""Question/engine binding across actual HTTP/MCP boundaries, without an LLM."""
import asyncio
import json
import os
import sys

import pytest

from test_claude_packages import backend, ROOT
from packages.claude import client


QUESTIONS=[
    'Čomu sa venuje diplomová práca Jána Holpa? Stručne ju predstav a uveď zdroj.',
    'Porovnaj  zadanie a výsledky Ondreja Megelu a Jána Holpa.\nPlán nie je výsledok; uveď zdroje.',
    'Ktorí študenti majú tag "RAG" a zároveň diplomový projekt z roku 2026?']


@pytest.mark.parametrize('engine',['rag','graphrag'])
def test_latest_question_and_all_conditions_cross_http_unchanged(backend,engine):
    for question in QUESTIONS:
        r=client.query_backend(question,engine,limit=7)
        assert r['accepted_for_delivery'] and r['query']==question and r['engine']==engine
        path,body,_=backend['requests'][-1]
        assert path=='/answer' and body['query']==question and body['engine']==engine and body['limit']==7
    assert len(backend['requests'])==len(QUESTIONS)


@pytest.mark.parametrize('engine',['rag','graphrag'])
def test_only_a_matching_leading_command_is_removed(backend,engine):
    question='Aké práce má Ján Holp v roku 2020? Zachovaj výraz "RAG".'
    r=client.query_backend('  /'+engine.upper()+'  '+question+'  ',engine)
    assert r['accepted_for_delivery'] and r['query']==question
    assert backend['requests'][-1][1]['query']==question
    other='graphrag' if engine=='rag' else 'rag'
    r=client.query_backend('/'+other+' '+question,engine)
    assert r['reason']=='engine_mismatch' and not r['accepted_for_delivery']
    assert len(backend['requests'])==1


@pytest.mark.parametrize('question',['/rag','/rag  ','/graphrag'])
def test_command_without_a_question_never_reaches_backend(backend,question):
    engine='graphrag' if question.startswith('/graphrag') else 'rag'
    assert client.query_backend(question,engine)['reason']=='invalid_query'
    assert not backend['requests']


@pytest.mark.parametrize('change',[
    {'query':'A previous unrelated question','engine':'rag'},
    {'query':QUESTIONS[0],'engine':'graphrag'},
])
def test_stale_question_or_wrong_engine_cannot_deliver_another_answer(backend,change):
    backend['raw']=json.dumps({**backend['checked'],'answer':'OTHER_ANSWER_SHOULD_NOT_ESCAPE',**change}).encode()
    r=client.query_backend(QUESTIONS[0],'rag')
    assert r['reason']=='question_mismatch' and not r['accepted_for_delivery']
    assert 'OTHER_ANSWER' not in json.dumps(r)


def test_missing_question_echo_is_not_accepted(backend):
    backend['raw']=json.dumps({**backend['checked'],'engine':'rag'}).encode()
    r=client.query_backend(QUESTIONS[0],'rag')
    assert r['reason']=='invalid_backend_response' and not r['accepted_for_delivery']


def test_consecutive_mcp_calls_do_not_reuse_previous_queries(backend):
    from mcp import ClientSession,StdioServerParameters
    from mcp.client.stdio import stdio_client
    from packages.claude.mcp_server import QUESTION_POLICY
    async def run():
        params=StdioServerParameters(command=sys.executable,args=[str(ROOT/'packages/claude/mcp_server.py')],
                                    env={**os.environ,'PYTHONDONTWRITEBYTECODE':'1'})
        async with stdio_client(params) as (read,write):
            async with ClientSession(read,write) as session:
                initialized=await session.initialize()
                assert QUESTION_POLICY in initialized.instructions
                for engine,question in [('rag',QUESTIONS[0]),('graphrag',QUESTIONS[2]),('rag',QUESTIONS[1])]:
                    result=await session.call_tool('zpwiki_'+engine,{'query':question})
                    assert not result.isError and result.structuredContent['query']==question
                    assert backend['requests'][-1][1]['query']==question
                    assert backend['requests'][-1][1]['engine']==engine
    asyncio.run(run())