@@ -16,10 +16,11 @@ import javax.swing.JList
16
16
import org.utbot.common.PathUtil
17
17
import org.utbot.framework.plugin.api.CodegenLanguage
18
18
import org.utbot.intellij.plugin.ui.GenerateTestsModel
19
+ import org.utbot.intellij.plugin.ui.utils.BaseTestsModel
19
20
import org.utbot.intellij.plugin.ui.utils.addDedicatedTestRoot
20
21
import org.utbot.intellij.plugin.ui.utils.suitableTestSourceRoots
21
22
22
- class TestFolderComboWithBrowseButton (private val model : GenerateTestsModel ) : ComboboxWithBrowseButton() {
23
+ class TestFolderComboWithBrowseButton (private val model : BaseTestsModel ) : ComboboxWithBrowseButton() {
23
24
24
25
private val SET_TEST_FOLDER = " set test folder"
25
26
@@ -71,7 +72,7 @@ class TestFolderComboWithBrowseButton(private val model: GenerateTestsModel) : C
71
72
}
72
73
}
73
74
74
- private fun createNewTestSourceRoot (model : GenerateTestsModel ): VirtualFile ? =
75
+ private fun createNewTestSourceRoot (model : BaseTestsModel ): VirtualFile ? =
75
76
ReadAction .compute<VirtualFile , RuntimeException > {
76
77
val desc = FileChooserDescriptor (false , true , false , false , false , false )
77
78
val initialFile = model.project.guessProjectDir()
@@ -92,7 +93,7 @@ class TestFolderComboWithBrowseButton(private val model: GenerateTestsModel) : C
92
93
childComponent.model = DefaultComboBoxModel (ArrayUtil .toObjectArray(comboItems))
93
94
}
94
95
95
- private fun formatUrl (virtualFile : VirtualFile , model : GenerateTestsModel ): String {
96
+ private fun formatUrl (virtualFile : VirtualFile , model : BaseTestsModel ): String {
96
97
var directoryUrl = if (virtualFile is FakeVirtualFile ) {
97
98
virtualFile.parent.presentableUrl + File .separatorChar + virtualFile.name
98
99
} else {
0 commit comments