1 files changed, 105 insertions, 6 deletions

diff --git a/support/texlab/crates/rename/src/tests.rs b/support/texlab/crates/rename/src/tests.rs
index 0f4263033e..4da866fb08 100644
--- a/support/texlab/crates/rename/src/tests.rs
+++ b/support/texlab/crates/rename/src/tests.rs
@@ -1,23 +1,37 @@
 use rustc_hash::FxHashMap;
 
-use crate::RenameParams;
+use base_db::Config;
+use parser::SyntaxConfig;
 
-fn check(input: &str) {
-    let fixture = test_utils::fixture::Fixture::parse(input);
+use crate::{RenameInformation, RenameParams};
+
+fn check_with_syntax_config(config: SyntaxConfig, input: &str) {
+    let mut fixture = test_utils::fixture::Fixture::parse(input);
+    fixture.workspace.set_config(Config {
+        syntax: config,
+        ..Config::default()
+    });
+    let fixture = fixture;
 
-    let mut expected = FxHashMap::default();
+    let mut expected: FxHashMap<_, Vec<RenameInformation>> = FxHashMap::default();
     for spec in &fixture.documents {
         if !spec.ranges.is_empty() {
             let document = fixture.workspace.lookup(&spec.uri).unwrap();
-            expected.insert(document, spec.ranges.clone());
+            expected.insert(
+                document,
+                spec.ranges.iter().map(|r| r.clone().into()).collect(),
+            );
         }
     }
-
     let (feature, offset) = fixture.make_params().unwrap();
     let actual = crate::rename(RenameParams { feature, offset });
     assert_eq!(actual.changes, expected);
 }
 
+fn check(input: &str) {
+    check_with_syntax_config(SyntaxConfig::default(), input)
+}
+
 #[test]
 fn test_command() {
     check(
@@ -78,12 +92,97 @@ fn test_label() {
        |
        ^^^
 
+%! baz.tex
+\ref{foo}
+
 %! bar.tex
 \ref{foo}
      ^^^
+"#,
+    )
+}
+
+#[test]
+fn test_custom_label_ref() {
+    let mut config = SyntaxConfig::default();
+    config
+        .label_definition_commands
+        .extend(vec!["asm", "goal"].into_iter().map(String::from));
+    config.label_definition_prefixes.extend(
+        vec![("asm", "asm:"), ("goal", "goal:")]
+            .into_iter()
+            .map(|(x, y)| (String::from(x), String::from(y))),
+    );
+    config
+        .label_reference_commands
+        .extend(vec!["asmref", "goalref"].into_iter().map(String::from));
+    config.label_reference_prefixes.extend(
+        vec![("asmref", "asm:"), ("goalref", "goal:")]
+            .into_iter()
+            .map(|(x, y)| (String::from(x), String::from(y))),
+    );
+    check_with_syntax_config(
+        config,
+        r#"
+%! foo.tex
+\goal{foo}
+
+\asm{foo}\include{bar}\include{baz}
+     |
+     ^^^
+
+%! bar.tex
+\asmref{foo}
+        ^^^
+
+%! baz.tex
+\ref{foo}
+
+\ref{asm:foo}
+     ^^^^^^^
+
+"#,
+    )
+}
+
+#[test]
+fn test_custom_label_def() {
+    let mut config = SyntaxConfig::default();
+    config
+        .label_definition_commands
+        .extend(vec!["asm", "goal"].into_iter().map(String::from));
+    config.label_definition_prefixes.extend(
+        vec![("asm", "asm:"), ("goal", "goal:")]
+            .into_iter()
+            .map(|(x, y)| (String::from(x), String::from(y))),
+    );
+    config
+        .label_reference_commands
+        .extend(vec!["asmref", "goalref"].into_iter().map(String::from));
+    config.label_reference_prefixes.extend(
+        vec![("asmref", "asm:"), ("goalref", "goal:")]
+            .into_iter()
+            .map(|(x, y)| (String::from(x), String::from(y))),
+    );
+    check_with_syntax_config(
+        config,
+        r#"
+%! foo.tex
+\goal{foo}
+
+\label{asm:foo}\include{bar}\include{baz}
+       |
+       ^^^^^^^
+
+%! bar.tex
+\asmref{foo}
+        ^^^
 
 %! baz.tex
 \ref{foo}
+
+\ref{asm:foo}
+     ^^^^^^^
 "#,
     )
 }