mirror of
https://github.com/microsoft/agent-framework.git
synced 2026-06-16 21:04:09 +08:00
Laveesh Rohra
151 lines
4.7 KiB
Python
151 lines
4.7 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
"""Unit tests for multi-agent support in AgentFunctionApp."""
|
|
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from agent_framework_azurefunctions import AgentFunctionApp
|
|
|
|
|
|
class TestMultiAgentInit:
|
|
"""Test suite for multi-agent initialization."""
|
|
|
|
def test_init_with_agents_list(self) -> None:
|
|
"""Test initialization with list of agents."""
|
|
agent1 = Mock()
|
|
agent1.name = "Agent1"
|
|
agent2 = Mock()
|
|
agent2.name = "Agent2"
|
|
|
|
app = AgentFunctionApp(agents=[agent1, agent2])
|
|
|
|
assert len(app.agents) == 2
|
|
assert "Agent1" in app.agents
|
|
assert "Agent2" in app.agents
|
|
assert app.agents["Agent1"] == agent1
|
|
assert app.agents["Agent2"] == agent2
|
|
|
|
def test_init_with_empty_agents_list(self) -> None:
|
|
"""Test initialization with empty list of agents."""
|
|
app = AgentFunctionApp(agents=[])
|
|
|
|
assert len(app.agents) == 0
|
|
|
|
def test_init_with_no_agents(self) -> None:
|
|
"""Test initialization without any agents."""
|
|
app = AgentFunctionApp()
|
|
|
|
assert len(app.agents) == 0
|
|
|
|
def test_init_with_duplicate_agent_names(self) -> None:
|
|
"""Test initialization with agents having the same name raises error."""
|
|
agent1 = Mock()
|
|
agent1.name = "TestAgent"
|
|
agent2 = Mock()
|
|
agent2.name = "TestAgent"
|
|
|
|
with pytest.raises(ValueError, match="already registered"):
|
|
AgentFunctionApp(agents=[agent1, agent2])
|
|
|
|
def test_init_with_agent_without_name(self) -> None:
|
|
"""Test initialization with agent missing name attribute raises error."""
|
|
agent1 = Mock()
|
|
agent1.name = "Agent1"
|
|
agent2 = Mock(spec=[]) # Mock without name attribute
|
|
|
|
with pytest.raises(ValueError, match="does not have a 'name' attribute"):
|
|
AgentFunctionApp(agents=[agent1, agent2])
|
|
|
|
|
|
class TestAddAgentMethod:
|
|
"""Test suite for add_agent() method."""
|
|
|
|
def test_add_agent_to_empty_app(self) -> None:
|
|
"""Test adding agent to app initialized without agents."""
|
|
app = AgentFunctionApp()
|
|
|
|
agent = Mock()
|
|
agent.name = "NewAgent"
|
|
|
|
app.add_agent(agent)
|
|
|
|
assert len(app.agents) == 1
|
|
assert "NewAgent" in app.agents
|
|
assert app.agents["NewAgent"] == agent
|
|
|
|
def test_add_multiple_agents(self) -> None:
|
|
"""Test adding multiple agents sequentially."""
|
|
app = AgentFunctionApp()
|
|
|
|
agent1 = Mock()
|
|
agent1.name = "Agent1"
|
|
agent2 = Mock()
|
|
agent2.name = "Agent2"
|
|
|
|
app.add_agent(agent1)
|
|
app.add_agent(agent2)
|
|
|
|
assert len(app.agents) == 2
|
|
assert "Agent1" in app.agents
|
|
assert "Agent2" in app.agents
|
|
|
|
def test_add_agent_with_duplicate_name_raises_error(self) -> None:
|
|
"""Test that adding agent with duplicate name raises ValueError."""
|
|
agent1 = Mock()
|
|
agent1.name = "MyAgent"
|
|
agent2 = Mock()
|
|
agent2.name = "MyAgent"
|
|
|
|
app = AgentFunctionApp(agents=[agent1])
|
|
|
|
# Try to add another agent with the same name
|
|
with pytest.raises(ValueError, match="already registered"):
|
|
app.add_agent(agent2)
|
|
|
|
def test_add_agent_to_app_with_existing_agents(self) -> None:
|
|
"""Test adding agent to app that already has agents."""
|
|
agent1 = Mock()
|
|
agent1.name = "Agent1"
|
|
agent2 = Mock()
|
|
agent2.name = "Agent2"
|
|
|
|
app = AgentFunctionApp(agents=[agent1])
|
|
app.add_agent(agent2)
|
|
|
|
assert len(app.agents) == 2
|
|
assert "Agent1" in app.agents
|
|
assert "Agent2" in app.agents
|
|
|
|
def test_add_agent_without_name_raises_error(self) -> None:
|
|
"""Test that adding agent without name attribute raises error."""
|
|
app = AgentFunctionApp()
|
|
|
|
agent = Mock(spec=[]) # Mock without name attribute
|
|
|
|
with pytest.raises(ValueError, match="does not have a 'name' attribute"):
|
|
app.add_agent(agent)
|
|
|
|
|
|
class TestHealthCheckWithMultipleAgents:
|
|
"""Test suite for health check with multiple agents."""
|
|
|
|
def test_health_check_returns_all_agents(self) -> None:
|
|
"""Test that health check returns information about all agents."""
|
|
agent1 = Mock()
|
|
agent1.name = "Agent1"
|
|
agent2 = Mock()
|
|
agent2.name = "Agent2"
|
|
|
|
app = AgentFunctionApp(agents=[agent1, agent2])
|
|
|
|
# Note: We can't easily test the actual health check endpoint without running the app
|
|
# But we can verify the agents dictionary is properly populated
|
|
assert len(app.agents) == 2
|
|
assert app.enable_health_check is True
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__, "-v", "--tb=short"])
|