reovim-driver-syntax 0.14.4

use {
    super::*,
    crate::{
        edit::SyntaxEdit,
        highlight::{Annotation, HighlightCategory},
    },
    std::ops::Range,
};

/// Minimal driver for testing.
struct MinimalDriver {
    language: &'static str,
}

impl SyntaxDriver for MinimalDriver {
    fn language(&self) -> &str {
        self.language
    }

    fn parse(&mut self, _content: &str) {}

    fn update(&mut self, _content: &str, _edit: &SyntaxEdit) {}

    fn highlights(&self, _byte_range: Range<usize>) -> Vec<Annotation> {
        vec![Annotation::new(0, 1, HighlightCategory::new("comment"))]
    }

    fn is_parsed(&self) -> bool {
        true
    }
}

/// Test factory implementation.
struct TestFactory;

impl SyntaxDriverFactory for TestFactory {
    fn create(&self, language_id: &str) -> Option<Box<dyn SyntaxDriver>> {
        match language_id {
            "rust" => Some(Box::new(MinimalDriver { language: "rust" })),
            "python" => Some(Box::new(MinimalDriver { language: "python" })),
            _ => None,
        }
    }

    fn supported_languages(&self) -> Vec<&str> {
        vec!["rust", "python"]
    }
}

#[test]
fn test_factory_create() {
    let factory = TestFactory;

    let rust = factory.create("rust");
    assert!(rust.is_some());
    assert_eq!(rust.unwrap().language(), "rust");

    let python = factory.create("python");
    assert!(python.is_some());
    assert_eq!(python.unwrap().language(), "python");

    let unknown = factory.create("unknown");
    assert!(unknown.is_none());
}

#[test]
fn test_factory_supported_languages() {
    let factory = TestFactory;

    let languages = factory.supported_languages();
    assert_eq!(languages.len(), 2);
    assert!(languages.contains(&"rust"));
    assert!(languages.contains(&"python"));
}

#[test]
fn test_factory_supports() {
    let factory = TestFactory;

    assert!(factory.supports("rust"));
    assert!(factory.supports("python"));
    assert!(!factory.supports("unknown"));
    assert!(!factory.supports("javascript"));
}

#[test]
#[cfg_attr(coverage_nightly, coverage(off))]
fn test_factory_is_object_safe() {
    fn _accepts_ref(_: &dyn SyntaxDriverFactory) {}
    fn _accepts_box(_: Box<dyn SyntaxDriverFactory>) {}
}

#[test]
fn test_factory_created_driver_produces_highlights() {
    let factory = TestFactory;
    let mut driver = factory.create("rust").unwrap();

    assert_eq!(driver.language(), "rust");

    // MinimalDriver starts already parsed (is_parsed returns true)
    driver.parse("some content");
    assert!(driver.is_parsed());

    let highlights = driver.highlights(0..10);
    assert_eq!(highlights.len(), 1);
}

/// Factory that overrides the default `supports` method.
struct CustomSupportFactory;

impl SyntaxDriverFactory for CustomSupportFactory {
    fn create(&self, language_id: &str) -> Option<Box<dyn SyntaxDriver>> {
        if language_id == "custom" {
            Some(Box::new(MinimalDriver { language: "custom" }))
        } else {
            None
        }
    }

    fn supported_languages(&self) -> Vec<&str> {
        vec!["custom"]
    }

    // Uses default supports() implementation
}

#[test]
fn test_factory_default_supports_implementation() {
    let factory = CustomSupportFactory;

    // Default supports() checks supported_languages()
    assert!(factory.supports("custom"));
    assert!(!factory.supports("other"));
}

#[test]
fn test_factory_created_driver_update_and_highlights() {
    let factory = TestFactory;
    let mut driver = factory.create("python").unwrap();

    assert_eq!(driver.language(), "python");
    driver.parse("x = 1");

    let edit = SyntaxEdit {
        start_byte: 0,
        old_end_byte: 1,
        new_end_byte: 2,
        start_row: 0,
        start_col: 0,
        old_end_row: 0,
        old_end_col: 1,
        new_end_row: 0,
        new_end_col: 2,
    };
    driver.update("xx = 1", &edit);

    // MinimalDriver always returns a highlight
    let highlights = driver.highlights(0..100);
    assert_eq!(highlights.len(), 1);
    assert_eq!(highlights[0].category.as_str(), "comment");
}

#[test]
fn test_custom_factory_create_supported() {
    let factory = CustomSupportFactory;
    let driver = factory.create("custom");
    assert!(driver.is_some());
    assert_eq!(driver.unwrap().language(), "custom");
}

#[test]
fn test_custom_factory_create_unsupported() {
    let factory = CustomSupportFactory;
    let driver = factory.create("unknown");
    assert!(driver.is_none());
}

#[test]
fn test_custom_factory_supported_languages() {
    let factory = CustomSupportFactory;
    let langs = factory.supported_languages();
    assert_eq!(langs.len(), 1);
    assert!(langs.contains(&"custom"));
}

#[test]
fn test_factory_send_sync() {
    fn assert_send_sync<T: Send + Sync>() {}
    assert_send_sync::<TestFactory>();
    assert_send_sync::<CustomSupportFactory>();
}