fraiseql-core 2.2.0

#![allow(clippy::unwrap_used)] // Reason: test code, panics are acceptable

//! End-to-End Tests: Python Schema Authoring → JSON Compilation → Runtime Execution
//!
//! This test suite validates the complete flow from Python federation decorator authoring
//! through JSON schema compilation to Rust runtime execution.
//!
//! Test scenarios:
//! 1. Basic federation key declarations (@key)
//! 2. Extended types with @extends
//! 3. External fields with @external
//! 4. Field requirements with @requires
//! 5. Field provisions with @provides
//! 6. Multi-key entities
//! 7. Entity resolution from compiled schema
//! 8. Cross-subgraph entity references
//! 9. Query execution with federation metadata
//! 10. Error handling for schema violations

use std::collections::HashMap;

use fraiseql_core::federation::types::{
    EntityRepresentation, FederatedType, FederationMetadata, FieldFederationDirectives,
    FieldPathSelection, KeyDirective,
};
use serde_json::json;

// ============================================================================
// Test: Basic Federation Key Declaration (Python: @key)
// ============================================================================

#[test]
fn test_python_basic_federation_key() {
    // TEST: Python @key decorator should generate correct FederatedType
    // PYTHON SCHEMA:
    //   @type
    //   @key("id")
    //   class User:
    //     id: str
    //     name: str
    //
    // WHEN: Compiled to FederationMetadata
    // THEN: Should have single key with id field

    let mut user_type = FederatedType::new("User".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    assert!(metadata.enabled);
    assert_eq!(metadata.version, "v2");
    assert_eq!(metadata.types.len(), 1);
    assert_eq!(metadata.types[0].keys.len(), 1);
    assert_eq!(metadata.types[0].keys[0].fields, vec!["id".to_string()]);
}

#[test]
fn test_python_multiple_federation_keys() {
    // TEST: Python @key @key pattern for multiple keys
    // PYTHON SCHEMA:
    //   @type
    //   @key("tenant_id")
    //   @key("id")
    //   class Account:
    //     tenant_id: str
    //     id: str
    //
    // WHEN: Compiled
    // THEN: Should have two keys

    let mut account_type = FederatedType::new("Account".to_string());
    account_type.keys.push(KeyDirective {
        fields:     vec!["tenant_id".to_string()],
        resolvable: true,
    });
    account_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![account_type],
    };

    assert_eq!(metadata.types[0].keys.len(), 2);
}

#[test]
fn test_python_composite_key() {
    // TEST: Python composite key with multiple fields
    // PYTHON SCHEMA:
    //   @type
    //   @key("org_id tenant_id")
    //   class TeamMember:
    //     org_id: str
    //     tenant_id: str
    //
    // WHEN: Compiled
    // THEN: Should have single key with two fields

    let mut team_member = FederatedType::new("TeamMember".to_string());
    team_member.keys.push(KeyDirective {
        fields:     vec!["org_id".to_string(), "tenant_id".to_string()],
        resolvable: true,
    });

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![team_member],
    };

    assert_eq!(metadata.types[0].keys[0].fields.len(), 2);
    assert!(metadata.types[0].keys[0].fields.contains(&"org_id".to_string()));
    assert!(metadata.types[0].keys[0].fields.contains(&"tenant_id".to_string()));
}

// ============================================================================
// Test: Extended Types (@extends)
// ============================================================================

#[test]
fn test_python_extended_type() {
    // TEST: Python @extends decorator for type extension
    // PYTHON SCHEMA IN ORDERS SUBGRAPH:
    //   @type
    //   @extends
    //   @key("id")
    //   class User:
    //     @external
    //     id: str
    //     orders: list[Order]
    //
    // WHEN: Extended type is compiled
    // THEN: Should mark as extended with external fields

    let mut user_type = FederatedType::new("User".to_string());
    user_type.is_extends = true;
    user_type.external_fields.push("id".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    assert!(metadata.types[0].is_extends);
    assert!(metadata.types[0].external_fields.contains(&"id".to_string()));
}

// ============================================================================
// Test: Field Requirements (@requires)
// ============================================================================

#[test]
fn test_python_requires_directive() {
    // TEST: Python field with @requires decorator
    // PYTHON SCHEMA:
    //   @type
    //   class Order:
    //     id: str
    //     @requires("weight")
    //     def shipping_estimate(self) -> float:
    //       pass
    //
    // WHEN: Compiled with @requires
    // THEN: Should generate field directives with requires

    let mut order_type = FederatedType::new("Order".to_string());
    order_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });
    order_type.set_field_directives(
        "shippingEstimate".to_string(),
        FieldFederationDirectives {
            requires:  vec![FieldPathSelection {
                path:     vec!["weight".to_string()],
                typename: "Order".to_string(),
            }],
            provides:  vec![],
            external:  false,
            shareable: false,
        },
    );

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![order_type],
    };

    let directives = metadata.types[0].get_field_directives("shippingEstimate");
    assert!(directives.is_some());
    assert_eq!(directives.unwrap().requires.len(), 1);
}

#[test]
fn test_python_multiple_requires() {
    // TEST: Python field with multiple @requires
    // PYTHON SCHEMA:
    //   @type
    //   class Order:
    //     @requires("weight")
    //     @requires("dimensions")
    //     def shipping_cost(self) -> float:
    //       pass
    //
    // WHEN: Compiled
    // THEN: Should have two requires directives

    let mut order_type = FederatedType::new("Order".to_string());
    order_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });
    order_type.set_field_directives(
        "shippingCost".to_string(),
        FieldFederationDirectives::new()
            .add_requires(FieldPathSelection {
                path:     vec!["weight".to_string()],
                typename: "Order".to_string(),
            })
            .add_requires(FieldPathSelection {
                path:     vec!["dimensions".to_string()],
                typename: "Order".to_string(),
            }),
    );

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![order_type],
    };

    let directives = metadata.types[0].get_field_directives("shippingCost");
    assert_eq!(directives.unwrap().requires.len(), 2);
}

// ============================================================================
// Test: Field Provisions (@provides)
// ============================================================================

#[test]
fn test_python_provides_directive() {
    // TEST: Python field with @provides decorator
    // PYTHON SCHEMA:
    //   @type
    //   class User:
    //     @provides("Order.userId")
    //     def orders(self) -> list[Order]:
    //       pass
    //
    // WHEN: Compiled with @provides
    // THEN: Should generate field directives with provides

    let mut user_type = FederatedType::new("User".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });
    user_type.set_field_directives(
        "orders".to_string(),
        FieldFederationDirectives::new().add_provides(FieldPathSelection {
            path:     vec!["userId".to_string()],
            typename: "Order".to_string(),
        }),
    );

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    let directives = metadata.types[0].get_field_directives("orders");
    assert!(directives.is_some());
    assert_eq!(directives.unwrap().provides.len(), 1);
}

// ============================================================================
// Test: Runtime Entity Resolution from Compiled Schema
// ============================================================================

#[test]
fn test_python_entity_resolution_basic() {
    // TEST: Resolve entity from compiled Python schema
    // GIVEN: User type with id key field
    // WHEN: Entity representation is provided
    // THEN: Should resolve against compiled schema

    let mut user_type = FederatedType::new("User".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });

    let _federation_metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    let mut entity_fields = HashMap::new();
    entity_fields.insert("id".to_string(), json!("user-123"));
    entity_fields.insert("name".to_string(), json!("Alice"));
    entity_fields.insert("email".to_string(), json!("alice@example.com"));

    let representation = EntityRepresentation {
        typename:   "User".to_string(),
        key_fields: {
            let mut m = HashMap::new();
            m.insert("id".to_string(), json!("user-123"));
            m
        },
        all_fields: entity_fields,
    };

    assert_eq!(representation.typename, "User");
    assert_eq!(representation.key_fields.get("id").unwrap(), &json!("user-123"));
    assert!(representation.has_field("name"));
    assert!(representation.has_field("email"));
}

#[test]
fn test_python_entity_resolution_with_requires() {
    // TEST: Resolve entity with @requires validation
    // GIVEN: Field with @requires("email")
    // WHEN: Entity includes email field
    // THEN: Resolution should succeed

    let mut user_type = FederatedType::new("User".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });
    user_type.set_field_directives(
        "premiumContent".to_string(),
        FieldFederationDirectives::new().add_requires(FieldPathSelection {
            path:     vec!["email".to_string()],
            typename: "User".to_string(),
        }),
    );

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    let mut entity_fields = HashMap::new();
    entity_fields.insert("id".to_string(), json!("user-123"));
    entity_fields.insert("email".to_string(), json!("alice@example.com"));

    let representation = EntityRepresentation {
        typename:   "User".to_string(),
        key_fields: {
            let mut m = HashMap::new();
            m.insert("id".to_string(), json!("user-123"));
            m
        },
        all_fields: entity_fields,
    };

    // Validate @requires is satisfied
    let directives = metadata.types[0].get_field_directives("premiumContent");
    assert!(directives.is_some());
    assert!(representation.has_field("email"));
}

// ============================================================================
// Test: Multi-Subgraph Federation
// ============================================================================

#[test]
fn test_python_cross_subgraph_reference() {
    // TEST: Two subgraphs with entity references
    // SUBGRAPH 1 (Users):
    //   @type @key("id") class User
    //
    // SUBGRAPH 2 (Orders):
    //   @type @key("id") class Order
    //   @type class User (extended reference)
    //
    // WHEN: Compiled to federation metadata
    // THEN: Both should be tracked

    let mut user_type = FederatedType::new("User".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });

    let mut order_type = FederatedType::new("Order".to_string());
    order_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type, order_type],
    };

    assert_eq!(metadata.types.len(), 2);
    assert!(metadata.types.iter().any(|t| t.name == "User"));
    assert!(metadata.types.iter().any(|t| t.name == "Order"));
}

// ============================================================================
// Test: Query Execution with Federation Metadata
// ============================================================================

#[test]
fn test_python_federation_query_types() {
    // TEST: Standard federation schema should have Query type
    // PYTHON SCHEMA:
    //   @type
    //   class Query:
    //     def user(self, id: str) -> User:
    //       pass
    //
    // WHEN: Compiled
    // THEN: Query type should exist in federation metadata

    let mut query_type = FederatedType::new("Query".to_string());
    query_type.keys.push(KeyDirective {
        fields:     vec!["__typename".to_string()],
        resolvable: false,
    });

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![query_type],
    };

    let query = metadata.types.iter().find(|t| t.name == "Query");
    assert!(query.is_some());
}

// ============================================================================
// Test: Error Handling
// ============================================================================

#[test]
fn test_python_missing_key_validation() {
    // TEST: Entity without key fields should fail validation
    // GIVEN: User type with id key requirement
    // WHEN: Entity representation lacks id
    // THEN: Should identify missing key field

    let _user_type = FederatedType::new("User".to_string());

    // Empty key fields means representation is incomplete
    let representation = EntityRepresentation {
        typename:   "User".to_string(),
        key_fields: HashMap::new(),
        all_fields: HashMap::new(),
    };

    assert!(representation.key_fields.is_empty());
}

#[test]
fn test_python_schema_with_no_federation() {
    // TEST: Standalone schema without federation should still work
    // PYTHON SCHEMA:
    //   @type
    //   class User:
    //     id: str
    //
    // WHEN: Compiled without federation directives
    // THEN: Should create type but not mark as federated

    let user_type = FederatedType::new("User".to_string());

    let federation_metadata = FederationMetadata {
        enabled: false,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    assert!(!federation_metadata.enabled);
}

// ============================================================================
// Test: Schema Compilation Verification
// ============================================================================

#[test]
fn test_python_schema_compilation_roundtrip() {
    // TEST: Compiled schema should be consistent
    // GIVEN: Python schema with federation directives
    // WHEN: Compiled to metadata
    // THEN: Should be readable and executable

    let mut user_type = FederatedType::new("User".to_string());
    user_type.keys.push(KeyDirective {
        fields:     vec!["id".to_string()],
        resolvable: true,
    });
    user_type.set_field_directives("email".to_string(), FieldFederationDirectives::new());

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type],
    };

    // Verify metadata is complete and consistent
    assert_eq!(metadata.types.len(), 1);
    assert_eq!(metadata.types[0].name, "User");
    assert!(!metadata.types[0].keys.is_empty());
    assert!(metadata.types[0].get_field_directives("email").is_some());
}

#[test]
fn test_python_multiple_types_in_schema() {
    // TEST: Schema with multiple types
    // PYTHON SCHEMA:
    //   @type @key("id") class User: ...
    //   @type @key("id") class Order: ...
    //   @type @key("id") class Product: ...
    //
    // WHEN: All compiled to single metadata
    // THEN: All types should be available

    let user_type = FederatedType::new("User".to_string());
    let order_type = FederatedType::new("Order".to_string());
    let product_type = FederatedType::new("Product".to_string());

    let metadata = FederationMetadata {
        enabled: true,
        version: "v2".to_string(),
        types:   vec![user_type, order_type, product_type],
    };

    assert_eq!(metadata.types.len(), 3);
    let type_names: Vec<_> = metadata.types.iter().map(|t| t.name.as_str()).collect();
    assert!(type_names.contains(&"User"));
    assert!(type_names.contains(&"Order"));
    assert!(type_names.contains(&"Product"));
}