fix: fxi test

unit-mesh · Dec 14, 2023 · 4bcde50 · 4bcde50
1 parent 8f2989e
commit 4bcde50
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/unit-picker/src/test/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtilTest.kt b/unit-picker/src/test/kotlin/cc/unitmesh/pick/ext/CodeDataStructUtilTest.kt
@@ -13,8 +13,8 @@ class CodeDataStructUtilTest {
         val codeDataStruct = CodeDataStruct(
             NodeName = "TestClass",
             Fields = listOf(
-                CodeField("Int", "id"),
-                CodeField("String", "name")
+                CodeField("Int", "", "id"),
+                CodeField("String", "", "name")
             ),
             Functions = listOf(
                 CodeFunction("getName", "", ""),

diff --git a/...icker/src/test/kotlin/cc/unitmesh/pick/prompt/builder/RelatedCodeCompletionBuilderTest.kt b/...icker/src/test/kotlin/cc/unitmesh/pick/prompt/builder/RelatedCodeCompletionBuilderTest.kt
@@ -132,10 +132,10 @@ public class HelloController {
 
         assertEquals(
             first.relatedCode, """// class BlogPost {
-//    : Long
-//    : String
-//    : String
-//    : String
+//    id: Long
+//    title: String
+//    content: String
+//    author: String
 // 
 //    'getter/setter: setAuthor
 //