syster-base 0.1.12-alpha

//! Tests for folding range extraction

use crate::core::{Position, Span};
use crate::semantic::adapters::folding_ranges::{
    extract_kerml_folding_ranges, extract_sysml_folding_ranges,
};
use crate::syntax::kerml::ast::{
    Comment as KerMLComment, Element as KerMLElement, KerMLFile, Package as KerMLPackage,
};
use crate::syntax::sysml::ast::{
    Comment as SysMLComment, Element as SysMLElement, Package as SysMLPackage, SysMLFile,
};

fn make_span(start_line: usize, end_line: usize) -> Span {
    Span {
        start: Position {
            line: start_line,
            column: 0,
        },
        end: Position {
            line: end_line,
            column: 1,
        },
    }
}

// =============================================================================
// SysML folding extraction tests
// =============================================================================

#[test]
fn test_sysml_extract_folding_ranges_empty_file() {
    let file = SysMLFile {
        namespace: None,
        namespaces: vec![],
        elements: vec![],
    };
    let ranges = extract_sysml_folding_ranges(&file);
    assert!(ranges.is_empty());
}

#[test]
fn test_sysml_extract_folding_ranges_filters_single_line() {
    let file = SysMLFile {
        namespace: None,
        namespaces: vec![],
        elements: vec![SysMLElement::Package(SysMLPackage {
            name: Some("Test".to_string()),
            elements: vec![],
            span: Some(make_span(1, 1)), // Same line = not foldable
        })],
    };
    let ranges = extract_sysml_folding_ranges(&file);
    assert!(ranges.is_empty());
}

#[test]
fn test_sysml_extract_folding_ranges_includes_multiline() {
    let file = SysMLFile {
        namespace: None,
        namespaces: vec![],
        elements: vec![SysMLElement::Package(SysMLPackage {
            name: Some("Test".to_string()),
            elements: vec![],
            span: Some(make_span(1, 5)), // Multi-line = foldable
        })],
    };
    let ranges = extract_sysml_folding_ranges(&file);
    assert_eq!(ranges.len(), 1);
    assert!(!ranges[0].is_comment); // Package is not a comment
}

#[test]
fn test_sysml_extract_folding_ranges_comment_kind() {
    let file = SysMLFile {
        namespace: None,
        namespaces: vec![],
        elements: vec![SysMLElement::Comment(SysMLComment::new(
            "A long comment",
            Some(make_span(1, 3)),
        ))],
    };
    let ranges = extract_sysml_folding_ranges(&file);
    assert_eq!(ranges.len(), 1);
    assert!(ranges[0].is_comment); // Comment should be marked as comment
}

#[test]
fn test_sysml_extract_folding_ranges_sorted_by_line() {
    let file = SysMLFile {
        namespace: None,
        namespaces: vec![],
        elements: vec![
            SysMLElement::Package(SysMLPackage {
                name: Some("Second".to_string()),
                elements: vec![],
                span: Some(make_span(10, 15)),
            }),
            SysMLElement::Package(SysMLPackage {
                name: Some("First".to_string()),
                elements: vec![],
                span: Some(make_span(1, 5)),
            }),
        ],
    };
    let ranges = extract_sysml_folding_ranges(&file);
    assert_eq!(ranges.len(), 2);
    assert!(ranges[0].span.start.line < ranges[1].span.start.line);
}

// =============================================================================
// KerML folding extraction tests
// =============================================================================

#[test]
fn test_kerml_extract_folding_ranges_empty_file() {
    let file = KerMLFile {
        namespace: None,
        elements: vec![],
    };
    let ranges = extract_kerml_folding_ranges(&file);
    assert!(ranges.is_empty());
}

#[test]
fn test_kerml_extract_folding_ranges_filters_single_line() {
    let file = KerMLFile {
        namespace: None,
        elements: vec![KerMLElement::Package(KerMLPackage {
            name: Some("Test".to_string()),
            elements: vec![],
            span: Some(make_span(1, 1)), // Same line = not foldable
        })],
    };
    let ranges = extract_kerml_folding_ranges(&file);
    assert!(ranges.is_empty());
}

#[test]
fn test_kerml_extract_folding_ranges_includes_multiline() {
    let file = KerMLFile {
        namespace: None,
        elements: vec![KerMLElement::Package(KerMLPackage {
            name: Some("Test".to_string()),
            elements: vec![],
            span: Some(make_span(1, 5)), // Multi-line = foldable
        })],
    };
    let ranges = extract_kerml_folding_ranges(&file);
    assert_eq!(ranges.len(), 1);
    assert!(!ranges[0].is_comment); // Package is not a comment
}

#[test]
fn test_kerml_extract_folding_ranges_comment_kind() {
    let file = KerMLFile {
        namespace: None,
        elements: vec![KerMLElement::Comment(KerMLComment {
            content: "A long comment".to_string(),
            about: vec![],
            locale: None,
            span: Some(make_span(1, 3)),
        })],
    };
    let ranges = extract_kerml_folding_ranges(&file);
    assert_eq!(ranges.len(), 1);
    assert!(ranges[0].is_comment); // Comment should be marked as comment
}

#[test]
fn test_kerml_extract_folding_ranges_sorted_by_line() {
    let file = KerMLFile {
        namespace: None,
        elements: vec![
            KerMLElement::Package(KerMLPackage {
                name: Some("Second".to_string()),
                elements: vec![],
                span: Some(make_span(10, 15)),
            }),
            KerMLElement::Package(KerMLPackage {
                name: Some("First".to_string()),
                elements: vec![],
                span: Some(make_span(1, 5)),
            }),
        ],
    };
    let ranges = extract_kerml_folding_ranges(&file);
    assert_eq!(ranges.len(), 2);
    assert!(ranges[0].span.start.line < ranges[1].span.start.line);
}