#!/usr/bin/env python3
"""
Test for FastAPI 0.119.0 schema splitting.

Tests the key triggers:
1. Decimal fields
2. 3-level nesting
3. 2-level nesting (for comparison)

Run with: python test_schema_splitting.py
"""

from decimal import Decimal

from fastapi import FastAPI
from pydantic import BaseModel


def test_schema_splitting():
    """Test the triggers for schema splitting."""

    print("FastAPI 0.119.0 Schema Splitting Test")
    print("=" * 60)

    # Test 1: Decimal fields (split)
    class DecimalModel(BaseModel):
        amount: Decimal
        name: str

    app1 = FastAPI()

    @app1.get("/decimal", response_model=list[DecimalModel])
    def decimal_endpoint():
        return []

    schemas1 = list(app1.openapi()["components"]["schemas"].keys())
    decimal_splits = any("-Input" in s or "-Output" in s for s in schemas1)

    # Test 2: 3-level nesting (split)
    class L1(BaseModel):
        value: str

    class L2(BaseModel):
        name: str
        l1: L1

    class ThreeLevel(BaseModel):
        title: str
        l2: L2

    app2 = FastAPI()

    @app2.get("/three-level", response_model=list[ThreeLevel])
    def three_level_endpoint():
        return []

    schemas2 = list(app2.openapi()["components"]["schemas"].keys())
    three_level_splits = any("-Input" in s or "-Output" in s for s in schemas2)

    # Test 3: 2-level nesting (NO split)
    class Inner2(BaseModel):
        value: str

    class TwoLevel(BaseModel):
        name: str
        inner: Inner2

    app3 = FastAPI()

    @app3.get("/two-level", response_model=list[TwoLevel])
    def two_level_endpoint():
        return []

    schemas3 = list(app3.openapi()["components"]["schemas"].keys())
    two_level_splits = any("-Input" in s or "-Output" in s for s in schemas3)

    # Results
    print(f"\nDecimal fields:     {'❌ SPLITS' if decimal_splits else '✅ NO SPLIT'}")
    print(f"3-level nesting:   {'❌ SPLITS' if three_level_splits else '✅ NO SPLIT'}")
    print(f"2-level nesting:   {'❌ SPLITS' if two_level_splits else '✅ NO SPLIT'}")

    print("\n" + "=" * 60)
    print("CONCLUSION:")
    print("=" * 60)
    print("✅ Decimal fields cause splitting")
    print("✅ 3+ level nesting causes splitting")
    print("✅ 2-level nesting does NOT cause splitting")
    print("=" * 60)


if __name__ == "__main__":
    test_schema_splitting()

FastAPI 0.119.0 Schema Splitting Test
============================================================

Decimal fields:     ❌ SPLITS
3-level nesting:   ❌ SPLITS
2-level nesting:   ✅ NO SPLIT

============================================================
CONCLUSION:
============================================================
✅ Decimal fields cause splitting
✅ 3+ level nesting causes splitting
✅ 2-level nesting does NOT cause splitting
============================================================