Introduce autowired class variables

Author: EgorkaKulikov

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

@@ -5,4 +5,9 @@ import org.utbot.framework.plugin.api.BuiltinClassId
 internal val autowiredClassId = BuiltinClassId(
     canonicalName = "org.springframework.beans.factory.annotation",
     simpleName = "Autowired",
+)
+
+internal val repositoryClassId = BuiltinClassId(
+    canonicalName = "org.springframework.data.repository.Repository",
+    simpleName = "Repository"
 )

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

@@ -1,5 +1,6 @@
 package org.utbot.framework.codegen.domain.models
 
+import org.utbot.framework.codegen.domain.UtModelWrapper
 import org.utbot.framework.codegen.domain.models.builders.TypedModelWrappers
 import org.utbot.framework.plugin.api.ClassId
 
@@ -22,15 +23,17 @@ class SimpleTestClassModel(
 
 /**
  * Extended [SimpleTestClassModel] for Spring analysis reasons
- *
- * @param injectingMocksClass a class to inject other mocks into
- * @param mockedClasses variables of test class to represent mocked instances
  */
 class SpringTestClassModel(
     classUnderTest: ClassId,
     methodTestSets: List<CgMethodTestSet>,
     nestedClasses: List<SimpleTestClassModel>,
-    val thisInstanceModels: TypedModelWrappers = mapOf(),
-    val thisInstanceDependentMocks: TypedModelWrappers = mapOf(),
+    val springSpecificInformation: SpringSpecificInformation,
 ): TestClassModel(classUnderTest, methodTestSets, nestedClasses)
 
+
+class SpringSpecificInformation(
+    val thisInstanceModels: TypedModelWrappers = mapOf(),
+    val thisInstanceDependentMocks: TypedModelWrappers = mapOf(),
+    val repositoryModels: TypedModelWrappers = mapOf(),
+)

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/domain/models/builders/SpringTestClassModelBuilder.kt

@@ -1,8 +1,11 @@
 package org.utbot.framework.codegen.domain.models.builders
 
+import com.jetbrains.rd.framework.base.deepClonePolymorphic
 import org.utbot.framework.codegen.domain.UtModelWrapper
+import org.utbot.framework.codegen.domain.builtin.repositoryClassId
 import org.utbot.framework.codegen.domain.context.CgContext
 import org.utbot.framework.codegen.domain.models.CgMethodTestSet
+import org.utbot.framework.codegen.domain.models.SpringSpecificInformation
 import org.utbot.framework.codegen.domain.models.SpringTestClassModel
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.UtArrayModel
@@ -25,20 +28,24 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
 
     override fun createTestClassModel(classUnderTest: ClassId, testSets: List<CgMethodTestSet>): SpringTestClassModel {
         val baseModel = SimpleTestClassModelBuilder(context).createTestClassModel(classUnderTest, testSets)
-        val (thisInstanceModels, dependentMockModels) = collectThisInstanceAndDependentModels(testSets)
+        val collectedModels = collectModelsForClassVariables(testSets)
 
         return SpringTestClassModel(
             classUnderTest = baseModel.classUnderTest,
             methodTestSets = baseModel.methodTestSets,
             nestedClasses = baseModel.nestedClasses,
-            thisInstanceModels = thisInstanceModels,
-            thisInstanceDependentMocks = dependentMockModels
+            springSpecificInformation = SpringSpecificInformation(
+                thisInstanceModels = collectedModels.thisInstanceModels,
+                thisInstanceDependentMocks = collectedModels.thisInstanceDependentMocks,
+                repositoryModels = collectedModels.repositoryModels,
+            )
         )
     }
 
-    private fun collectThisInstanceAndDependentModels(testSets: List<CgMethodTestSet>): Pair<TypedModelWrappers, TypedModelWrappers> {
+    private fun collectModelsForClassVariables(testSets: List<CgMethodTestSet>): SpringSpecificInformation {
         val thisInstanceModels = mutableSetOf<UtModelWrapper>()
         val thisInstancesDependentModels = mutableSetOf<UtModelWrapper>()
+        val repositoryModels = mutableSetOf<UtModelWrapper>()
 
         with(context) {
             for ((testSetIndex, testSet) in testSets.withIndex()) {
@@ -51,6 +58,7 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
                                     thisInstanceModels += model.wrap()
                                     thisInstancesDependentModels += collectByThisInstanceModel(model)
                                 }
+                            // TODO: collect repository models from execution after merging prototype branch
                         }
                     }
                 }
@@ -63,7 +71,11 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
                     cgModel.model.isMockModel() && cgModel !in thisInstanceModels
                 }
 
-        return thisInstanceModels.groupByClassId() to dependentMockModels.groupByClassId()
+        return SpringSpecificInformation(
+            thisInstanceModels.groupByClassId(),
+            dependentMockModels.groupByClassId(),
+            mapOf(repositoryClassId to repositoryModels),
+        )
     }
 
     private fun collectByThisInstanceModel(model: UtModel): Set<UtModelWrapper> {

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgAbstractSpringTestClassConstructor.kt

@@ -1,6 +1,7 @@
 package org.utbot.framework.codegen.tree
 
 import org.utbot.framework.codegen.domain.builtin.TestClassUtilMethodProvider
+import org.utbot.framework.codegen.domain.builtin.autowiredClassId
 import org.utbot.framework.codegen.domain.builtin.injectMocksClassId
 import org.utbot.framework.codegen.domain.builtin.mockClassId
 import org.utbot.framework.codegen.domain.context.CgContext
@@ -84,8 +85,8 @@ abstract class CgAbstractSpringTestClassConstructor(context: CgContext):
     abstract fun constructAdditionalMethods(): CgMethodsCluster
 
     protected fun constructFieldsWithAnnotation(
+        annotationClassId: ClassId,
         groupedModelsByClassId: TypedModelWrappers,
-        annotationClassId: ClassId
     ): List<CgFieldDeclaration> {
         require(annotationClassId == injectMocksClassId || annotationClassId == mockClassId) {
             error("Unexpected annotation classId -- $annotationClassId")
@@ -109,6 +110,7 @@ abstract class CgAbstractSpringTestClassConstructor(context: CgContext):
             when (annotationClassId) {
                 injectMocksClassId -> variableConstructor.injectedMocksModelsVariables += listOfUtModels
                 mockClassId -> variableConstructor.mockedModelsVariables += listOfUtModels
+                autowiredClassId -> variableConstructor.autowiredVariables += listOfUtModels
             }
         }
 
@@ -129,7 +131,8 @@ abstract class CgAbstractSpringTestClassConstructor(context: CgContext):
         val whiteListOfModels =
             listOf(
                 variableConstructor.mockedModelsVariables,
-                variableConstructor.injectedMocksModelsVariables
+                variableConstructor.injectedMocksModelsVariables,
+                variableConstructor.autowiredVariables,
             ).flatten()
 
         valueByUtModelWrapper

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringIntegrationTestClassConstructor.kt

@@ -9,7 +9,14 @@ import org.utbot.framework.codegen.domain.models.SpringTestClassModel
 class CgSpringIntegrationTestClassConstructor(context: CgContext) : CgAbstractSpringTestClassConstructor(context) {
 
     override fun constructClassFields(testClassModel: SpringTestClassModel): List<CgFieldDeclaration> {
-        return constructFieldsWithAnnotation(testClassModel.thisInstanceModels, autowiredClassId)
+        val thisIntances = testClassModel.springSpecificInformation.thisInstanceModels
+        val repositories = testClassModel.springSpecificInformation.repositoryModels
+
+        val fields = mutableListOf<CgFieldDeclaration>()
+        fields += constructFieldsWithAnnotation(autowiredClassId, thisIntances)
+        fields += constructFieldsWithAnnotation(autowiredClassId, repositories)
+
+        return fields
     }
 
     override fun constructAdditionalMethods() = CgMethodsCluster(

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringUnitTestClassConstructor.kt

@@ -27,10 +27,12 @@ class CgSpringUnitTestClassConstructor(context: CgContext) : CgAbstractSpringTes
 
     override fun constructClassFields(testClassModel: SpringTestClassModel): List<CgFieldDeclaration> {
         val fields = mutableListOf<CgFieldDeclaration>()
+        val mocks = testClassModel.springSpecificInformation.thisInstanceDependentMocks
+        val mockedFields = constructFieldsWithAnnotation(mockClassId, mocks)
 
         if (testClassModel.thisInstanceDependentMocks.isNotEmpty()) {
-            val mockedFields = constructFieldsWithAnnotation(testClassModel.thisInstanceDependentMocks, mockClassId)
-            val injectingMocksFields = constructFieldsWithAnnotation(testClassModel.thisInstanceModels, injectMocksClassId)
+            val mockedFields = constructFieldsWithAnnotation(mockClassId, mocks)
+            val injectingMocksFields = constructFieldsWithAnnotation(injectMocksClassId, thisInstances)
 
             fields += injectingMocksFields
             fields += mockedFields

utbot-framework/src/main/kotlin/org/utbot/framework/codegen/tree/CgSpringVariableConstructor.kt

@@ -12,6 +12,7 @@ import org.utbot.framework.plugin.api.isMockModel
 class CgSpringVariableConstructor(context: CgContext) : CgVariableConstructor(context) {
     val injectedMocksModelsVariables: MutableSet<UtModelWrapper> = mutableSetOf()
     val mockedModelsVariables: MutableSet<UtModelWrapper> = mutableSetOf()
+    val autowiredVariables: MutableSet<UtModelWrapper> = mutableSetOf()
 
     override fun getOrCreateVariable(model: UtModel, name: String?): CgValue {
         val alreadyCreatedInjectMocks = findCgValueByModel(model, injectedMocksModelsVariables)
@@ -39,6 +40,11 @@ class CgSpringVariableConstructor(context: CgContext) : CgVariableConstructor(co
             return alreadyCreatedMock
         }
 
+        val alreadyCreatedAutowired = findCgValueByModel(model, autowiredVariables)
+        if (alreadyCreatedAutowired != null) {
+            return alreadyCreatedAutowired
+        }
+
         return super.getOrCreateVariable(model, name)
     }