Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_group_chat_manager.py

@@ -1,13 +1,7 @@
 import os
 import sys
-import re
-import asyncio
-import json
 import pytest
-import logging
-from datetime import datetime, date
 from unittest.mock import AsyncMock, MagicMock, patch
-from pydantic import BaseModel
 
 # Adjust sys.path so that the project root is found.
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../")))
@@ -156,7 +150,7 @@ def group_chat_manager():
 class DummyStepMissingAgent(Step):
     @property
     def agent(self):
-        return ""  # Force missing agent
+        return ""  
 
 # ---------------------- Tests ----------------------
 
@@ -359,4 +353,4 @@ def agent(self):
         human_feedback="",
         human_approval_status=HumanFeedbackStatus.requested,
     )
-   
+      

src/backend/tests/agents/test_human.py

@@ -34,23 +34,28 @@ def patch_instantiation_context(monkeypatch):
     monkeypatch.setattr(AgentInstantiationContext, "current_runtime", lambda: dummy_runtime)
     monkeypatch.setattr(AgentInstantiationContext, "current_agent_id", lambda: dummy_agent_id)
 
+
 # --- Patch ApprovalRequest so that required fields get default values ---
 from src.backend.models.messages import ApprovalRequest as RealApprovalRequest, Plan
+
+
 class DummyApprovalRequest(RealApprovalRequest):
     def __init__(self, **data):
         # Provide default values for missing fields.
         data.setdefault("action", "dummy_action")
         data.setdefault("agent", "dummy_agent")
         super().__init__(**data)
 
+
 @pytest.fixture(autouse=True)
 def patch_approval_request(monkeypatch):
     monkeypatch.setattr("src.backend.agents.human.ApprovalRequest", DummyApprovalRequest)
 
 # Now import the module under test.
 from autogen_core.base import MessageContext, AgentId
 from src.backend.agents.human import HumanAgent
-from src.backend.models.messages import HumanFeedback, Step, StepStatus, AgentMessage, ApprovalRequest, BAgentType
+from src.backend.models.messages import HumanFeedback, Step, StepStatus, BAgentType
+
 
 # Define a minimal dummy MessageContext implementation.
 class DummyMessageContext(MessageContext):
@@ -60,6 +65,7 @@ def __init__(self, sender="dummy_sender", topic_id="dummy_topic", is_rpc=False,
         self.is_rpc = is_rpc
         self.cancellation_token = cancellation_token
 
+
 # Define a fake memory implementation.
 class FakeMemory:
     def __init__(self):
@@ -92,6 +98,7 @@ async def get_plan_by_session(self, session_id: str) -> Plan:
             human_clarification_response=None,
         )
 
+
 # Fixture to create a HumanAgent instance with fake memory.
 @pytest.fixture
 def human_agent():
@@ -101,8 +108,8 @@ def human_agent():
     agent = HumanAgent(memory=fake_memory, user_id=user_id, group_chat_manager_id=group_chat_manager_id)
     return agent, fake_memory
 
-# ------------------- Existing Tests -------------------
 
+# ------------------- Existing Tests -------------------
 def test_human_agent_init():
     fake_memory = MagicMock()
     user_id = "test_user"
@@ -112,6 +119,7 @@ def test_human_agent_init():
     assert agent.group_chat_manager_id == group_chat_manager_id
     assert agent._memory == fake_memory
 
+
 @pytest.mark.asyncio
 async def test_handle_step_feedback_no_step_found(human_agent):
     """
@@ -169,6 +177,7 @@ async def test_handle_step_feedback_update_exception(human_agent):
     with pytest.raises(Exception, match="Update failed"):
         await agent.handle_step_feedback(feedback, ctx)
 
+
 @pytest.mark.asyncio
 async def test_handle_step_feedback_add_item_exception(human_agent):
     """