Introduce Spring codegen in pipeline

Author: EgorkaKulikov

utbot-testing/src/main/kotlin/org/utbot/testing/TestCodeGeneratorPipeline.kt

@@ -12,14 +12,11 @@ import org.utbot.framework.codegen.domain.ProjectType
 import org.utbot.framework.codegen.domain.StaticsMocking
 import org.utbot.framework.codegen.domain.TestFramework
 import org.utbot.framework.codegen.generator.CodeGenerator
+import org.utbot.framework.codegen.generator.SpringCodeGenerator
 import org.utbot.framework.codegen.services.language.CgLanguageAssistant
 import org.utbot.framework.codegen.tree.ututils.UtilClassKind
 import org.utbot.framework.codegen.tree.ututils.UtilClassKind.Companion.UT_UTILS_INSTANCE_NAME
-import org.utbot.framework.plugin.api.CodegenLanguage
-import org.utbot.framework.plugin.api.ExecutableId
-import org.utbot.framework.plugin.api.MockFramework
-import org.utbot.framework.plugin.api.MockStrategyApi
-import org.utbot.framework.plugin.api.UtMethodTestSet
+import org.utbot.framework.plugin.api.*
 import org.utbot.framework.plugin.api.util.UtContext
 import org.utbot.framework.plugin.api.util.description
 import org.utbot.framework.plugin.api.util.id
@@ -263,21 +260,46 @@ class TestCodeGeneratorPipeline(private val testInfrastructureConfiguration: Tes
 
         withUtContext(UtContext(classUnderTest.java.classLoader)) {
             val codeGenerator = with(testInfrastructureConfiguration) {
-                CodeGenerator(
-                    classUnderTest.id,
-                    projectType = ProjectType.PureJvm,
-                    generateUtilClassFile = generateUtilClassFile,
-                    paramNames = params,
-                    testFramework = testFramework,
-                    staticsMocking = staticsMocking,
-                    forceStaticMocking = forceStaticMocking,
-                    generateWarningsForStaticMocking = false,
-                    codegenLanguage = codegenLanguage,
-                    cgLanguageAssistant = CgLanguageAssistant.getByCodegenLanguage(codegenLanguage),
-                    parameterizedTestSource = parametrizedTestSource,
-                    runtimeExceptionTestsBehaviour = runtimeExceptionTestsBehaviour,
-                    enableTestsTimeout = enableTestsTimeout
-                )
+                when (projectType) {
+                    ProjectType.Spring -> SpringCodeGenerator(
+                        classUnderTest.id,
+                        projectType = ProjectType.Spring,
+                        generateUtilClassFile = generateUtilClassFile,
+                        paramNames = params,
+                        testFramework = testFramework,
+                        staticsMocking = staticsMocking,
+                        forceStaticMocking = forceStaticMocking,
+                        generateWarningsForStaticMocking = false,
+                        codegenLanguage = codegenLanguage,
+                        cgLanguageAssistant = CgLanguageAssistant.getByCodegenLanguage(codegenLanguage),
+                        parameterizedTestSource = parametrizedTestSource,
+                        runtimeExceptionTestsBehaviour = runtimeExceptionTestsBehaviour,
+                        enableTestsTimeout = enableTestsTimeout,
+                        codeGenerationContext = SpringApplicationContext(
+                            mockInstalled = true,
+                            staticsMockingIsConfigured = true,
+                            shouldUseImplementors = false,
+                            springTestType = SpringTestType.UNIT_TEST,
+                            springSettings = SpringSettings.AbsentSpringSettings(),
+                        )
+                    )
+                    ProjectType.PureJvm -> CodeGenerator(
+                        classUnderTest.id,
+                        projectType = ProjectType.PureJvm,
+                        generateUtilClassFile = generateUtilClassFile,
+                        paramNames = params,
+                        testFramework = testFramework,
+                        staticsMocking = staticsMocking,
+                        forceStaticMocking = forceStaticMocking,
+                        generateWarningsForStaticMocking = false,
+                        codegenLanguage = codegenLanguage,
+                        cgLanguageAssistant = CgLanguageAssistant.getByCodegenLanguage(codegenLanguage),
+                        parameterizedTestSource = parametrizedTestSource,
+                        runtimeExceptionTestsBehaviour = runtimeExceptionTestsBehaviour,
+                        enableTestsTimeout = enableTestsTimeout
+                    )
+                    else -> error("Unsupported project type $projectType in code generator instantiation")
+                }
             }
             val testClassCustomName = "${classUnderTest.java.simpleName}GeneratedTest"
 
@@ -327,7 +349,10 @@ class TestCodeGeneratorPipeline(private val testInfrastructureConfiguration: Tes
                 testFramework = TestFramework.defaultItem,
                 codegenLanguage = configuration.language,
                 mockFramework = MockFramework.defaultItem,
-                mockStrategy = MockStrategyApi.defaultItem,
+                mockStrategy = when (configuration.projectType) {
+                    ProjectType.Spring -> MockStrategyApi.springDefaultItem
+                    else -> MockStrategyApi.defaultItem
+                },
                 staticsMocking = StaticsMocking.defaultItem,
                 parametrizedTestSource = configuration.parametrizedTestSource,
                 forceStaticMocking = ForceStaticMocking.defaultItem,