Skip to content

Commit b8ff429

Browse files
committed
Add BaseModel support
1 parent 464e3ca commit b8ff429

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,11 @@ import javax.swing.JList
1616
import org.utbot.common.PathUtil
1717
import org.utbot.framework.plugin.api.CodegenLanguage
1818
import org.utbot.intellij.plugin.ui.GenerateTestsModel
19+
import org.utbot.intellij.plugin.ui.utils.BaseTestsModel
1920
import org.utbot.intellij.plugin.ui.utils.addDedicatedTestRoot
2021
import org.utbot.intellij.plugin.ui.utils.suitableTestSourceRoots
2122

22-
class TestFolderComboWithBrowseButton(private val model: GenerateTestsModel) : ComboboxWithBrowseButton() {
23+
class TestFolderComboWithBrowseButton(private val model: BaseTestsModel) : ComboboxWithBrowseButton() {
2324

2425
private val SET_TEST_FOLDER = "set test folder"
2526

@@ -71,7 +72,7 @@ class TestFolderComboWithBrowseButton(private val model: GenerateTestsModel) : C
7172
}
7273
}
7374

74-
private fun createNewTestSourceRoot(model: GenerateTestsModel): VirtualFile? =
75+
private fun createNewTestSourceRoot(model: BaseTestsModel): VirtualFile? =
7576
ReadAction.compute<VirtualFile, RuntimeException> {
7677
val desc = FileChooserDescriptor(false, true, false, false, false, false)
7778
val initialFile = model.project.guessProjectDir()
@@ -92,7 +93,7 @@ class TestFolderComboWithBrowseButton(private val model: GenerateTestsModel) : C
9293
childComponent.model = DefaultComboBoxModel(ArrayUtil.toObjectArray(comboItems))
9394
}
9495

95-
private fun formatUrl(virtualFile: VirtualFile, model: GenerateTestsModel): String {
96+
private fun formatUrl(virtualFile: VirtualFile, model: BaseTestsModel): String {
9697
var directoryUrl = if (virtualFile is FakeVirtualFile) {
9798
virtualFile.parent.presentableUrl + File.separatorChar + virtualFile.name
9899
} else {

0 commit comments

Comments
 (0)