ts_query_ls/tests/lint.rs

#[cfg(test)]
mod test {
    use rstest::rstest;
    use std::{
        collections::{BTreeMap, HashMap},
        path::Path,
        process::Command,
        sync::LazyLock,
    };
    use ts_query_ls::{Options, Predicate, PredicateParameter};

    static CONFIG: LazyLock<Options> = LazyLock::new(|| Options {
        parser_install_directories: Vec::new(),
        language_retrieval_patterns: Vec::new(),
        valid_directives: BTreeMap::new(),
        parser_aliases: BTreeMap::new(),
        valid_predicates: BTreeMap::from([(
            String::from("pred-name"),
            Predicate {
                description: String::from("A predicate"),
                parameters: vec![PredicateParameter {
                    description: None,
                    type_: ts_query_ls::PredicateParameterType::Any,
                    arity: ts_query_ls::PredicateParameterArity::Variadic,
                }],
            },
        )]),
        valid_captures: HashMap::from([
            (
                String::from("after_trailing_whitespace"),
                BTreeMap::from([(String::from("capture"), String::from("A capture."))]),
            ),
            (
                String::from("before_predicates"),
                BTreeMap::from([(String::from("type"), String::from("A type."))]),
            ),
        ]),
    });

    #[rstest]
    #[case(
        concat!(env!("CARGO_MANIFEST_DIR"), "/queries/formatting_test_files/after_trailing_whitespace.scm"),
        Some(["Unsupported capture name \"@cap\", consider prefixing with '_'"].as_slice())
    )]
    #[case(
        concat!(env!("CARGO_MANIFEST_DIR"), "/queries/formatting_test_files/before_predicates.scm"),
        Some(["Unrecognized predicate \"lua-match\""].as_slice())
    )]
    #[case(
        concat!(env!("CARGO_MANIFEST_DIR"), "/queries/formatting_test_files/before_missing.scm"),
        None
    )]
    #[case(
        concat!(env!("CARGO_MANIFEST_DIR"), "/queries/formatting_test_files/before_syntax_error.scm"),
        Some(["Invalid syntax"].as_slice())
    )]
    fn cli_lint(#[case] path_str: &str, #[case] warning_messages: Option<&[&str]>) {
        // Arrange
        let path = Path::new(path_str);

        // Act
        let output = Command::new(env!("CARGO_BIN_EXE_ts_query_ls"))
            .arg("lint")
            .arg(path)
            .arg("--config")
            .arg(serde_json::to_string::<Options>(&CONFIG).unwrap())
            .output()
            .expect("Failed to wait on ts-query-ls format command");

        // Assert
        if let Some(messages) = warning_messages {
            let string_output = String::from_utf8(output.stderr).unwrap();
            for message in messages {
                assert!(string_output.contains(message));
            }
        } else {
            let string_output = String::from_utf8(output.stderr.clone()).unwrap();
            println!("{string_output}");
            assert!(output.stderr.is_empty());
        }
    }
}