From 04bb4f3fcb83f40d8fa51e9458b5e47ead486453 Mon Sep 17 00:00:00 2001
From: he1pa <18012015693@163.com>
Date: Thu, 21 Nov 2024 14:56:41 +0800
Subject: [PATCH] add ut

Signed-off-by: he1pa <18012015693@163.com>
---
 kclvm/tools/src/LSP/src/completion.rs                     | 8 ++++++++
 ...ver__completion__tests__complete_unimport_schemas.snap | 5 +++++
 .../src/test_data/completion_test/unimport/pkg/kcl.mod    | 6 ++++++
 .../LSP/src/test_data/completion_test/unimport/pkg/main.k | 2 ++
 .../test_data/completion_test/unimport/unimport/kcl.mod   | 8 ++++++++
 .../test_data/completion_test/unimport/unimport/main.k    | 0
 .../completion_test/unimport/unimport/subpkg/subpkg.k     | 2 ++
 7 files changed, 31 insertions(+)
 create mode 100644 kclvm/tools/src/LSP/src/snapshots/kcl_language_server__completion__tests__complete_unimport_schemas.snap
 create mode 100644 kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/kcl.mod
 create mode 100644 kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/main.k
 create mode 100644 kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/kcl.mod
 create mode 100644 kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/main.k
 create mode 100644 kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/subpkg/subpkg.k

diff --git a/kclvm/tools/src/LSP/src/completion.rs b/kclvm/tools/src/LSP/src/completion.rs
index 9bd572928..19c3ed03c 100644
--- 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/kclvm/tools/src/LSP/src/snapshots/kcl_language_server__completion__tests__complete_unimport_schemas.snap b/kclvm/tools/src/LSP/src/snapshots/kcl_language_server__completion__tests__complete_unimport_schemas.snap
new file mode 100644
index 000000000..55e233391
--- /dev/null
+++ b/kclvm/tools/src/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
new file mode 100644
index 000000000..950182eb1
--- /dev/null
+++ 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
new file mode 100644
index 000000000..ac99847b1
--- /dev/null
+++ b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/pkg/main.k
@@ -0,0 +1,2 @@
+schema Name:
+    name: str
\ No newline at end of file
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
new file mode 100644
index 000000000..21f756fbe
--- /dev/null
+++ 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" }
\ No newline at end of file
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
new file mode 100644
index 000000000..e69de29bb
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
new file mode 100644
index 000000000..9dfbc5485
--- /dev/null
+++ b/kclvm/tools/src/LSP/src/test_data/completion_test/unimport/unimport/subpkg/subpkg.k
@@ -0,0 +1,2 @@
+schema SubPKg:
+    name: str
\ No newline at end of file