Attempts to improve the logic

Author: EgorkaKulikov

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/components/TestFolderComboWithBrowseButton.kt

@@ -51,13 +51,9 @@ class TestFolderComboWithBrowseButton(private val model: GenerateTestsModel) : C
             }
         }
 
-        val testRoots = if (model.project.isGradle()) {
-            val allRoots = mutableSetOf<VirtualFile>()
-            model.project.allModules().map { it.suitableTestSourceRoots() }.forEach(allRoots::addAll)
-            allRoots.toMutableList()
-        } else {
-            model.potentialTestModules.flatMap { it.suitableTestSourceRoots().toMutableList() }.toMutableList()
-        }
+        val testRoots = model.potentialTestModules
+            .flatMap { it.suitableTestSourceRoots().toList() }
+            .toMutableList()
 
         // this method is blocked for Gradle, where multiple test modules can exist
         model.testModule.addDedicatedTestRoot(testRoots)

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/utils/ModuleUtils.kt

@@ -31,6 +31,7 @@ import org.jetbrains.android.sdk.AndroidSdkType
 import org.jetbrains.jps.model.module.JpsModuleSourceRootType
 import org.jetbrains.kotlin.config.KotlinFacetSettingsProvider
 import org.jetbrains.kotlin.config.TestResourceKotlinRootType
+import org.jetbrains.kotlin.idea.util.projectStructure.allModules
 import org.jetbrains.kotlin.platform.TargetPlatformVersion
 
 private val logger = KotlinLogging.logger {}
@@ -86,6 +87,10 @@ fun Module.getOrCreateSarifReportsPath(testSourceRoot: VirtualFile?): Path {
  * Find test modules by current source module.
  */
 fun Module.testModules(project: Project): List<Module> {
+    if (project.isGradle()) {
+        return project.allModules()
+    }
+
     var testModules = findPotentialModulesForTests(project, this)
     val testRootUrls = testModules.flatMap { it.suitableTestSourceRoots() }