Some suggestions to Spring variables creation

Author: EgorkaKulikov

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

@@ -66,7 +66,7 @@ class CgSpringTestClassConstructor(context: CgContext): CgAbstractTestClassConst
 
                 fields += CgFieldDeclaration(mockClassId, declaration, annotationInjectMock)
                 val value = vc.getOrCreateVariable(listOfUtModels.first(), name)
-                CgSpringVariableConstructor.fromListToVariableList += Pair(listOfUtModels, value)
+                CgSpringVariableConstructor.injectingMocksModelsVariables += listOfUtModels to value
             }
 
             // endregion
@@ -89,7 +89,7 @@ class CgSpringTestClassConstructor(context: CgContext): CgAbstractTestClassConst
                     )
                 fields += CgFieldDeclaration(mockClassId, declaration, annotationMock)
                 val value = vc.getOrCreateVariable(listOfUtModels.first(), name)
-                CgSpringVariableConstructor.fromListToVariableList += Pair(listOfUtModels, value)
+                CgSpringVariableConstructor.mockedModelsVariables += listOfUtModels to value
             }
 
             // endregion

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

@@ -1,19 +1,51 @@
 package org.utbot.framework.codegen.tree
 
+import com.jetbrains.rd.util.firstOrNull
 import org.utbot.framework.codegen.domain.context.CgContext
 import org.utbot.framework.codegen.domain.models.CgValue
+import org.utbot.framework.codegen.domain.models.CgVariable
+import org.utbot.framework.plugin.api.UtCompositeModel
 import org.utbot.framework.plugin.api.UtModel
+import org.utbot.framework.plugin.api.isMockModel
 
 class CgSpringVariableConstructor(context: CgContext) : CgVariableConstructor(context) {
+
+    private val mockFrameworkManager = CgComponents.getMockFrameworkManagerBy(context)
+
     override fun getOrCreateVariable(model: UtModel, name: String?): CgValue {
-//        val existingValue = fromListToVariableList[model]
-        val existingValue = fromListToVariableList.find { it.first.contains(model) }?.second
+        val alreadyCreatedMock = mockedModelsVariables
+            .filter { it.key.contains(model) }
+            .firstOrNull()
+            ?.value
+
+        val alreadyCreatedInjectMocks = injectingMocksModelsVariables
+            .filter { it.key.contains(model) }
+            .firstOrNull()
+            ?.value
+
+        if (alreadyCreatedInjectMocks != null) {
+            val compModel = model as UtCompositeModel
+            for (mock in compModel.fields.values) {
+                getOrCreateVariable(mock)
+            }
+        }
+
+        if (alreadyCreatedMock != null) {
+            if (model.isMockModel()) {
+                mockFrameworkManager.createMockForVariable(
+                    model as UtCompositeModel,
+                    alreadyCreatedMock as CgVariable,
+                )
+            }
+
+            return alreadyCreatedMock
+        }
 
-        return existingValue ?: super.getOrCreateVariable(model, name)
+        return super.getOrCreateVariable(model, name)
     }
 
     companion object {
-//        val fromListToVariableList: MutableMap<List<UtModel>, CgValue> = mutableMapOf()
-        val fromListToVariableList: MutableList<Pair<List<UtModel>, CgValue>> = mutableListOf()
+        val mockedModelsVariables: MutableMap<List<UtModel>, CgValue> = mutableMapOf()
+        val injectingMocksModelsVariables: MutableMap<List<UtModel>, CgValue> = mutableMapOf()
     }
 }