Support Spring guided type replacements in symbolic engine

Author: EgorkaKulikov

utbot-framework-api/src/main/kotlin/org/utbot/framework/plugin/api/Api.kt

@@ -55,6 +55,8 @@ import java.io.File
 import kotlin.contracts.ExperimentalContracts
 import kotlin.contracts.contract
 import org.utbot.common.isAbstract
+import org.utbot.common.isStatic
+import org.utbot.framework.plugin.api.util.isSubtypeOf
 import org.utbot.framework.plugin.api.util.utContext
 
 const val SYMBOLIC_NULL_ADDR: Int = 0
@@ -1151,12 +1153,21 @@ class WildcardTypeParameter : TypeParameters(emptyList())
 /**
  * Additional data describing user project.
  */
-interface ApplicationContext
+interface ApplicationContext {
+
+    /**
+     * Finds a type to replace the original during symbolic
+     * analysis if it is guided with some additional information.
+     */
+    fun replaceTypeIfNeeded(type: RefType): ClassId?
+}
 
 /**
  * A context to use when no additional data is required.
  */
-object EmptyApplicationContext: ApplicationContext
+object EmptyApplicationContext: ApplicationContext {
+    override fun replaceTypeIfNeeded(type: RefType): ClassId? = null
+}
 
 /**
  * Data we get from Spring application context
@@ -1167,9 +1178,24 @@ object EmptyApplicationContext: ApplicationContext
 data class SpringApplicationContext(
     val beanQualifiedNames: List<String> = emptyList(),
 ): ApplicationContext {
-    private val springInjectedClasses: List<ClassId> by lazy {
-        beanQualifiedNames.map { fqn -> utContext.classLoader.loadClass(fqn).id }
+
+    val springInjectedClasses: List<ClassId> by lazy {
+        beanQualifiedNames
+            .map { fqn -> utContext.classLoader.loadClass(fqn) }
+            .filterNot { it.isAbstract || it.isInterface || it.isLocalClass || it.isMemberClass && !it.isStatic }
+            .map { it.id }
     }
+
+    /**
+     * Replaces an interface type with its implementor type
+     * if there is the unique implementor in bean definitions.
+     */
+    override fun replaceTypeIfNeeded(type: RefType): ClassId? =
+        if (type.sootClass.isInterface) {
+            springInjectedClasses.singleOrNull { it.isSubtypeOf(type.id) }
+        } else {
+            null
+        }
 }
 
 interface CodeGenerationSettingItem {

utbot-framework/src/main/kotlin/org/utbot/engine/Traverser.kt

@@ -1376,6 +1376,18 @@ class Traverser(
     ): ObjectValue {
         touchAddress(addr)
 
+        //TODO: remove !! after PR-1889 merge, context cannot be null any more
+        val concreteClassId = applicationContext!!.replaceTypeIfNeeded(type)
+        concreteClassId?.let {
+            val sootType = typeResolver.classOrDefault(it.canonicalName)
+            val typeStorage = typeResolver.constructTypeStorage(sootType, useConcreteType = false)
+
+            val typeHardConstraint = typeRegistry.typeConstraint(addr, typeStorage).all().asHardConstraint()
+            queuedSymbolicStateUpdates += typeHardConstraint
+
+            return ObjectValue(typeStorage, addr)
+        }
+
         val nullEqualityConstraint = mkEq(addr, nullObjectAddr)
 
         if (mockInfoGenerator != null) {

utbot-framework/src/main/kotlin/org/utbot/engine/types/TypeResolver.kt

@@ -95,7 +95,7 @@ class TypeResolver(private val typeRegistry: TypeRegistry, private val hierarchy
             .toSet()
     }
 
-    private fun classOrDefault(typeName: String): RefType =
+    fun classOrDefault(typeName: String): RefType =
         runCatching { Scene.v().getRefType(typeName) }.getOrDefault(OBJECT_TYPE)
 
     fun findFields(type: RefType) = typeRegistry.findFields(type) {