add ut

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

Hide file tree

Showing 7 changed files with 31 additions and 0 deletions.
diff --git a/kclvm/tools/src/LSP/src/completion.rs b/kclvm/tools/src/LSP/src/completion.rs
@@ -2298,4 +2298,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)"
+---
+["HelloWorldSchema{}(import helloworld)", "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,8 @@
+[package]
+name = "unimport"
+edition = "v0.9.0"
+version = "0.0.1"
+
+[dependencies]
+helloworld = "0.1.4"
+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