Adapt force mock listener for plugin

Author: sofurihafe

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/CgMethodTestSet.kt

@@ -8,6 +8,7 @@ import org.utbot.framework.plugin.api.UtExecution
 import org.utbot.framework.plugin.api.UtExecutionFailure
 import org.utbot.framework.plugin.api.UtExecutionSuccess
 import org.utbot.framework.plugin.api.UtMethodTestSet
+import org.utbot.framework.plugin.api.UtSymbolicExecution
 import org.utbot.framework.plugin.api.util.executableId
 import org.utbot.framework.plugin.api.util.objectClassId
 import org.utbot.framework.plugin.api.util.voidClassId
@@ -66,6 +67,12 @@ data class CgMethodTestSet private constructor(
         return executionsByStaticsUsage.map { (_, executions) -> substituteExecutions(executions) }
     }
 
+    fun extractSymbolicExecutions(): CgMethodTestSet {
+        val executionsBySource = executions.filterIsInstance<UtSymbolicExecution>()
+
+        return substituteExecutions(executionsBySource)
+    }
+
     /**
      * Finds a [ClassId] of all result models in executions.
      *

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/model/constructor/tree/CgTestClassConstructor.kt

@@ -147,14 +147,20 @@ internal class CgTestClassConstructor(val context: CgContext) :
                 }
             }
             ParametrizedTestSource.PARAMETRIZE -> {
-                for (splitByExecutionTestSet in testSet.splitExecutionsByResult()) {
-                    for (splitByChangedStaticsTestSet in splitByExecutionTestSet.splitExecutionsByChangedStatics()) {
-                        createParametrizedTestAndDataProvider(
-                            splitByChangedStaticsTestSet,
-                            requiredFields,
-                            regions,
-                            methodUnderTest
-                        )
+                // We filter out Fuzzer executions because sometimes we need to turn off
+                // Symbolic Engine and stop the test generation in case of mocking taking place
+                // and do not collect this execution. But having Fuzzer executions may cause
+                // unwanted executions being present.
+                testSet.extractSymbolicExecutions().also {
+                    for (splitByExecutionTestSet in it.splitExecutionsByResult()) {
+                        for (splitByChangedStaticsTestSet in splitByExecutionTestSet.splitExecutionsByChangedStatics()) {
+                            createParametrizedTestAndDataProvider(
+                                splitByChangedStaticsTestSet,
+                                requiredFields,
+                                regions,
+                                methodUnderTest
+                            )
+                        }
                     }
                 }
             }

utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/TestCaseGenerator.kt

@@ -159,6 +159,7 @@ open class TestCaseGenerator(
                         )
 
                         engineActions.map { engine.apply(it) }
+                        engineActions.clear()
 
                         generate(engine)
                             .catch {

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt

@@ -54,6 +54,7 @@ import java.nio.file.Paths
 import java.util.concurrent.TimeUnit
 import org.utbot.engine.util.mockListeners.ForceStaticMockListener
 import org.utbot.framework.PathSelectorType
+import org.utbot.framework.codegen.ParametrizedTestSource
 import org.utbot.framework.plugin.services.WorkingDirService
 import org.utbot.intellij.plugin.models.packageName
 import org.utbot.intellij.plugin.settings.Settings
@@ -206,13 +207,32 @@ object UtTestsDialogProcessor {
 
                                 withStaticsSubstitutionRequired(true) {
                                     val mockFrameworkInstalled = model.mockFramework?.isInstalled ?: true
+                                    var forceMockListener: ForceMockListener? = null
+                                    var forceStaticMockListener: ForceStaticMockListener? = null
+
+                                    when (model.parametrizedTestSource) {
+                                        ParametrizedTestSource.PARAMETRIZE -> {
+                                            forceMockListener = ForceMockListener.create(testCaseGenerator, model.conflictTriggers)
+                                            forceStaticMockListener = ForceStaticMockListener.create(
+                                                testCaseGenerator,
+                                                model.conflictTriggers
+                                            )
+                                        }
 
-                                    if (!mockFrameworkInstalled) {
-                                        ForceMockListener.create(testCaseGenerator, model.conflictTriggers)
-                                    }
+                                        ParametrizedTestSource.DO_NOT_PARAMETRIZE -> {
+                                            when {
+                                                !mockFrameworkInstalled -> forceMockListener = ForceMockListener.create(
+                                                    testCaseGenerator,
+                                                    model.conflictTriggers
+                                                )
 
-                                    if (!model.staticsMocking.isConfigured) {
-                                        ForceStaticMockListener.create(testCaseGenerator, model.conflictTriggers)
+                                                !model.staticsMocking.isConfigured -> forceStaticMockListener =
+                                                    ForceStaticMockListener.create(
+                                                        testCaseGenerator,
+                                                        model.conflictTriggers
+                                                    )
+                                            }
+                                        }
                                     }
 
                                     val notEmptyCases = runCatching {
@@ -235,6 +255,9 @@ object UtTestsDialogProcessor {
                                         }
                                     }.getOrDefault(listOf())
 
+                                    forceMockListener?.detach(testCaseGenerator, forceMockListener)
+                                    forceStaticMockListener?.detach(testCaseGenerator, forceStaticMockListener)
+
                                     if (notEmptyCases.isEmpty()) {
                                         if (model.srcClasses.size > 1) {
                                             logger.error { "Failed to generate any tests cases for class $className" }