Apply review fixes

Author: EgorkaKulikov

utbot-analytics/src/test/kotlin/org/utbot/features/FeatureProcessorWithRepetitionTest.kt

@@ -5,13 +5,13 @@ import org.junit.jupiter.api.Assertions
 import org.junit.jupiter.api.BeforeAll
 import org.junit.jupiter.api.Test
 import org.utbot.analytics.EngineAnalyticsContext
-import org.utbot.examples.UtTestCaseChecker
+import org.utbot.examples.UtValueTestCaseChecker
 import org.utbot.examples.eq
 import org.utbot.examples.withFeaturePath
 import java.io.File
 import java.io.FileInputStream
 
-class FeatureProcessorWithRepetitionTest : UtTestCaseChecker(OnePath::class, false) {
+class FeatureProcessorWithRepetitionTest : UtValueTestCaseChecker(OnePath::class, false) {
     companion object {
         const val featureDir = "src/test/resources/features"
         fun reward(coverage: Double, time: Double) = RewardEstimator.reward(coverage, time)

utbot-cli/src/main/kotlin/org/utbot/cli/BunchTestGeneratorCommand.kt

@@ -99,7 +99,7 @@ class BunchTestGeneratorCommand : GenerateTestsAbstractCommand(
 
             initializeEngine(workingDirectory)
 
-            // utContext is used in `generateTestCases`, `generateTest`, `generateReport`
+            // utContext is used in `generate`, `generateTest`, `generateReport`
             withUtContext(UtContext(classLoader)) {
 
                 val testClassName = "${classUnderTest.simpleName}Test"

utbot-cli/src/main/kotlin/org/utbot/cli/GenerateTestsAbstractCommand.kt

@@ -159,7 +159,7 @@ abstract class GenerateTestsAbstractCommand(name: String, help: String) :
         searchDirectory: Path,
         chosenClassesToMockAlways: Set<ClassId>
     ): List<UtTestCase> =
-        TestCaseGenerator.generateTestCases(
+        TestCaseGenerator.generate(
             targetMethods,
             mockStrategy,
             chosenClassesToMockAlways,

utbot-cli/src/main/kotlin/org/utbot/cli/GenerateTestsCommand.kt

@@ -97,7 +97,7 @@ class GenerateTestsCommand :
             if (targetMethods.isEmpty()) {
                 throw Exception("Nothing to process. No methods were provided")
             }
-            // utContext is used in `generateTestCases`, `generateTest`, `generateReport`
+            // utContext is used in `generate`, `generateTest`, `generateReport`
             withUtContext(UtContext(targetMethods.first().clazz.java.classLoader)) {
 
                 val testClassName = output?.toPath()?.toFile()?.nameWithoutExtension

utbot-framework/src/main/kotlin/org/utbot/external/api/UtBotJavaApi.kt

@@ -127,7 +127,7 @@ object UtBotJavaApi {
                         FileUtil.isolateClassFiles(classUnderTest.kotlin).toPath(), classpath, dependencyClassPath
                     )
                 }
-                .generateTestCases(
+                .generate(
                     methodsForAutomaticGeneration.map {
                         toUtMethod(
                             it.methodToBeTestedFromUserInput,
@@ -191,7 +191,7 @@ object UtBotJavaApi {
                     init(
                         FileUtil.isolateClassFiles(classUnderTest.kotlin).toPath(), classpath, dependencyClassPath
                     )
-                }.generateTestCases(
+                }.generate(
                     methodsForAutomaticGeneration.map {
                         toUtMethod(
                             it.methodToBeTestedFromUserInput,

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

@@ -125,7 +125,7 @@ object TestCaseGenerator {
         }
 
     @Throws(CancellationException::class)
-    fun generateTestCasesAsync(
+    fun generateAsync(
         controller: EngineController,
         method: UtMethod<*>,
         mockStrategy: MockStrategyApi,
@@ -136,7 +136,7 @@ object TestCaseGenerator {
         return createDefaultFlow(engine)
     }
 
-    fun generateTestCases(
+    fun generate(
         methods: List<UtMethod<*>>,
         mockStrategy: MockStrategyApi,
         chosenClassesToMockAlways: Set<ClassId> = Mocker.javaDefaultClasses.mapTo(mutableSetOf()) { it.id },

utbot-framework/src/main/kotlin/org/utbot/framework/plugin/sarif/GenerateTestsAndSarifReportFacade.kt

@@ -70,7 +70,7 @@ class GenerateTestsAndSarifReportFacade(
     }
 
     private fun generateTestCases(targetClass: TargetClassWrapper, workingDirectory: Path): List<UtTestCase> =
-        TestCaseGenerator.generateTestCases(
+        TestCaseGenerator.generate(
             targetClass.targetMethods(),
             sarifProperties.mockStrategy,
             sarifProperties.classesToMockAlways,

utbot-framework/src/test/kotlin/org/utbot/examples/TestSpecificTestCaseGenerator.kt

@@ -5,6 +5,7 @@ import mu.KotlinLogging
 import org.utbot.common.runBlockingWithCancellationPredicate
 import org.utbot.common.runIgnoringCancellationException
 import org.utbot.engine.EngineController
+import org.utbot.engine.UtBotSymbolicEngine
 import org.utbot.framework.UtSettings
 import org.utbot.framework.plugin.api.MockStrategyApi
 import org.utbot.framework.plugin.api.TestCaseGenerator
@@ -13,10 +14,22 @@ import org.utbot.framework.plugin.api.UtExecution
 import org.utbot.framework.plugin.api.UtMethod
 import org.utbot.framework.plugin.api.UtTestCase
 import org.utbot.framework.util.jimpleBody
+import java.nio.file.Path
 
+/**
+ * Special [UtTestCase] generator for test methods that has a correct
+ * wrapper for suspend function [TestCaseGenerator.generateAsync].
+ */
 object TestSpecificTestCaseGenerator {
     private val logger = KotlinLogging.logger {}
 
+    fun init(buildDir: Path,
+             classpath: String?,
+             dependencyPaths: String,
+             engineActions: MutableList<(UtBotSymbolicEngine) -> Unit> = mutableListOf(),
+             isCanceled: () -> Boolean = { false },
+    ) = TestCaseGenerator.init(buildDir, classpath, dependencyPaths, engineActions, isCanceled)
+
     fun generate(method: UtMethod<*>, mockStrategy: MockStrategyApi): UtTestCase {
         logger.trace { "UtSettings:${System.lineSeparator()}" + UtSettings.toString() }
 
@@ -27,7 +40,7 @@ object TestSpecificTestCaseGenerator {
 
         runIgnoringCancellationException {
             runBlockingWithCancellationPredicate(TestCaseGenerator.isCanceled) {
-                TestCaseGenerator.generateTestCasesAsync(EngineController(), method, mockStrategy).collect {
+                TestCaseGenerator.generateAsync(EngineController(), method, mockStrategy).collect {
                     when (it) {
                         is UtExecution -> executions += it
                         is UtError -> errors.merge(it.description, 1, Int::plus)

utbot-framework/src/test/kotlin/org/utbot/examples/UtModelChecker.kt renamed to utbot-framework/src/test/kotlin/org/utbot/examples/UtModelTestCaseChecker.kt

@@ -15,7 +15,6 @@ import org.utbot.framework.plugin.api.FieldId
 import org.utbot.framework.plugin.api.MockStrategyApi
 import org.utbot.framework.plugin.api.MockStrategyApi.NO_MOCKS
 import org.utbot.framework.plugin.api.UtAssembleModel
-import org.utbot.framework.plugin.api.TestCaseGenerator
 import org.utbot.framework.plugin.api.UtCompositeModel
 import org.utbot.framework.plugin.api.UtDirectSetFieldModel
 import org.utbot.framework.plugin.api.UtExecution
@@ -39,7 +38,7 @@ import kotlin.reflect.KFunction2
 import kotlin.reflect.KFunction3
 import org.junit.jupiter.api.Assertions.assertTrue
 
-internal abstract class UtModelChecker(
+internal abstract class UtModelTestCaseChecker(
     testClass: KClass<*>,
     testCodeGeneration: Boolean = true,
     languagePipelines: List<CodeGenerationLanguageLastStage> = listOf(
@@ -136,7 +135,7 @@ internal abstract class UtModelChecker(
             buildDir = findPathToClassFiles(classLocation)
             previousClassLocation = classLocation
         }
-        TestCaseGenerator.init(buildDir, classpath = null, dependencyPaths = System.getProperty("java.class.path"))
+        TestSpecificTestCaseGenerator.init(buildDir, classpath = null, dependencyPaths = System.getProperty("java.class.path"))
         return TestSpecificTestCaseGenerator.generate(method, mockStrategy)
     }
 

utbot-framework/src/test/kotlin/org/utbot/examples/UtTestCaseChecker.kt renamed to utbot-framework/src/test/kotlin/org/utbot/examples/UtValueTestCaseChecker.kt

@@ -42,7 +42,6 @@ import org.utbot.framework.plugin.api.MockStrategyApi
 import org.utbot.framework.plugin.api.MockStrategyApi.NO_MOCKS
 import org.utbot.framework.plugin.api.ObjectMockTarget
 import org.utbot.framework.plugin.api.ParameterMockTarget
-import org.utbot.framework.plugin.api.TestCaseGenerator
 import org.utbot.framework.plugin.api.UtCompositeModel
 import org.utbot.framework.plugin.api.UtConcreteValue
 import org.utbot.framework.plugin.api.UtInstrumentation
@@ -72,7 +71,7 @@ import kotlin.reflect.KFunction5
 
 val logger = KotlinLogging.logger {}
 
-abstract class UtTestCaseChecker(
+abstract class UtValueTestCaseChecker(
     testClass: KClass<*>,
     testCodeGeneration: Boolean = true,
     languagePipelines: List<CodeGenerationLanguageLastStage> = listOf(
@@ -2493,7 +2492,7 @@ abstract class UtTestCaseChecker(
         mockStrategy: MockStrategyApi,
         additionalDependenciesClassPath: String
     ): UtTestCase {
-        TestCaseGenerator.init(buildDir, additionalDependenciesClassPath, System.getProperty("java.class.path"))
+        TestSpecificTestCaseGenerator.init(buildDir, additionalDependenciesClassPath, System.getProperty("java.class.path"))
         return TestSpecificTestCaseGenerator.generate(method, mockStrategy)
     }
 
@@ -2504,7 +2503,7 @@ abstract class UtTestCaseChecker(
     ): UtTestCase {
         val additionalDependenciesClassPath =
             computeAdditionalDependenciesClasspathAndBuildDir(method, additionalDependencies)
-        TestCaseGenerator.init(buildDir, additionalDependenciesClassPath, System.getProperty("java.class.path"))
+        TestSpecificTestCaseGenerator.init(buildDir, additionalDependenciesClassPath, System.getProperty("java.class.path"))
         withUtContext(UtContext(method.clazz.java.classLoader)) {
             return TestSpecificTestCaseGenerator.generate(method, mockStrategy)
         }