docs: update test example

unit-mesh · Dec 29, 2023 · 90a5c91 · 90a5c91
1 parent 1e411a7
commit 90a5c91
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unit-picker/src/test/kotlin/cc/unitmesh/pick/SingleProjectCodePickerTest.kt b/unit-picker/src/test/kotlin/cc/unitmesh/pick/SingleProjectCodePickerTest.kt
@@ -39,9 +39,9 @@ class SingleProjectCodePickerTest {
                 language = "kotlin",
                 url = root,
                 maxTokenLength = 8192,
-                buildPlan = listOf(BuildPlanType.COMMENT),
+                buildPlan = listOf(BuildPlanType.COMMENT, BuildPlanType.RELATED_CODE),
                 completionTypes = listOf(
-                    CompletionBuilderType.DOCUMENTATION
+                    CompletionBuilderType.DOCUMENTATION, CompletionBuilderType.TEST_CODE_GEN
                 ),
             )
         )