Improve adding of ReplacedFuzzedTypeFlag

UnitTestBot · Aug 29, 2023 · 1ba7d74 · 1ba7d74
1 parent 775db7b
commit 1ba7d74
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...mework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt b/...mework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt
@@ -88,10 +88,12 @@ class SpringApplicationContextImpl(
                             .withFallback(origValueProvider)
                             .replaceTypes { description, type ->
                                 typeReplacer.replaceTypeIfNeeded(type.classId)
-                                    ?.takeIf { it != type.classId }
-                                    ?.let { replacement ->
+                                    ?.let { replacementClassId ->
                                         // TODO infer generic type of replacement
-                                        toFuzzerType(replacement.jClass, description.typeCache).addProperties(
+                                        val replacement =
+                                            if (type.classId == replacementClassId) type
+                                            else toFuzzerType(replacementClassId.jClass, description.typeCache)
+                                        replacement.addProperties(
                                             dynamicPropertiesOf(ReplacedFuzzedTypeFlag.withValue(Unit))
                                         )
                                     } ?: type