From 1ba7d749b6a8409e354efb091c7717b9b3c03cb9 Mon Sep 17 00:00:00 2001
From: IlyaMuravjov <muravjovilya@gmail.com>
Date: Tue, 29 Aug 2023 16:16:12 +0300
Subject: [PATCH] Improve adding of `ReplacedFuzzedTypeFlag`

---
 .../context/spring/SpringApplicationContextImpl.kt        | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/utbot-framework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt b/utbot-framework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt
index f0548e523d..9f86c8ae23 100644
--- a/utbot-framework/src/main/kotlin/org/utbot/framework/context/spring/SpringApplicationContextImpl.kt
+++ b/utbot-framework/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