Small hotfixes

Author: zishkaz

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/js/JsActionMethods.kt

@@ -48,6 +48,7 @@ object JsActionMethods {
         val editor = e.getData(CommonDataKeys.EDITOR) ?: return null
         val file = e.getData(CommonDataKeys.PSI_FILE) as? JSFile ?: return null
         val element = findPsiElement(file, editor) ?: return null
+        val kek = element.module
         val module = element.module ?: return null
         val focusedMethod = getContainingMethod(element)
         containingClass(element)?.let {

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/js/JsDialogWindow.kt

@@ -2,74 +2,49 @@ package org.utbot.intellij.plugin.js
 
 import com.intellij.lang.javascript.refactoring.ui.JSMemberSelectionTable
 import com.intellij.lang.javascript.refactoring.util.JSMemberInfo
-import com.intellij.openapi.ui.ComboBox
 import com.intellij.openapi.ui.DialogPanel
 import com.intellij.openapi.ui.DialogWrapper
-import com.intellij.ui.ContextHelpLabel
-import com.intellij.ui.components.Panel
-import com.intellij.ui.layout.CellBuilder
-import com.intellij.ui.layout.Row
 import com.intellij.ui.layout.panel
 import com.intellij.util.ui.JBUI
-import org.utbot.framework.codegen.TestFramework
-import org.utbot.framework.plugin.api.CodeGenerationSettingItem
 import org.utbot.intellij.plugin.ui.components.TestFolderComboWithBrowseButton
-import java.awt.BorderLayout
-import javax.swing.DefaultComboBoxModel
 import javax.swing.JComponent
-import javax.swing.JPanel
+
 
 class JsDialogWindow(val model: JsTestsModel): DialogWrapper(model.project) {
 
     private val items = model.fileMethods
 
     private val functionsTable = JSMemberSelectionTable(items, null, null).apply {
-        items?.let {
-            val height = this.rowHeight * (items.size.coerceAtMost(12) + 1)
-            this.preferredScrollableViewportSize = JBUI.size(-1, height)
-        }
+        val height = this.rowHeight * (items.size.coerceAtMost(12) + 1)
+        this.preferredScrollableViewportSize = JBUI.size(-1, height)
     }
 
     private val testSourceFolderField = TestFolderComboWithBrowseButton(model)
 
-    private val testFrameworks = ComboBox(DefaultComboBoxModel(TestFramework.allItems.toTypedArray()))
-
     private lateinit var panel: DialogPanel
 
-    @Suppress("UNCHECKED_CAST")
-    private val itemsToHelpTooltip = hashMapOf(
-        (testFrameworks as ComboBox<CodeGenerationSettingItem>) to ContextHelpLabel.create(""),
-    )
-
     init {
         title = "Generate tests with UtBot"
         setResizable(false)
         init()
     }
 
-    @Suppress("UNCHECKED_CAST")
     override fun createCenterPanel(): JComponent {
         panel = panel {
             row("Test source root:") {
                 component(testSourceFolderField)
             }
-            row("Test framework:") {
-                makePanelWithHelpTooltip(
-                    testFrameworks as ComboBox<CodeGenerationSettingItem>,
-                    itemsToHelpTooltip[testFrameworks]
-                )
-            }
             row("Generate test methods for:") {}
             row {
                 scrollPane(functionsTable)
             }
         }
-        checkBoxUpdate()
+        updateMembersTable()
         return panel
     }
 
-    private fun checkBoxUpdate() {
-        if (items == null) return
+    private fun updateMembersTable() {
+        if (items.isEmpty()) isOKActionEnabled = false
         val focusedNames = model.focusedMethod?.map { it.name }
         val selectedMethods = items.filter {
             focusedNames?.contains(it.member.name) ?: false
@@ -82,13 +57,4 @@ class JsDialogWindow(val model: JsTestsModel): DialogWrapper(model.project) {
     }
 
     private fun checkMembers(members: Collection<JSMemberInfo>) = members.forEach { it.isChecked = true }
-
-    private fun Row.makePanelWithHelpTooltip(
-        mainComponent: JComponent,
-        contextHelpLabel: ContextHelpLabel?
-    ): CellBuilder<JPanel> =
-        component(Panel().apply {
-            add(mainComponent, BorderLayout.LINE_START)
-            contextHelpLabel?.let { add(it, BorderLayout.LINE_END) }
-        })
 }

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/js/JsTestsModel.kt

@@ -12,7 +12,7 @@ class JsTestsModel(
     project: Project,
     srcModule: Module,
     testModule: Module,
-    val fileMethods: Set<JSMemberInfo>?,
+    val fileMethods: Set<JSMemberInfo>,
     val focusedMethod: Set<JSFunction>?,
 ) : BaseTestsModel(
     project,