Default value of codegen language and test framework is improved in UI

Author: EgorkaKulikov

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

@@ -117,6 +117,7 @@ import javax.swing.JList
 import javax.swing.JPanel
 import kotlin.streams.toList
 import org.jetbrains.concurrency.thenRun
+import org.jetbrains.kotlin.asJava.classes.KtUltraLightClass
 import org.utbot.intellij.plugin.ui.utils.allLibraries
 
 private const val RECENTS_KEY = "org.utbot.recents"
@@ -176,6 +177,15 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  title = "Generate tests with UtBot"
  setResizable(false)
 
+ TestFramework.allItems.forEach {
+ it.isInstalled = findFrameworkLibrary(model.project, model.testModule, it) != null
+ }
+ MockFramework.allItems.forEach {
+ it.isInstalled = findFrameworkLibrary(model.project, model.testModule, it) != null
+ }
+ StaticsMocking.allItems.forEach { it.isConfigured = staticsMockingConfigured() }
+
+
  // Configure notification urls callbacks
  TestsReportNotifier.urlOpeningListener.callbacks[TestReportUrlOpeningListener.mockitoSuffix]?.plusAssign {
  if (createMockFrameworkNotificationDialog() == Messages.YES) {
@@ -420,7 +430,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  .toSet()
 
  val selectedMethods = selectedMembers.filter { it.member is PsiMethod }.toSet()
- model.selectedMethods = if (selectedMethods.isEmpty()) null else selectedMethods
+ model.selectedMethods = selectedMethods.ifEmpty { null }
 
  model.testFramework = testFrameworks.item
  model.mockStrategy = mockStrategies.item
@@ -566,7 +576,6 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  cbSpecifyTestPackage.isEnabled = model.srcClasses.all { cl -> cl.packageName.isNotEmpty() }
 
  val settings = model.project.service<Settings>()
- codegenLanguages.item = settings.codegenLanguage
  mockStrategies.item = settings.mockStrategy
  staticsMocking.item = settings.staticsMocking
  parametrizedTestSources.item = settings.parametrizedTestSource
@@ -575,15 +584,19 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  mockStrategies.isEnabled = areMocksSupported
  staticsMocking.isEnabled = areMocksSupported && mockStrategies.item != MockStrategyApi.NO_MOCKS
 
- //We do not support parameterized tests for JUnit4
+ codegenLanguages.item =
+ if (model.srcClasses.all { it is KtUltraLightClass }) CodegenLanguage.KOTLIN else CodegenLanguage.JAVA
+
+
+ val installedTestFramework = TestFramework.allItems.singleOrNull { it.isInstalled }
  currentFrameworkItem = when (parametrizedTestSources.item) {
- ParametrizedTestSource.DO_NOT_PARAMETRIZE -> settings.testFramework
- ParametrizedTestSource.PARAMETRIZE ->
- if (settings.testFramework == Junit4) TestFramework.parametrizedDefaultItem else settings.testFramework
+ ParametrizedTestSource.DO_NOT_PARAMETRIZE -> installedTestFramework ?: settings.testFramework
+ ParametrizedTestSource.PARAMETRIZE -> installedTestFramework
+ ?: if (settings.testFramework != Junit4) settings.testFramework else TestFramework.parametrizedDefaultItem
  }
 
  updateTestFrameworksList(settings.parametrizedTestSource)
- updateParametrizationVisibility(settings.testFramework)
+ updateParametrizationEnabled(currentFrameworkItem)
 
  updateMockStrategyList()
  updateStaticMockingStrategyList()
@@ -833,7 +846,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  val item = comboBox.item as TestFramework
 
  currentFrameworkItem = item
- updateParametrizationVisibility(currentFrameworkItem)
+ updateParametrizationEnabled(currentFrameworkItem)
  }
 
  parametrizedTestSources.addActionListener { event ->
@@ -883,10 +896,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  ParametrizedTestSource.DO_NOT_PARAMETRIZE -> TestFramework.defaultItem
  ParametrizedTestSource.PARAMETRIZE -> TestFramework.parametrizedDefaultItem
  }
- enabledTestFrameworks.forEach {
- it.isInstalled = findFrameworkLibrary(model.project, model.testModule, it) != null
- if (it.isInstalled && !defaultItem.isInstalled) defaultItem = it
- }
+ enabledTestFrameworks.forEach { if (it.isInstalled && !defaultItem.isInstalled) defaultItem = it }
 
  testFrameworks.model = DefaultComboBoxModel(enabledTestFrameworks.toTypedArray())
  testFrameworks.item = if (currentFrameworkItem in enabledTestFrameworks) currentFrameworkItem else defaultItem
@@ -906,7 +916,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  }
 
  //We would like to disable parametrization options for JUnit4
- private fun updateParametrizationVisibility(testFramework: TestFramework) {
+ private fun updateParametrizationEnabled(testFramework: TestFramework) {
  when (testFramework) {
  Junit4 -> parametrizedTestSources.isEnabled = false
  Junit5,
@@ -915,9 +925,6 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  }
 
  private fun updateMockStrategyList() {
- MOCKITO.isInstalled =
- findFrameworkLibrary(model.project, model.testModule, MOCKITO) != null
-
  mockStrategies.renderer = object : ColoredListCellRenderer<MockStrategyApi>() {
  override fun customizeCellRenderer(
  list: JList<out MockStrategyApi>, value: MockStrategyApi?,
@@ -932,8 +939,6 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
  }
 
  private fun updateStaticMockingStrategyList() {
- val staticsMockingConfigured = staticsMockingConfigured()
- StaticsMocking.allItems.forEach { it.isConfigured = staticsMockingConfigured }
  staticsMocking.renderer = object : ColoredListCellRenderer<StaticsMocking>() {
  override fun customizeCellRenderer(
  list: JList<out StaticsMocking>, value: StaticsMocking?,