from app.utils.climate_intelligence import (
    build_climate_intelligence_context,
    extract_nigeria_location,
    infer_crop_and_growth_stage,
)


def test_build_climate_intelligence_context_contains_climate_risk_and_recommendations():
    context = build_climate_intelligence_context(
        latitude=10.0,
        longitude=7.0,
        crop="maize",
        growth_stage="pre-planting",
        days=7,
    )

    assert isinstance(context, dict)
    assert "location" in context
    assert "current_weather" in context
    assert "forecast" in context
    assert "climate_risk" in context
    assert "recommendations" in context
    assert context["location"]["country"] == "Nigeria"
    assert context["climate_risk"] in {"low", "medium", "high"}
    assert isinstance(context["recommendations"], list)
    assert context["recommendations"]


def test_extract_nigeria_location_handles_lagos_and_ikeja():
    assert extract_nigeria_location("What is the weather like in Lagos?")[0] == "Lagos"
    assert extract_nigeria_location("Ikeja weather forecast")[0] == "Ikeja"


def test_infer_crop_and_growth_stage_handles_farm_questions():
    crop, stage = infer_crop_and_growth_stage("Should I plant maize in Kaduna before the rains?")
    assert crop == "maize"
    assert stage in {"pre-planting", "planting"}