test_performance.py

import pytest

from legacy_calculator import calculate_total, process_order

# Sample test data
SAMPLE_ITEMS = [
    {"category": "electronics", "price": 1200},
    {"category": "clothing", "price": 50},
    {"category": "books", "price": 30},
]


def test_calculate_total_basic(benchmark):
    """Implement benchmark test for calculate_total."""
    result = benchmark(calculate_total, SAMPLE_ITEMS, "SUMMER", "premium", True)
    assert result > 0


def test_process_order_benchmark(benchmark):
    """Benchmark the process_order function."""
    order = {
        "items": SAMPLE_ITEMS,
        "customer": {"type": "premium"},
        "discount_code": "SUMMER",
        "is_holiday": True,
    }

    result = benchmark(process_order, order)
    assert result is not None


# Add more test cases for edge cases
def test_process_order_none():
    assert process_order(None) is None


def test_process_order_empty_items():
    order = {
        "items": [],
        "customer": {"type": "premium"},
        "discount_code": "SUMMER",
        "is_holiday": False,
    }
    assert process_order(order) == 0


def test_process_order_missing_customer():
    order = {
        "items": SAMPLE_ITEMS,
        "discount_code": "SUMMER",
        "is_holiday": False,
    }
    assert process_order(order) is None