Add usvm engine option on UI

UnitTestBot · EgorkaKulikov · Dec 11, 2023 · Dec 11, 2023 · Dec 11, 2023 · Dec 13, 2023
commit dbeaf15317dd2023c69b81d48e187d5a2196714c
diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/Domain.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/Domain.kt
@@ -691,6 +691,30 @@ enum class ParametrizedTestSource(
     }
 }
 
+enum class SymbolicEngineSource(
+    override val id: String,
+    override val displayName: String,
+    override val description: String = "Use $displayName symbolic engine"
+) : CodeGenerationSettingItem {
+    UnitTestBot(
+        id = "UnitTestBot",
+        displayName = "UnitTestBot",
+        description = "Use UnitTestBot symbolic engine",
+    ),
+    Usvm(
+        id = "USVM",
+        displayName = "USVM",
+        description = "Use USVM symbolic engine",
+    );
+
+    override fun toString(): String = id
+
+    companion object : CodeGenerationSettingBox {
+        override val defaultItem: SymbolicEngineSource = SymbolicEngineSource.UnitTestBot
+        override val allItems: List<SymbolicEngineSource> = SymbolicEngineSource.values().toList()
+    }
+}
+
 enum class ProjectType {
     /**
      * Standard JVM project without DI frameworks

diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/models/GenerateTestsModel.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/models/GenerateTestsModel.kt
@@ -26,6 +26,7 @@ import org.jetbrains.kotlin.psi.KtFile
 import org.utbot.common.PathUtil.fileExtension
 import org.utbot.framework.SummariesGenerationType
 import org.utbot.framework.UtSettings
+import org.utbot.framework.codegen.domain.SymbolicEngineSource
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.MockFramework
 import org.utbot.framework.plugin.api.MockStrategyApi
@@ -77,6 +78,7 @@ class GenerateTestsModel(
     lateinit var mockFramework: MockFramework
     lateinit var staticsMocking: StaticsMocking
     lateinit var parametrizedTestSource: ParametrizedTestSource
+    lateinit var symbolicEngineSource: SymbolicEngineSource
     lateinit var runtimeExceptionTestsBehaviour: RuntimeExceptionTestsBehaviour
     lateinit var hangingTestsTimeout: HangingTestsTimeout
     var useTaintAnalysis: Boolean = false

diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/GenerateTestsDialogWindow.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/GenerateTestsDialogWindow.kt
@@ -88,6 +88,7 @@ import org.utbot.framework.codegen.domain.ParametrizedTestSource
 import org.utbot.framework.codegen.domain.ProjectType
 import org.utbot.framework.codegen.domain.SpringModule.*
 import org.utbot.framework.codegen.domain.StaticsMocking
+import org.utbot.framework.codegen.domain.SymbolicEngineSource
 import org.utbot.framework.codegen.domain.TestFramework
 import org.utbot.framework.codegen.domain.TestNg
 import org.utbot.framework.plugin.api.MockStrategyApi
@@ -215,6 +216,8 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
         }
     private val parametrizedTestSources = JCheckBox("Parameterized tests")
 
+    private val useExperimentalEngine = JCheckBox("Experimental symbolic engine")
+
     private lateinit var panel: DialogPanel
 
     @Suppress("UNCHECKED_CAST")
@@ -412,6 +415,13 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
                 cell(testFrameworks)
             }
 
+            row {
+                cell(useExperimentalEngine)
+                contextHelp("USVM symbolic engine will be used")
+            }.enabledIf(ComboBoxPredicate(springConfig) {
+                model.projectType == ProjectType.PureJvm
+            })
+
             if (model.projectType == ProjectType.Spring) {
                 row("Spring configuration:") {
                     cell(springConfig)
@@ -705,6 +715,8 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
         model.mockStrategy = mockStrategies.item
         model.parametrizedTestSource =
             if (parametrizedTestSources.isSelected) ParametrizedTestSource.PARAMETRIZE else ParametrizedTestSource.DO_NOT_PARAMETRIZE
+        model.symbolicEngineSource =
+            if (useExperimentalEngine.isSelected) SymbolicEngineSource.Usvm else SymbolicEngineSource.UnitTestBot
 
         model.mockFramework = MOCKITO
         model.staticsMocking = if (staticsMocking.isSelected) MockitoStaticMocking else NoStaticMocking
@@ -903,6 +915,9 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
         parametrizedTestSources.isSelected = (settings.parametrizedTestSource == ParametrizedTestSource.PARAMETRIZE
                 && model.projectType == ProjectType.PureJvm)
 
+        useExperimentalEngine.isSelected = (settings.symbolicEngineSource == SymbolicEngineSource.Usvm
+                && model.projectType == ProjectType.PureJvm)
+
         codegenLanguages.item = model.codegenLanguage
 
         val installedTestFramework = TestFramework.allItems.singleOrNull { it.isInstalled }
@@ -1212,6 +1227,18 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
             updateControlsEnabledStatus()
         }
 
+        useExperimentalEngine.addActionListener {_ ->
+            if (useExperimentalEngine.isSelected) {
+                mockStrategies.isEnabled = false
+                mockStrategies.item = MockStrategyApi.NO_MOCKS
+
+                staticsMocking.isEnabled = false
+                staticsMocking.isSelected = false
+            } else {
+                updateControlsEnabledStatus()
+            }
+        }
+
         springTestType.addActionListener { event ->
             val comboBox = event.source as ComboBox<*>
             val item = comboBox.item as SpringTestType
@@ -1382,7 +1409,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
     }
 
     private fun updateControlsEnabledStatus() {
-        mockStrategies.isEnabled = true
+        mockStrategies.isEnabled = !useExperimentalEngine.isSelected
 
         updateParametrizationEnabled()
         updateStaticMockEnabled()

diff --git a/utbot-ui-commons/src/main/kotlin/org/utbot/intellij/plugin/settings/CommonSettings.kt b/utbot-ui-commons/src/main/kotlin/org/utbot/intellij/plugin/settings/CommonSettings.kt
@@ -37,6 +37,7 @@ import java.util.concurrent.CompletableFuture
 import kotlin.reflect.KClass
 import org.utbot.common.isWindows
 import org.utbot.framework.SummariesGenerationType
+import org.utbot.framework.codegen.domain.SymbolicEngineSource
 import org.utbot.framework.codegen.domain.UnknownTestFramework
 import org.utbot.framework.plugin.api.SpringTestType
 import org.utbot.framework.plugin.api.isSummarizationCompatible
@@ -66,6 +67,7 @@ class Settings(val project: Project) : PersistentStateComponent<Settings.State>
         var forceStaticMocking: ForceStaticMocking = ForceStaticMocking.defaultItem,
         var treatOverflowAsError: TreatOverflowAsError = TreatOverflowAsError.defaultItem,
         var parametrizedTestSource: ParametrizedTestSource = ParametrizedTestSource.defaultItem,
+        var symbolicEngineSource: SymbolicEngineSource = SymbolicEngineSource.defaultItem,
         var classesToMockAlways: Array<String> = Mocker.defaultSuperClassesToMockAlwaysNames.toTypedArray(),
         var springTestType: SpringTestType = SpringTestType.defaultItem,
         var springConfig: String = SpringConfig.defaultItem,
@@ -98,6 +100,7 @@ class Settings(val project: Project) : PersistentStateComponent<Settings.State>
             if (forceStaticMocking != other.forceStaticMocking) return false
             if (treatOverflowAsError != other.treatOverflowAsError) return false
             if (parametrizedTestSource != other.parametrizedTestSource) return false
+            if (symbolicEngineSource != other.symbolicEngineSource) return false
             if (!classesToMockAlways.contentEquals(other.classesToMockAlways)) return false
             if (springTestType != other.springTestType) return false
             if (springConfig != other.springConfig) return false
@@ -124,6 +127,7 @@ class Settings(val project: Project) : PersistentStateComponent<Settings.State>
             result = 31 * result + forceStaticMocking.hashCode()
             result = 31 * result + treatOverflowAsError.hashCode()
             result = 31 * result + parametrizedTestSource.hashCode()
+            result = 31 * result + symbolicEngineSource.hashCode()
             result = 31 * result + classesToMockAlways.contentHashCode()
             result = 31 * result + springTestType.hashCode()
             result = 31 * result + springConfig.hashCode()
@@ -174,6 +178,8 @@ class Settings(val project: Project) : PersistentStateComponent<Settings.State>
 
     val parametrizedTestSource: ParametrizedTestSource get() = state.parametrizedTestSource
 
+    val symbolicEngineSource: SymbolicEngineSource get() = state.symbolicEngineSource
+
     val classesToMockAlways: Set<String> get() = state.classesToMockAlways.toSet()
 
     val springTestType: SpringTestType get() = state.springTestType