From da9205ebc8ee6e87f423fc92545400e8a5cdee56 Mon Sep 17 00:00:00 2001
From: Riley Bruins <ribru17@hotmail.com>
Date: Mon, 5 May 2025 20:02:23 -0700
Subject: [PATCH] test: coverage for *some* lint warnings

---
 tests/lint.rs | 82 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 tests/lint.rs

diff --git a/tests/lint.rs b/tests/lint.rs
new file mode 100644
index 0000000..eb26337
--- /dev/null
+++ b/tests/lint.rs
@@ -0,0 +1,82 @@
+#[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());
+        }
+    }
+}