"""
Tests for PlanningAgent.
"""

import pytest
from dcos.agents.planning import PlanningAgent
from dcos.agents.base import AgentRole


class TestPlanningAgent:
    def test_initialization(self):
        agent = PlanningAgent()
        assert agent.identity.name == "planner"
        assert agent.identity.role == AgentRole.PLANNER

    def test_act_returns_plan(self):
        agent = PlanningAgent()
        result = agent.act({"goal": "test_goal"})
        assert "plan" in result
        assert "steps" in result

    def test_decompose_goal(self):
        agent = PlanningAgent()
        tasks = agent.decompose("complex_goal")
        assert len(tasks) > 0
        assert tasks[0]["id"] == "research"

    def test_capabilities(self):
        agent = PlanningAgent()
        caps = agent.get_capabilities()
        assert "strategic_planning" in caps
        assert "task_decomposition" in caps