Avoid mocking replaced types

Author: IlyaMuravjov

utbot-java-fuzzing/src/main/kotlin/org/utbot/fuzzing/spring/unit/Mocks.kt

@@ -18,11 +18,15 @@ import org.utbot.fuzzing.Routine
 import org.utbot.fuzzing.Scope
 import org.utbot.fuzzing.ScopeProperty
 import org.utbot.fuzzing.Seed
+import org.utbot.fuzzing.spring.FuzzedTypeFlag
+import org.utbot.fuzzing.spring.properties
 import org.utbot.fuzzing.spring.utils.jType
 import org.utbot.fuzzing.spring.utils.toTypeParametrizedByTypeVariables
 import org.utbot.fuzzing.spring.utils.typeToken
 import org.utbot.fuzzing.toFuzzerType
 
+object NeverMockFlag : FuzzedTypeFlag
+
 val methodsToMockProperty = ScopeProperty<Set<MethodId>>(
     description = "Method ids that can be mocked by `MockValueProvider`"
 )
@@ -37,6 +41,8 @@ class MockValueProvider(private val idGenerator: IdGenerator<Int>) : JavaValuePr
 
     private val methodsToMock = mutableSetOf<MethodId>()
 
+    override fun accept(type: FuzzedType) = NeverMockFlag !in type.properties
+
     override fun enrich(description: FuzzedDescription, type: FuzzedType, scope: Scope) {
         val publicMethods = type.classId.jClass.methods.map { it.executableId }
         publicMethods.intersect(methodsToMock).takeIf { it.isNotEmpty() }?.let {

utbot-spring-framework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt

@@ -1,8 +1,10 @@
 package org.utbot.framework.context.spring
 
 import mu.KotlinLogging
+import org.utbot.common.dynamicPropertiesOf
 import org.utbot.common.isAbstract
 import org.utbot.common.isStatic
+import org.utbot.common.withValue
 import org.utbot.framework.codegen.generator.AbstractCodeGenerator
 import org.utbot.framework.codegen.generator.CodeGeneratorParams
 import org.utbot.framework.codegen.generator.SpringCodeGenerator
@@ -25,8 +27,10 @@ import org.utbot.framework.plugin.api.util.allSuperTypes
 import org.utbot.framework.plugin.api.util.id
 import org.utbot.framework.plugin.api.util.jClass
 import org.utbot.framework.plugin.api.util.utContext
+import org.utbot.fuzzing.spring.addProperties
 import org.utbot.fuzzing.spring.decorators.replaceTypes
 import org.utbot.fuzzing.spring.unit.InjectMockValueProvider
+import org.utbot.fuzzing.spring.unit.NeverMockFlag
 import org.utbot.fuzzing.toFuzzerType
 
 class SpringApplicationContextImpl(
@@ -75,10 +79,14 @@ class SpringApplicationContextImpl(
                         )
                             .withFallback(origValueProvider)
                             .replaceTypes { description, type ->
-                                typeReplacer.replaceTypeIfNeeded(type.classId)?.let { replacement ->
-                                    // TODO infer generic type
-                                    toFuzzerType(replacement.jClass, description.typeCache)
-                                } ?: type
+                                typeReplacer.replaceTypeIfNeeded(type.classId)
+                                    ?.takeIf { it != type.classId }
+                                    ?.let { replacement ->
+                                        // TODO infer generic type of replacement
+                                        toFuzzerType(replacement.jClass, description.typeCache).addProperties(
+                                            dynamicPropertiesOf(NeverMockFlag.withValue(Unit))
+                                        )
+                                    } ?: type
                             }
                     }
             }