Add BaseTestsModel suppport

Author: rudolf101

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

@@ -1,36 +1,35 @@
 package org.utbot.intellij.plugin.ui
 
-import org.utbot.framework.codegen.ForceStaticMocking
-import org.utbot.framework.codegen.HangingTestsTimeout
-import org.utbot.framework.codegen.ParametrizedTestSource
-import org.utbot.framework.codegen.RuntimeExceptionTestsBehaviour
-import org.utbot.framework.codegen.StaticsMocking
-import org.utbot.framework.codegen.TestFramework
-import org.utbot.framework.plugin.api.ClassId
-import org.utbot.framework.plugin.api.CodegenLanguage
-import org.utbot.framework.plugin.api.MockFramework
-import org.utbot.framework.plugin.api.MockStrategyApi
 import com.intellij.openapi.module.Module
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.projectRoots.JavaSdkVersion
 import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.psi.PsiClass
 import com.intellij.refactoring.util.classMembers.MemberInfo
 import org.jetbrains.kotlin.idea.core.getPackage
+import org.utbot.framework.codegen.*
+import org.utbot.framework.plugin.api.ClassId
+import org.utbot.framework.plugin.api.CodegenLanguage
+import org.utbot.framework.plugin.api.MockFramework
+import org.utbot.framework.plugin.api.MockStrategyApi
+import org.utbot.intellij.plugin.ui.utils.BaseTestsModel
 
-data class GenerateTestsModel(
-    val project: Project,
-    val srcModule: Module,
-    val testModule: Module,
+class GenerateTestsModel(
+    project: Project,
+    srcModule: Module,
+    testModule: Module,
     val jdkVersion: JavaSdkVersion,
     var srcClasses: Set<PsiClass>,
     var selectedMethods: Set<MemberInfo>?,
-    var timeout:Long,
+    var timeout: Long,
     var generateWarningsForStaticMocking: Boolean = false,
     var forceMockHappened: Boolean = false,
     var hasTestFrameworkConflict: Boolean = false,
+) : BaseTestsModel(
+    project,
+    srcModule,
+    testModule
 ) {
-    var testSourceRoot: VirtualFile? = null
     var testPackageName: String? = null
     lateinit var testFramework: TestFramework
     lateinit var mockStrategy: MockStrategyApi