Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/agents/test_procurement.py

@@ -1,20 +1,8 @@
 import os
+import sys
 import pytest
 from unittest.mock import MagicMock
 
-# Mock modules and environment variables
-import sys
-sys.modules['azure.monitor.events.extension'] = MagicMock()
-
-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 procurement tools for testing
 from src.backend.agents.procurement import (
     order_hardware,
@@ -41,6 +29,16 @@
     track_procurement_metrics,
 )
 
+sys.modules["azure.monitor.events.extension"] = MagicMock()
+
+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 the async functions
 @pytest.mark.asyncio
@@ -70,7 +68,9 @@ async def test_process_purchase_order():
 @pytest.mark.asyncio
 async def test_initiate_contract_negotiation():
     result = await initiate_contract_negotiation("VendorX", "Exclusive deal for 2025")
-    assert "Contract negotiation initiated with VendorX: Exclusive deal for 2025" in result
+    assert (
+        "Contract negotiation initiated with VendorX: Exclusive deal for 2025" in result
+    )
 
 
 @pytest.mark.asyncio
@@ -93,7 +93,9 @@ async def test_manage_vendor_relationship():
 
 @pytest.mark.asyncio
 async def test_update_procurement_policy():
-    result = await update_procurement_policy("Policy2025", "Updated terms and conditions")
+    result = await update_procurement_policy(
+        "Policy2025", "Updated terms and conditions"
+    )
     assert "Procurement policy 'Policy2025' updated." in result
 
 
@@ -335,7 +337,9 @@ async def test_track_order_invalid_number():
 
 @pytest.mark.asyncio
 async def test_initiate_contract_negotiation_long_details():
-    long_details = "This is a very long contract negotiation detail for testing purposes. " * 10
+    long_details = (
+        "This is a very long contract negotiation detail for testing purposes. " * 10
+    )
     result = await initiate_contract_negotiation("VendorY", long_details)
     assert "Contract negotiation initiated with VendorY" in result
     assert long_details in result
@@ -513,7 +517,9 @@ async def test_handle_return_negative_and_zero_quantity():
     result_negative = await handle_return("Laptop", -5, "Damaged")
     result_zero = await handle_return("Laptop", 0, "Packaging Issue")
     assert "Processed return of -5 units of Laptop due to Damaged." in result_negative
-    assert "Processed return of 0 units of Laptop due to Packaging Issue." in result_zero
+    assert (
+        "Processed return of 0 units of Laptop due to Packaging Issue." in result_zero
+    )
 
 
 @pytest.mark.asyncio
@@ -666,4 +672,3 @@ async def test_order_software_license_invalid_cases():
     result_zero_quantity = await order_software_license("Photoshop", "Single User", 0)
     assert "Ordered 5  licenses of Photoshop." in result_empty_type
     assert "Ordered 0 Single User licenses of Photoshop." in result_zero_quantity
-

src/backend/tests/agents/test_product.py

@@ -1,9 +1,33 @@
 import os
+import sys
 import pytest
 from unittest.mock import MagicMock
+# Import functions directly from product.py for testing
+from src.backend.agents.product import (
+    add_mobile_extras_pack,
+    get_product_info,
+    update_inventory,
+    schedule_product_launch,
+    analyze_sales_data,
+    get_customer_feedback,
+    manage_promotions,
+    set_reorder_level,
+    check_inventory,
+    update_product_price,
+    provide_product_recommendations,
+    handle_product_recall,
+    set_product_discount,
+    manage_supply_chain,
+    forecast_product_demand,
+    handle_product_complaints,
+    monitor_market_trends,
+    generate_product_report,
+    develop_new_product_ideas,
+    optimize_product_page,
+    track_product_shipment,
+    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
@@ -15,6 +39,7 @@
 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,
@@ -39,7 +64,6 @@
     optimize_product_page,
     track_product_shipment,
     evaluate_product_performance,
-    
 )
 
 
@@ -227,7 +251,7 @@ async def test_handle_product_recall_valid():
     result = await handle_product_recall("Product B", "Safety concerns")
     assert "Product recall for" in result
     assert "Product B" in result
-    assert "Safety concerns" in result    
+    assert "Safety concerns" in result
 
 
 @pytest.mark.asyncio
@@ -346,7 +370,7 @@ async def test_manage_supply_chain_empty_supplier():
 @pytest.mark.asyncio
 async def test_analyze_sales_data_invalid_period():
     result = await analyze_sales_data("Product R", "InvalidPeriod")
-    assert "Sales data for **'Product R'** over **InvalidPeriod** analyzed." in result    
+    assert "Sales data for **'Product R'** over **InvalidPeriod** analyzed." in result
 
 
 # Test `update_product_price` with zero price
@@ -478,4 +502,3 @@ async def test_generate_product_report_detailed_type():
 async def test_update_product_price_high_precision():
     result = await update_product_price("Product AG", 123.456789)
     assert "Price for **'Product AG'** updated to **$123.46**." in result
-