Support JS projects in CodeGenerator also

Author: EgorkaKulikov

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/CodeGenerator.kt

@@ -5,6 +5,7 @@ import org.utbot.framework.codegen.domain.ForceStaticMocking
 import org.utbot.framework.codegen.domain.HangingTestsTimeout
 import org.utbot.framework.codegen.domain.ParametrizedTestSource
 import org.utbot.framework.codegen.domain.ProjectType
+import org.utbot.framework.codegen.domain.ProjectType.*
 import org.utbot.framework.codegen.domain.RuntimeExceptionTestsBehaviour
 import org.utbot.framework.codegen.domain.StaticsMocking
 import org.utbot.framework.codegen.domain.TestFramework
@@ -30,7 +31,7 @@ import java.time.format.DateTimeFormatter
 open class CodeGenerator(
     val classUnderTest: ClassId,
     //TODO: support setting `projectType` in Sarif plugins, UtBotJava api, etc.
-    val projectType: ProjectType = ProjectType.PURE_JVM,
+    val projectType: ProjectType = PureJvm,
     paramNames: MutableMap<ExecutableId, List<String>> = mutableMapOf(),
     generateUtilClassFile: Boolean = false,
     testFramework: TestFramework = TestFramework.defaultItem,
@@ -81,9 +82,10 @@ open class CodeGenerator(
         return withCustomContext(testClassCustomName) {
             context.withTestClassFileScope {
                 when (context.projectType) {
-                    ProjectType.PURE_JVM,
-                    ProjectType.PYTHON -> generateForSimpleClass(cgTestSets)
-                    ProjectType.SPRING -> generateForSpringClass(cgTestSets)
+                    PureJvm,
+                    Python,
+                    JavaScript -> generateForSimpleClass(cgTestSets)
+                    Spring -> generateForSpringClass(cgTestSets)
                 }
             }
         }

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/Domain.kt

@@ -716,18 +716,22 @@ enum class ProjectType {
     /**
      * Standard JVM project without DI frameworks
      */
-    PURE_JVM,
+    PureJvm,
 
     /**
      * Spring or Spring Boot project
      */
-    SPRING,
+    Spring,
 
     /**
      * Python project
      */
-    PYTHON,
+    Python,
 
+    /**
+     * JavaScript project
+     */
+    JavaScript,
 }
 
 enum class TypeReplacementApproach {

utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonDialogWindow.kt

@@ -172,7 +172,7 @@ class PythonDialogWindow(val model: PythonTestsModel) : DialogWrapper(model.proj
         model.testFramework = testFrameworks.item
         model.timeout = TimeUnit.SECONDS.toMillis(timeoutSpinnerForTotalTimeout.number.toLong())
         model.testSourceRootPath = testSourceFolderField.text
-        model.projectType = ProjectType.PYTHON
+        model.projectType = ProjectType.Python
 
         val settings = model.project.service<Settings>()
         with(settings) {

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt

@@ -25,7 +25,7 @@ import org.utbot.framework.CancellationStrategyType.CANCEL_EVERYTHING
 import org.utbot.framework.CancellationStrategyType.NONE
 import org.utbot.framework.CancellationStrategyType.SAVE_PROCESSED_RESULTS
 import org.utbot.framework.UtSettings
-import org.utbot.framework.codegen.domain.ProjectType
+import org.utbot.framework.codegen.domain.ProjectType.*
 import org.utbot.framework.codegen.domain.TypeReplacementApproach
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.ApplicationContext
@@ -177,9 +177,10 @@ object UtTestsDialogProcessor {
                         val staticMockingConfigured = model.staticsMocking.isConfigured
 
                         val applicationContext = when (model.projectType) {
-                            ProjectType.PURE_JVM,
-                            ProjectType.PYTHON -> ApplicationContext(mockFrameworkInstalled, staticMockingConfigured)
-                            ProjectType.SPRING -> {
+                            PureJvm,
+                            Python,
+                            JavaScript -> ApplicationContext(mockFrameworkInstalled, staticMockingConfigured)
+                            Spring -> {
                                 val shouldUseImplementors = when (model.typeReplacementApproach) {
                                     TypeReplacementApproach.DO_NOT_REPLACE -> false
                                     TypeReplacementApproach.REPLACE_IF_POSSIBLE -> true

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

@@ -241,7 +241,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
         DependencyInjectionFramework.allItems.forEach {
             it.isInstalled = findDependencyInjectionLibrary(model.project, model.testModule, it) != null
         }
-        model.projectType = if (SpringBeans.isInstalled) ProjectType.SPRING else ProjectType.PURE_JVM
+        model.projectType = if (SpringBeans.isInstalled) ProjectType.Spring else ProjectType.PureJvm
 
         StaticsMocking.allItems.forEach {
             it.isConfigured = staticsMockingConfigured()
@@ -526,7 +526,7 @@ class GenerateTestsDialogWindow(val model: GenerateTestsModel) : DialogWrapper(m
         model.testSourceRoot?.apply { model.updateSourceRootHistory(this.toNioPath().toString()) }
 
         //TODO: obtain this values from UI controls https://github.com/UnitTestBot/UTBotJava/issues/1929
-        model.projectType = ProjectType.PURE_JVM
+        model.projectType = ProjectType.PureJvm
         model.typeReplacementApproach = TypeReplacementApproach.DO_NOT_REPLACE
 
         val settings = model.project.service<Settings>()

utbot-js/src/main/kotlin/codegen/JsCodeGenerator.kt

@@ -33,6 +33,7 @@ class JsCodeGenerator(
 ) {
     private var context: CgContext = CgContext(
         classUnderTest = classUnderTest,
+        projectType = ProjectType.JavaScript,
         paramNames = paramNames,
         testFramework = testFramework,
         mockFramework = MockFramework.MOCKITO,

utbot-python/src/main/kotlin/org/utbot/python/framework/codegen/model/PythonCodeGenerator.kt

@@ -43,7 +43,6 @@ import org.utbot.python.newtyping.pythonAnyType
 import org.utbot.python.newtyping.pythonModules
 import org.utbot.python.newtyping.pythonTypeRepresentation
 import org.utbot.python.framework.codegen.toPythonRawString
-import org.utbot.python.newtyping.pythonDescription
 import org.utbot.python.newtyping.pythonName
 
 class PythonCodeGenerator(
@@ -61,7 +60,7 @@ class PythonCodeGenerator(
     testClassPackageName: String = classUnderTest.packageName
 ) : CodeGenerator(
     classUnderTest = classUnderTest,
-    projectType = ProjectType.PYTHON,
+    projectType = ProjectType.Python,
     paramNames = paramNames,
     generateUtilClassFile = true,
     testFramework = testFramework,