Fix build directory for classes from Kotlin files

Author: volivan239

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

@@ -197,7 +197,7 @@ abstract class GenerateTestsAbstractCommand(name: String, help: String) :
         UtSettings.treatOverflowAsError = treatOverflowAsError == TreatOverflowAsError.AS_ERROR
 
         return TestCaseGenerator(
-            workingDirectory,
+            listOf(workingDirectory),
             classPathNormalized,
             System.getProperty("java.class.path"),
             JdkInfoDefaultProvider().info

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

@@ -113,7 +113,7 @@ object UtBotJavaApi {
 
         testSets.addAll(withUtContext(utContext) {
             val buildPath = FileUtil.isolateClassFiles(classUnderTest).toPath()
-            TestCaseGenerator(buildPath, classpath, dependencyClassPath, jdkInfo = JdkInfoDefaultProvider().info)
+            TestCaseGenerator(listOf(buildPath), classpath, dependencyClassPath, jdkInfo = JdkInfoDefaultProvider().info)
                 .generate(
                     methodsForAutomaticGeneration.map {
                         it.methodToBeTestedFromUserInput.executableId
@@ -173,7 +173,7 @@ object UtBotJavaApi {
 
         return withUtContext(UtContext(classUnderTest.classLoader)) {
             val buildPath = FileUtil.isolateClassFiles(classUnderTest).toPath()
-            TestCaseGenerator(buildPath, classpath, dependencyClassPath, jdkInfo = JdkInfoDefaultProvider().info)
+            TestCaseGenerator(listOf(buildPath), classpath, dependencyClassPath, jdkInfo = JdkInfoDefaultProvider().info)
                 .generate(
                     methodsForAutomaticGeneration.map {
                         it.methodToBeTestedFromUserInput.executableId

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

@@ -52,14 +52,14 @@ import kotlin.reflect.KCallable
  * Generates test cases: one by one or a whole set for the method under test.
  *
  * Note: the instantiating of [TestCaseGenerator] may take some time,
- * because it requires initializing Soot for the current [buildDir] and [classpath].
+ * because it requires initializing Soot for the current [buildDirs] and [classpath].
  *
  * @param jdkInfo specifies the JRE and the runtime library version used for analysing system classes and user's code.
- * @param forceSootReload forces to reinitialize Soot even if the previous buildDir equals to [buildDir] and previous
+ * @param forceSootReload forces to reinitialize Soot even if the previous buildDirs equals to [buildDirs] and previous
  * classpath equals to [classpath]. This is the case for plugin scenario, as the source code may be modified.
  */
 open class TestCaseGenerator(
-    private val buildDir: Path,
+    private val buildDirs: List<Path>,
     private val classpath: String?,
     private val dependencyPaths: String,
     private val jdkInfo: JdkInfo,
@@ -71,21 +71,21 @@ open class TestCaseGenerator(
     private val timeoutLogger: KLogger = KotlinLogging.logger(logger.name + ".timeout")
 
     private val classpathForEngine: String
-        get() = buildDir.toString() + (classpath?.let { File.pathSeparator + it } ?: "")
+        get() = (buildDirs + listOfNotNull(classpath)).joinToString(File.pathSeparator)
 
     init {
         if (!isCanceled()) {
             checkFrameworkDependencies(dependencyPaths)
 
-            logger.trace("Initializing ${this.javaClass.name} with buildDir = $buildDir, classpath = $classpath")
+            logger.trace("Initializing ${this.javaClass.name} with buildDir = $buildDirs, classpath = $classpath")
 
 
             if (disableCoroutinesDebug) {
                 System.setProperty(kotlinx.coroutines.DEBUG_PROPERTY_NAME, kotlinx.coroutines.DEBUG_PROPERTY_VALUE_OFF)
             }
 
             timeoutLogger.trace().bracket("Soot initialization") {
-                SootUtils.runSoot(buildDir, classpath, forceSootReload, jdkInfo)
+                SootUtils.runSoot(buildDirs, classpath, forceSootReload, jdkInfo)
             }
 
             //warmup

utbot-framework/src/main/kotlin/org/utbot/framework/util/SootUtils.kt

@@ -19,7 +19,7 @@ import org.utbot.engine.overrides.UtOverrideMock
 import org.utbot.engine.overrides.collections.AbstractCollection
 import org.utbot.engine.overrides.collections.AssociativeArray
 import org.utbot.engine.overrides.collections.Collection
-import org.utbot.engine.overrides.collections.List
+import org.utbot.engine.overrides.collections.List as UtList
 import org.utbot.engine.overrides.collections.RangeModifiableUnlimitedArray
 import org.utbot.engine.overrides.collections.UtArrayList
 import org.utbot.engine.overrides.collections.UtGenericAssociative
@@ -58,40 +58,40 @@ object SootUtils {
      *
      * @param jdkInfo specifies the JRE and the runtime library version used for analysing system classes and user's
      * code.
-     * @param forceReload forces to reinitialize Soot even if the [previousBuildDir] equals to the class buildDir.
+     * @param forceReload forces to reinitialize Soot even if the [previousBuildDirs] equals to the class buildDir.
      */
     fun runSoot(clazz: java.lang.Class<*>, forceReload: kotlin.Boolean, jdkInfo: JdkInfo) {
         val buildDir = FileUtil.locateClassPath(clazz) ?: FileUtil.isolateClassFiles(clazz)
         val buildDirPath = buildDir.toPath()
 
-        runSoot(buildDirPath, null, forceReload, jdkInfo)
+        runSoot(listOf(buildDirPath), null, forceReload, jdkInfo)
     }
 
 
     /**
      * @param jdkInfo specifies the JRE and the runtime library version used for analysing system classes and user's
      * code.
-     * @param forceReload forces to reinitialize Soot even if the [previousBuildDir] equals to [buildDirPath] and
+     * @param forceReload forces to reinitialize Soot even if the [previousBuildDirs] equals to [buildDirPaths] and
      * [previousClassPath] equals to [classPath].
      */
-    fun runSoot(buildDirPath: Path, classPath: String?, forceReload: kotlin.Boolean, jdkInfo: JdkInfo) {
+    fun runSoot(buildDirPaths: List<Path>, classPath: String?, forceReload: kotlin.Boolean, jdkInfo: JdkInfo) {
         synchronized(this) {
-            if (buildDirPath != previousBuildDir || classPath != previousClassPath || forceReload) {
-                initSoot(buildDirPath, classPath, jdkInfo)
-                previousBuildDir = buildDirPath
+            if (buildDirPaths != previousBuildDirs || classPath != previousClassPath || forceReload) {
+                initSoot(buildDirPaths, classPath, jdkInfo)
+                previousBuildDirs = buildDirPaths
                 previousClassPath = classPath
             }
         }
     }
 
-    private var previousBuildDir: Path? = null
+    private var previousBuildDirs: List<Path>? = null
     private var previousClassPath: String? = null
 }
 
 /**
  * Convert code to Jimple
  */
-private fun initSoot(buildDir: Path, classpath: String?, jdkInfo: JdkInfo) {
+private fun initSoot(buildDirs: List<Path>, classpath: String?, jdkInfo: JdkInfo) {
     G.reset()
     val options = Options.v()
 
@@ -107,7 +107,7 @@ private fun initSoot(buildDir: Path, classpath: String?, jdkInfo: JdkInfo) {
                     + if (!classpath.isNullOrEmpty()) File.pathSeparator + "$classpath" else ""
         )
         set_src_prec(Options.src_prec_only_class)
-        set_process_dir(listOf("$buildDir"))
+        set_process_dir(buildDirs.map { "$it" })
         set_keep_line_number(true)
         set_ignore_classpath_errors(true) // gradle/build/resources/main does not exists, but it's not a problem
         set_output_format(Options.output_format_jimple)
@@ -196,7 +196,7 @@ private val classesToLoad = arrayOf(
     Stream::class,
     Arrays::class,
     Collection::class,
-    List::class,
+    UtList::class,
     UtStream::class,
     UtStream.UtStreamIterator::class
 ).map { it.java }.toTypedArray()

utbot-framework/src/main/kotlin/org/utbot/tests/infrastructure/TestSpecificTestCaseGenerator.kt

@@ -33,7 +33,7 @@ class TestSpecificTestCaseGenerator(
     engineActions: MutableList<(UtBotSymbolicEngine) -> Unit> = mutableListOf(),
     isCanceled: () -> Boolean = { false },
 ): TestCaseGenerator(
-    buildDir,
+    listOf(buildDir),
     classpath,
     dependencyPaths,
     JdkInfoDefaultProvider().info,

utbot-gradle/src/main/kotlin/org/utbot/gradle/plugin/GenerateTestsAndSarifReportTask.kt

@@ -90,7 +90,7 @@ open class GenerateTestsAndSarifReportTask @Inject constructor(
             withUtContext(UtContext(sourceSet.classLoader)) {
                 val testCaseGenerator =
                     TestCaseGenerator(
-                        sourceSet.workingDirectory,
+                        listOf(sourceSet.workingDirectory),
                         sourceSet.runtimeClasspath,
                         dependencyPaths,
                         JdkInfoDefaultProvider().info