"""Tests for QueryRouter intent classification."""

import pytest
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent.parent / "backend"))


class TestQueryRouterClassification:
    """Test the intent classification prompt format."""

    def test_classify_prompt_structure(self):
        """Verify the classification prompt contains all three categories."""
        from orchestrator.query_router import QueryRouter
        import inspect
        source = inspect.getsource(QueryRouter.classify)

        assert "sql_query" in source
        assert "conversational" in source
        assert "clarification_needed" in source
