@@ -2,7 +2,6 @@ package org.utbot.intellij.plugin.language.js
22
33import com.intellij.lang.javascript.refactoring.ui.JSMemberSelectionTable
44import com.intellij.lang.javascript.refactoring.util.JSMemberInfo
5- import com.intellij.openapi.ui.ComboBox
65import com.intellij.openapi.ui.DialogPanel
76import com.intellij.openapi.ui.DialogWrapper
87import com.intellij.openapi.ui.Messages
@@ -11,18 +10,15 @@ import com.intellij.openapi.vfs.VirtualFile
1110import com.intellij.openapi.vfs.VirtualFileManager
1211import com.intellij.ui.ContextHelpLabel
1312import com.intellij.ui.JBIntSpinner
14- import com.intellij.ui.components.Panel
13+ import com.intellij.ui.dsl.builder.panel
1514import com.intellij.ui.layout.Cell
16- import com.intellij.ui.layout.panel
1715import com.intellij.util.ui.JBUI
1816import framework.codegen.Mocha
19- import org.utbot.framework.plugin.api.CodeGenerationSettingItem
17+ import org.utbot.framework.codegen.domain.TestFramework
2018import org.utbot.intellij.plugin.ui.components.TestSourceDirectoryChooser
2119import settings.JsTestGenerationSettings.defaultTimeout
22- import java.awt.BorderLayout
2320import java.io.File
2421import java.nio.file.Paths
25- import javax.swing.DefaultComboBoxModel
2622import javax.swing.JComboBox
2723import javax.swing.JComponent
2824
@@ -36,7 +32,7 @@ class JsDialogWindow(val model: JsTestsModel) : DialogWrapper(model.project) {
3632 }
3733
3834 private val testSourceFolderField = TestSourceDirectoryChooser (model, model.file.virtualFile)
39- private val testFrameworks = ComboBox ( DefaultComboBoxModel (arrayOf( Mocha )) )
35+ private val testFrameworks = listOf< TestFramework >( Mocha )
4036 private val nycSourceFileChooserField = NycSourceFileChooser (model)
4137 private val coverageMode = CoverageModeButtons
4238
@@ -62,34 +58,28 @@ class JsDialogWindow(val model: JsTestsModel) : DialogWrapper(model.project) {
6258 override fun createCenterPanel (): JComponent {
6359 panel = panel {
6460 row(" Test source root:" ) {
65- component (testSourceFolderField)
61+ cell (testSourceFolderField)
6662 }
6763 row(" Test framework:" ) {
68- component(
69- Panel ().apply {
70- add(testFrameworks as ComboBox <CodeGenerationSettingItem >, BorderLayout .LINE_START )
71- }
72- )
64+ comboBox(testFrameworks, null )
7365 }
7466 row(" Nyc source path:" ) {
75- component (nycSourceFileChooserField)
67+ cell (nycSourceFileChooserField)
7668 }
7769 row(" Coverage mode:" ) {
78- coverageMode.fastButton()
79- coverageMode.baseButton( )
70+ cell( coverageMode.fastButton). apply { this .component.isSelected == true }
71+ cell( coverageMode.baseButton)
8072// panelWithHelpTooltip("Fast mode can't find timeouts, but works faster") {
8173// component(coverageMode.fastButton, coverageMode.baseButton)
8274// component(coverageMode.baseButton)
8375// }
8476 }
8577 row(" Timeout for Node.js (in seconds):" ) {
86- panelWithHelpTooltip(" The execution timeout" ) {
87- component(timeoutSpinner)
88- }
78+ cell(timeoutSpinner)
8979 }
9080 row(" Generate test methods for:" ) {}
9181 row {
92- scrollPane (functionsTable)
82+ scrollCell (functionsTable)
9383 }
9484 }
9585 updateMembersTable()
@@ -108,7 +98,7 @@ class JsDialogWindow(val model: JsTestsModel) : DialogWrapper(model.project) {
10898 override fun doOKAction () {
10999 val selected = functionsTable.selectedMemberInfos.toSet()
110100 model.selectedMethods = if (selected.any()) selected else emptySet()
111- model.testFramework = testFrameworks.item
101+ model.testFramework = testFrameworks
112102 model.timeout = timeoutSpinner.number.toLong()
113103 model.pathToNYC = nycSourceFileChooserField.text
114104 model.coverageMode = coverageMode.mode
0 commit comments