fix: fix tets

unit-mesh · Dec 25, 2023 · ae7b669 · ae7b669
1 parent 9bddd2b
commit ae7b669
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...-picker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeServiceTest.kt b/...-picker/src/test/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeServiceTest.kt
@@ -174,7 +174,7 @@ public class CompanyService {
                 "com.rest.springbootemployee.service.CompanyServiceTest" to testFileJob,
             ),
             insOutputConfig = InsOutputConfig(),
-            completionBuilderTypes = listOf(CompletionBuilderType.IN_BLOCK_COMPLETION),
+            completionBuilderTypes = listOf(CompletionBuilderType.TEST_CODE_GEN),
             maxCompletionInOneFile = 3,
             insQualityThreshold = InsQualityThreshold()
         )