Support XML configs in Spring projects #2041 (#2075)

Add XML configs gathering for Spring projects

Author: sofurihafe

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

@@ -190,7 +190,11 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
     private val codegenLanguages = createComboBox(CodegenLanguage.values())
     private val testFrameworks = createComboBox(TestFramework.allItems.toTypedArray())
 
-    private val modelSpringConfigs = (listOf(NO_SPRING_CONFIGURATION_OPTION) + model.getSortedSpringConfigurationClasses()).toTypedArray()
+    private val modelSpringConfigs = (
+            listOf(NO_SPRING_CONFIGURATION_OPTION) +
+                    model.getSortedSpringConfigurationClasses() +
+                    model.getSpringXMLConfigurationFiles()
+            ).toTypedArray()
     private val springConfig = createComboBox(modelSpringConfigs)
 
     private val mockStrategies = createComboBox(MockStrategyApi.values())

utbot-ui-commons/src/main/kotlin/org/utbot/intellij/plugin/models/BaseTestModel.kt

@@ -15,12 +15,20 @@ import org.jetbrains.kotlin.idea.core.getPackage
 import org.jetbrains.kotlin.idea.util.projectStructure.allModules
 import org.jetbrains.kotlin.idea.util.rootManager
 import org.jetbrains.kotlin.idea.util.sourceRoot
+import org.jetbrains.kotlin.utils.addToStdlib.flatMapToNullable
+import org.utbot.common.PathUtil.fileExtension
 import org.utbot.framework.codegen.domain.ProjectType
 import org.utbot.framework.plugin.api.CodegenLanguage
 import org.utbot.intellij.plugin.ui.utils.ITestSourceRoot
+import org.utbot.intellij.plugin.ui.utils.getResourcesPaths
 import org.utbot.intellij.plugin.ui.utils.getSortedTestRoots
 import org.utbot.intellij.plugin.ui.utils.isBuildWithGradle
 import org.utbot.intellij.plugin.ui.utils.suitableTestSourceRoots
+import java.nio.file.Files
+import java.nio.file.Path
+import javax.xml.parsers.DocumentBuilderFactory
+import kotlin.streams.asSequence
+import kotlin.streams.toList
 
 val PsiClass.packageName: String get() = this.containingFile.containingDirectory.getPackage()?.qualifiedName ?: ""
 const val HISTORY_LIMIT = 10
@@ -111,6 +119,27 @@ open class BaseTestsModel(
         }.mapNotNull { it.qualifiedName }
     }
 
+    fun getSpringXMLConfigurationFiles(): List<String> {
+        val resourcesPaths = srcModule.getResourcesPaths()
+        val xmlFilePaths = resourcesPaths.flatMapTo(mutableListOf()) { path ->
+            Files.list(path)
+                .asSequence()
+                .filter { it.fileExtension == ".xml" }
+        }
+
+        val builder = DocumentBuilderFactory.newInstance().newDocumentBuilder()
+        return xmlFilePaths.mapNotNull { path ->
+            val doc = builder.parse(path.toFile())
+
+            val isBeanTagName = doc.documentElement.tagName == "beans"
+            val hasAttribute = doc.documentElement.getAttribute("xmlns") == "http://www.springframework.org/schema/beans"
+            when {
+                isBeanTagName && hasAttribute -> path.toString()
+                else -> null
+            }
+        }
+    }
+
     fun updateSourceRootHistory(path: String) {
         sourceRootHistory.apply {
             remove(path)//Remove existing entry if any

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

@@ -24,6 +24,7 @@ import com.intellij.openapi.vfs.newvfs.impl.FakeVirtualFile
 import com.intellij.util.PathUtil.getParentPath
 import java.nio.file.Path
 import mu.KotlinLogging
+import org.jetbrains.jps.model.java.JavaResourceRootType
 import org.jetbrains.jps.model.module.JpsModuleSourceRootType
 import org.jetbrains.kotlin.config.KotlinFacetSettingsProvider
 import org.jetbrains.kotlin.config.TestResourceKotlinRootType
@@ -64,6 +65,18 @@ fun Module.kotlinTargetPlatform(): TargetPlatformVersion {
         ?.singleOrNull() ?: error("Can't determine target platform for module $this")
 }
 
+/**
+ * Gets paths to project resources source roots.
+ *
+ * E.g. src/main/resources
+ */
+fun Module.getResourcesPaths(): List<Path> =
+    ModuleRootManager.getInstance(this)
+        .contentEntries
+        .flatMap { it.sourceFolders.toList() }
+        .filter { it.rootType is JavaResourceRootType && !it.isTestSource }
+        .mapNotNull { it.file?.toNioPath() }
+
 /**
  * Gets a path to test resources source root.
  *
@@ -230,7 +243,8 @@ private fun getOrCreateTestResourcesUrl(module: Module, testSourceRoot: VirtualF
         if (!rootModel.isDisposed && rootModel.isWritable) rootModel.dispose()
     }
 }
-fun SourceFolder.getModifiableContentEntry() : ContentEntry? {
+
+private fun SourceFolder.getModifiableContentEntry() : ContentEntry? {
     return ModuleRootManager.getInstance(contentEntry.rootModel.module).modifiableModel.contentEntries.find { entry -> entry.url == url }
 }