from app.modules.rag.contracts.enums import RetrievalMode
from app.modules.rag.retrieval.query_router import RagQueryRouter


def test_query_router_uses_docs_by_default() -> None:
    router = RagQueryRouter()
    assert router.resolve_mode("Какие есть требования по биллингу?") == RetrievalMode.DOCS


def test_query_router_switches_to_code_on_explicit_code_requests() -> None:
    router = RagQueryRouter()
    assert router.resolve_mode("Объясни как работает код endpoint create invoice") == RetrievalMode.CODE