MCPizza - Enhanced

"""
Mock order object creation for bypassing pizzapi menu parsing bugs
"""
import types
from typing import Any, Optional, Dict


def create_mock_order(store: Any, customer: Any, address: Any, order_type: str) -> Any:
    """
    Create a mock order object that has the essential attributes
    but skips menu loading (which has bugs in pizzapi).

    Args:
        store: Store object from pizzapi
        customer: Customer object from pizzapi
        address: Address object from pizzapi
        order_type: "Delivery" or "Carryout"

    Returns:
        Mock order object with data dict and helper methods
    """
    order = types.SimpleNamespace()
    order.store = store
    order.customer = customer
    order.address = address
    order.data = {
        'Address': {
            'Street': address.street,
            'City': address.city,
            'Region': address.region,
            'PostalCode': address.zip,
            'Type': 'House'
        },
        'Coupons': [],
        'CustomerID': '',
        'Extension': '',
        'OrderChannel': 'OLO',
        'OrderID': '',
        'NoCombine': True,
        'OrderMethod': 'Web',
        'OrderTaker': None,
        'Payments': [],
        'Products': [],
        'Market': '',
        'Currency': '',
        'ServiceMethod': order_type,
        'Tags': {},
        'Version': '1.0',
        'SourceOrganizationURI': 'order.dominos.com',
        'LanguageCode': 'en',
        'Partners': {},
        'NewUser': True,
        'metaData': {},
        'Amounts': {},
        'BusinessDate': '',
        'EstimatedWaitMinutes': '',
        'PriceOrderTime': '',
        'AmoundsBreakdown': {}
    }

    # Add menu attribute (unused but expected)
    order.menu = None

    # Add helper methods
    def add_coupon(coupon_code: str):
        """Add coupon to order"""
        order.data['Coupons'].append({
            'Code': coupon_code,
            'Qty': 1,
            'ID': len(order.data['Coupons']) + 1
        })

    def add_product(product_code: str, options: Optional[Dict] = None):
        """Add product with toppings to order"""
        order.data['Products'].append({
            'Code': product_code,
            'Qty': 1,
            'ID': len(order.data['Products']) + 1,
            'isNew': True,
            'Options': options or {}
        })

    order.add_coupon = add_coupon
    order.add_product = add_product

    return order