Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_tech_support.py

@@ -4,18 +4,6 @@
 from unittest.mock import MagicMock
 from autogen_core.components.tools import FunctionTool
 
-# Mock the azure.monitor.events.extension module globally
-sys.modules["azure.monitor.events.extension"] = MagicMock()
-
-# Set environment variables to mock Config 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 the functions under test
 from src.backend.agents.tech_support import (
     send_welcome_email,
@@ -64,6 +52,18 @@
     get_tech_support_tools,
 )
 
+# Mock the azure.monitor.events.extension module globally
+sys.modules["azure.monitor.events.extension"] = MagicMock()
+
+# Set environment variables to mock Config 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"
+
 @pytest.mark.asyncio
 async def test_collaborate_with_code_deployment():
     result = await collaborate_with_code_deployment("AI Deployment Project")

src/backend/tests/test_otlp_tracing.py

@@ -1,45 +1,37 @@
 import sys
 import os
 from unittest.mock import patch, MagicMock
+from src.backend.otlp_tracing import configure_oltp_tracing  # Import directly since it's in backend
 
 # Add the backend directory to the Python path
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
-from otlp_tracing import configure_oltp_tracing  # Import directly since it's in backend
-
-
-@patch("otlp_tracing.OTLPSpanExporter")
-@patch("otlp_tracing.BatchSpanProcessor")
-@patch("otlp_tracing.TracerProvider")
-@patch("otlp_tracing.trace")
-@patch("otlp_tracing.Resource")
+@patch("src.backend.otlp_tracing.TracerProvider")
+@patch("src.backend.otlp_tracing.OTLPSpanExporter")
+@patch("src.backend.otlp_tracing.Resource")
 def test_configure_oltp_tracing(
     mock_resource,
-    mock_trace,
-    mock_tracer_provider,
-    mock_batch_processor,
     mock_otlp_exporter,
+    mock_tracer_provider,
 ):
-    # Mock objects
-    mock_resource.return_value = {"service.name": "macwe"}
+    # Mock the Resource
+    mock_resource_instance = MagicMock()
+    mock_resource.return_value = mock_resource_instance
+
+    # Mock TracerProvider
     mock_tracer_provider_instance = MagicMock()
     mock_tracer_provider.return_value = mock_tracer_provider_instance
-    mock_batch_processor.return_value = MagicMock()
-    mock_otlp_exporter.return_value = MagicMock()
+
+    # Mock OTLPSpanExporter
+    mock_otlp_exporter_instance = MagicMock()
+    mock_otlp_exporter.return_value = mock_otlp_exporter_instance
 
     # Call the function
     endpoint = "mock-endpoint"
     tracer_provider = configure_oltp_tracing(endpoint=endpoint)
 
     # Assertions
-    mock_tracer_provider.assert_called_once_with(resource={"service.name": "macwe"})
-    mock_otlp_exporter.assert_called_once()
-    mock_batch_processor.assert_called_once_with(mock_otlp_exporter.return_value)
-    mock_tracer_provider_instance.add_span_processor.assert_called_once_with(
-        mock_batch_processor.return_value
-    )
-    mock_trace.set_tracer_provider.assert_called_once_with(
-        mock_tracer_provider_instance
-    )
-
+    mock_tracer_provider.assert_called_once_with(resource=mock_resource_instance)
+    mock_otlp_exporter.assert_called_once_with()
+    mock_tracer_provider_instance.add_span_processor.assert_called_once()
     assert tracer_provider == mock_tracer_provider_instance