gts/

schema.rs

//! Runtime schema generation traits for GTS types.
//!
//! This module provides the `GtsSchema` trait which enables runtime schema
//! composition for nested generic types like `BaseEventV1<AuditPayloadV1<PlaceOrderDataV1>>`.

use serde_json::Value;

/// The JSON Schema **draft-07** dialect URI that GTS Type Schemas declare via
/// `$schema`. Single source of truth for the value emitted by the schema
/// generators (the `struct_to_gts_schema` macro, the CLI generator).
pub const JSON_SCHEMA_DRAFT_07: &str = "http://json-schema.org/draft-07/schema#";

/// Chain-aggregated state of a type's `x-gts-traits-schema`, under JSON Schema
/// `allOf` composition over the `$id` chain. Drives the macro's compile-time
/// "traits values need a usable schema" guard.
#[derive(Clone, Copy, PartialEq, Eq, Debug)]
pub enum TraitSchemaState {
    /// No `x-gts-traits-schema` declared anywhere in the chain.
    Absent,
    /// A satisfiable trait shape exists (`true`, an object subschema, or a
    /// `$ref`); trait values are permitted and their *contents* are left to
    /// runtime validation (OP#13), not checked here.
    Open,
    /// Some layer declares `false`, so the composed `allOf` is unsatisfiable —
    /// any trait values are prohibited.
    Prohibited,
}

impl TraitSchemaState {
    /// Compose this (ancestor-side) state with a descendant layer's own state
    /// under `allOf` semantics: `false` anywhere wins (`Prohibited`); otherwise
    /// any satisfiable schema makes the chain `Open`; otherwise `Absent`.
    #[must_use]
    pub const fn join(self, own: TraitSchemaState) -> TraitSchemaState {
        match (self, own) {
            (TraitSchemaState::Prohibited, _) | (_, TraitSchemaState::Prohibited) => {
                TraitSchemaState::Prohibited
            }
            (TraitSchemaState::Open, _) | (_, TraitSchemaState::Open) => TraitSchemaState::Open,
            _ => TraitSchemaState::Absent,
        }
    }
}

/// Trait for types that have a GTS schema.
///
/// This trait enables runtime schema composition for nested generic types.
/// When you have `BaseEventV1<P>` where `P: GtsSchema`, the composed schema
/// can be generated at runtime with proper nesting.
///
/// # Example
///
/// ```ignore
/// use gts::GtsSchema;
///
/// // Get the composed schema for a nested type
/// let schema = BaseEventV1::<AuditPayloadV1<PlaceOrderDataV1>>::gts_schema();
/// // The schema will have payload field containing AuditPayloadV1's schema,
/// // which in turn has data field containing PlaceOrderDataV1's schema
/// ```
pub trait GtsSchema {
    /// The GTS type ID for this type (formerly `SCHEMA_ID`).
    const TYPE_ID: &'static str;

    /// Deprecated alias for [`Self::TYPE_ID`].
    ///
    /// Defaults to `Self::TYPE_ID` so existing implementations that only set
    /// `TYPE_ID` keep working. Reading `T::SCHEMA_ID` from downstream code
    /// raises a deprecation warning pointing at the new name.
    #[deprecated(since = "0.10.0", note = "use `TYPE_ID` instead")]
    const SCHEMA_ID: &'static str = Self::TYPE_ID;

    /// The name of the field that contains the generic type parameter, if any.
    /// For example, `BaseEventV1<P>` has `payload` as the generic field.
    const GENERIC_FIELD: Option<&'static str> = None;

    /// `true` if this type declares `x-gts-final` (not inheritable). Set by
    /// `#[struct_to_gts_schema]` from `gts_final = true`; read by the
    /// derive-from-final compile-time guard.
    const GTS_FINAL: bool = false;

    /// `true` if this type declares `x-gts-abstract` (not directly
    /// instantiable). Set by `#[struct_to_gts_schema]` from `gts_abstract =
    /// true`; read by the `gts_instance!` compile-time guard.
    const GTS_ABSTRACT: bool = false;

    /// Chain-aggregated `x-gts-traits-schema` state (this type's own layer
    /// `allOf`-composed with its ancestors'). Set by `#[struct_to_gts_schema]`;
    /// read by the compile-time guard that rejects `traits` values when the
    /// chain has no usable trait shape ([`TraitSchemaState::Absent`]) or
    /// prohibits traits ([`TraitSchemaState::Prohibited`]).
    const TRAIT_SCHEMA: TraitSchemaState = TraitSchemaState::Absent;

    /// Returns the JSON schema for this type with $ref references intact.
    fn gts_schema_with_refs() -> Value;

    /// Returns the composed JSON schema for this type.
    /// For types with generic parameters that implement `GtsSchema`,
    /// this returns the schema with the generic field's type replaced
    /// by the nested type's schema.
    #[must_use]
    fn gts_schema() -> Value {
        Self::gts_schema_with_refs()
    }

    /// Generate a GTS-style schema with allOf and $ref to base type.
    ///
    /// This produces a schema like:
    /// ```json
    /// {
    ///   "$id": "gts://innermost_type_id",
    ///   "allOf": [
    ///     { "$ref": "gts://base_type_id" },
    ///     { "properties": { "payload": { nested_schema } } }
    ///   ]
    /// }
    /// ```
    #[must_use]
    fn gts_schema_with_refs_allof() -> Value {
        Self::gts_schema_with_refs()
    }

    /// Get the innermost type ID in a nested generic chain.
    /// For `BaseEventV1<AuditPayloadV1<PlaceOrderDataV1>>`, returns `PlaceOrderDataV1`'s ID.
    #[must_use]
    fn innermost_type_id() -> &'static str {
        Self::TYPE_ID
    }

    /// Deprecated alias for [`Self::innermost_type_id`].
    #[deprecated(since = "0.10.0", note = "renamed to `innermost_type_id`")]
    #[must_use]
    fn innermost_schema_id() -> &'static str {
        Self::innermost_type_id()
    }

    /// Get the innermost (leaf) type's raw schema.
    /// For `BaseEventV1<AuditPayloadV1<PlaceOrderDataV1>>`, returns `PlaceOrderDataV1`'s schema.
    #[must_use]
    fn innermost_schema() -> Value {
        Self::gts_schema_with_refs()
    }

    /// This type's *own* declared `x-gts-traits-schema`, or `None` if it
    /// declares none. The single layer this type contributes — not the
    /// chain-aggregated effective trait-schema (the registry composes those
    /// along the `$id` chain via `allOf`). Overridden by
    /// `#[struct_to_gts_schema]` when `traits_schema = …` is set.
    #[must_use]
    fn gts_traits_schema() -> Option<Value> {
        None
    }

    /// This type's *own* declared `x-gts-traits` values, or `None` if it
    /// resolves none. The single layer this type contributes — not the
    /// chain-merged effective traits object. Overridden by
    /// `#[struct_to_gts_schema]` when `traits = …` is set.
    #[must_use]
    fn gts_traits() -> Option<Value> {
        None
    }

    /// Collect the nesting path (generic field names) from outer to inner types.
    /// For `BaseEventV1<AuditPayloadV1<PlaceOrderDataV1>>`, returns `["payload", "data"]`.
    #[must_use]
    fn collect_nesting_path() -> Vec<&'static str> {
        Vec::new()
    }

    /// Path of generic-slot field names from the document root (the
    /// outermost base type in the chain) down to where this type's own
    /// properties live in a composed instance.
    ///
    /// For a base type (no parent): `[]`.
    /// For each derived level: parent's path plus the parent's
    /// `GENERIC_FIELD`. So for the chain
    /// `BaseEventV1<P> -> AuditPayloadV1<D> -> PlaceOrderDataV1<E> -> PlaceOrderDataPayloadV1`:
    ///
    /// | Type | `outer_generic_path()` |
    /// |---|---|
    /// | `BaseEventV1` | `[]` |
    /// | `AuditPayloadV1` | `["payload"]` |
    /// | `PlaceOrderDataV1` | `["payload", "data"]` |
    /// | `PlaceOrderDataPayloadV1` | `["payload", "data", "last"]` |
    ///
    /// Used by derived emitters to wrap their overlay properties at the
    /// correct depth, so a derived schema's `allOf` overlay declares its
    /// fields nested under the parent chain's generic slots rather than
    /// at the top level (which would violate the base's
    /// `additionalProperties: false` per gts-spec sec 3.1).
    #[must_use]
    fn outer_generic_path() -> Vec<&'static str> {
        Vec::new()
    }

    /// Wrap properties in a nested structure following the nesting path.
    /// For path `["payload", "data"]` and properties `{order_id, product_id, last}`,
    /// returns `{ "payload": { "type": "object", "properties": { "data": { "type": "object", "additionalProperties": false, "properties": {...}, "required": [...] } } } }`
    ///
    /// The `additionalProperties: false` is placed on the object that contains the current type's
    /// own properties. Generic fields that will be extended by children are just `{"type": "object"}`.
    ///
    /// # Arguments
    /// * `path` - The nesting path from outer to inner (e.g., `["payload", "data"]`)
    /// * `properties` - The properties of the current type
    /// * `required` - The required fields of the current type
    /// * `generic_field` - The name of the generic field in the current type (if any), which should NOT have additionalProperties: false
    #[must_use]
    fn wrap_in_nesting_path(
        path: &[&str],
        properties: Value,
        required: Value,
        generic_field: Option<&str>,
    ) -> Value {
        if path.is_empty() {
            return properties;
        }

        // Build the innermost schema - this contains the current type's own properties
        // Set additionalProperties: false on this level (the object containing our properties)
        let mut current = serde_json::json!({
            "type": "object",
            "additionalProperties": false,
            "properties": properties,
            "required": required
        });

        // If we have a generic field, ensure it's just {"type": "object"} without additionalProperties
        // This field will be extended by child schemas
        if let Some(gf) = generic_field
            && let Some(props) = current
                .get_mut("properties")
                .and_then(|v| v.as_object_mut())
            && props.contains_key(gf)
        {
            props.insert(gf.to_owned(), serde_json::json!({"type": "object"}));
        }

        // Wrap from inner to outer - parent levels don't need additionalProperties: false
        for field in path.iter().rev() {
            current = serde_json::json!({
                "type": "object",
                "properties": {
                    *field: current
                }
            });
        }

        // Extract just the properties object from the outermost wrapper
        // since the caller will put this in a "properties" field
        if let Some(props) = current.get("properties") {
            return props.clone();
        }

        current
    }
}

/// Marker implementation for () to allow `BaseEventV1<()>` etc.
impl GtsSchema for () {
    const TYPE_ID: &'static str = "";

    fn gts_schema_with_refs() -> Value {
        serde_json::json!({
            "type": "object"
        })
    }
}

/// Marker implementation for [`serde_json::Value`] — the same "I am a
/// placeholder" protocol as `impl GtsSchema for ()`, except the carrier
/// holds actual JSON data rather than being empty.
///
/// Use as the default generic parameter in `Base<P>` types whose payload
/// is heterogeneous at runtime — e.g. a multi-provider catalog where the
/// concrete leaf shape is selected by an `info.gts_type` field on the
/// data, not by the Rust type parameter. Consumers narrow to a typed
/// view (`Base<ConcreteLeaf>` or `Base<Intermediate<ConcreteLeaf>>`)
/// by matching on the runtime `gts_type` and deserialising the JSON
/// payload into the chosen target.
///
/// Like `impl GtsSchema for ()`, `TYPE_ID` is the empty sentinel that
/// signals "no own identity — read the real id from data".
impl GtsSchema for Value {
    const TYPE_ID: &'static str = "";

    fn gts_schema_with_refs() -> Value {
        serde_json::json!({
            "type": "object",
            "description": "Opaque JSON payload; the concrete schema is \
                            identified by the carrying type's runtime \
                            discriminator field (typically `gts_type`)."
        })
    }
}

/// Private trait for nested GTS struct serialization.
///
/// Nested structs implement this instead of `serde::Serialize` to prevent
/// direct serialization (which would produce incomplete JSON without base struct fields).
/// Base structs use `#[serde(serialize_with)]` to call this trait internally.
pub trait GtsSerialize {
    /// Serialize this value using the GTS serialization protocol.
    ///
    /// # Errors
    ///
    /// Returns an error if serialization fails.
    fn gts_serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer;
}

/// Private trait for nested GTS struct deserialization.
///
/// Nested structs implement this instead of `serde::Deserialize` to prevent
/// direct deserialization.
pub trait GtsDeserialize<'de>: Sized {
    /// Deserialize this value using the GTS deserialization protocol.
    ///
    /// # Errors
    ///
    /// Returns an error if deserialization fails.
    fn gts_deserialize<__D>(deserializer: __D) -> Result<Self, __D::Error>
    where
        __D: serde::Deserializer<'de>;
}

/// Internal marker trait to block direct serde serialization on nested GTS structs.
///
/// The macro implements this for nested structs; any direct `Serialize` impl then
/// conflicts with the blanket impl below, producing a compile-time error.
#[doc(hidden)]
pub trait GtsNoDirectSerialize {}

/// Internal marker trait to block direct serde deserialization on nested GTS structs.
#[doc(hidden)]
pub trait GtsNoDirectDeserialize {}

impl<T: serde::Serialize> GtsNoDirectSerialize for T {}

impl<T> GtsNoDirectDeserialize for T where for<'de> T: serde::Deserialize<'de> {}

/// Blanket impl: anything with Serialize also has `GtsSerialize`.
/// This allows standard serde types (String, i32, etc.) to be used in GTS structs.
impl<T: serde::Serialize> GtsSerialize for T {
    fn gts_serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        serde::Serialize::serialize(self, serializer)
    }
}

/// Blanket impl: anything with Deserialize also has `GtsDeserialize`.
impl<'de, T: serde::Deserialize<'de>> GtsDeserialize<'de> for T {
    fn gts_deserialize<__D>(deserializer: __D) -> Result<Self, __D::Error>
    where
        __D: serde::Deserializer<'de>,
    {
        <T as serde::Deserialize<'de>>::deserialize(deserializer)
    }
}

/// Serialize a value via `GtsSerialize` trait.
///
/// Used with `#[serde(serialize_with = "gts::serialize_gts")]` on generic fields in base structs.
///
/// # Errors
///
/// Returns an error if serialization fails.
pub fn serialize_gts<T: GtsSerialize, S: serde::Serializer>(
    value: &T,
    serializer: S,
) -> Result<S::Ok, S::Error> {
    value.gts_serialize(serializer)
}

/// Deserialize a value via `GtsDeserialize` trait.
///
/// Used with `#[serde(deserialize_with = "gts::deserialize_gts")]` on generic fields in base structs.
///
/// # Errors
///
/// Returns an error if deserialization fails.
pub fn deserialize_gts<'de, T: GtsDeserialize<'de>, D: serde::Deserializer<'de>>(
    deserializer: D,
) -> Result<T, D::Error> {
    T::gts_deserialize(deserializer)
}

/// Wrapper to serialize a GtsSerialize type using serde's Serialize trait.
///
/// This is used internally by the macro to serialize generic fields in nested structs.
/// Generic fields may not implement Serialize directly (only GtsSerialize), so this
/// wrapper bridges the gap.
#[doc(hidden)]
pub struct GtsSerializeWrapper<'a, T: GtsSerialize + ?Sized>(pub &'a T);

impl<T: GtsSerialize + ?Sized> serde::Serialize for GtsSerializeWrapper<'_, T> {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        self.0.gts_serialize(serializer)
    }
}

/// Wrapper for deserializing into a GtsDeserialize type.
///
/// Used internally by the macro for generic field deserialization in nested structs.
#[doc(hidden)]
pub struct GtsDeserializeWrapper<T>(pub T);

impl<'de, T: GtsDeserialize<'de>> serde::Deserialize<'de> for GtsDeserializeWrapper<T> {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        T::gts_deserialize(deserializer).map(GtsDeserializeWrapper)
    }
}

/// Generate a GTS-style schema for a nested type with allOf and $ref to base.
///
/// This macro generates a schema where:
/// - `$id` is the innermost type's schema ID
/// - `allOf` contains a `$ref` to the base (outermost) type's schema ID
/// - The nested types' properties are placed in the payload fields
///
/// # Example
///
/// ```ignore
/// use gts::gts_schema_for;
///
/// let schema = gts_schema_for!(BaseEventV1<AuditPayloadV1<PlaceOrderDataV1>>);
/// // Produces:
/// // {
/// //   "$id": "gts://...PlaceOrderDataV1...",
/// //   "allOf": [
/// //     { "$ref": "gts://BaseEventV1..." },
/// //     { "properties": { "payload": { ... } } }
/// //   ]
/// // }
/// ```
#[macro_export]
macro_rules! gts_schema_for {
    ($base:ty) => {{
        use $crate::GtsSchema;
        <$base as GtsSchema>::gts_schema_with_refs_allof()
    }};
}

/// Strip schema metadata fields ($id, $schema, title, description) for cleaner nested schemas.
#[must_use]
pub fn strip_schema_metadata(schema: &Value) -> Value {
    let mut result = schema.clone();
    if let Some(obj) = result.as_object_mut() {
        obj.remove("$id");
        obj.remove("$schema");
        obj.remove("title");
        obj.remove("description");

        // Recursively strip from nested properties
        if let Some(props) = obj.get_mut("properties").and_then(|v| v.as_object_mut()) {
            let keys: Vec<String> = props.keys().cloned().collect();
            for key in keys {
                if let Some(prop_value) = props.get(&key) {
                    let cleaned = strip_schema_metadata(prop_value);
                    props.insert(key, cleaned);
                }
            }
        }
    }
    result
}

/// Build a GTS schema with allOf structure referencing base type.
///
/// # Arguments
/// * `innermost_type_id` - The $id for the generated schema (innermost type)
/// * `base_type_id` - The $ref target (base/outermost type)
/// * `title` - Schema title
/// * `own_properties` - Properties specific to this composed type
/// * `required` - Required fields
#[must_use]
pub fn build_gts_allof_schema(
    innermost_type_id: &str,
    base_type_id: &str,
    title: &str,
    own_properties: &Value,
    required: &[&str],
) -> Value {
    serde_json::json!({
        "$id": format!("gts://{}", innermost_type_id),
        "$schema": "http://json-schema.org/draft-07/schema#",
        "title": title,
        "type": "object",
        "allOf": [
            { "$ref": format!("gts://{}", base_type_id) },
            {
                "type": "object",
                "properties": own_properties,
                "required": required
            }
        ]
    })
}

#[cfg(test)]
#[allow(clippy::unwrap_used, clippy::expect_used)]
mod tests {
    use super::*;
    use serde_json::json;

    #[test]
    fn trait_schema_state_join_truth_table() {
        use TraitSchemaState::{Absent, Open, Prohibited};

        // Full 3×3 lattice under `allOf` composition:
        // - `Prohibited` (a `false` subschema) annihilates anything → Prohibited.
        // - otherwise any `Open` (satisfiable schema) makes the chain Open.
        // - otherwise (both sides Absent) the chain stays Absent.
        let cases = [
            (Absent, Absent, Absent),
            (Absent, Open, Open),
            (Absent, Prohibited, Prohibited),
            (Open, Absent, Open),
            (Open, Open, Open),
            (Open, Prohibited, Prohibited),
            (Prohibited, Absent, Prohibited),
            (Prohibited, Open, Prohibited),
            (Prohibited, Prohibited, Prohibited),
        ];

        for (ancestor, own, expected) in cases {
            assert_eq!(
                ancestor.join(own),
                expected,
                "join({ancestor:?}, {own:?}) should be {expected:?}"
            );
        }
    }

    #[test]
    fn test_unit_type_properties() {
        // Test all unit type properties in one test
        let schema = <()>::gts_schema();
        assert_eq!(schema, json!({"type": "object"}));
        assert_eq!(<()>::TYPE_ID, "");
        assert_eq!(<()>::GENERIC_FIELD, None);
    }

    #[test]
    fn test_wrap_in_nesting_path_empty_path() {
        let properties = json!({"field1": {"type": "string"}});
        let required = json!(["field1"]);

        let result = <()>::wrap_in_nesting_path(&[], properties.clone(), required, None);

        assert_eq!(result, properties);
    }

    #[test]
    fn test_wrap_in_nesting_path_single_level() {
        let properties = json!({"field1": {"type": "string"}});
        let required = json!(["field1"]);

        let result = <()>::wrap_in_nesting_path(&["payload"], properties, required.clone(), None);

        assert_eq!(
            result,
            json!({
                "payload": {
                    "type": "object",
                    "additionalProperties": false,
                    "properties": {"field1": {"type": "string"}},
                    "required": required
                }
            })
        );
    }

    #[test]
    fn test_wrap_in_nesting_path_multi_level() {
        let properties = json!({"field1": {"type": "string"}});
        let required = json!(["field1"]);

        let result =
            <()>::wrap_in_nesting_path(&["payload", "data"], properties, required.clone(), None);

        assert_eq!(
            result,
            json!({
                "payload": {
                    "type": "object",
                    "properties": {
                        "data": {
                            "type": "object",
                            "additionalProperties": false,
                            "properties": {"field1": {"type": "string"}},
                            "required": required
                        }
                    }
                }
            })
        );
    }

    #[test]
    fn test_wrap_in_nesting_path_with_generic_field() {
        let properties = json!({
            "field1": {"type": "string"},
            "generic_field": {"type": "number"}
        });
        let required = json!(["field1"]);

        let result =
            <()>::wrap_in_nesting_path(&["payload"], properties, required, Some("generic_field"));

        let result_obj = result.as_object().unwrap();
        let payload = result_obj.get("payload").unwrap();
        let props = payload.get("properties").unwrap();

        // Generic field should be just {"type": "object"}
        assert_eq!(
            props.get("generic_field").unwrap(),
            &json!({"type": "object"})
        );
        // Other fields should be preserved
        assert_eq!(props.get("field1").unwrap(), &json!({"type": "string"}));
    }

    #[test]
    fn test_strip_schema_metadata_removes_all_metadata() {
        // Test removal of all metadata fields including $id, $schema, title, description
        let schema = json!({
            "$id": "gts://test",
            "$schema": "http://json-schema.org/draft-07/schema#",
            "title": "Test Schema",
            "description": "A test",
            "type": "object",
            "properties": {"field": {"type": "string"}}
        });

        let result = strip_schema_metadata(&schema);

        // All metadata should be removed
        assert!(result.get("$id").is_none());
        assert!(result.get("$schema").is_none());
        assert!(result.get("title").is_none());
        assert!(result.get("description").is_none());
        // Non-metadata should be preserved
        assert_eq!(result.get("type").unwrap(), "object");
        assert!(result.get("properties").is_some());
    }

    #[test]
    fn test_strip_schema_metadata_recursive() {
        let schema = json!({
            "$id": "gts://test",
            "properties": {
                "nested": {
                    "$id": "gts://nested",
                    "type": "string",
                    "description": "Nested field"
                }
            }
        });

        let result = strip_schema_metadata(&schema);

        assert!(result.get("$id").is_none());
        let props = result.get("properties").unwrap();
        let nested = props.get("nested").unwrap();
        assert!(nested.get("$id").is_none());
        assert!(nested.get("description").is_none());
        assert_eq!(nested.get("type").unwrap(), "string");
    }

    #[test]
    fn test_strip_schema_metadata_preserves_non_metadata() {
        let schema = json!({
            "$id": "gts://test",
            "type": "object",
            "properties": {"field": {"type": "string"}},
            "required": ["field"],
            "additionalProperties": false
        });

        let result = strip_schema_metadata(&schema);

        assert_eq!(result.get("type").unwrap(), "object");
        assert!(result.get("properties").is_some());
        assert!(result.get("required").is_some());
        assert_eq!(result.get("additionalProperties").unwrap(), &json!(false));
    }

    #[test]
    fn test_build_gts_allof_schema_structure() {
        let properties = json!({"field1": {"type": "string"}});
        let required = vec!["field1"];

        let result = build_gts_allof_schema(
            "vendor.package.namespace.child.1",
            "vendor.package.namespace.base.1",
            "Child Schema",
            &properties,
            &required,
        );

        assert_eq!(
            result.get("$id").unwrap(),
            "gts://vendor.package.namespace.child.1"
        );
        assert_eq!(
            result.get("$schema").unwrap(),
            "http://json-schema.org/draft-07/schema#"
        );
        assert_eq!(result.get("title").unwrap(), "Child Schema");
        assert_eq!(result.get("type").unwrap(), "object");

        let allof = result.get("allOf").unwrap().as_array().unwrap();
        assert_eq!(allof.len(), 2);
    }

    #[test]
    fn test_build_gts_allof_schema_ref_format() {
        let properties = json!({"field1": {"type": "string"}});
        let required = vec!["field1"];

        let result = build_gts_allof_schema(
            "vendor.package.namespace.child.1",
            "vendor.package.namespace.base.1",
            "Child Schema",
            &properties,
            &required,
        );

        let allof = result.get("allOf").unwrap().as_array().unwrap();
        let ref_obj = &allof[0];

        assert_eq!(
            ref_obj.get("$ref").unwrap(),
            "gts://vendor.package.namespace.base.1"
        );
    }

    #[test]
    fn test_build_gts_allof_schema_properties_in_allof() {
        let properties = json!({"field1": {"type": "string"}, "field2": {"type": "number"}});
        let required = vec!["field1", "field2"];

        let result = build_gts_allof_schema(
            "vendor.package.namespace.child.1",
            "vendor.package.namespace.base.1",
            "Child Schema",
            &properties,
            &required,
        );

        let allof = result.get("allOf").unwrap().as_array().unwrap();
        let props_obj = &allof[1];

        assert_eq!(props_obj.get("type").unwrap(), "object");
        assert_eq!(props_obj.get("properties").unwrap(), &properties);
        assert_eq!(props_obj.get("required").unwrap(), &json!(required));
    }
}