Added the ability to shorten the paths to SpringConfigClasses and SpringConfigXml in the UI

Author: tepa46

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

@@ -152,8 +152,11 @@ import org.utbot.intellij.plugin.ui.utils.parseVersion
 import org.utbot.intellij.plugin.ui.utils.testResourceRootTypes
 import org.utbot.intellij.plugin.ui.utils.testRootType
 import org.utbot.intellij.plugin.util.IntelliJApiHelper
+import org.utbot.intellij.plugin.util.SpringConfigurationsHelper
 import org.utbot.intellij.plugin.util.extractFirstLevelMembers
 import org.utbot.intellij.plugin.util.findSdkVersion
+import java.awt.Component
+import java.io.File
 import java.time.LocalDateTime
 import java.time.format.DateTimeFormatter
 import java.util.*
@@ -169,6 +172,8 @@ private const val NO_SPRING_CONFIGURATION_OPTION = "No configuration"
 private const val ACTION_GENERATE = "Generate Tests"
 private const val ACTION_GENERATE_AND_RUN = "Generate and Run"
 
+private const val JAVA_CONFIG_SEPARATOR = "."
+
 class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(model.project) {
     companion object {
         const val minSupportedSdkVersion = 8
@@ -192,12 +197,25 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
     private val codegenLanguages = createComboBox(CodegenLanguage.values())
     private val testFrameworks = createComboBox(TestFramework.allItems.toTypedArray())
 
-    private val modelSpringConfigs = setOf(
+    private val shortenedSortedSpringConfigurationClasses =
+        SpringConfigurationsHelper.shortenSpringConfigNames(
+            model.getSortedSpringConfigurationClasses(),
+            JAVA_CONFIG_SEPARATOR
+        )
+
+    private val shortenedSpringXMLConfigurationFiles =
+        SpringConfigurationsHelper.shortenSpringConfigNames(
+            model.getSpringXMLConfigurationFiles(),
+            File.separator
+        )
+
+    private val shortenedSpringConfigNames = setOf(
         null to listOf(NO_SPRING_CONFIGURATION_OPTION),
-        "Java-based configurations" to model.getSortedSpringConfigurationClasses(),
-        "XML-based configurations" to model.getSpringXMLConfigurationFiles()
+        "Java-based configurations" to shortenedSortedSpringConfigurationClasses,
+        "XML-based configurations" to shortenedSpringXMLConfigurationFiles
     )
-    private val springConfig = createComboBoxWithSeparatorsForSpringConfigs(modelSpringConfigs)
+
+    private val springConfig = createComboBoxWithSeparatorsForSpringConfigs(shortenedSpringConfigNames)
 
     private val mockStrategies = createComboBox(MockStrategyApi.values())
     private val staticsMocking = JCheckBox("Mock static methods")
@@ -1116,7 +1134,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
     private fun updateSpringConfigurationEnabled() {
         // We check for > 1 because there is already extra-dummy NO_SPRING_CONFIGURATION_OPTION option
         springConfig.isEnabled = model.projectType == ProjectType.Spring
-                && modelSpringConfigs.size > 1
+                && shortenedSpringConfigNames.size > 1
     }
 
     private fun staticsMockingConfigured(): Boolean {

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/util/SpringConfigurationsHelper.kt

@@ -0,0 +1,91 @@
+package org.utbot.intellij.plugin.util
+
+/**
+ * Getting from spring Configuration Classes and Spring XML Configuration Files shortened paths
+ *
+ * How is this done:
+ * - Parent directories are appended to the file name until the path becomes unique
+ *
+ *  Example:
+ *  - [["config.web.WebConfig", "config.web2.WebConfig", "config.web.AnotherConfig"]] ->
+ *  [["web.WebConfig", "web2.WebConfig", "AnotherConfig"]]
+ */
+object SpringConfigurationsHelper {
+
+    private var separator = ""
+
+    data class PathData(private var shortenedPath: String) {
+
+        private val pathConstructor: MutableList<String> = shortenedPath.split(separator).toMutableList()
+
+        init {
+            try {
+                shortenedPath = pathConstructor.last()
+                pathConstructor.removeLast()
+            } catch (e: Exception) {
+                println("Path [$shortenedPath] can't be extended")
+            }
+        }
+
+        fun getShortenedPath() = shortenedPath
+
+        fun addPathParentDir(): Boolean {
+            return try {
+                shortenedPath = pathConstructor.last() + separator + shortenedPath
+                pathConstructor.removeLast()
+                true
+            } catch (e: Exception) {
+                println("Path [$shortenedPath] can't be extended")
+                false
+            }
+        }
+    }
+
+    private fun preparePathsData(
+        paths: Set<String>,
+        pathsData: MutableList<PathData>,
+        pathsDataMap: MutableMap<String, PathData>
+    ) {
+        for (path in paths) {
+            pathsDataMap[path] = PathData(path)
+        }
+        pathsData.addAll(pathsDataMap.values)
+    }
+
+    private fun getMinimizedPaths(pathDataMap: MutableMap<String, PathData>): Set<String> {
+        val shortenedPaths = mutableSetOf<String>()
+        for (elem in pathDataMap.values) {
+            shortenedPaths.add(elem.getShortenedPath())
+        }
+        return shortenedPaths.toSet()
+    }
+
+    fun shortenSpringConfigNames(paths: Set<String>, separator: String): Set<String> {
+        SpringConfigurationsHelper.separator = separator
+
+        val pathsDataMap = mutableMapOf<String, PathData>()
+        var pathsData = mutableListOf<PathData>()
+
+        preparePathsData(paths, pathsData, pathsDataMap)
+
+        while (pathsData.size != pathsData.distinct().size) {
+            pathsData = pathsData.sortedBy { it.getShortenedPath() }.toMutableList()
+            for (ind in pathsData.indices) {
+                val curShortenedPath = pathsData[ind].getShortenedPath()
+
+                var maxIndWithSamePath = ind
+                while (maxIndWithSamePath < pathsData.size) {
+                    if (pathsData[maxIndWithSamePath].getShortenedPath() == curShortenedPath) maxIndWithSamePath++
+                    else break
+                }
+
+                if (ind == maxIndWithSamePath - 1) continue
+                for (i in ind until maxIndWithSamePath) {
+                    if (!pathsData[i].addPathParentDir()) return paths
+                }
+                break
+            }
+        }
+        return getMinimizedPaths(pathsDataMap)
+    }
+}