fix: fix empty issue

unit-mesh · Dec 24, 2023 · c8706eb · c8706eb
1 parent fef5b88
commit c8706eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/ClassTestCodeBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/ClassTestCodeBuilder.kt
@@ -21,9 +21,9 @@ class ClassTestCodeBuilder(private val context: JobContext) : TestCodeBuilder {
         underTestFile: CodeDataStruct,
         relevantClasses: List<CodeDataStruct>,
     ): List<BasicTestIns> {
-        var generatedCode = dataStruct.toSourceCode()
+        val generatedCode = dataStruct.toSourceCode()
         if (generatedCode.lines().size > context.insQualityThreshold.maxLineInCode) {
-            generatedCode = generatedCode.lines().take(context.insQualityThreshold.maxLineInCode).joinToString("\n")
+            return emptyList()
         }
 
         val namingStyle = dataStruct.checkNamingStyle()