Added option for turning off symbolic or fuzzing

tochilinak · tochilinak · commit 88ef3332da53 · 2023-12-12T16:57:29.000+03:00
diff --git a/utbot-cli-python/src/main/kotlin/org/utbot/cli/language/python/sbft/SbftGenerateTestsCommand.kt b/utbot-cli-python/src/main/kotlin/org/utbot/cli/language/python/sbft/SbftGenerateTestsCommand.kt
@@ -8,6 +8,7 @@ import com.github.ajalt.clikt.parameters.options.option
 import com.github.ajalt.clikt.parameters.options.required
 import com.github.ajalt.clikt.parameters.options.split
 import com.github.ajalt.clikt.parameters.types.choice
+import com.github.ajalt.clikt.parameters.types.enum
 import com.github.ajalt.clikt.parameters.types.long
 import mu.KotlinLogging
 import org.parsers.python.PythonParser
@@ -16,12 +17,7 @@ import org.utbot.cli.language.python.findCurrentPythonModule
 import org.utbot.cli.language.python.toAbsolutePath
 import org.utbot.cli.language.python.writeToFileAndSave
 import org.utbot.framework.codegen.domain.RuntimeExceptionTestsBehaviour
-import org.utbot.python.MypyConfig
-import org.utbot.python.PythonMethodHeader
-import org.utbot.python.PythonTestGenerationConfig
-import org.utbot.python.PythonTestGenerationProcessor
-import org.utbot.python.TestFileInformation
-import org.utbot.python.UsvmConfig
+import org.utbot.python.*
 import org.utbot.python.code.PythonCode
 import org.utbot.python.coverage.CoverageOutputFormat
 import org.utbot.python.coverage.PythonCoverageMode
@@ -115,6 +111,10 @@ class SbftGenerateTestsCommand : CliktCommand(
     )
         .flag(default = false)
 
+    private val searchMode by option("--mode")
+        .enum<InputSearchMode> { it.name }
+        .default(InputSearchMode.BOTH)
+
     private val javaCmd by option(
         "--java-cmd",
         help = "(required) Path to Java command (ONLY FOR USVM)."
@@ -273,6 +273,7 @@ class SbftGenerateTestsCommand : CliktCommand(
                 prohibitedExceptions = if (prohibitedExceptions == listOf("-")) emptyList() else prohibitedExceptions,
                 checkUsvm = checkUsvm,
                 doNotGenerateStateAssertions = doNotGenerateStateAssertions,
+                inputSearchMode = searchMode
             )
             val processor = SbftCliProcessor(config)
 
diff --git a/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationConfig.kt b/utbot-python/src/main/kotlin/org/utbot/python/PythonTestGenerationConfig.kt
@@ -34,6 +34,7 @@ class PythonTestGenerationConfig(
     val prohibitedExceptions: List<String> = defaultProhibitedExceptions,
     val checkUsvm: Boolean = false,
     val doNotGenerateStateAssertions: Boolean = false,
+    val inputSearchMode: InputSearchMode = InputSearchMode.BOTH
 ) {
     companion object {
         val defaultProhibitedExceptions = listOf(
@@ -60,4 +61,14 @@ data class UsvmConfig(
         val usvmJavaCmd = "${System.getProperty("java.home")}/bin/java"
         val defaultConfig = UsvmConfig(usvmJavaCmd, usvmDirectory)
     }
+}
+
+@Suppress("unused")
+enum class InputSearchMode(
+    val fuzzingEnabled: Boolean,
+    val symbolicEnabled: Boolean
+) {
+    FUZZING(true, false),
+    SYMBOLIC(false, true),
+    BOTH(true, true)
 }
diff --git a/utbot-python/src/main/kotlin/org/utbot/python/engine/GlobalPythonEngine.kt b/utbot-python/src/main/kotlin/org/utbot/python/engine/GlobalPythonEngine.kt
@@ -105,26 +105,36 @@ class GlobalPythonEngine(
     }
 
     fun run() {
-        val fuzzing = thread(
-            start = true,
-            isDaemon = false,
-            name = "Fuzzer"
-        ) {
-            logger.info { " >>>>>>> Start fuzzer >>>>>>> " }
-            runFuzzing()
-            logger.info { " <<<<<<< Finish fuzzer <<<<<<< " }
+        val threads = mutableListOf<Thread>()
+        if (configuration.inputSearchMode.fuzzingEnabled) {
+            val fuzzing = thread(
+                start = true,
+                isDaemon = false,
+                name = "Fuzzer"
+            ) {
+                logger.info { " >>>>>>> Start fuzzer >>>>>>> " }
+                runFuzzing()
+                logger.info { " <<<<<<< Finish fuzzer <<<<<<< " }
+            }
+            threads.add(fuzzing)
+        } else {
+            logger.info { "Fuzzing disabled." }
         }
-        val symbolic = thread(
-            start = true,
-            isDaemon = false,
-            name = "Symbolic"
-        ) {
-            logger.info { " ------- Start symbolic ------- " }
-            runSymbolic()
-            logger.info { " ======= Finish symbolic ======= " }
+        if (configuration.inputSearchMode.symbolicEnabled) {
+            val symbolic = thread(
+                start = true,
+                isDaemon = false,
+                name = "Symbolic"
+            ) {
+                logger.info { " ------- Start symbolic ------- " }
+                runSymbolic()
+                logger.info { " ======= Finish symbolic ======= " }
+            }
+            threads.add(symbolic)
+        } else {
+            logger.info { "Symbolic disabled." }
         }
-        fuzzing.join()
-        symbolic.join()
+        threads.forEach { it.join() }
     }
 
     fun debugUsvmRun() {
