fix: fix more options issues

unit-mesh · Dec 25, 2023 · 77ae49b · 77ae49b
1 parent dd25185
commit 77ae49b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/BasicTestIns.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/BasicTestIns.kt
@@ -4,7 +4,6 @@ import cc.unitmesh.core.Instruction
 import cc.unitmesh.core.SupportedLang
 import cc.unitmesh.core.unittest.TestCodeBuilderType
 import cc.unitmesh.core.unittest.TypedTestIns
-import cc.unitmesh.pick.builder.ins.RelatedCodeIns
 import kotlinx.serialization.Serializable
 import kotlinx.serialization.json.Json
 

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeService.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/lang/JavaTestCodeService.kt
@@ -1,6 +1,8 @@
 package cc.unitmesh.pick.builder.unittest.lang
 
+import cc.unitmesh.core.completion.CompletionBuilderType
 import cc.unitmesh.core.unittest.TypedTestIns
+import cc.unitmesh.pick.builder.testBuilders
 import cc.unitmesh.pick.builder.unittest.ClassTestCodeBuilder
 import cc.unitmesh.pick.builder.unittest.MethodTestCodeBuilder
 import cc.unitmesh.pick.worker.job.InstructionFileJob
@@ -65,6 +67,10 @@ class JavaTestCodeService(val context: JobContext) : UnitTestService {
     }.flatten()
 
     override fun build(dataStruct: CodeDataStruct): List<TypedTestIns> {
+        if (!context.completionBuilderTypes.contains(CompletionBuilderType.TEST_CODE_GEN)) {
+            return emptyList()
+        }
+
         val underTestFile = this.findUnderTestFile(dataStruct).firstOrNull() ?: return emptyList()
         val relevantClasses = this.lookupRelevantClass(dataStruct)
         val classTestIns = ClassTestCodeBuilder(context)