Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_product.py

@@ -1,21 +1,6 @@
 import os
 import pytest
 from unittest.mock import MagicMock
-
-# Mock the azure.monitor.events.extension module globally
-import sys
-
-sys.modules["azure.monitor.events.extension"] = MagicMock()
-
-# Set environment variables to mock dependencies
-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"
-
 # Import functions directly from product.py for testing
 from src.backend.agents.product import (
     add_mobile_extras_pack,
@@ -42,6 +27,20 @@
     evaluate_product_performance,
 )
 
+# Mock the azure.monitor.events.extension module globally
+import sys
+
+sys.modules["azure.monitor.events.extension"] = MagicMock()
+
+# Set environment variables to mock dependencies
+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"
+
 
 # Test cases for existing functions
 @pytest.mark.asyncio

src/backend/tests/agents/test_tech_support.py

@@ -64,6 +64,7 @@
 os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
 os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
 
+
 @pytest.mark.asyncio
 async def test_collaborate_with_code_deployment():
     result = await collaborate_with_code_deployment("AI Deployment Project")

src/backend/tests/test_app.py

@@ -1,26 +1,24 @@
 import os
 import sys
 import pytest
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, patch
 from fastapi.testclient import TestClient
 
 # Mock Azure dependencies
 sys.modules["azure.monitor"] = MagicMock()
 sys.modules["azure.monitor.events.extension"] = MagicMock()
 sys.modules["azure.monitor.opentelemetry"] = MagicMock()
 
-# Mock the configure_azure_monitor function
-configure_azure_monitor = MagicMock()
-
 # Set up environment variables
 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["APPLICATIONINSIGHTS_INSTRUMENTATION_KEY"] = "mock-instrumentation-key"
 
-# Import FastAPI app
-from src.backend.app import app
+# Mock telemetry initialization in the app
+with patch("src.backend.app.configure_azure_monitor", MagicMock()):
+    from src.backend.app import app
 
 # Initialize FastAPI test client
 client = TestClient(app)

src/backend/tests/test_otlp_tracing.py

@@ -6,6 +6,7 @@
 # Add the backend directory to the Python path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
+
 @patch("src.backend.otlp_tracing.TracerProvider")
 @patch("src.backend.otlp_tracing.OTLPSpanExporter")
 @patch("src.backend.otlp_tracing.Resource")