add ut

Signed-off-by: he1pa <18012015693@163.com>
kcl-lang · Nov 21, 2024 · 3eab378 · 3eab378
1 parent 15e2ef0
commit 3eab378
Show file tree

Hide file tree

Showing 7 changed files with 46 additions and 2 deletions.
diff --git a/kclvm/tools/src/LSP/src/completion.rs b/kclvm/tools/src/LSP/src/completion.rs
@@ -1006,8 +1006,22 @@ mod tests {
 
         let mut expected_labels: Vec<String> = vec![
             "", // generate from error recovery of "pkg."
-            "subpkg", "math", "Person", "Person{}", "P", "P{}", "p", "p1", "p2", "p3", "p4",
-            "aaaa", "Config", "Config{}", "n",
+            "subpkg",
+            "math",
+            "Person",
+            "Person1{}",
+            "Person{}",
+            "P",
+            "P{}",
+            "p",
+            "p1",
+            "p2",
+            "p3",
+            "p4",
+            "aaaa",
+            "Config",
+            "Config{}",
+            "n",
         ]
         .iter()
         .map(|s| s.to_string())
@@ -2298,4 +2312,12 @@ mod tests {
         23,
         Some('.')
     );
+
+    completion_label_without_builtin_func_test_snapshot!(
+        complete_unimport_schemas,
+        "src/test_data/completion_test/unimport/unimport/main.k",
+        1,
+        1,
+        None
+    );
 }
diff --git a/.../LSP/src/snapshots/kcl_language_server__completion__tests__complete_unimport_schemas.snap b/.../LSP/src/snapshots/kcl_language_server__completion__tests__complete_unimport_schemas.snap
@@ -0,0 +1,5 @@
+---
+source: tools/src/LSP/src/completion.rs
+expression: "format! (\"{:?}\", got_labels)"
+---
+["Name{}(import pkg)", "SubPKg{}(import subpkg)"]
diff --git a/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/kcl.mod b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/kcl.mod
@@ -0,0 +1,6 @@
+[package]
+name = "pkg"
+edition = "v0.9.0"
+version = "0.0.1"
+
+
diff --git a/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/main.k b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/main.k
@@ -0,0 +1,2 @@
+schema Name:
+    name: str
diff --git a/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/kcl.mod b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/kcl.mod
@@ -0,0 +1,7 @@
+[package]
+name = "unimport"
+edition = "v0.9.0"
+version = "0.0.1"
+
+[dependencies]
+pkg = { path = "../pkg" }
diff --git a/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/main.k b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/main.k
diff --git a/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/subpkg/subpkg.k b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/subpkg/subpkg.k
@@ -0,0 +1,2 @@
+schema SubPKg:
+    name: str