Fix execution import handler

tamarinvs19 · tamarinvs19 · commit 78aca4c1458e · 2022-09-22T11:32:31.000+03:00
diff --git a/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt b/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationProcessor.kt
@@ -144,15 +144,18 @@ object PythonTestGenerationProcessor {
                 }
             }
             val importResultModules = notEmptyTests.flatMap { testSet ->
-                testSet.executions.flatMap { execution ->
-                    (execution.result as UtExecutionSuccess).let { result ->
-                        (result.model as PythonModel).let {
-                            it.allContainingClassIds.map { classId ->
-                                PythonUserImport(classId.moduleName)
+                testSet.executions.mapNotNull { execution ->
+                    if (execution.result is UtExecutionSuccess) {
+                        (execution.result as UtExecutionSuccess).let { result ->
+                            (result.model as PythonModel).let {
+                                it.allContainingClassIds.map { classId ->
+                                    PythonUserImport(classId.moduleName)
+                                }
                             }
                         }
                     }
-                }
+                    else null
+                }.flatten()
             }
             val testRootModules = notEmptyTests.mapNotNull { testSet ->
                 methodIds[testSet.method]?.rootModuleName?.let { PythonUserImport(it) }

Original file line number	Diff line number	Diff line change
`@@ -144,15 +144,18 @@ object PythonTestGenerationProcessor {`
`144`	`144`	`}`
`145`	`145`	`}`
`146`	`146`	`val importResultModules = notEmptyTests.flatMap { testSet ->`
`147`		`- testSet.executions.flatMap { execution ->`
`148`		`- (execution.result as UtExecutionSuccess).let { result ->`
`149`		`- (result.model as PythonModel).let {`
`150`		`- it.allContainingClassIds.map { classId ->`
`151`		`- PythonUserImport(classId.moduleName)`
	`147`	`+ testSet.executions.mapNotNull { execution ->`
	`148`	`+ if (execution.result is UtExecutionSuccess) {`
	`149`	`+ (execution.result as UtExecutionSuccess).let { result ->`
	`150`	`+ (result.model as PythonModel).let {`
	`151`	`+ it.allContainingClassIds.map { classId ->`
	`152`	`+ PythonUserImport(classId.moduleName)`
	`153`	`+ }`
`152`	`154`	`}`
`153`	`155`	`}`
`154`	`156`	`}`
`155`		`- }`
	`157`	`+ else null`
	`158`	`+ }.flatten()`
`156`	`159`	`}`
`157`	`160`	`val testRootModules = notEmptyTests.mapNotNull { testSet ->`
`158`	`161`	`methodIds[testSet.method]?.rootModuleName?.let { PythonUserImport(it) }`