Mocking

kex-core/src/main/kotlin/org/vorpal/research/kex/asm/manager/ClassInstantiationManager.kt

@@ -3,11 +3,13 @@ package org.vorpal.research.kex.asm.manager
 import org.vorpal.research.kex.ExecutionContext
 import org.vorpal.research.kex.asm.util.AccessModifier
 import org.vorpal.research.kex.asm.util.accessModifier
+import org.vorpal.research.kex.config.kexConfig
 import org.vorpal.research.kex.ktype.KexClass
 import org.vorpal.research.kex.ktype.KexReference
 import org.vorpal.research.kex.ktype.KexRtManager.rtMapped
 import org.vorpal.research.kex.ktype.KexType
 import org.vorpal.research.kex.ktype.kexType
+import org.vorpal.research.kex.util.KfgTargetFilter
 import org.vorpal.research.kex.util.isSubtypeOfCached
 import org.vorpal.research.kfg.ClassManager
 import org.vorpal.research.kfg.ir.Class
@@ -24,7 +26,7 @@ import kotlin.random.Random
 
 val instantiationManager: ClassInstantiationManager get() = StringClassInstantiationManagerImpl
 
-class NoConcreteInstanceException(val klass: Class) : Exception()
+class NoConcreteInstanceException(val klass: Class) : Exception("No concrete instance for $klass")
 
 interface ClassInstantiationManager {
     fun isDirectlyInstantiable(klass: Class, accessLevel: AccessModifier): Boolean
@@ -287,6 +289,15 @@ private object StringClassInstantiationManagerImpl : ClassInstantiationManager {
     }
 }
 
+
+private val ignoredInstantiations: Set<KfgTargetFilter> by lazy {
+    kexConfig.getMultipleStringValue("testGen", "ignoreInstantiation").flatMapTo(mutableSetOf()) {
+        val filter = KfgTargetFilter.parse(it)
+        listOf(filter, filter.rtMapped)
+    }
+}
+
+
 class ClassInstantiationDetector(
     private val ctx: ExecutionContext,
     private val baseAccessLevel: AccessModifier = AccessModifier.Private
@@ -296,6 +307,11 @@ class ClassInstantiationDetector(
     override fun cleanup() {}
 
     override fun visit(klass: Class) {
+        if (ignoredInstantiations.any { it.matches(klass) }) {
+            super.visit(klass)
+            return
+        }
+
         if (StringClassInstantiationManagerImpl.isDirectlyInstantiable(klass, baseAccessLevel))
             addInstantiableClass(klass, klass)
         super.visit(klass)

kex-core/src/main/kotlin/org/vorpal/research/kex/descriptor/converter.kt

@@ -16,9 +16,7 @@ import org.vorpal.research.kex.ktype.KexShort
 import org.vorpal.research.kex.ktype.KexString
 import org.vorpal.research.kex.ktype.KexType
 import org.vorpal.research.kex.ktype.asArray
-import org.vorpal.research.kex.util.allFields
-import org.vorpal.research.kex.util.isStatic
-import org.vorpal.research.kex.util.kex
+import org.vorpal.research.kex.util.*
 import org.vorpal.research.kfg.type.SystemTypeNames
 import org.vorpal.research.kthelper.assert.unreachable
 import org.vorpal.research.kthelper.logging.log
@@ -66,13 +64,21 @@ class Object2DescriptorConverter : DescriptorBuilder() {
                 is LongArray -> KexLong.asArray()
                 is FloatArray -> KexFloat.asArray()
                 is DoubleArray -> KexDouble.asArray()
-                is Array<*> -> any.javaClass.componentType.kex.asArray()
+                is Array<*> -> any.javaClass.componentType.kexTypeMockitoMockFixed.asArray()
                 is String -> KexString()
-                else -> any.javaClass.kex
+                else -> any.javaClass.kexTypeMockitoMockFixed
             }
         }
     }
 
+    private val Class<*>.kexTypeMockitoMockFixed: KexType
+        get() = when {
+            !kexConfig.isMockingEnabled || !kexConfig.isMockitoClassesWorkaroundEnabled -> this.kex
+
+            !this.name.containsMockitoMock -> this.kex
+            else -> KexClass(this.kex.name.removeMockitoMockSuffix())
+        }
+
     fun convert(any: Any?, depth: Int = 0): Descriptor {
         if (any == null) return `null`
         if (any in objectToDescriptor) return objectToDescriptor[any]!!