"""
Tests for ResearchAgent.
"""

import pytest
from dcos.agents.research import ResearchAgent


class TestResearchAgent:
    def test_initialization(self):
        agent = ResearchAgent()
        assert agent.identity.name == "researcher"

    def test_research(self):
        agent = ResearchAgent()
        result = agent.research("AI safety", depth=3)
        assert result["topic"] == "AI safety"
        assert len(result["key_findings"]) == 3

    def test_act(self):
        agent = ResearchAgent()
        result = agent.act({"query": "test query"})
        assert "findings" in result
        assert result["confidence"] > 0