wehub-resource-sync
db620d33df
CodeQL / Analyze (csharp) (push) Waiting to run
CodeQL / Analyze (python) (push) Waiting to run
dotnet-build-and-test / dotnet-test-functions (push) Has been cancelled
dotnet-build-and-test / paths-filter (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Debug, windows-latest, net9.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net8.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-foundry-hosted-it (push) Has been cancelled
dotnet-build-and-test / dotnet-build-and-test-check (push) Has been cancelled
dotnet-build-and-test / Integration Test Report (push) Has been cancelled
88 lines
3.3 KiB
Python
88 lines
3.3 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
"""Unit tests for DurableAIAgentOrchestrationContext.
|
|
|
|
Focuses on critical orchestration workflows: agent retrieval and integration.
|
|
Run with: pytest tests/test_orchestration_context.py -v
|
|
"""
|
|
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
from agent_framework import SupportsAgentRun
|
|
|
|
from agent_framework_durabletask import DurableAgentSession
|
|
from agent_framework_durabletask._orchestration_context import DurableAIAgentOrchestrationContext
|
|
from agent_framework_durabletask._shim import DurableAIAgent
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_orchestration_context() -> Mock:
|
|
"""Create a mock OrchestrationContext for testing."""
|
|
return Mock()
|
|
|
|
|
|
@pytest.fixture
|
|
def agent_context(mock_orchestration_context: Mock) -> DurableAIAgentOrchestrationContext:
|
|
"""Create a DurableAIAgentOrchestrationContext with mock context."""
|
|
return DurableAIAgentOrchestrationContext(mock_orchestration_context)
|
|
|
|
|
|
class TestDurableAIAgentOrchestrationContextGetAgent:
|
|
"""Test core workflow: retrieving agents from orchestration context."""
|
|
|
|
def test_get_agent_returns_durable_agent_shim(self, agent_context: DurableAIAgentOrchestrationContext) -> None:
|
|
"""Verify get_agent returns a DurableAIAgent instance."""
|
|
agent = agent_context.get_agent("assistant")
|
|
|
|
assert isinstance(agent, DurableAIAgent)
|
|
assert isinstance(agent, SupportsAgentRun) # pyrefly: ignore[unsafe-overlap]
|
|
|
|
def test_get_agent_shim_has_correct_name(self, agent_context: DurableAIAgentOrchestrationContext) -> None:
|
|
"""Verify retrieved agent has the correct name."""
|
|
agent = agent_context.get_agent("my_agent")
|
|
|
|
assert agent.name == "my_agent"
|
|
|
|
def test_get_agent_multiple_times_returns_new_instances(
|
|
self, agent_context: DurableAIAgentOrchestrationContext
|
|
) -> None:
|
|
"""Verify multiple get_agent calls return independent instances."""
|
|
agent1 = agent_context.get_agent("assistant")
|
|
agent2 = agent_context.get_agent("assistant")
|
|
|
|
assert agent1 is not agent2 # Different object instances
|
|
|
|
def test_get_agent_different_agents(self, agent_context: DurableAIAgentOrchestrationContext) -> None:
|
|
"""Verify context can retrieve multiple different agents."""
|
|
agent1 = agent_context.get_agent("agent1")
|
|
agent2 = agent_context.get_agent("agent2")
|
|
|
|
assert agent1.name == "agent1"
|
|
assert agent2.name == "agent2"
|
|
|
|
|
|
class TestDurableAIAgentOrchestrationContextIntegration:
|
|
"""Test integration scenarios between orchestration context and agent shim."""
|
|
|
|
def test_orchestration_agent_has_working_run_method(
|
|
self, agent_context: DurableAIAgentOrchestrationContext
|
|
) -> None:
|
|
"""Verify agent from context has callable run method (even if not yet implemented)."""
|
|
agent = agent_context.get_agent("assistant")
|
|
|
|
assert hasattr(agent, "run")
|
|
assert callable(agent.run)
|
|
|
|
def test_orchestration_agent_can_create_sessions(self, agent_context: DurableAIAgentOrchestrationContext) -> None:
|
|
"""Verify agent from context can create DurableAgentSession instances."""
|
|
agent = agent_context.get_agent("assistant")
|
|
|
|
session = agent.create_session()
|
|
|
|
assert isinstance(session, DurableAgentSession)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__, "-v", "--tb=short"])
|