import pytest

from aegis.core import Agent, AgentContext, AgentResult
from aegis.core.registry import (
    AgentNotFoundError,
    AgentRegistry,
    DuplicateAgentError,
)


class StubAgent(Agent):
    def __init__(self, name: str) -> None:
        self.name = name

    async def run(self, context: AgentContext) -> AgentResult:
        return AgentResult(
            agent_name=self.name,
            success=True,
            output={"task": context.task.name},
        )


def test_registry_registers_and_resolves_agent() -> None:
    agent = StubAgent("planner")
    registry = AgentRegistry()

    registry.register(agent)

    assert registry.get("planner") is agent
    assert registry.contains("planner") is True
    assert len(registry) == 1


def test_registry_rejects_duplicate_agent_names() -> None:
    registry = AgentRegistry([StubAgent("planner")])

    with pytest.raises(DuplicateAgentError):
        registry.register(StubAgent("planner"))


def test_registry_raises_when_agent_is_missing() -> None:
    registry = AgentRegistry()

    with pytest.raises(AgentNotFoundError):
        registry.get("missing")


def test_registry_returns_sorted_names() -> None:
    registry = AgentRegistry(
        [
            StubAgent("research"),
            StubAgent("planner"),
        ]
    )

    assert registry.names() == ("planner", "research")