"""
Tests for the context-aware intent gate (is_continuation_reply).

Verifies that short/vague student replies during active tutoring sessions
are NOT rejected as off-topic, while genuinely off-topic messages still are.
"""

import sys
import os
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

import pytest
from unittest.mock import MagicMock
from main import is_continuation_reply


def _make_active_state(active_topic: str = "", current_problem: str = ""):
    """Create a mock WorkingMemoryState."""
    state = MagicMock()
    state.active_topic = active_topic
    state.current_problem = current_problem
    return state


def _make_history(messages: list[tuple[str, str]]) -> list[dict]:
    """Create history list from (role, content) tuples."""
    return [{"role": role, "content": content} for role, content in messages]


# ─── Tests: Should return True (continuation) ─────────────────────

class TestContinuationPositive:
    def test_di_ko_alam_with_active_topic(self):
        state = _make_active_state(active_topic="Functions and Their Graphs")
        assert is_continuation_reply("di ko alam", state, []) is True

    def test_idk_with_current_problem(self):
        state = _make_active_state(current_problem="Solve x^2 + 3x - 4 = 0")
        assert is_continuation_reply("idk", state, []) is True

    def test_huh_after_tutor_question(self):
        history = _make_history([
            ("user", "what is a function?"),
            ("assistant", "A function maps each input to exactly one output. Can you give me an example?"),
        ])
        assert is_continuation_reply("huh?", None, history) is True

    def test_thanks_with_active_topic(self):
        state = _make_active_state(active_topic="Quadratic Equations")
        assert is_continuation_reply("thanks", state, []) is True

    def test_hindi_ko_gets_no_state_but_phrase_match(self):
        # Filipino phrase match should trigger even without active state
        assert is_continuation_reply("hindi ko gets", None, []) is True

    def test_short_reply_with_active_context(self):
        state = _make_active_state(active_topic="Linear Equations")
        assert is_continuation_reply("I'm confused", state, []) is True

    def test_ano_with_active_problem(self):
        state = _make_active_state(current_problem="Find the derivative of f(x)=3x^2")
        assert is_continuation_reply("ano?", state, []) is True

    def test_paano_po(self):
        assert is_continuation_reply("paano po", None, []) is True

    def test_sige_with_context(self):
        state = _make_active_state(active_topic="Statistics")
        assert is_continuation_reply("sige", state, []) is True

    def test_help_me(self):
        assert is_continuation_reply("help me", None, []) is True

    def test_short_msg_after_solve_prompt(self):
        history = _make_history([
            ("assistant", "Try to solve this: What is 2x + 5 = 11?"),
        ])
        assert is_continuation_reply("what", None, history) is True

    def test_ok_with_active_topic(self):
        state = _make_active_state(active_topic="Trigonometry")
        assert is_continuation_reply("ok", state, []) is True


# ─── Tests: Should return False (not a continuation) ──────────────

class TestContinuationNegative:
    def test_explicit_offtopic_no_context(self):
        assert is_continuation_reply(
            "who is the president of the philippines?", None, []
        ) is False

    def test_write_essay_no_context(self):
        assert is_continuation_reply("write my essay", None, []) is False

    def test_long_offtopic_no_context(self):
        assert is_continuation_reply(
            "can you tell me about the history of the roman empire", None, []
        ) is False

    def test_empty_message(self):
        assert is_continuation_reply("", None, []) is False

    def test_long_non_math_no_state(self):
        assert is_continuation_reply(
            "what is the meaning of life and why are we here on earth",
            None,
            [],
        ) is False

    def test_explicit_offtopic_even_with_history_no_question(self):
        history = _make_history([
            ("assistant", "Great job solving that equation!"),
        ])
        # Long off-topic message, no active state, last assistant msg not a question
        assert is_continuation_reply(
            "tell me about basketball players in the NBA this season",
            None,
            history,
        ) is False