mig-assembly 0.1.60

//! Shared test helpers for constructing MIG schema test fixtures.
//!
//! Available to both unit tests (`#[cfg(test)]` modules) and integration tests.

use mig_types::schema::mig::{MigSegment, MigSegmentGroup};

/// Create a minimal `MigSegment` for testing, with only the `id` set meaningfully.
pub fn make_mig_segment(id: &str) -> MigSegment {
    MigSegment {
        id: id.to_string(),
        name: id.to_string(),
        description: None,
        counter: None,
        level: 0,
        number: None,
        max_rep_std: 1,
        max_rep_spec: 1,
        status_std: Some("M".to_string()),
        status_spec: Some("M".to_string()),
        example: None,
        data_elements: vec![],
        composites: vec![],
    }
}

/// Create a `MigSegment` with a MIG `Number` attribute for testing mig_number assignment.
pub fn make_mig_segment_numbered(id: &str, number: &str) -> MigSegment {
    let mut seg = make_mig_segment(id);
    seg.number = Some(number.to_string());
    seg
}

/// Create a minimal `MigSegmentGroup` for testing.
pub fn make_mig_group(
    id: &str,
    segments: Vec<&str>,
    nested: Vec<MigSegmentGroup>,
) -> MigSegmentGroup {
    MigSegmentGroup {
        id: id.to_string(),
        name: id.to_string(),
        description: None,
        counter: None,
        level: 1,
        max_rep_std: 99,
        max_rep_spec: 99,
        status_std: Some("M".to_string()),
        status_spec: Some("M".to_string()),
        segments: segments.into_iter().map(make_mig_segment).collect(),
        nested_groups: nested,
        variant_code: None,
        variant_qualifier_position: None,
        variant_codes: vec![],
        merged_variant_count: None,
    }
}

/// Create a `MigSegmentGroup` with a variant code for testing variant-aware assembly.
pub fn make_mig_group_with_variant(
    id: &str,
    segments: Vec<&str>,
    nested: Vec<MigSegmentGroup>,
    variant_code: &str,
) -> MigSegmentGroup {
    let mut group = make_mig_group(id, segments, nested);
    group.variant_code = Some(variant_code.to_string());
    group.variant_codes = vec![variant_code.to_string()];
    group
}