import pytest

from aegis.agents import PlannerAgent
from aegis.core import AgentContext, Task


@pytest.mark.asyncio
async def test_planner_returns_structured_plan() -> None:
    planner = PlannerAgent()
    context = AgentContext(
        task=Task(
            name="Build API",
            description="Create and validate a small API endpoint.",
        )
    )

    result = await planner.run(context)

    assert result.success is True
    assert result.agent_name == "planner"
    assert result.output["task_name"] == "Build API"
    assert len(result.output["steps"]) == 3


@pytest.mark.asyncio
async def test_planner_steps_have_sequential_ids() -> None:
    planner = PlannerAgent()
    context = AgentContext(
        task=Task(
            name="Validate repository",
            description="Run repository validation checks.",
        )
    )

    result = await planner.run(context)
    step_ids = [step["id"] for step in result.output["steps"]]

    assert step_ids == [1, 2, 3]


@pytest.mark.asyncio
async def test_planner_output_is_bounded() -> None:
    planner = PlannerAgent()
    context = AgentContext(
        task=Task(
            name="Small task",
            description="Produce a concise execution plan.",
        )
    )

    result = await planner.run(context)

    assert len(result.output["steps"]) <= 3