fix wrapper concrete types

Author: SBOne-Kenobi

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

@@ -25,7 +25,6 @@ import org.utbot.framework.plugin.api.UtModel
 import org.utbot.framework.plugin.api.UtNullModel
 import org.utbot.framework.plugin.api.UtPrimitiveModel
 import org.utbot.framework.plugin.api.getIdOrThrow
-import org.utbot.framework.plugin.api.idOrNull
 import org.utbot.framework.plugin.api.util.id
 import org.utbot.framework.plugin.api.util.objectArrayClassId
 import org.utbot.framework.plugin.api.util.objectClassId
@@ -34,6 +33,7 @@ import soot.Scene
 import soot.SootClass
 import soot.SootField
 import soot.SootMethod
+import soot.Type
 
 val rangeModifiableArrayId: ClassId = RangeModifiableUnlimitedArray::class.id
 
@@ -264,6 +264,9 @@ class RangeModifiableUnlimitedArrayWrapper : WrapperInterface {
         return resultModel
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(ARRAY_OBJECT_TYPE)
+
     companion object {
         internal val rangeModifiableArrayClass: SootClass
             get() = Scene.v().getSootClass(rangeModifiableArrayId.name)
@@ -277,6 +280,7 @@ class RangeModifiableUnlimitedArrayWrapper : WrapperInterface {
 }
 
 val associativeArrayId: ClassId = AssociativeArray::class.id
+val associativeArrayType: Type = Scene.v().getSootClass(AssociativeArray::class.java.canonicalName).type
 
 class AssociativeArrayWrapper : WrapperInterface {
 
@@ -415,6 +419,9 @@ class AssociativeArrayWrapper : WrapperInterface {
         return model
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(associativeArrayType)
+
     private fun Traverser.getStorageArrayField(addr: UtAddrExpression) =
         getArrayField(addr, associativeArrayClass, storageField)
 

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

@@ -26,20 +26,22 @@ import org.utbot.framework.plugin.api.UtNullModel
 import org.utbot.framework.plugin.api.UtStatementModel
 import org.utbot.framework.plugin.api.classId
 import org.utbot.framework.plugin.api.getIdOrThrow
-import org.utbot.framework.util.graph
 import org.utbot.framework.plugin.api.id
 import org.utbot.framework.plugin.api.util.booleanClassId
 import org.utbot.framework.plugin.api.util.constructorId
 import org.utbot.framework.plugin.api.util.id
 import org.utbot.framework.plugin.api.util.methodId
 import org.utbot.framework.plugin.api.util.objectClassId
+import org.utbot.framework.util.graph
 import org.utbot.framework.util.nextModelName
+import soot.ArrayType
 import soot.IntType
 import soot.RefType
 import soot.Scene
 import soot.SootClass
 import soot.SootField
 import soot.SootMethod
+import soot.Type
 
 abstract class BaseOverriddenWrapper(protected val overriddenClassName: String) : WrapperInterface {
     val overriddenClass: SootClass = Scene.v().getSootClass(overriddenClassName)
@@ -137,6 +139,9 @@ abstract class BaseContainerWrapper(containerClassName: String) : BaseOverridden
             }
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(Scene.v().getSootClass(chooseClassIdWithConstructor(type.classId).canonicalName).type)
+
     /**
      * Method returns list of parameter models that will be passed to [modificationMethodId]
      * while construction modification chain in [UtAssembleModel] for the specified [wrapper]
@@ -405,6 +410,9 @@ private val UT_GENERIC_ASSOCIATIVE_CLASS
 private val UT_GENERIC_ASSOCIATIVE_SET_EQUAL_GENERIC_TYPE_SIGNATURE =
     UT_GENERIC_ASSOCIATIVE_CLASS.getMethodByName(UtGenericAssociative<*, *>::setEqualGenericType.name).signature
 
+val ARRAY_OBJECT_TYPE: Type
+    get() = ArrayType.v(OBJECT_TYPE, 1)
+
 val ARRAY_LIST_TYPE: RefType
     get() = Scene.v().getSootClass(java.util.ArrayList::class.java.canonicalName).type
 val LINKED_LIST_TYPE: RefType

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

@@ -1,11 +1,15 @@
 package org.utbot.engine
 
+import java.util.concurrent.atomic.AtomicInteger
+import kotlin.reflect.KFunction2
+import kotlin.reflect.KFunction5
+import kotlinx.collections.immutable.persistentListOf
 import org.utbot.api.mock.UtMock
-import org.utbot.common.packageName
 import org.utbot.engine.overrides.UtArrayMock
 import org.utbot.engine.overrides.UtLogicMock
 import org.utbot.engine.overrides.UtOverrideMock
 import org.utbot.engine.pc.UtAddrExpression
+import org.utbot.engine.util.mockListeners.MockListenerController
 import org.utbot.framework.plugin.api.ClassId
 import org.utbot.framework.plugin.api.FieldId
 import org.utbot.framework.plugin.api.MethodId
@@ -14,17 +18,13 @@ import org.utbot.framework.plugin.api.id
 import org.utbot.framework.plugin.api.util.id
 import org.utbot.framework.plugin.api.util.isRefType
 import org.utbot.framework.util.executableId
-import java.util.concurrent.atomic.AtomicInteger
-import kotlin.reflect.KFunction2
-import kotlin.reflect.KFunction5
-import kotlinx.collections.immutable.persistentListOf
-import org.utbot.engine.util.mockListeners.MockListenerController
 import org.utbot.framework.util.isInaccessibleViaReflection
 import soot.BooleanType
 import soot.RefType
 import soot.Scene
 import soot.SootClass
 import soot.SootMethod
+import soot.Type
 
 /**
  * Generates mock with address provided.
@@ -307,6 +307,9 @@ class UtMockWrapper(
     override fun value(resolver: Resolver, wrapper: ObjectValue): UtModel =
         TODO("value on mock called: $this")
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(this.type)
+
     override fun toString() = "UtMock(type=$type, target=$mockInfo)"
 }
 

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

@@ -1,5 +1,11 @@
 package org.utbot.engine
 
+import java.util.Optional
+import java.util.OptionalDouble
+import java.util.OptionalInt
+import java.util.OptionalLong
+import java.util.concurrent.CopyOnWriteArrayList
+import kotlin.reflect.KClass
 import org.utbot.common.WorkaroundReason.MAKE_SYMBOLIC
 import org.utbot.common.workaround
 import org.utbot.engine.UtListClass.UT_ARRAY_LIST
@@ -34,12 +40,7 @@ import soot.RefType
 import soot.Scene
 import soot.SootClass
 import soot.SootMethod
-import java.util.Optional
-import java.util.OptionalDouble
-import java.util.OptionalInt
-import java.util.OptionalLong
-import java.util.concurrent.CopyOnWriteArrayList
-import kotlin.reflect.KClass
+import soot.Type
 
 typealias TypeToBeWrapped = RefType
 typealias WrapperType = RefType
@@ -222,6 +223,11 @@ interface WrapperInterface {
     ): List<InvokeResult>
 
     fun value(resolver: Resolver, wrapper: ObjectValue): UtModel
+
+    /**
+     * Returns list of types that can be produced by [value] method.
+     */
+    fun getPotentialPossibleTypes(type: Type): Set<Type>
 }
 
 // TODO: perhaps we have to have wrapper around concrete value here
@@ -255,4 +261,7 @@ data class ThrowableWrapper(val throwable: Throwable) : WrapperInterface {
                 }
             }
     }
+
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(Scene.v().getSootClass(throwable.javaClass.canonicalName).type)
 }

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

@@ -27,6 +27,7 @@ import org.utbot.framework.plugin.api.util.objectClassId
 import org.utbot.framework.util.nextModelName
 import soot.Scene
 import soot.SootMethod
+import soot.Type
 
 /**
  * Auxiliary enum class for specifying an implementation for [OptionalWrapper], that it will use.
@@ -98,6 +99,9 @@ class OptionalWrapper(private val utOptionalClass: UtOptionalClass) : BaseOverri
             }
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(type)
+
     private fun Resolver.instantiationFactoryCallModel(classId: ClassId, wrapper: ObjectValue, model: UtAssembleModel) : UtExecutableCallModel {
         val valueField = FieldId(overriddenClass.id, "value")
         val isPresentFieldId = FieldId(overriddenClass.id, "isPresent")

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

@@ -9,6 +9,7 @@ import org.utbot.framework.util.nextModelName
 import soot.Scene
 import soot.SootClass
 import soot.SootMethod
+import soot.Type
 
 class SecurityManagerWrapper : BaseOverriddenWrapper(utSecurityManagerClass.name) {
     private val baseModelName: String = "securityManager"
@@ -32,6 +33,9 @@ class SecurityManagerWrapper : BaseOverriddenWrapper(utSecurityManagerClass.name
         return UtAssembleModel(addr, classId, modelName, instantiationChain, modificationChain)
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(type)
+
     companion object {
         val utSecurityManagerClass: SootClass
             get() = Scene.v().getSootClass(UtSecurityManager::class.qualifiedName)

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

@@ -16,6 +16,7 @@ import org.utbot.framework.plugin.api.util.objectClassId
 import org.utbot.framework.util.nextModelName
 import soot.RefType
 import soot.Scene
+import soot.Type
 
 /**
  * Auxiliary enum class for specifying an implementation for [CommonStreamWrapper], that it will use.
@@ -81,6 +82,9 @@ abstract class StreamWrapper(
             }
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(STREAM_TYPE)
+
     override fun chooseClassIdWithConstructor(classId: ClassId): ClassId = error("No constructor for Stream")
 
     override val modificationMethodId: MethodId

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

@@ -1,6 +1,9 @@
 package org.utbot.engine
 
 import com.github.curiousoddman.rgxgen.RgxGen
+import kotlin.math.max
+import kotlinx.collections.immutable.persistentListOf
+import kotlinx.collections.immutable.persistentSetOf
 import org.utbot.common.unreachableBranch
 import org.utbot.engine.overrides.strings.UtNativeString
 import org.utbot.engine.overrides.strings.UtString
@@ -43,15 +46,13 @@ import org.utbot.framework.plugin.api.util.charClassId
 import org.utbot.framework.plugin.api.util.constructorId
 import org.utbot.framework.plugin.api.util.defaultValueModel
 import org.utbot.framework.util.nextModelName
-import kotlin.math.max
-import kotlinx.collections.immutable.persistentListOf
-import kotlinx.collections.immutable.persistentSetOf
 import soot.CharType
 import soot.IntType
 import soot.Scene
 import soot.SootClass
 import soot.SootField
 import soot.SootMethod
+import soot.Type
 
 val utStringClass: SootClass
     get() = Scene.v().getSootClass(UtString::class.qualifiedName)
@@ -191,6 +192,9 @@ class StringWrapper : BaseOverriddenWrapper(utStringClass.name) {
                 )
             }
     }
+
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(STRING_TYPE)
 }
 
 internal val utNativeStringClass = Scene.v().getSootClass(UtNativeString::class.qualifiedName)
@@ -286,6 +290,9 @@ class UtNativeStringWrapper : WrapperInterface {
         }
 
     override fun value(resolver: Resolver, wrapper: ObjectValue): UtModel = UtNullModel(STRING_TYPE.classId)
+
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(STRING_TYPE)
 }
 
 sealed class UtAbstractStringBuilderWrapper(className: String) : BaseOverriddenWrapper(className) {
@@ -348,6 +355,9 @@ sealed class UtAbstractStringBuilderWrapper(className: String) : BaseOverriddenW
             }
     }
 
+    override fun getPotentialPossibleTypes(type: Type): Set<Type> =
+        setOf(STRING_TYPE)
+
     private val SootClass.valueField: SootField
         get() = getField("value", CharType.v().arrayType)
 

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

@@ -1,5 +1,6 @@
 package org.utbot.engine
 
+import java.util.Objects
 import org.utbot.engine.pc.UtAddrExpression
 import org.utbot.engine.pc.UtExpression
 import org.utbot.engine.pc.isConcrete
@@ -12,7 +13,6 @@ import org.utbot.engine.pc.mkInt
 import org.utbot.engine.pc.mkLong
 import org.utbot.engine.pc.mkShort
 import org.utbot.engine.pc.toConcrete
-import java.util.Objects
 import soot.ArrayType
 import soot.BooleanType
 import soot.ByteType
@@ -179,7 +179,10 @@ fun SymbolicValue.toConcrete(): Any = when (this) {
 
 // TODO: one more constructor?
 fun objectValue(type: RefType, addr: UtAddrExpression, implementation: WrapperInterface) =
-    ObjectValue(TypeStorage(type), addr, Concrete(implementation))
+    ObjectValue(
+        TypeStorage(type, implementation.getPotentialPossibleTypes(type)),
+        addr, Concrete(implementation)
+    )
 
 val voidValue
     get() = PrimitiveValue(VoidType.v(), nullObjectAddr)