"""Integration tests for Azure Pricing MCP Server.
These tests make actual API calls to Azure's pricing API.
Run with: pytest tests/test_integration.py -v
"""
import pytest
from azure_pricing_mcp.client import AzurePricingClient
from azure_pricing_mcp.services import PricingService, SKUService
from azure_pricing_mcp.services.retirement import RetirementService
@pytest.fixture
async def services():
"""Create all services for integration testing."""
async with AzurePricingClient() as client:
retirement_service = RetirementService(client)
pricing_service = PricingService(client, retirement_service)
sku_service = SKUService(pricing_service)
yield {
"pricing": pricing_service,
"sku": sku_service,
"retirement": retirement_service,
}
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_vm_price_search(services):
"""Test real VM price search against Azure API."""
result = await services["pricing"].search_prices(
service_name="Virtual Machines",
region="eastus",
sku_name="D4s v3",
limit=5,
)
assert result["count"] > 0
assert result["currency"] == "USD"
assert len(result["items"]) > 0
# Verify structure of returned items
item = result["items"][0]
assert "skuName" in item
assert "retailPrice" in item
assert "armRegionName" in item
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_storage_price_search(services):
"""Test real storage price search."""
result = await services["pricing"].search_prices(
service_name="Storage",
region="eastus",
limit=10,
)
assert result["count"] > 0
assert len(result["items"]) > 0
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_price_comparison(services):
"""Test real price comparison across regions."""
result = await services["pricing"].compare_prices(
service_name="Virtual Machines",
sku_name="D4s v3",
regions=["eastus", "westus", "westeurope"],
)
assert len(result["comparisons"]) > 0
assert result["comparison_type"] == "regions"
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_cost_estimate(services):
"""Test real cost estimation."""
result = await services["pricing"].estimate_costs(
service_name="Virtual Machines",
sku_name="D4s v3",
region="eastus",
hours_per_month=730,
)
assert "on_demand_pricing" in result
assert result["on_demand_pricing"]["hourly_rate"] > 0
assert result["on_demand_pricing"]["monthly_cost"] > 0
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_sku_discovery(services):
"""Test real SKU discovery."""
result = await services["sku"].discover_skus(
service_name="Virtual Machines",
region="eastus",
limit=20,
)
assert result["total_skus"] > 0
assert len(result["skus"]) > 0
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_fuzzy_service_discovery(services):
"""Test real fuzzy service name matching."""
result = await services["sku"].discover_service_skus(
service_hint="app service",
limit=10,
)
# Should find Azure App Service
assert result["service_found"] is not None
assert "App Service" in result["service_found"] or len(result.get("suggestions", [])) > 0
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_discount_application(services):
"""Test discount application on real data."""
result = await services["pricing"].search_prices(
service_name="Virtual Machines",
sku_name="D4s v3",
region="eastus",
discount_percentage=10.0,
limit=5,
)
assert "discount_applied" in result
assert result["discount_applied"]["percentage"] == 10.0
# Verify discount was applied to items
if result["items"]:
item = result["items"][0]
assert "originalPrice" in item
assert item["retailPrice"] < item["originalPrice"]
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_customer_discount(services):
"""Test customer discount retrieval."""
result = await services["pricing"].get_customer_discount()
assert result["discount_percentage"] > 0
assert result["customer_id"] is not None
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_invalid_sku_validation(services):
"""Test SKU validation with invalid SKU."""
result = await services["pricing"].search_prices(
service_name="Virtual Machines",
sku_name="InvalidSKUName123",
validate_sku=True,
limit=5,
)
# Should provide suggestions or validation info
assert "sku_validation" in result or result["count"] == 0
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_multiple_currencies(services):
"""Test pricing in different currencies."""
currencies = ["USD", "EUR", "GBP"]
for currency in currencies:
result = await services["pricing"].search_prices(
service_name="Virtual Machines",
sku_name="D4s v3",
region="eastus",
currency_code=currency,
limit=1,
)
assert result["currency"] == currency
if result["items"]:
assert result["items"][0]["currencyCode"] == currency
@pytest.mark.integration
@pytest.mark.asyncio
async def test_real_savings_plans(services):
"""Test retrieval of savings plan pricing."""
result = await services["pricing"].estimate_costs(
service_name="Virtual Machines",
sku_name="D4s v3",
region="eastus",
hours_per_month=730,
)
# Some VMs have savings plans
if result.get("savings_plans"):
assert len(result["savings_plans"]) > 0
plan = result["savings_plans"][0]
assert "term" in plan
assert "hourly_rate" in plan
assert "savings_percent" in plan
