Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_group_chat_manager.py

@@ -19,7 +19,6 @@
 sys.modules["azure.monitor.events.extension"] = MagicMock()
 
 
-from src.backend.event_utils import track_event_if_configured
 from autogen_core.base._agent_instantiation import AgentInstantiationContext
 
 
@@ -29,6 +28,7 @@ def dummy_agent_instantiation_context():
     yield
     AgentInstantiationContext.AGENT_INSTANTIATION_CONTEXT_VAR.reset(token)
 
+
 # --- Import production classes ---
 from src.backend.agents.group_chat_manager import GroupChatManager
 from src.backend.models.messages import (
@@ -43,7 +43,7 @@ def dummy_agent_instantiation_context():
     BAgentType,
 )
 from autogen_core.base import AgentId, MessageContext
-from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+
 
 # --- Define a DummyMessageContext that supplies required parameters ---
 class DummyMessageContext(MessageContext):
@@ -75,7 +75,6 @@ async def get_plan_by_session(self, session_id: str) -> Plan:
             human_clarification_response="Plan feedback",
         )
 
-
     async def get_steps_by_plan(self, plan_id: str) -> list:
         step1 = Step.model_construct(
             id="step1",
@@ -107,9 +106,6 @@ async def add_plan(self, plan: Plan):
     async def update_plan(self, plan: Plan):
         pass
 
-    async def update_step(self, step: Step):
-        self.updated_steps.append(step)
-
 
 # --- Fake send_message for GroupChatManager ---
 async def fake_send_message(message, agent_id):
@@ -148,11 +144,12 @@ def group_chat_manager():
     manager.send_message = AsyncMock(side_effect=fake_send_message)
     return manager, fake_memory
 
+
 # --- To simulate a missing agent in a step, define a dummy subclass ---
 class DummyStepMissingAgent(Step):
     @property
     def agent(self):
-        return ""  
+        return "" 
 
 
 # ---------------------- Tests ----------------------
@@ -346,12 +343,13 @@ async def test_execute_step_human_agent(group_chat_manager):
 @pytest.mark.asyncio
 async def test_execute_step_missing_agent_raises(group_chat_manager):
     manager, fake_memory = group_chat_manager
+
     # Create a dummy step using a subclass that forces agent to be an empty string.
     class DummyStepMissingAgent(Step):
         @property
         def agent(self):
             return ""
-    step = DummyStepMissingAgent.model_construct(
+    DummyStepMissingAgent.model_construct(
         id="step_missing",
         plan_id="plan1",
         action="Do something",

src/backend/tests/agents/test_human.py

@@ -20,7 +20,6 @@
 # Patch azure module so that event_utils imports correctly.
 sys.modules["azure.monitor.events.extension"] = MagicMock()
 
-from src.backend.event_utils import track_event_if_configured
 from autogen_core.base._agent_instantiation import AgentInstantiationContext
 dummy_runtime = MagicMock()
 dummy_agent_id = "dummy_agent_id"

src/backend/tests/agents/test_marketing.py

@@ -20,7 +20,6 @@
 sys.modules["azure.monitor.events.extension"] = MagicMock()
 
 # Import the marketing functions and MarketingAgent from the module.
-from autogen_core.components.tools import FunctionTool
 from src.backend.agents.marketing import (
     create_marketing_campaign,
     analyze_market_trends,

src/backend/tests/agents/test_planner.py

@@ -3,7 +3,7 @@
 import sys
 import json
 import pytest
-from unittest.mock import AsyncMock, MagicMock, patch
+from unittest.mock import AsyncMock, MagicMock
 
 # --- Setup environment and module search path ---
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
@@ -18,11 +18,7 @@
 # Ensure the project root is in sys.path.
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../")))
 
-# --- Patch event tracking to be a no-op ---
 from src.backend.event_utils import track_event_if_configured
-track_event_if_configured = lambda event, props: None
-
-# --- Patch AgentInstantiationContext to bypass instantiation errors ---
 from autogen_core.base._agent_instantiation import AgentInstantiationContext
 
 

src/backend/tests/agents/test_procurement.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import pytest
-from datetime import datetime
 from unittest.mock import MagicMock
 
 # --- Fake missing Azure modules ---
@@ -138,6 +137,7 @@ async def test_procurement_functions(func, args, expected):
     else:
         assert result == expected
 
+
 # --- Test get_procurement_tools ---
 def test_get_procurement_tools():
     tools = get_procurement_tools()

src/backend/tests/test_app.py

@@ -129,7 +129,7 @@ def override_dependencies(monkeypatch):
         "src.backend.auth.auth_utils.get_authenticated_user_details",
         lambda headers: {"user_principal_id": "mock-user-id"},
     )
-        
+  
     monkeypatch.setattr(
         "src.backend.utils.retrieve_all_agent_tools",
         lambda: [{