UnitTestBot · nikitavlaev · Jul 7, 2022 · Jul 6, 2022
diff --git a/utbot-framework/src/main/kotlin/org/utbot/engine/Mocks.kt b/utbot-framework/src/main/kotlin/org/utbot/engine/Mocks.kt
@@ -168,7 +168,7 @@ class Mocker(
     fun shouldMock(
         type: RefType,
         mockInfo: UtMockInfo,
-    ): Boolean = checkIfShouldMock(type, mockInfo).also { if (it) mockListenerController?.onShouldMock(strategy) }
+    ): Boolean = checkIfShouldMock(type, mockInfo).also { if (it) mockListenerController?.onShouldMock(strategy, mockInfo) }
 
     private fun checkIfShouldMock(
         type: RefType,

diff --git a/utbot-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/Exceptions.kt b/utbot-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/Exceptions.kt
@@ -0,0 +1,13 @@
+package org.utbot.engine.util.mockListeners
+
+import kotlinx.coroutines.CancellationException
+
+/**
+ * Exception used in [org.utbot.engine.util.mockListeners.ForceMockListener].
+ */
+class ForceMockCancellationException: CancellationException("Forced mocks without Mockito")
+
+/**
+ * Exception used in [org.utbot.engine.util.mockListeners.ForceStaticMockListener].
+ */
+class ForceStaticMockCancellationException: CancellationException("Forced static mocks without Mockito-inline")
diff --git a/utbot-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/ForceMockListener.kt b/utbot-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/ForceMockListener.kt
@@ -1,7 +1,7 @@
 package org.utbot.engine.util.mockListeners
 import org.utbot.engine.EngineController
 import org.utbot.engine.MockStrategy
-import org.utbot.engine.util.mockListeners.exceptions.ForceMockCancellationException
+import org.utbot.engine.UtMockInfo
 
 /**
  * Listener for mocker events in [org.utbot.engine.UtBotSymbolicEngine].
@@ -13,7 +13,7 @@ class ForceMockListener: MockListener {
     var forceMockHappened = false
         private set
 
-    override fun onShouldMock(controller: EngineController, strategy: MockStrategy) {
+    override fun onShouldMock(controller: EngineController, strategy: MockStrategy, mockInfo: UtMockInfo) {
         // If force mocking happened -- сancel engine job
         controller.job?.cancel(ForceMockCancellationException())
         forceMockHappened = true

diff --git a/...-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/ForceStaticMockListener.kt b/...-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/ForceStaticMockListener.kt
@@ -0,0 +1,29 @@
+package org.utbot.engine.util.mockListeners
+
+import org.utbot.engine.EngineController
+import org.utbot.engine.MockStrategy
+import org.utbot.engine.UtMockInfo
+import org.utbot.engine.UtNewInstanceMockInfo
+import org.utbot.engine.UtStaticMethodMockInfo
+import org.utbot.engine.UtStaticObjectMockInfo
+
+/**
+ * Listener for mocker events in [org.utbot.engine.UtBotSymbolicEngine].
+ * If forced static mock happened, cancels the engine job.
+ *
+ * Supposed to be created only if Mockito inline is not installed.
+ */
+class ForceStaticMockListener: MockListener {
+    var forceStaticMockHappened = false
+        private set
+
+    override fun onShouldMock(controller: EngineController, strategy: MockStrategy, mockInfo: UtMockInfo) {
+        if (mockInfo is UtNewInstanceMockInfo
+            || mockInfo is UtStaticMethodMockInfo
+            || mockInfo is UtStaticObjectMockInfo) {
+            // If force static mocking happened -- сancel engine job
+            controller.job?.cancel(ForceStaticMockCancellationException())
+            forceStaticMockHappened = true
+        }
+    }
+}
diff --git a/utbot-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/MockListener.kt b/utbot-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/MockListener.kt
@@ -2,10 +2,11 @@ package org.utbot.engine.util.mockListeners
 
 import org.utbot.engine.EngineController
 import org.utbot.engine.MockStrategy
+import org.utbot.engine.UtMockInfo
 
 /**
  * Listener that can be attached using [MockListenerController] to mocker in [org.utbot.engine.UtBotSymbolicEngine].
  */
 interface MockListener {
-    fun onShouldMock(controller: EngineController, strategy: MockStrategy)
+    fun onShouldMock(controller: EngineController, strategy: MockStrategy, mockInfo: UtMockInfo)
 }
diff --git a/...t-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/MockListenerController.kt b/...t-framework/src/main/kotlin/org/utbot/engine/util/mockListeners/MockListenerController.kt
@@ -2,6 +2,7 @@ package org.utbot.engine.util.mockListeners
 
 import org.utbot.engine.EngineController
 import org.utbot.engine.MockStrategy
+import org.utbot.engine.UtMockInfo
 
 /**
  * Controller that allows to attach listeners to mocker in [org.utbot.engine.UtBotSymbolicEngine].
@@ -13,7 +14,7 @@ class MockListenerController(private val controller: EngineController) {
         listeners += listener
     }
 
-    fun onShouldMock(strategy: MockStrategy) {
-        listeners.map { it.onShouldMock(controller, strategy) }
+    fun onShouldMock(strategy: MockStrategy, mockInfo: UtMockInfo) {
+        listeners.map { it.onShouldMock(controller, strategy, mockInfo) }
     }
 }
diff --git a/...n/kotlin/org/utbot/engine/util/mockListeners/exceptions/ForceMockCancellationException.kt b/...n/kotlin/org/utbot/engine/util/mockListeners/exceptions/ForceMockCancellationException.kt
diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/UtBotTestCaseGenerator.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/plugin/api/UtBotTestCaseGenerator.kt
@@ -58,7 +58,7 @@ object UtBotTestCaseGenerator : TestCaseGenerator {
     private val logger = KotlinLogging.logger {}
     private val timeoutLogger = KotlinLogging.logger(logger.name + ".timeout")
 
-    lateinit var configureEngine: (UtBotSymbolicEngine) -> Unit
+    lateinit var engineActions: MutableList<(UtBotSymbolicEngine) -> Unit>
     lateinit var isCanceled: () -> Boolean
 
     //properties to save time on soot initialization
@@ -76,19 +76,19 @@ object UtBotTestCaseGenerator : TestCaseGenerator {
         buildDir,
         classpath,
         dependencyPaths,
-        configureEngine = {},
+        engineActions = mutableListOf(),
         isCanceled
     )
 
     fun init(
         buildDir: Path,
         classpath: String?,
         dependencyPaths: String,
-        configureEngine: (UtBotSymbolicEngine) -> Unit,
+        engineActions: MutableList<(UtBotSymbolicEngine) -> Unit>,
         isCanceled: () -> Boolean
     ) {
         this.isCanceled = isCanceled
-        this.configureEngine = configureEngine
+        this.engineActions = engineActions
         if (isCanceled()) return
 
         checkFrameworkDependencies(dependencyPaths)
@@ -293,7 +293,9 @@ object UtBotTestCaseGenerator : TestCaseGenerator {
                             mockStrategy,
                             chosenClassesToMockAlways,
                             executionTimeEstimator
-                        ).apply(configureEngine)
+                        )
+
+                        engineActions.map { engine.apply(it) }
 
                         generate(engine).collect {
                             when (it) {

diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerator.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerator.kt
@@ -33,15 +33,15 @@ class CodeGenerator(
     buildDir: String,
     classpath: String,
     pluginJarsPath: String,
-    configureEngine: (UtBotSymbolicEngine) -> Unit = {},
+    engineActions: MutableList<(UtBotSymbolicEngine) -> Unit> = mutableListOf(),
     isCanceled: () -> Boolean,
 ) {
     init {
         UtSettings.testMinimizationStrategyType = TestSelectionStrategyType.COVERAGE_STRATEGY
     }
 
     val generator = (project.service<Settings>().testCasesGenerator as UtBotTestCaseGenerator).apply {
-        init(Paths.get(buildDir), classpath, pluginJarsPath, configureEngine, isCanceled)
+        init(Paths.get(buildDir), classpath, pluginJarsPath, engineActions, isCanceled)
     }
 
     private val settingsState = project.service<Settings>().state

diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/TestGenerator.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/TestGenerator.kt
@@ -405,11 +405,11 @@ object TestGenerator {
             """.trimIndent()
             appendHtmlLine(savedFileMessage)
         }
-        TestsReportNotifier.notify(notifyMessage, model.project, model.testModule)
+        TestsReportNotifier.notify(notifyMessage)
     }
 
     private fun processInitialWarnings(testsCodeWithTestReport: TestsCodeWithTestReport, model: GenerateTestsModel) {
-        val hasInitialWarnings = model.forceMockHappened || model.hasTestFrameworkConflict
+        val hasInitialWarnings = model.forceMockHappened || model.forceStaticMockHappened || model.hasTestFrameworkConflict
         if (!hasInitialWarnings) {
             return
         }
@@ -425,6 +425,14 @@ object TestGenerator {
                 """.trimIndent()
                 }
             }
+            if (model.forceStaticMockHappened) {
+                initialWarnings.add {
+                    """
+                    <b>Warning</b>: Some test cases were ignored, because mockito-inline is not installed in the project.<br>
+                    Better results could be achieved by <a href="${TestReportUrlOpeningListener.prefix}${TestReportUrlOpeningListener.mockitoInlineSuffix}">configuring mockito-inline</a>.
+                """.trimIndent()
+                }
+            }
             if (model.hasTestFrameworkConflict) {
                 initialWarnings.add {
                     """

diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/ConfigureWindowCommon.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/ConfigureWindowCommon.kt