"""Tests for quiz-maker skill: JSON parse, fallback, and export smoke."""

from pathlib import Path

from agent.models import QuizMakerInput, QuizOutline, QuizQuestion
from agent.prompts import fallback_quiz, quiz_max_tokens, quiz_to_markdown
from agent.runner import AgentRunner
from agent.tools.quiz import create_quiz, create_quiz_docx, create_quiz_html


def test_quiz_max_tokens_scales_with_question_count():
    assert quiz_max_tokens(5) == 1020
    assert quiz_max_tokens(3) == 660
    assert quiz_max_tokens(12) == 1536


def test_parse_quiz_outline_normalizes_count():
    runner = AgentRunner()
    raw = (
        '{"title": "Science Quiz", "instructions": "Circle one.", "questions": ['
        '{"prompt": "Q1?", "choices": ["a", "b", "c", "d"], "correct_index": 0, "explanation": "e1"},'
        '{"prompt": "Q2?", "choices": ["a", "b", "c", "d"], "correct_index": 1, "explanation": "e2"},'
        '{"prompt": "Q3?", "choices": ["a", "b", "c", "d"], "correct_index": 2, "explanation": "e3"}'
        "]}"
    )
    outline = runner._parse_quiz_outline(raw, expected_questions=5)
    assert len(outline.questions) == 5
    assert outline.title == "Science Quiz"


def test_parse_quiz_outline_trims_extra_questions():
    runner = AgentRunner()
    questions = ",".join(
        f'{{"prompt": "Q{i}?", "choices": ["a","b","c","d"], "correct_index": 0, "explanation": ""}}'
        for i in range(1, 8)
    )
    raw = f'{{"title": "Long", "questions": [{questions}]}}'
    outline = runner._parse_quiz_outline(raw, expected_questions=5)
    assert len(outline.questions) == 5


def test_parse_quiz_outline_or_error_empty():
    runner = AgentRunner()
    outline, err = runner._parse_quiz_outline_or_error("", 5, None)
    assert outline is None
    assert "empty" in err.lower()


def test_fallback_quiz_has_requested_count():
    req = QuizMakerInput(topic="Fractions", grade="5", question_count=7)
    outline = fallback_quiz(req)
    assert len(outline.questions) == 7
    assert "Fractions" in outline.title
    assert all(len(q.choices) == 4 for q in outline.questions)


def test_quiz_to_markdown_includes_answers():
    outline = QuizOutline(
        title="Test",
        instructions="Read carefully.",
        questions=[
            QuizQuestion(
                prompt="2+2?",
                choices=["4", "3", "5", "6"],
                correct_index=0,
                explanation="Basic addition.",
            ),
            QuizQuestion(
                prompt="3+3?",
                choices=["6", "5", "7", "8"],
                correct_index=0,
                explanation="Also addition.",
            ),
            QuizQuestion(
                prompt="1+1?",
                choices=["2", "1", "3", "4"],
                correct_index=0,
                explanation="Easy.",
            ),
        ],
    )
    md = quiz_to_markdown(outline)
    assert "2+2?" in md
    assert "**Answer:** A" in md


def test_create_quiz_docx_and_html(tmp_path: Path):
    outline = QuizOutline(
        title="Smoke Quiz",
        instructions="Circle the best answer.",
        questions=[
            QuizQuestion(
                prompt="Sample?",
                choices=["Yes", "No", "Maybe", "Sometimes"],
                correct_index=0,
                explanation="Because.",
            ),
            QuizQuestion(
                prompt="Another?",
                choices=["A", "B", "C", "D"],
                correct_index=2,
                explanation="C is correct.",
            ),
            QuizQuestion(
                prompt="Third?",
                choices=["1", "2", "3", "4"],
                correct_index=1,
                explanation="Two.",
            ),
        ],
    )
    docx_path = tmp_path / "quiz.docx"
    html_path = tmp_path / "quiz.html"
    create_quiz_docx(outline, docx_path)
    create_quiz_html(outline, html_path)
    assert docx_path.stat().st_size > 100
    html_text = html_path.read_text(encoding="utf-8")
    assert "Smoke Quiz" in html_text
    assert "Answer key" in html_text

    paths = create_quiz(outline, tmp_path / "out", stem="worksheet")
    assert paths["docx"].exists()
    assert paths["html"].exists()