import pytest

def test_billing_module_imports():
    from rbox.billing import models, stripe_client, usage_tracker, invoice_generator, scheduler
    assert models is not None
    assert stripe_client is not None
    assert usage_tracker is not None
    assert invoice_generator is not None
    assert scheduler is not None

def test_billing_models_exist():
    from rbox.billing.models import (
        SubscriptionPlan, UserSubscription, UsageRecord, UsageAggregate,
        Invoice, InvoiceLineItem, PricingConfig, PaymentMethod, BillingEvent
    )
    assert SubscriptionPlan is not None
    assert UserSubscription is not None
    assert UsageRecord is not None
    assert UsageAggregate is not None
    assert Invoice is not None
    assert InvoiceLineItem is not None
    assert PricingConfig is not None
    assert PaymentMethod is not None
    assert BillingEvent is not None

def test_stripe_client_exists():
    from rbox.billing.stripe_client import StripeClient
    assert StripeClient is not None
    assert hasattr(StripeClient, 'create_customer')
    assert hasattr(StripeClient, 'create_invoice')
    assert hasattr(StripeClient, 'finalize_invoice')

def test_usage_tracker_exists():
    from rbox.billing.usage_tracker import UsageTracker
    assert UsageTracker is not None
    assert hasattr(UsageTracker, 'track_storage')
    assert hasattr(UsageTracker, 'track_bandwidth')
    assert hasattr(UsageTracker, 'aggregate_daily_usage')
    assert hasattr(UsageTracker, 'get_current_storage')
    assert hasattr(UsageTracker, 'get_monthly_usage')

def test_invoice_generator_exists():
    from rbox.billing.invoice_generator import InvoiceGenerator
    assert InvoiceGenerator is not None
    assert hasattr(InvoiceGenerator, 'generate_monthly_invoice')
    assert hasattr(InvoiceGenerator, 'finalize_invoice')
    assert hasattr(InvoiceGenerator, 'mark_invoice_paid')

def test_scheduler_exists():
    from rbox.billing.scheduler import scheduler, start_scheduler, stop_scheduler
    assert scheduler is not None
    assert callable(start_scheduler)
    assert callable(stop_scheduler)

def test_routers_exist():
    from rbox.routers import billing, admin_billing
    assert billing is not None
    assert admin_billing is not None
    assert hasattr(billing, 'router')
    assert hasattr(admin_billing, 'router')

def test_middleware_exists():
    from rbox.middleware.usage_tracking import UsageTrackingMiddleware
    assert UsageTrackingMiddleware is not None

def test_settings_updated():
    from rbox.settings import settings
    assert hasattr(settings, 'STRIPE_SECRET_KEY')
    assert hasattr(settings, 'STRIPE_PUBLISHABLE_KEY')
    assert hasattr(settings, 'STRIPE_WEBHOOK_SECRET')
    assert hasattr(settings, 'BILLING_ENABLED')

def test_main_includes_billing():
    from rbox.main import app
    routes = [route.path for route in app.routes]
    billing_routes = [r for r in routes if '/billing' in r]
    assert len(billing_routes) > 0