qubit-metadata 0.7.2

// =============================================================================
//    Copyright (c) 2025 - 2026 Haixing Hu.
//
//    SPDX-License-Identifier: Apache-2.0
//
//    Licensed under the Apache License, Version 2.0.
// =============================================================================
//! Integration tests for examples documented in the README and user guide.

#![cfg(any(feature = "json", feature = "schema", feature = "filter"))]

#[cfg(feature = "json")]
use qubit_budget::ResourceLimit;
#[cfg(feature = "json")]
use qubit_budget::json::JsonResource;
#[cfg(feature = "json")]
use qubit_metadata::Metadata;
#[cfg(feature = "json")]
use qubit_metadata::MetadataWireDecodeError;
#[cfg(feature = "json")]
use qubit_metadata::MetadataWireEncodeError;
#[cfg(feature = "json")]
use qubit_metadata::default_json_encode_limits;

mod support {
    #[cfg(feature = "json")]
    pub mod readme_json_roundtrip;
    #[cfg(feature = "json")]
    pub mod user_guide_bounded_decode;
    #[cfg(feature = "json")]
    pub mod user_guide_bounded_encode_tests;
    #[cfg(feature = "schema")]
    pub mod user_guide_chunk_workflow;
    #[cfg(feature = "json")]
    pub mod user_guide_directional_limits;
    #[cfg(feature = "filter")]
    pub mod user_guide_empty_sets;
    #[cfg(feature = "filter")]
    #[cfg(feature = "json")]
    pub mod user_guide_filter_bounded_encode;
    #[cfg(feature = "filter")]
    pub mod user_guide_query_translation;
}

#[cfg(feature = "filter")]
#[test]
fn test_user_guide_empty_sets_example() {
    support::user_guide_empty_sets::main().expect("empty set guide example should run");
}

#[cfg(feature = "filter")]
#[test]
fn test_query_translation_example() {
    support::user_guide_query_translation::main().expect("query translation guide example should run");
}

#[cfg(feature = "filter")]
#[test]
fn test_query_translation_source_matches() {
    for (name, document) in [
        ("English guide", include_str!("../doc/user_guide.md")),
        ("Chinese guide", include_str!("../doc/user_guide.zh_CN.md")),
    ] {
        assert_documented_example(
            name,
            document,
            "query-translation-example",
            include_str!("support/user_guide_query_translation.rs"),
        );
    }
}

#[cfg(feature = "filter")]
#[test]
fn test_user_guide_empty_sets_source_matches() {
    for (name, document) in [
        ("English guide", include_str!("../doc/user_guide.md")),
        ("Chinese guide", include_str!("../doc/user_guide.zh_CN.md")),
    ] {
        assert_documented_example(
            name,
            document,
            "empty-set-example",
            include_str!("support/user_guide_empty_sets.rs"),
        );
    }
}

#[cfg(feature = "json")]
#[test]
fn test_readme_json_roundtrip_example() {
    support::readme_json_roundtrip::main().expect("README JSON round-trip example should run");
}

#[cfg(feature = "json")]
#[test]
fn test_user_guide_directional_limits_example() {
    support::user_guide_directional_limits::main().expect("directional budget guide example should run");
}

#[cfg(feature = "json")]
#[test]
fn test_user_guide_domain_limits_example() -> Result<(), MetadataWireDecodeError> {
    support::user_guide_bounded_decode::main()
}

#[cfg(feature = "json")]
#[test]
fn test_user_guide_bounded_decode_source_matches() {
    for (name, document) in [
        ("English README", include_str!("../README.md")),
        ("Chinese README", include_str!("../README.zh_CN.md")),
        ("English guide", include_str!("../doc/user_guide.md")),
        ("Chinese guide", include_str!("../doc/user_guide.zh_CN.md")),
    ] {
        assert!(
            document.contains("qubit-metadata = { version = \"0.7\", features = [\"json\"] }"),
            "{name}: metadata json dependency"
        );
        if name.contains("README") {
            assert_readme_json_roundtrip(name, document);
        } else {
            assert!(
                document.contains("qubit-budget = { version = \"0.10\", features = [\"json\"] }"),
                "{name}: budget json dependency"
            );
            assert_documented_example(
                name,
                document,
                "directional-json-limits-example",
                marked_example(
                    include_str!("support/user_guide_directional_limits.rs"),
                    "user-guide-directional-limits-example",
                ),
            );
            assert_documented_example(
                name,
                document,
                "bounded-json-decoding-example",
                include_str!("support/user_guide_bounded_decode.rs"),
            );
            assert_documented_example(
                name,
                document,
                "bounded-json-encoding-example",
                include_str!("support/user_guide_bounded_encode_tests.rs"),
            );
            #[cfg(feature = "filter")]
            assert_documented_example(
                name,
                document,
                "bounded-filter-json-encoding-example",
                include_str!("support/user_guide_filter_bounded_encode.rs"),
            );
        }
    }
}

#[cfg(feature = "filter")]
#[cfg(feature = "json")]
#[test]
fn test_user_guide_filter_bounded_json_encoding_example() {
    support::user_guide_filter_bounded_encode::main().expect("filter encoding guide example should run");
}

#[cfg(feature = "schema")]
#[test]
fn test_user_guide_chunk_workflow_example() {
    support::user_guide_chunk_workflow::main().expect("chunk workflow should run");
    for (name, text) in [
        ("English guide", include_str!("../doc/user_guide.md")),
        ("Chinese guide", include_str!("../doc/user_guide.zh_CN.md")),
    ] {
        assert_documented_example(
            name,
            text,
            "chunk-workflow-example",
            include_str!("support/user_guide_chunk_workflow.rs"),
        );
    }
}

#[cfg(feature = "json")]
fn assert_readme_json_roundtrip(name: &str, document: &str) {
    let source = include_str!("support/readme_json_roundtrip.rs");
    assert_documented_example(
        name,
        document,
        "readme-json-roundtrip-example",
        marked_example(source, "readme-json-roundtrip-example"),
    );
}

#[cfg(feature = "json")]
fn marked_example<'a>(source: &'a str, marker: &str) -> &'a str {
    let start = format!("// {marker}:start");
    let end = format!("// {marker}:end");
    assert_eq!(source.matches(&start).count(), 1, "helper start marker for {marker}");
    assert_eq!(source.matches(&end).count(), 1, "helper end marker for {marker}");
    let a = source.find(&start).expect("helper start marker") + start.len();
    let b = source.find(&end).expect("helper end marker");
    assert!(a < b, "helper markers must be ordered for {marker}");
    source[a..b].trim()
}

fn assert_documented_example(name: &str, document: &str, marker: &str, expected: &str) {
    let start = format!("<!-- {marker}:start -->");
    let end = format!("<!-- {marker}:end -->");
    assert_eq!(document.matches(&start).count(), 1, "{name}: start marker for {marker}");
    assert_eq!(document.matches(&end).count(), 1, "{name}: end marker for {marker}");
    let start_index = document.find(&start).expect("start marker");
    let end_index = document.find(&end).expect("end marker");
    assert!(start_index < end_index, "{name}: markers out of order for {marker}");
    let block = &document[start_index + start.len()..end_index];
    assert_eq!(
        block.matches("```rust").count(),
        1,
        "{name}: one Rust block for {marker}"
    );
    assert_eq!(
        block.matches("```").count(),
        2,
        "{name}: no extra fenced blocks for {marker}"
    );
    let code = block
        .trim()
        .strip_prefix("```rust")
        .and_then(|value| value.strip_suffix("```"))
        .expect("one fenced Rust block")
        .trim();
    assert_eq!(code, expected.trim(), "{name}: {marker} must match compiled example");
}

#[cfg(feature = "json")]
#[test]
fn test_user_guide_bounded_json_encoding_example() {
    support::user_guide_bounded_encode_tests::main().expect("encoding guide example should run");
}

#[cfg(feature = "json")]
#[test]
fn test_user_guide_output_budget_rejects_metadata() {
    let limits = default_json_encode_limits()
        .into_builder()
        .output_bytes_limit(ResourceLimit::new(JsonResource::OutputBytes, 1))
        .build();
    let metadata = Metadata::new().with("tenant_id", "acme");

    assert!(matches!(
        metadata.to_json_vec_with_limits(limits),
        Err(MetadataWireEncodeError::Budget(_))
    ));
}