Skip to content

Commit 8d365c5

Browse files
committed
Add BaseModel support
1 parent b8ff429 commit 8d365c5

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

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

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,19 @@ import com.intellij.lang.javascript.refactoring.util.JSMemberInfo
55
import com.intellij.openapi.module.Module
66
import com.intellij.openapi.project.Project
77
import com.intellij.openapi.vfs.VirtualFile
8+
import com.intellij.testIntegration.TestFramework
9+
import org.utbot.intellij.plugin.ui.utils.BaseTestsModel
810

9-
data class JsTestsModel(
10-
val project: Project,
11-
val srcModule: Module,
12-
val testModule: Module,
11+
class JsTestsModel(
12+
project: Project,
13+
srcModule: Module,
14+
testModule: Module,
1315
val fileMethods: Set<JSMemberInfo>?,
1416
val focusedMethod: Set<JSFunction>?,
17+
) : BaseTestsModel(
18+
project,
19+
srcModule,
20+
testModule
1521
) {
16-
var testSourceRoot: VirtualFile? = null
17-
// TODO testPackageName
18-
// TODO testFramework
22+
lateinit var testFramework: TestFramework
1923
}

0 commit comments

Comments
 (0)