from memabra.candidate_types import CandidateObject
from memabra.router import RuleBasedRouter, TaskContext


def test_router_prefers_memory_for_preference_queries():
    router = RuleBasedRouter()
    decision = router.choose(
        TaskContext(user_input="Remember my preferred deployment region"),
        memory_candidates=[
            CandidateObject(
                id="mem-1",
                type="memory",
                title="Preferred region",
                summary="User prefers us-west-2",
                confidence=0.9,
                freshness=0.8,
                success_rate=0.9,
            )
        ],
        skill_candidates=[],
        tool_candidates=[],
    )

    assert decision.decision_type == "inject_memory"
    assert decision.selected_ids == ["mem-1"]