Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_group_chat_manager.py

@@ -28,7 +28,7 @@
     Plan,
     ActionRequest,
 )
-from autogen_core.base import AgentInstantiationContext, AgentRuntime
+from autogen_core.base import MessageContext, AgentInstantiationContext, AgentRuntime
 from autogen_core.components.models import AzureOpenAIChatCompletionClient
 from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 from autogen_core.base import AgentId
@@ -102,91 +102,6 @@ async def test_update_step_status(mock_track_event, setup_group_chat_manager):
     )
 
 
-@pytest.mark.asyncio
-@patch("src.backend.agents.group_chat_manager.track_event_if_configured")
-async def test_execute_step(mock_track_event, setup_group_chat_manager):
-    """
-    Test the `_execute_step` method.
-    """
-    group_chat_manager, mock_memory, session_id, user_id, mock_agent_ids = setup_group_chat_manager
-
-    # Create a mock Step
-    step = Step(
-        id="test_step_id",
-        session_id=session_id,
-        plan_id="test_plan_id",
-        user_id=user_id,
-        action="Test Action",
-        agent=BAgentType.planner_agent,
-        status=StepStatus.planned,
-    )
-
-    # Mock memory responses
-    mock_plan = Plan(
-        id="test_plan_id",
-        session_id=session_id,
-        user_id=user_id,
-        summary="Test Plan Summary",
-        initial_goal="Test Initial Goal",
-    )
-    mock_memory.get_plan_by_session.return_value = mock_plan
-    mock_memory.get_steps_by_plan.return_value = [step]
-
-    # Mock helper methods
-    group_chat_manager.send_message = AsyncMock()
-
-    # Call the method
-    await group_chat_manager._execute_step(session_id, step)
-
-    # Assertions
-    step.status = StepStatus.action_requested
-    mock_memory.update_step.assert_called_once_with(step)
-
-    mock_track_event.assert_has_calls([
-        call(
-            "Group Chat Manager - Update step to action_requested and updated into the cosmos",
-            {
-                "status": step.status,
-                "session_id": step.session_id,
-                "user_id": user_id,
-                "source": step.agent,
-            },
-        ),
-        call(
-            "Group Chat Manager - Requesting Planneragent to perform the action and added into the cosmos",
-            {
-                "session_id": session_id,
-                "user_id": user_id,
-                "plan_id": "test_plan_id",
-                "content": f"Requesting Planneragent to perform action: {step.action}",
-                "source": "GroupChatManager",
-                "step_id": step.id,
-            },
-        ),
-    ])
-
-    # Adjusted expected ActionRequest
-    expected_action_request = ActionRequest(
-        step_id="test_step_id",
-        plan_id="test_plan_id",
-        session_id="test_session_id",
-        action=(
-            "<conversation_history>Here is the conversation history so far for the current plan. "
-            "This information may or may not be relevant to the step you have been asked to execute."
-            "The user's task was:\nTest Plan Summary\n\n"
-            "The conversation between the previous agents so far is below:\n<conversation_history \\>. "
-            "Here is the step to action: Test Action. ONLY perform the steps and actions required to complete this specific step, "
-            "the other steps have already been completed. Only use the conversational history for additional information, "
-            "if it's required to complete the step you have been assigned."
-        ),
-        agent=BAgentType.planner_agent,
-    )
-
-    group_chat_manager.send_message.assert_called_once_with(
-        expected_action_request, mock_agent_ids[BAgentType.planner_agent]
-    )
-
-
 @pytest.mark.asyncio
 async def test_update_step_invalid_feedback_status(setup_group_chat_manager):
     """

src/backend/tests/agents/test_hr.py

@@ -51,6 +51,7 @@
 )
 # pylint: enable=C0413
 
+
 @pytest.mark.asyncio
 async def test_schedule_orientation_session():
     """Test scheduling an orientation session."""