added test for CompileError quick fix

Signed-off-by: Shashank Mittal <[email protected]>

kclvm/driver/src/test_data/test_vendor/.kpm/config/kpm.json

@@ -0,0 +1 @@
+{"DefaultOciRegistry":"ghcr.io","DefaultOciRepo":"kcl-lang","DefaultOciPlainHttp":false}

kclvm/tools/src/LSP/src/test_data/diagnostics.k

@@ -7,3 +7,5 @@ c: Person = Person {
 
 d = 1
 d = 2
+number = 2
+count = nu

kclvm/tools/src/LSP/src/tests.rs

@@ -1,6 +1,7 @@
 use crossbeam_channel::after;
 use crossbeam_channel::select;
 use indexmap::IndexSet;
+use kclvm_error::diagnostic;
 use kclvm_sema::core::global_state::GlobalState;
 use lsp_server::RequestId;
 use lsp_server::Response;
@@ -140,6 +141,7 @@ fn build_lsp_diag(
     severity: Option<DiagnosticSeverity>,
     related_info: Vec<(String, (u32, u32, u32, u32), String)>,
     code: Option<NumberOrString>,
+    data: Option<serde_json::Value>
 ) -> Diagnostic {
     let related_information = if related_info.is_empty() {
         None
@@ -184,7 +186,7 @@ fn build_lsp_diag(
         message,
         related_information,
         tags: None,
-        data: None,
+        data,
     }
 }
 
@@ -201,13 +203,15 @@ fn build_expect_diags() -> Vec<Diagnostic> {
             Some(DiagnosticSeverity::ERROR),
             vec![],
             Some(NumberOrString::String("InvalidSyntax".to_string())),
+            None,
         ),
         build_lsp_diag(
             (0, 0, 0, 10),
             "pkgpath abc not found in the program".to_string(),
             Some(DiagnosticSeverity::ERROR),
             vec![],
             Some(NumberOrString::String("CannotFindModule".to_string())),
+            None,
         ),
         build_lsp_diag(
             (0, 0, 0, 10),
@@ -218,6 +222,7 @@ fn build_expect_diags() -> Vec<Diagnostic> {
             Some(DiagnosticSeverity::ERROR),
             vec![],
             Some(NumberOrString::String("CannotFindModule".to_string())),
+            None,
         ),
         build_lsp_diag(
             (8, 0, 8, 1),
@@ -229,6 +234,7 @@ fn build_expect_diags() -> Vec<Diagnostic> {
                 "The variable 'd' is declared here".to_string(),
             )],
             Some(NumberOrString::String("ImmutableError".to_string())),
+            None,
         ),
         build_lsp_diag(
             (7, 0, 7, 1),
@@ -240,21 +246,33 @@ fn build_expect_diags() -> Vec<Diagnostic> {
                 "Can not change the value of 'd', because it was declared immutable".to_string(),
             )],
             Some(NumberOrString::String("ImmutableError".to_string())),
+            None,
         ),
         build_lsp_diag(
             (2, 0, 2, 1),
             "expected str, got int(1)".to_string(),
             Some(DiagnosticSeverity::ERROR),
             vec![],
             Some(NumberOrString::String("TypeError".to_string())),
+            None,
+        ),
+        build_lsp_diag(
+            (10, 8, 10, 10),
+            "name 'nu' is not defined, did you mean '[\"number\", \"n\", \"num\"]'?".to_string(),
+            Some(DiagnosticSeverity::ERROR),
+            vec![],
+            Some(NumberOrString::String("CompileError".to_string())),
+            Some(serde_json::json!({ "suggested_replacement": ["number"] })),
         ),
         build_lsp_diag(
             (0, 0, 0, 10),
             "Module 'abc' imported but unused".to_string(),
             Some(DiagnosticSeverity::WARNING),
             vec![],
             Some(NumberOrString::String("UnusedImportWarning".to_string())),
+            None,
         ),
+
     ];
     expected_diags
 }

kclvm/tools/src/LSP/src/to_lsp.rs

@@ -47,6 +47,7 @@ fn kcl_msg_to_lsp_diags(
     let data = msg
         .suggested_replacement
         .as_ref()
+        .filter(|s| !s.is_empty())
         .map(|s| json!({ "suggested_replacement": [s] }));
 
     let related_information = if related_msg.is_empty() {