tmp

Author: EgorkaKulikov

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

@@ -57,6 +57,12 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
                                 .forEach { model ->
                                     thisInstanceModels += model.wrap()
                                     thisInstancesDependentModels += collectByThisInstanceModel(model)
+
+                                }
+                            //TODO: собрать рекурснивно все модели, как в моках, и из них уже выбрать что надо
+                            execution.stateBefore.parameters + setOf(execution.stateBefore.thisInstance)
+                                .filterNotNull()
+                                .forEach { model ->
                                     applicationContextModels += collectApplicationContextModels(model)
                                 }
 
@@ -153,7 +159,7 @@ class SpringTestClassModelBuilder(val context: CgContext): TestClassModelBuilder
             model.modificationsChain.mapNotNull { it.instance }.forEach { innerModels += it }
 
             return innerModels
-                .filter { it.classId == applicationContextClassId }
+                .filterIsInstance<UtSpringContextModel>()
                 .map { it.wrap() }
                 .toSet()
         }