Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_tech_support.py

@@ -1,7 +1,7 @@
 import os
 import sys
-import pytest
 from unittest.mock import MagicMock, AsyncMock, patch
+import pytest
 from autogen_core.components.tools import FunctionTool
 
 # Mock the azure.monitor.events.extension module globally
@@ -18,8 +18,6 @@
 os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
 os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
 
-
-# Import the functions under test
 from src.backend.agents.tech_support import (
     send_welcome_email,
     set_up_office_365_account,
@@ -67,6 +65,7 @@
     get_tech_support_tools,
 )
 
+
 # Mock Azure DefaultAzureCredential
 @pytest.fixture(autouse=True)
 def mock_azure_credentials():

src/backend/tests/context/test_cosmos_memory.py

@@ -1,19 +1,9 @@
-import os
 import pytest
 from unittest.mock import AsyncMock, patch
 from azure.cosmos.partition_key import PartitionKey
-
-# Set environment variables before importing modules that depend on them
-os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
-os.environ["COSMOSDB_KEY"] = "mock-key"
-os.environ["COSMOSDB_DATABASE"] = "mock-database"
-os.environ["COSMOSDB_CONTAINER"] = "mock-container"
-os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"] = "mock-deployment-name"
-os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
-os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
-
 from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
+
 # Helper to create async iterable
 async def async_iterable(mock_items):
     """Helper to create an async iterable."""
@@ -47,7 +37,9 @@ def mock_cosmos_client():
     # Mocking context methods
     mock_context = AsyncMock()
     mock_context.store_message = AsyncMock()
-    mock_context.retrieve_messages = AsyncMock(return_value=async_iterable([{"id": "test_id", "content": "test_content"}]))
+    mock_context.retrieve_messages = AsyncMock(
+        return_value=async_iterable([{"id": "test_id", "content": "test_content"}])
+    )
 
     return mock_client, mock_container, mock_context
 
@@ -74,4 +66,3 @@ async def test_initialize(mock_config, mock_cosmos_client):
         id="mock-container", partition_key=PartitionKey(path="/session_id")
     )
     assert context._container == mock_container
-

src/backend/tests/test_app.py

@@ -1,12 +1,9 @@
 import os
 import sys
-import pytest
 from unittest.mock import MagicMock, patch
+import pytest
 from fastapi.testclient import TestClient
 
-# Ensure the `src` folder is included in the Python path
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../../..")))
-
 # Mock Azure dependencies to prevent import errors
 sys.modules["azure.monitor"] = MagicMock()
 sys.modules["azure.monitor.events.extension"] = MagicMock()
@@ -17,16 +14,16 @@
 os.environ["COSMOSDB_KEY"] = "mock-key"
 os.environ["COSMOSDB_DATABASE"] = "mock-database"
 os.environ["COSMOSDB_CONTAINER"] = "mock-container"
-os.environ["APPLICATIONINSIGHTS_INSTRUMENTATION_KEY"] = "mock-instrumentation-key"
+os.environ[
+    "APPLICATIONINSIGHTS_INSTRUMENTATION_KEY"
+] = "InstrumentationKey=mock-instrumentation-key;IngestionEndpoint=https://mock-ingestion-endpoint"
 os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"] = "mock-deployment-name"
 os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
 os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
 
 # Mock telemetry initialization to prevent errors
-patch("src.backend.app.configure_azure_monitor", MagicMock()).start()
-
-# Import the FastAPI app after mocking dependencies
-from src.backend.app import app
+with patch("azure.monitor.opentelemetry.configure_azure_monitor", MagicMock()):
+    from src.backend.app import app
 
 # Initialize FastAPI test client
 client = TestClient(app)