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
 import org.utbot.framework.process.OpenModulesContainer
 
@@ -1177,6 +1179,12 @@ open class StandardApplicationContext(
             this.staticsMockingIsConfigured = false
         }
     }
+
+    /**
+     * Finds a type to replace the original during symbolic
+     * analysis if it is guided with some additional information.
+     */
+    open fun replaceTypeIfNeeded(type: RefType): ClassId? = null
 }
 
 /**
@@ -1191,8 +1199,22 @@ class SpringApplicationContext(
     val beanQualifiedNames: List<String> = emptyList(),
 ): StandardApplicationContext(mockInstalled, staticsMockingIsConfigured) {
     private val springInjectedClasses: List<ClassId> by lazy {
-        beanQualifiedNames.map { fqn -> utContext.classLoader.loadClass(fqn).id }
+        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

@@ -1373,6 +1373,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) {