fix: fix content error issue

unit-mesh · Dec 27, 2023 · c31cd93 · c31cd93
1 parent eb0d128
commit c31cd93
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/MethodTestCodeBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/builder/unittest/MethodTestCodeBuilder.kt
@@ -32,9 +32,9 @@ class MethodTestCodeBuilder(private val context: JobContext) : TestCodeBuilder {
             function.FunctionCalls.map {
                 if (it.NodeName == underTestFile.NodeName) {
                     val canonicalName = it.Package + "." + it.NodeName + ":" + it.FunctionName
-                    val originalContent = underTestFunctionMap[canonicalName]
+                    val originalContent = underTestFunctionMap[canonicalName] ?: return@map
 
-                    if (originalContent != null) {
+                    if (originalContent.isNotBlank() && function.Content.isNotBlank()) {
                         val testIns = BasicTestIns(
                             lang = context.project.language,
                             underTestCode = originalContent,