test_comprehensive.py

#!/usr/bin/env python3
import requests
import json
import sys
from decimal import Decimal

BASE_URL = "http://127.0.0.1:8000"
TENANT_ID = "test_tenant_001"


def test_endpoint(method, url, data=None, files=None, headers=None, expected_status=200, description=""):
    """Test an endpoint and return the response"""
    try:
        if method.upper() == "GET":
            response = requests.get(url, headers=headers)
        elif method.upper() == "POST":
            if files:
                response = requests.post(url, data=data, files=files, headers=headers)
            else:
                response = requests.post(url, json=data, headers=headers)
        elif method.upper() == "PUT":
            if files:
                response = requests.put(url, data=data, files=files, headers=headers)
            else:
                response = requests.put(url, json=data, headers=headers)
        elif method.upper() == "PATCH":
            if files:
                response = requests.patch(url, data=data, files=files, headers=headers)
            else:
                response = requests.patch(url, json=data, headers=headers)
        elif method.upper() == "DELETE":
            response = requests.delete(url, headers=headers)

        status_ok = response.status_code == expected_status
        print(f"{'✓' if status_ok else '✗'} {description}")
        print(f"  {method} {url}")
        print(f"  Status: {response.status_code} (expected: {expected_status})")

        if not status_ok:
            print(f"  Response: {response.text[:200]}...")

        print()
        return response, status_ok
    except Exception as e:
        print(f"✗ {description}")
        print(f"  Error: {e}")
        print()
        return None, False


def main():
    print("Comprehensive DoxiiCMS Backend API Testing")
    print("=" * 60)

    results = {"passed": 0, "failed": 0}

    # Test 1: Basic endpoints
    print("\n1. BASIC ENDPOINTS")
    print("-" * 20)

    _, ok = test_endpoint("GET", f"{BASE_URL}/", description="Root endpoint")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    _, ok = test_endpoint("GET", f"{BASE_URL}/health", description="Health check")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    _, ok = test_endpoint("OPTIONS", f"{BASE_URL}/any-path", expected_status=400, description="CORS OPTIONS")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 2: Tenant middleware
    print("\n2. TENANT MIDDLEWARE")
    print("-" * 20)

    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/nonexistent/products", expected_status=404, description="Nonexistent tenant"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    _, ok = test_endpoint("GET", f"{BASE_URL}/invalid-path", expected_status=400, description="Invalid path format")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 3: Categories API
    print("\n3. CATEGORIES API")
    print("-" * 20)

    # List categories
    _, ok = test_endpoint("GET", f"{BASE_URL}/tenant/{TENANT_ID}/categories", description="List categories")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Create parent category
    parent_data = {
        "name": "Electronics",
        "slug": "electronics",
        "description": "Electronic devices",
        "is_active": True,
        "sort_order": 10,
    }
    resp, ok = test_endpoint(
        "POST", f"{BASE_URL}/tenant/{TENANT_ID}/categories", data=parent_data, description="Create parent category"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    parent_id = resp.json()["id"] if resp and resp.status_code == 201 else None

    # Create child category
    child_data = {
        "name": "Smartphones",
        "slug": "smartphones",
        "description": "Mobile phones",
        "parent_id": parent_id,
        "is_active": True,
        "sort_order": 1,
    }
    resp, ok = test_endpoint(
        "POST", f"{BASE_URL}/tenant/{TENANT_ID}/categories", data=child_data, description="Create child category"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    child_id = resp.json()["id"] if resp and resp.status_code == 201 else None

    # Test category tree
    _, ok = test_endpoint("GET", f"{BASE_URL}/tenant/{TENANT_ID}/categories/tree", description="Category tree")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test filtering
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/categories?is_active=eq.true", description="Filter active categories"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test ordering
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/categories?order=name.asc", description="Order categories by name"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test pagination
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/categories?limit=5&offset=0", description="Paginate categories"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 4: Products API
    print("\n4. PRODUCTS API")
    print("-" * 20)

    # List products (empty)
    _, ok = test_endpoint("GET", f"{BASE_URL}/tenant/{TENANT_ID}/products", description="List products (empty)")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Create product with multipart data
    product_data = {
        "name": "iPhone 15",
        "slug": "iphone-15",
        "description": "Latest iPhone model",
        "short_description": "New iPhone",
        "sku": "IPH15-001",
        "price": 999.99,
        "compare_at_price": 1099.99,
        "cost_price": 600.00,
        "inventory_quantity": 50,
        "track_inventory": True,
        "continue_selling_when_out_of_stock": False,
        "weight": 0.2,
        "weight_unit": "kg",
        "status": "active",
        "is_featured": True,
        "seo_title": "iPhone 15 - Latest Model",
        "seo_description": "Buy the latest iPhone 15",
        "category_id": child_id,
    }

    # Create product with multipart form data
    files = {
        "product_data": (None, json.dumps(product_data), "application/json"),
    }
    # Note: For testing without actual files, we'll create product without media

    resp, ok = test_endpoint(
        "POST",
        f"{BASE_URL}/tenant/{TENANT_ID}/products",
        data=files,
        files=files,
        description="Create product without media",
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    product_id = resp.json()["id"] if resp and resp.status_code == 201 else None

    # Get product by ID
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/products/{product_id}", description="Get product by ID"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test product filtering
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/products?status=eq.active", description="Filter active products"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test product search
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/products?name=ilike.*iPhone*", description="Search products by name"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 5: Orders API
    print("\n5. ORDERS API")
    print("-" * 20)

    # Create customer first (using auth register as customer)
    customer_data = {
        "username": "customer1",
        "email": "customer1@example.com",
        "password": "password123",
        "first_name": "John",
        "last_name": "Doe",
    }
    resp, ok = test_endpoint(
        "POST", f"{BASE_URL}/tenant/{TENANT_ID}/auth/register", data=customer_data, description="Register customer"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    customer_id = resp.json()["data"]["id"] if resp and resp.status_code == 201 else 1

    # Create order
    order_data = {
        "customer_id": customer_id,
        "order_number": "ORD-001",
        "status": "pending",
        "payment_status": "pending",
        "fulfillment_status": "unfulfilled",
        "notes": "Test order",
        "items": [{"product_id": product_id, "quantity": 2, "unit_price": 999.99}],
    }

    resp, ok = test_endpoint(
        "POST", f"{BASE_URL}/tenant/{TENANT_ID}/orders", data=order_data, description="Create order"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    order_id = resp.json()["id"] if resp and resp.status_code == 201 else None

    # List orders
    _, ok = test_endpoint("GET", f"{BASE_URL}/tenant/{TENANT_ID}/orders", description="List orders")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Get order by ID
    _, ok = test_endpoint("GET", f"{BASE_URL}/tenant/{TENANT_ID}/orders/{order_id}", description="Get order by ID")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test order filtering
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/orders?status=eq.pending", description="Filter orders by status"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 6: Auth API
    print("\n6. AUTH API")
    print("-" * 20)

    # Login
    login_data = {"username": "customer1", "password": "password123"}
    resp, ok = test_endpoint(
        "POST", f"{BASE_URL}/tenant/{TENANT_ID}/auth/login", data=login_data, description="Login user"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    token = resp.json()["data"]["token"] if resp and resp.status_code == 200 else None

    # Test authenticated endpoint
    headers = {"Authorization": f"Bearer {token}"} if token else None
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/auth/me", headers=headers, description="Get current user"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 7: Edge cases and error handling
    print("\n7. EDGE CASES & ERROR HANDLING")
    print("-" * 20)

    # Duplicate category slug
    _, ok = test_endpoint(
        "POST",
        f"{BASE_URL}/tenant/{TENANT_ID}/categories",
        data=parent_data,
        expected_status=400,
        description="Duplicate category slug",
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Duplicate product SKU
    _, ok = test_endpoint(
        "POST",
        f"{BASE_URL}/tenant/{TENANT_ID}/products",
        data=files,
        files=files,
        expected_status=400,
        description="Duplicate product SKU",
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Invalid product ID
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/products/99999", expected_status=404, description="Invalid product ID"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Invalid order ID
    _, ok = test_endpoint(
        "GET", f"{BASE_URL}/tenant/{TENANT_ID}/orders/99999", expected_status=404, description="Invalid order ID"
    )
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Test 8: Internal API
    print("\n8. INTERNAL API")
    print("-" * 20)

    # List database configs
    _, ok = test_endpoint("GET", f"{BASE_URL}/internal/database/configs", description="List database configs")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Get database status
    _, ok = test_endpoint("GET", f"{BASE_URL}/internal/database/{TENANT_ID}/status", description="Get database status")
    if ok:
        results["passed"] += 1
    else:
        results["failed"] += 1

    # Summary
    print("\n" + "=" * 60)
    print("TEST SUMMARY")
    print("=" * 60)
    print(f"Passed: {results['passed']}")
    print(f"Failed: {results['failed']}")
    print(f"Total: {results['passed'] + results['failed']}")
    print(f"Success Rate: {(results['passed'] / (results['passed'] + results['failed']) * 100):.1f}%")


if __name__ == "__main__":
    main()