Fix python coverage database bug with access to database directory (#1553)

Author: tamarinvs19

utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonDialogProcessor.kt

@@ -233,7 +233,10 @@ object PythonDialogProcessor {
                     if (installResult.exitValue != 0) {
                         showErrorDialogLater(
                             project,
-                            "Requirements installing failed",
+                            "Requirements installing failed.<br>" +
+                                    "${installResult.stderr}<br><br>" +
+                                    "Try to install with pip:<br>" +
+                                    " ${requirements.joinToString("<br>")}",
                             "Requirements error"
                         )
                     }

utbot-python/src/main/kotlin/org/utbot/python/PythonEngine.kt

@@ -335,13 +335,16 @@ class PythonEngine(
             when (val evaluationResult = jobResult.evalResult) {
                 is PythonEvaluationError -> {
                     if (evaluationResult.status != 0) {
+                        val errorMessage = "Error evaluation: ${evaluationResult.status}, ${evaluationResult.message}, ${evaluationResult.stackTrace}"
+                        logger.info { errorMessage }
                         emit(
                             UtError(
-                                "Error evaluation: ${evaluationResult.status}, ${evaluationResult.message}",
+                                errorMessage,
                                 Throwable(evaluationResult.stackTrace.joinToString("\n"))
                             )
                         )
                     } else {
+                        logger.info { "Python evaluation error: ${evaluationResult.message}" }
                         emit(
                             UtError(
                                 evaluationResult.message,

utbot-python/src/main/kotlin/org/utbot/python/PythonEvaluation.kt

@@ -15,7 +15,6 @@ import org.utbot.python.utils.getResult
 import org.utbot.python.utils.startProcess
 import java.io.File
 
-
 sealed class PythonEvaluationResult
 
 class PythonEvaluationError(
@@ -64,13 +63,18 @@ fun startEvaluationProcess(input: EvaluationInput): EvaluationProcess {
         tag = "out_" + input.method.name + ".py",
         addToCleaner = false
     )
+    val coverageDatabasePath = TemporaryFileManager.assignTemporaryFile(
+        tag = "coverage_db_" + input.method.name,
+        addToCleaner = false,
+    )
     val runCode = PythonCodeGenerator.generateRunFunctionCode(
         input.method,
         input.methodArguments,
         input.directoriesForSysPath,
         input.moduleToImport,
         input.additionalModulesToImport,
-        fileForOutput.path.replace("\\", "\\\\")
+        fileForOutput.path.replace("\\", "\\\\"),
+        coverageDatabasePath.absolutePath.replace("\\", "\\\\")
     )
     val fileWithCode = TemporaryFileManager.createTemporaryFile(
         runCode,

utbot-python/src/main/kotlin/org/utbot/python/code/CodeGen.kt

@@ -20,7 +20,8 @@ object PythonCodeGenerator {
         directoriesForSysPath: Set<String>,
         moduleToImport: String,
         additionalModules: Set<String> = emptySet(),
-        fileForOutputName: String
+        fileForOutputName: String,
+        coverageDatabasePath: String,
     ): String {
         val context = UtContext(this::class.java.classLoader)
         withUtContext(context) {
@@ -36,7 +37,8 @@ object PythonCodeGenerator {
                 directoriesForSysPath,
                 moduleToImport,
                 additionalModules,
-                fileForOutputName
+                fileForOutputName,
+                coverageDatabasePath,
             )
         }
     }

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/PythonCodeGenerator.kt

@@ -108,7 +108,8 @@ class PythonCodeGenerator(
         directoriesForSysPath: Set<String>,
         moduleToImport: String,
         additionalModules: Set<String> = emptySet(),
-        fileForOutputName: String
+        fileForOutputName: String,
+        coverageDatabasePath: String,
     ): String = withCustomContext(testClassCustomName = null) {
         context.withTestClassFileScope {
             clearContextRelatedStorage()
@@ -160,21 +161,22 @@ class PythonCodeGenerator(
                 arguments.associateBy { argument -> CgLiteral(pythonStrClassId, "'${argument.name}'") }
             )
 
-            val fullpath = CgLiteral(pythonStrClassId, "'${method.moduleFilename}'")
-
-            val outputPath = CgLiteral(pythonStrClassId, "'$fileForOutputName'")
-
-            val executorCall = CgPythonFunctionCall(
-                pythonNoneClassId,
-                executorFunctionName,
-                listOf(
-                    functionName,
-                    args,
-                    kwargs,
-                    fullpath,
-                    outputPath,
-                )
+        val fullpath = CgLiteral(pythonStrClassId, "'${method.moduleFilename.replace("\\", "\\\\")}'")
+        val outputPath = CgLiteral(pythonStrClassId, "'$fileForOutputName'")
+        val databasePath = CgLiteral(pythonStrClassId, "'$coverageDatabasePath'")
+
+        val executorCall = CgPythonFunctionCall(
+            pythonNoneClassId,
+            executorFunctionName,
+            listOf(
+                databasePath,
+                functionName,
+                args,
+                kwargs,
+                fullpath,
+                outputPath,
             )
+        )
 
             parameters.forEach { it.accept(renderer) }
             executorCall.accept(renderer)

utbot-python/src/main/kotlin/org/utbot/python/utils/TemporaryFileManager.kt

@@ -4,15 +4,14 @@ import org.utbot.common.FileUtil
 import java.io.File
 import java.nio.file.Path
 import java.nio.file.Paths
-import kotlin.io.path.deleteExisting
 
 object TemporaryFileManager {
     private lateinit var tmpDirectory: Path
     private var nextId = 0
 
     fun setup() {
         tmpDirectory = FileUtil.createTempDirectory("python-test-generation-${nextId++}")
-        Cleaner.addFunction { tmpDirectory.deleteExisting() }
+        Cleaner.addFunction { tmpDirectory.toFile().deleteRecursively() }
     }
 
     fun assignTemporaryFile(fileName_: String? = null, tag: String? = null, addToCleaner: Boolean = true): File {

utbot-python/src/main/resources/requirements.txt

@@ -2,5 +2,5 @@ mypy==0.971
 astor
 typeshed-client
 coverage
-utbot-executor==0.1.4
+utbot-executor==0.1.7
 utbot-mypy-runner==0.1.10